276 lines
9.1 KiB
Python
276 lines
9.1 KiB
Python
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi import HTTPException, status
|
|
from pydantic import BaseModel
|
|
from typing import List, Optional
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
app = FastAPI()
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
class ContactPerson(BaseModel):
|
|
firstname: str
|
|
lastname: str
|
|
phone_number: str
|
|
email: str
|
|
|
|
|
|
class Address(BaseModel):
|
|
street: str
|
|
city: str
|
|
zipcode: str
|
|
country: str
|
|
|
|
class Proposal(BaseModel):
|
|
id: int
|
|
number: str
|
|
|
|
class Dewar(BaseModel):
|
|
id: str
|
|
dewar_name: str
|
|
tracking_number: str
|
|
number_of_pucks: int
|
|
number_of_samples: int
|
|
return_address: List[Address]
|
|
contact_person: List[ContactPerson]
|
|
status: str
|
|
ready_date: Optional[str] = None
|
|
shipping_date: Optional[str] = None
|
|
arrival_date: Optional[str] = None
|
|
shippingStatus: str
|
|
arrivalStatus: str
|
|
qrcode: str
|
|
|
|
|
|
class Shipment(BaseModel):
|
|
shipment_id: str
|
|
shipment_name: str
|
|
shipment_date: str
|
|
shipment_status: str
|
|
contact_person: List[ContactPerson]
|
|
proposal_number: List[Proposal]
|
|
return_address: List[Address]
|
|
comments: Optional[str] = None
|
|
dewars: List[Dewar]
|
|
|
|
def get_number_of_dewars(self) -> int:
|
|
return len(self.dewars)
|
|
|
|
def get_shipment_contact_persons(self) -> List[ContactPerson]:
|
|
return self.contact_person
|
|
|
|
def get_shipment_return_addresses(self) -> List[Address]:
|
|
return self.return_address
|
|
|
|
def get_proposals(self) -> List[Proposal]:
|
|
return self.proposal_number
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
# Example data for contacts
|
|
contacts = [
|
|
ContactPerson(firstname="Frodo", lastname="Baggins", phone_number="123-456-7890", email="frodo.baggins@lotr.com"),
|
|
ContactPerson(firstname="Samwise", lastname="Gamgee", phone_number="987-654-3210", email="samwise.gamgee@lotr.com"),
|
|
ContactPerson(firstname="Aragorn", lastname="Elessar", phone_number="123-333-4444",
|
|
email="aragorn.elessar@lotr.com"),
|
|
ContactPerson(firstname="Legolas", lastname="Greenleaf", phone_number="555-666-7777",
|
|
email="legolas.greenleaf@lotr.com"),
|
|
ContactPerson(firstname="Gimli", lastname="Son of Gloin", phone_number="888-999-0000",
|
|
email="gimli.sonofgloin@lotr.com"),
|
|
ContactPerson(firstname="Gandalf", lastname="The Grey", phone_number="222-333-4444",
|
|
email="gandalf.thegrey@lotr.com"),
|
|
ContactPerson(firstname="Boromir", lastname="Son of Denethor", phone_number="111-222-3333",
|
|
email="boromir.sonofdenethor@lotr.com"),
|
|
ContactPerson(firstname="Galadriel", lastname="Lady of Lothlórien", phone_number="444-555-6666",
|
|
email="galadriel.lothlorien@lotr.com"),
|
|
ContactPerson(firstname="Elrond", lastname="Half-elven", phone_number="777-888-9999",
|
|
email="elrond.halfelven@lotr.com"),
|
|
ContactPerson(firstname="Eowyn", lastname="Shieldmaiden of Rohan", phone_number="000-111-2222",
|
|
email="eowyn.rohan@lotr.com")
|
|
]
|
|
|
|
# Example data for return addresses
|
|
return_addresses = [
|
|
Address(street='123 Hobbiton St', city='Shire', zipcode='12345', country='Middle Earth'),
|
|
Address(street='456 Rohan Rd', city='Edoras', zipcode='67890', country='Middle Earth')
|
|
]
|
|
|
|
# Example data for dewars
|
|
dewars = [
|
|
Dewar(
|
|
id='DEWAR001',
|
|
dewar_name='Dewar One',
|
|
tracking_number='TRACK123',
|
|
number_of_pucks=7,
|
|
number_of_samples=70,
|
|
return_address=[return_addresses[0]],
|
|
contact_person=[contacts[0]],
|
|
status='Ready',
|
|
ready_date='2023-09-30',
|
|
shipping_date='2023-10-01',
|
|
arrival_date='2023-10-02',
|
|
shippingStatus='Shipped',
|
|
arrivalStatus='Arrived',
|
|
qrcode='QR123DEWAR001'
|
|
),
|
|
Dewar(
|
|
id='DEWAR002',
|
|
dewar_name='Dewar Two',
|
|
tracking_number='TRACK124',
|
|
number_of_pucks=3,
|
|
number_of_samples=33,
|
|
return_address=[return_addresses[1]],
|
|
contact_person=[contacts[1]],
|
|
status='In Transit',
|
|
ready_date='2023-10-01',
|
|
shipping_date='2023-10-02',
|
|
arrival_date='2023-10-04',
|
|
shippingStatus='In Transit',
|
|
arrivalStatus='Pending',
|
|
qrcode='QR123DEWAR002'
|
|
),
|
|
Dewar(
|
|
id='DEWAR003',
|
|
dewar_name='Dewar Three',
|
|
tracking_number='TRACK125',
|
|
number_of_pucks=4,
|
|
number_of_samples=47,
|
|
return_address=[return_addresses[0]],
|
|
contact_person=[contacts[2]],
|
|
status='Pending',
|
|
shippingStatus='Ready for Shipping',
|
|
arrivalStatus='Pending',
|
|
qrcode='QR123DEWAR003'
|
|
),
|
|
]
|
|
|
|
# Proposal data inspired by the Lord of the Rings
|
|
proposals = [
|
|
Proposal(id=1, number="PROPOSAL-FRODO-001"), # "The Quest for the Ring"
|
|
Proposal(id=2, number="PROPOSAL-GANDALF-002"), # "The Fellowship's Journey"
|
|
Proposal(id=3, number="PROPOSAL-ARAGORN-003"), # "Return of the King"
|
|
Proposal(id=4, number="PROPOSAL-SAURON-004"), # "The Dark Lord's Plot"
|
|
Proposal(id=5, number="PROPOSAL-MORDOR-005"), # "The Road to Mount Doom"
|
|
]
|
|
|
|
# Example: Attach specific Dewars by their ids to shipments
|
|
specific_dewar_ids1 = ['DEWAR003'] # The IDs of the Dewars you want to attach to the first shipment
|
|
specific_dewar_ids2 = ['DEWAR001', 'DEWAR002'] # The IDs of the Dewars you want to attach to the second shipment
|
|
|
|
# Find the Dewars with the matching ids
|
|
specific_dewars1 = [dewar for dewar in dewars if dewar.id in specific_dewar_ids1]
|
|
specific_dewars2 = [dewar for dewar in dewars if dewar.id in specific_dewar_ids2]
|
|
|
|
# Define shipments with the selected Dewars
|
|
shipments = [
|
|
Shipment(
|
|
shipment_id='SHIPMORDOR',
|
|
shipment_date='2024-10-10',
|
|
shipment_name='Shipment from Mordor',
|
|
shipment_status='Delivered',
|
|
contact_person=[contacts[1]],
|
|
proposal_number=[proposals[1]],
|
|
return_address=[return_addresses[0]],
|
|
comments='Handle with care',
|
|
dewars=specific_dewars1 # Attach specific Dewars for this shipment
|
|
),
|
|
Shipment(
|
|
shipment_id='SHIPMORDOR2',
|
|
shipment_date='2024-10-24',
|
|
shipment_name='Shipment from Mordor',
|
|
shipment_status='In Transit',
|
|
contact_person=[contacts[3]],
|
|
proposal_number=[proposals[2]],
|
|
return_address=[return_addresses[1]], # Changed index to a valid one
|
|
comments='Contains the one ring',
|
|
dewars=specific_dewars2 # Attach specific Dewars for this shipment
|
|
)
|
|
]
|
|
|
|
@app.get("/contacts", response_model=List[ContactPerson])
|
|
async def get_contacts():
|
|
return contacts
|
|
|
|
@app.get("/return_addresses", response_model=List[Address])
|
|
async def get_return_addresses():
|
|
return return_addresses
|
|
|
|
@app.get("/proposals", response_model=List[Proposal])
|
|
async def get_proposals():
|
|
return proposals
|
|
|
|
|
|
@app.get("/shipments", response_model=List[Shipment])
|
|
async def get_shipments():
|
|
return shipments
|
|
|
|
|
|
@app.get("/dewars", response_model=List[Dewar])
|
|
async def get_dewars():
|
|
return dewars
|
|
|
|
|
|
# Endpoint to get the number of dewars in each shipment
|
|
@app.get("/shipment_dewars")
|
|
async def get_shipment_dewars():
|
|
return [{"shipment_id": shipment.shipment_id, "number_of_dewars": shipment.get_number_of_dewars()} for shipment in
|
|
shipments]
|
|
|
|
@app.get("/shipment_contact_persons")
|
|
async def get_shipment_contact_persons():
|
|
return [{"shipment_id": shipment.shipment_id, "contact_person": shipment.get_shipment_contact_persons()} for shipment in
|
|
shipments]
|
|
|
|
# Creation of a new shipment
|
|
@app.post("/shipments", response_model=Shipment, status_code=status.HTTP_201_CREATED)
|
|
async def create_shipment(shipment: Shipment):
|
|
# Check for duplicate shipment_id
|
|
if any(s.shipment_id == shipment.shipment_id for s in shipments):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Shipment with this ID already exists."
|
|
)
|
|
|
|
shipments.append(shipment)
|
|
return shipment
|
|
|
|
# Creation of a new contact
|
|
@app.post("/contacts", response_model=ContactPerson, status_code=status.HTTP_201_CREATED)
|
|
async def create_contact(contact: ContactPerson):
|
|
logging.info(f"Received contact creation request: {contact}")
|
|
# Check for duplicate contact by email (or other unique fields)
|
|
if any(c.email == contact.email for c in contacts):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Contact with this email already exists."
|
|
)
|
|
|
|
contacts.append(contact)
|
|
return contact
|
|
|
|
# Creation of a return address
|
|
@app.post("/return_addresses", response_model=Address, status_code=status.HTTP_201_CREATED)
|
|
async def create_return_address(address: Address):
|
|
logging.info(f"Received contact creation request: {address}")
|
|
# Check for duplicate address by city
|
|
if any(a.city == address.city for a in return_addresses):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Address in this city already exists."
|
|
)
|
|
|
|
return_addresses.append(address)
|
|
return address
|