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 
9# In-memory stores
10users = {}
11user_id_counter = 1
12tokens = {}
13booths = {}
14booth_id_counter = 1
15 
16# Auth helper
17def get_user_from_token(authorization: str = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing Authorization header")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if user_id is None:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26# Models
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class BoothCreate(BaseModel):
36    vendor_name: str
37    product_type: str
38    location: Optional[str] = None
39    is_premium: Optional[bool] = False
40 
41class BoothUpdate(BaseModel):
42    vendor_name: Optional[str] = None
43    product_type: Optional[str] = None
44    location: Optional[str] = None
45    is_premium: Optional[bool] = None
46 
47# Signup
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global user_id_counter
51    user_id = user_id_counter
52    user_id_counter += 1
53    users[user_id] = {"username": req.username, "password": req.password}
54    token = secrets.token_hex(16)
55    tokens[token] = user_id
56    return {"user_id": user_id, "token": token}
57 
58# Login
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, u in users.items():
62        if u["username"] == req.username and u["password"] == req.password:
63            token = secrets.token_hex(16)
64            tokens[token] = uid
65            return {"token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68# Create booth
69@app.post("/booths")
70def create_booth(booth: BoothCreate, authorization: str = Header(None)):
71    get_user_from_token(authorization)
72    global booth_id_counter
73    booth_id = booth_id_counter
74    booth_id_counter += 1
75    booths[booth_id] = {
76        "id": booth_id,
77        "vendor_name": booth.vendor_name,
78        "product_type": booth.product_type,
79        "location": booth.location,
80        "is_premium": booth.is_premium
81    }
82    return booths[booth_id]
83 
84# Get booth
85@app.get("/booths/{booth_id}")
86def get_booth(booth_id: int, authorization: str = Header(None)):
87    get_user_from_token(authorization)
88    booth = booths.get(booth_id)
89    if not booth:
90        raise HTTPException(status_code=404, detail="Booth not found")
91    return booth
92 
93# Update booth (PATCH)
94@app.patch("/booths/{booth_id}")
95def update_booth(booth_id: int, updates: BoothUpdate, authorization: str = Header(None)):
96    get_user_from_token(authorization)
97    booth = booths.get(booth_id)
98    if not booth:
99        raise HTTPException(status_code=404, detail="Booth not found")
100    if updates.vendor_name is not None:
101        booth["vendor_name"] = updates.vendor_name
102    if updates.product_type is not None:
103        booth["product_type"] = updates.product_type
104    if updates.location is not None:
105        booth["location"] = updates.location
106    if updates.is_premium is not None:
107        booth["is_premium"] = updates.is_premium
108    return booth

1fastapi
2uvicorn