Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10trucks = {}
11orders = {}
12tokens = {}
13next_user_id = 1
14next_truck_id = 1
15next_order_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class TruckCreate(BaseModel):
26    name: str
27    cuisine: str
28    location: str
29    is_open: bool = True
30    owner_id: int
31 
32class TruckUpdate(BaseModel):
33    name: Optional[str] = None
34    cuisine: Optional[str] = None
35    location: Optional[str] = None
36    is_open: Optional[bool] = None
37    owner_id: Optional[int] = None
38 
39class OrderCreate(BaseModel):
40    truck_id: int
41    items: list
42    total: float
43 
44def get_current_user(authorization: str = Header(...)):
45    token = authorization.replace("Bearer ", "")
46    if token not in tokens:
47        raise HTTPException(status_code=401, detail="Invalid token")
48    return tokens[token]
49 
50@app.post("/signup")
51def signup(req: SignupRequest):
52    global next_user_id
53    user_id = next_user_id
54    next_user_id += 1
55    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
56    return {"id": user_id, "username": req.username}
57 
58@app.post("/login")
59def login(req: LoginRequest):
60    for uid, u in users.items():
61        if u["username"] == req.username and u["password"] == req.password:
62            token = secrets.token_hex(16)
63            tokens[token] = uid
64            return {"token": token}
65    raise HTTPException(status_code=401, detail="Invalid credentials")
66 
67@app.get("/trucks/{truck_id}")
68def get_truck(truck_id: int):
69    if truck_id not in trucks:
70        raise HTTPException(status_code=404, detail="Truck not found")
71    return trucks[truck_id]
72 
73@app.post("/trucks")
74def create_truck(truck: TruckCreate, authorization: str = Header(...)):
75    global next_truck_id
76    current_user = get_current_user(authorization)
77    truck_id = next_truck_id
78    next_truck_id += 1
79    trucks[truck_id] = {
80        "id": truck_id,
81        "name": truck.name,
82        "cuisine": truck.cuisine,
83        "location": truck.location,
84        "is_open": truck.is_open,
85        "owner_id": truck.owner_id
86    }
87    return trucks[truck_id]
88 
89@app.patch("/trucks/{truck_id}")
90def update_truck(truck_id: int, truck: TruckUpdate, authorization: str = Header(...)):
91    current_user = get_current_user(authorization)
92    if truck_id not in trucks:
93        raise HTTPException(status_code=404, detail="Truck not found")
94    t = trucks[truck_id]
95    if truck.name is not None:
96        t["name"] = truck.name
97    if truck.cuisine is not None:
98        t["cuisine"] = truck.cuisine
99    if truck.location is not None:
100        t["location"] = truck.location
101    if truck.is_open is not None:
102        t["is_open"] = truck.is_open
103    if truck.owner_id is not None:
104        t["owner_id"] = truck.owner_id
105    return t
106 
107@app.get("/orders/{order_id}")
108def get_order(order_id: int):
109    if order_id not in orders:
110        raise HTTPException(status_code=404, detail="Order not found")
111    return orders[order_id]
112 
113@app.post("/orders")
114def create_order(order: OrderCreate, authorization: str = Header(...)):
115    global next_order_id
116    current_user = get_current_user(authorization)
117    if order.truck_id not in trucks:
118        raise HTTPException(status_code=404, detail="Truck not found")
119    order_id = next_order_id
120    next_order_id += 1
121    orders[order_id] = {
122        "id": order_id,
123        "truck_id": order.truck_id,
124        "items": order.items,
125        "total": order.total,
126        "user_id": current_user
127    }
128    return orders[order_id]

1fastapi
2uvicorn