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 = {}
10tokens = {}
11spots = {}
12orders = {}
13next_user_id = 1
14next_spot_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 SpotCreate(BaseModel):
26    location: str
27    price: float
28    status: str = "available"
29    owner_id: int
30 
31class SpotUpdate(BaseModel):
32    location: Optional[str] = None
33    price: Optional[float] = None
34    status: Optional[str] = None
35    owner_id: Optional[int] = None
36 
37class OrderCreate(BaseModel):
38    spot_id: int
39    renter_id: int
40 
41def get_current_user(authorization: str = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="Missing auth header")
44    token = authorization.replace("Bearer ", "")
45    if token not in tokens:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    return tokens[token]
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    for u in users.values():
53        if u["username"] == req.username:
54            raise HTTPException(status_code=400, detail="Username taken")
55    user_id = next_user_id
56    next_user_id += 1
57    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
58    return {"id": user_id, "username": req.username}
59 
60@app.post("/login")
61def login(req: LoginRequest):
62    for u in users.values():
63        if u["username"] == req.username and u["password"] == req.password:
64            token = secrets.token_hex(16)
65            tokens[token] = u["id"]
66            return {"token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69@app.get("/spots/{spot_id}")
70def get_spot(spot_id: int):
71    if spot_id not in spots:
72        raise HTTPException(status_code=404, detail="Spot not found")
73    return spots[spot_id]
74 
75@app.post("/spots")
76def create_spot(spot: SpotCreate, authorization: str = Header(None)):
77    global next_spot_id
78    get_current_user(authorization)
79    spot_id = next_spot_id
80    next_spot_id += 1
81    spots[spot_id] = {"id": spot_id, "location": spot.location, "price": spot.price, "status": spot.status, "owner_id": spot.owner_id}
82    return spots[spot_id]
83 
84@app.patch("/spots/{spot_id}")
85def update_spot(spot_id: int, update: SpotUpdate, authorization: str = Header(None)):
86    get_current_user(authorization)
87    if spot_id not in spots:
88        raise HTTPException(status_code=404, detail="Spot not found")
89    spot = spots[spot_id]
90    if update.location is not None:
91        spot["location"] = update.location
92    if update.price is not None:
93        spot["price"] = update.price
94    if update.status is not None:
95        spot["status"] = update.status
96    if update.owner_id is not None:
97        spot["owner_id"] = update.owner_id
98    return spot
99 
100@app.get("/orders/{order_id}")
101def get_order(order_id: int):
102    if order_id not in orders:
103        raise HTTPException(status_code=404, detail="Order not found")
104    return orders[order_id]
105 
106@app.post("/orders")
107def create_order(order: OrderCreate, authorization: str = Header(None)):
108    global next_order_id
109    get_current_user(authorization)
110    if order.spot_id not in spots:
111        raise HTTPException(status_code=400, detail="Spot not found")
112    if spots[order.spot_id]["status"] != "available":
113        raise HTTPException(status_code=400, detail="Spot not available")
114    order_id = next_order_id
115    next_order_id += 1
116    orders[order_id] = {"id": order_id, "spot_id": order.spot_id, "renter_id": order.renter_id}
117    spots[order.spot_id]["status"] = "booked"
118    return orders[order_id]
119 
120if __name__ == "__main__":
121    uvicorn.run(app, host="127.0.0.1", port=8000)

1fastapi
2uvicorn