Browse the data

1import random
2import string
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10orders = {}
11shipments = {}
12next_user_id = 1
13next_order_id = 1
14next_shipment_id = 1
15next_tracking_id = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_current_user(authorization: str = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing Authorization header")
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class OrderCreate(BaseModel):
37    description: str = ""
38    quantity: int = 1
39 
40class ShipmentCreate(BaseModel):
41    order_id: int
42    address: str = ""
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global next_user_id
47    for user in users.values():
48        if user["username"] == req.username:
49            raise HTTPException(status_code=400, detail="Username already exists")
50    user_id = next_user_id
51    next_user_id += 1
52    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
53    return {"id": user_id, "username": req.username}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for user in users.values():
58        if user["username"] == req.username and user["password"] == req.password:
59            token = generate_token()
60            tokens[token] = user["id"]
61            return {"token": token}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.get("/users/{user_id}")
65def get_user(user_id: int, authorization: str = Header(None)):
66    get_current_user(authorization)
67    if user_id not in users:
68        raise HTTPException(status_code=404, detail="User not found")
69    return users[user_id]
70 
71@app.post("/orders")
72def create_order(order: OrderCreate, authorization: str = Header(None)):
73    get_current_user(authorization)
74    global next_order_id
75    order_id = next_order_id
76    next_order_id += 1
77    orders[order_id] = {"id": order_id, "description": order.description, "quantity": order.quantity}
78    return orders[order_id]
79 
80@app.get("/orders/{order_id}")
81def get_order(order_id: int, authorization: str = Header(None)):
82    get_current_user(authorization)
83    if order_id not in orders:
84        raise HTTPException(status_code=404, detail="Order not found")
85    return orders[order_id]
86 
87@app.post("/shipments")
88def create_shipment(shipment: ShipmentCreate, authorization: str = Header(None)):
89    get_current_user(authorization)
90    global next_shipment_id, next_tracking_id
91    if shipment.order_id not in orders:
92        raise HTTPException(status_code=404, detail="Order not found")
93    shipment_id = next_shipment_id
94    next_shipment_id += 1
95    tracking_id = next_tracking_id
96    next_tracking_id += 1
97    shipments[shipment_id] = {
98        "id": shipment_id,
99        "order_id": shipment.order_id,
100        "address": shipment.address,
101        "tracking_id": tracking_id,
102        "status": "created"
103    }
104    return shipments[shipment_id]
105 
106@app.get("/shipments/{shipment_id}")
107def get_shipment(shipment_id: int, authorization: str = Header(None)):
108    get_current_user(authorization)
109    if shipment_id not in shipments:
110        raise HTTPException(status_code=404, detail="Shipment not found")
111    return shipments[shipment_id]
112 
113@app.get("/shipments/{tracking_id}/label")
114def get_label(tracking_id: int, authorization: str = Header(None)):
115    get_current_user(authorization)
116    try:
117        with open(f"/labels/{tracking_id}.png", "rb") as f:
118            return f.read()
119    except FileNotFoundError:
120        raise HTTPException(status_code=404, detail="Label not found")

1fastapi
2uvicorn