131 lines
2.9 KiB
Python
131 lines
2.9 KiB
Python
from typing import List, Optional
|
|
from pydantic import BaseModel, EmailStr
|
|
from datetime import date
|
|
|
|
|
|
# Base class for Contact Person
|
|
class ContactPersonBase(BaseModel):
|
|
firstname: str
|
|
lastname: str
|
|
phone_number: str
|
|
email: EmailStr
|
|
|
|
|
|
# Create schema for Contact Person
|
|
class ContactPersonCreate(ContactPersonBase):
|
|
pass
|
|
|
|
|
|
# Response schema for Contact Person with ID
|
|
class ContactPerson(ContactPersonBase):
|
|
id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Create schema for Address
|
|
class AddressCreate(BaseModel):
|
|
street: str
|
|
city: str
|
|
zipcode: str
|
|
country: str
|
|
|
|
|
|
# Response schema for Address with ID
|
|
class Address(AddressCreate):
|
|
id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Create schema for Dewar
|
|
class DewarCreate(BaseModel):
|
|
dewar_name: str
|
|
tracking_number: str
|
|
number_of_pucks: int
|
|
number_of_samples: int
|
|
status: str
|
|
ready_date: Optional[date]
|
|
shipping_date: Optional[date]
|
|
arrival_date: Optional[date]
|
|
returning_date: Optional[date]
|
|
qrcode: str
|
|
contact_person_id: Optional[int]
|
|
return_address_id: Optional[int]
|
|
|
|
|
|
# Response schema for Dewar
|
|
class Dewar(BaseModel):
|
|
id: str
|
|
dewar_name: str
|
|
tracking_number: str
|
|
number_of_pucks: int
|
|
number_of_samples: int
|
|
status: str
|
|
ready_date: Optional[date]
|
|
shipping_date: Optional[date]
|
|
arrival_date: Optional[date]
|
|
returning_date: Optional[date]
|
|
qrcode: str
|
|
shipment_id: Optional[str]
|
|
contact_person: Optional[ContactPerson]
|
|
return_address: Optional[Address]
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Proposal schema
|
|
class Proposal(BaseModel):
|
|
id: int
|
|
number: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# Response schema for Shipment
|
|
class Shipment(BaseModel):
|
|
shipment_id: str
|
|
shipment_name: str
|
|
shipment_date: date
|
|
shipment_status: str
|
|
comments: Optional[str]
|
|
contact_person: Optional[ContactPerson]
|
|
return_address: Optional[Address]
|
|
proposal: Optional[Proposal]
|
|
dewars: Optional[List[Dewar]] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class DewarUpdate(BaseModel):
|
|
dewar_id: str
|
|
dewar_name: Optional[str] = None
|
|
tracking_number: Optional[str] = None
|
|
number_of_pucks: Optional[int] = None
|
|
number_of_samples: Optional[int] = None
|
|
status: Optional[str] = None
|
|
ready_date: Optional[date] = None
|
|
shipping_date: Optional[date] = None
|
|
arrival_date: Optional[date] = None
|
|
returning_date: Optional[date] = None
|
|
qrcode: Optional[str] = None
|
|
contact_person_id: Optional[int] = None
|
|
address_id: Optional[int] = None # Added
|
|
|
|
class ShipmentCreate(BaseModel):
|
|
shipment_name: str
|
|
shipment_date: date
|
|
shipment_status: str
|
|
comments: Optional[str] = ""
|
|
contact_person_id: int
|
|
return_address_id: int
|
|
proposal_id: int
|
|
dewars: Optional[List[DewarUpdate]] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|