Browse the data

1import secrets
2from datetime import datetime, timedelta
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10spots = {}
11orders = {}
12order_id_counter = 1
13spot_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class SpotCreate(BaseModel):
24    name: str
25 
26class OrderCreate(BaseModel):
27    spot_id: int
28    start_time: str
29    end_time: str
30 
31def get_current_user(authorization: str = Header(None)):
32    if not authorization or not authorization.startswith("Bearer "):
33        raise HTTPException(status_code=401, detail="Invalid auth")
34    token = authorization.split(" ")[1]
35    if token not in tokens:
36        raise HTTPException(status_code=401, detail="Invalid token")
37    return tokens[token]
38 
39def compute_price(start_hour, end_hour):
40    total = 0
41    for h in range(start_hour, end_hour):
42        if 7 <= h < 10:
43            total += 10
44        elif 10 <= h < 17:
45            total += 5
46        elif 17 <= h < 20:
47            total += 8
48        else:
49            total += 3
50    return total
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    if req.username in users:
55        raise HTTPException(status_code=400, detail="User exists")
56    users[req.username] = req.password
57    token = secrets.token_hex(16)
58    tokens[token] = req.username
59    return {"token": token}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    if req.username not in users or users[req.username] != req.password:
64        raise HTTPException(status_code=401, detail="Bad credentials")
65    token = secrets.token_hex(16)
66    tokens[token] = req.username
67    return {"token": token}
68 
69@app.post("/spots")
70def create_spot(spot: SpotCreate, authorization: str = Header(None)):
71    get_current_user(authorization)
72    global spot_id_counter
73    spot_id = spot_id_counter
74    spot_id_counter += 1
75    spots[spot_id] = {"id": spot_id, "name": spot.name, "available": True}
76    return spots[spot_id]
77 
78@app.get("/spots/{spot_id}")
79def get_spot(spot_id: int, authorization: str = Header(None)):
80    get_current_user(authorization)
81    if spot_id not in spots:
82        raise HTTPException(status_code=404, detail="Spot not found")
83    return spots[spot_id]
84 
85@app.get("/spots")
86def list_spots(authorization: str = Header(None)):
87    get_current_user(authorization)
88    return list(spots.values())
89 
90@app.post("/orders")
91def create_order(order: OrderCreate, authorization: str = Header(None)):
92    user = get_current_user(authorization)
93    global order_id_counter
94    if order.spot_id not in spots:
95        raise HTTPException(status_code=404, detail="Spot not found")
96    if not spots[order.spot_id]["available"]:
97        raise HTTPException(status_code=400, detail="Spot not available")
98    try:
99        start = datetime.fromisoformat(order.start_time)
100        end = datetime.fromisoformat(order.end_time)
101    except:
102        raise HTTPException(status_code=400, detail="Invalid datetime format")
103    if start >= end:
104        raise HTTPException(status_code=400, detail="Start must be before end")
105    total_hours = (end - start).seconds // 3600
106    if total_hours < 1:
107        raise HTTPException(status_code=400, detail="Minimum 1 hour")
108    price = compute_price(start.hour, end.hour)
109    order_id = order_id_counter
110    order_id_counter += 1
111    spots[order.spot_id]["available"] = False
112    orders[order_id] = {
113        "id": order_id,
114        "user": user,
115        "spot_id": order.spot_id,
116        "start_time": order.start_time,
117        "end_time": order.end_time,
118        "price": price
119    }
120    return orders[order_id]
121 
122@app.get("/orders/{order_id}")
123def get_order(order_id: int, authorization: str = Header(None)):
124    get_current_user(authorization)
125    if order_id not in orders:
126        raise HTTPException(status_code=404, detail="Order not found")
127    return orders[order_id]
128 
129@app.get("/orders")
130def list_orders(authorization: str = Header(None)):
131    get_current_user(authorization)
132    return list(orders.values())

1fastapi
2uvicorn
3pydantic