Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10spaces = {}
11rentals = {}
12 
13user_counter = 0
14space_counter = 0
15rental_counter = 0
16 
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22    class Config:
23        extra = "allow"
24 
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30 
31class SpaceRequest(BaseModel):
32    name: str
33    location: str
34    price_per_day: float
35 
36    class Config:
37        extra = "allow"
38 
39 
40class RentalRequest(BaseModel):
41    space_id: int
42    days: int
43 
44    class Config:
45        extra = "allow"
46 
47 
48def get_user_from_token(authorization: Optional[str]):
49    if not authorization:
50        raise HTTPException(status_code=401, detail="Missing token")
51    token = authorization.replace("Bearer ", "").strip()
52    if token not in tokens:
53        raise HTTPException(status_code=401, detail="Invalid token")
54    return tokens[token]
55 
56 
57@app.post("/signup")
58def signup(req: SignupRequest):
59    global user_counter
60    data = req.dict()
61    if data["username"] in [u["username"] for u in users.values()]:
62        raise HTTPException(status_code=400, detail="Username taken")
63    user_counter += 1
64    user = {
65        "id": user_counter,
66        "role": "vendor",
67        "is_admin": False,
68    }
69    user.update(data)
70    user["id"] = user_counter
71    users[user_counter] = user
72    return user
73 
74 
75@app.post("/login")
76def login(req: LoginRequest):
77    for uid, user in users.items():
78        if user["username"] == req.username and user["password"] == req.password:
79            token = secrets.token_hex(16)
80            tokens[token] = uid
81            return {"token": token}
82    raise HTTPException(status_code=401, detail="Invalid credentials")
83 
84 
85@app.post("/spaces")
86def create_space(req: SpaceRequest, authorization: Optional[str] = Header(None)):
87    global space_counter
88    uid = get_user_from_token(authorization)
89    space_counter += 1
90    data = req.dict()
91    space = {"id": space_counter, "owner_id": uid}
92    space.update(data)
93    space["id"] = space_counter
94    space["owner_id"] = data.get("owner_id", uid)
95    spaces[space_counter] = space
96    return space
97 
98 
99@app.get("/spaces/{space_id}")
100def get_space(space_id: int):
101    if space_id not in spaces:
102        raise HTTPException(status_code=404, detail="Not found")
103    return spaces[space_id]
104 
105 
106@app.post("/rentals")
107def create_rental(req: RentalRequest, authorization: Optional[str] = Header(None)):
108    global rental_counter
109    uid = get_user_from_token(authorization)
110    data = req.dict()
111    if data["space_id"] not in spaces:
112        raise HTTPException(status_code=404, detail="Space not found")
113    space = spaces[data["space_id"]]
114    rental_counter += 1
115    rental = {
116        "id": rental_counter,
117        "vendor_id": uid,
118        "total_price": space["price_per_day"] * data["days"],
119    }
120    rental.update(data)
121    rental["id"] = rental_counter
122    rental["vendor_id"] = data.get("vendor_id", uid)
123    rentals[rental_counter] = rental
124    return rental
125 
126 
127@app.get("/rentals/{rental_id}")
128def get_rental(rental_id: int):
129    if rental_id not in rentals:
130        raise HTTPException(status_code=404, detail="Not found")
131    return rentals[rental_id]
132 
133 
134@app.get("/users/{user_id}")
135def get_user(user_id: int):
136    if user_id not in users:
137        raise HTTPException(status_code=404, detail="Not found")
138    return users[user_id]

1fastapi
2uvicorn
3pydantic