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 = {}
11sizes_db = {}
12fit_logs = {}
13next_user_id = 1
14next_size_id = 1
15next_fit_log_id = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth token")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid auth token")
24    return user_id
25 
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class SizeRequest(BaseModel):
35    foot_length_cm: float
36    foot_width_cm: float
37    brand: str
38 
39class FitLogRequest(BaseModel):
40    shoe_size: str
41    brand: str
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="User already exists")
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    return {"id": user_id, "username": req.username}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for u in users.values():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = u["id"]
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/sizes/{size_id}")
64def get_size(size_id: int, authorization: Optional[str] = Header(None)):
65    get_current_user(authorization)
66    size = sizes_db.get(size_id)
67    if not size:
68        raise HTTPException(status_code=404, detail="Size not found")
69    return size
70 
71@app.post("/sizes")
72def recommend_size(req: SizeRequest, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_size_id
75    # Dumb conversion: length to EU size
76    length_cm = req.foot_length_cm
77    width_cm = req.foot_width_cm
78    brand = req.brand.lower()
79    # Very rough size table
80    if length_cm < 22:
81        eu_size = 35
82    elif length_cm < 22.5:
83        eu_size = 36
84    elif length_cm < 23.5:
85        eu_size = 37
86    elif length_cm < 24.5:
87        eu_size = 38
88    elif length_cm < 25.5:
89        eu_size = 39
90    elif length_cm < 26.5:
91        eu_size = 40
92    elif length_cm < 27.5:
93        eu_size = 41
94    elif length_cm < 28.5:
95        eu_size = 42
96    elif length_cm < 29.5:
97        eu_size = 43
98    else:
99        eu_size = 44
100    # Adjust for width (wide = half size up)
101    if width_cm > 10:
102        eu_size += 0.5
103    # Brand adjustment
104    if brand == "nike":
105        eu_size -= 0.5
106    elif brand == "adidas":
107        eu_size += 0.5
108    elif brand == "new balance":
109        eu_size += 1
110    size_id = next_size_id
111    next_size_id += 1
112    result = {
113        "id": size_id,
114        "recommended_size": f"{eu_size:.1f}",
115        "brand": req.brand,
116        "foot_length_cm": length_cm,
117        "foot_width_cm": width_cm
118    }
119    sizes_db[size_id] = result
120    return result
121 
122@app.get("/fit-log/{fit_log_id}")
123def get_fit_log(fit_log_id: int, authorization: Optional[str] = Header(None)):
124    get_current_user(authorization)
125    log = fit_logs.get(fit_log_id)
126    if not log:
127        raise HTTPException(status_code=404, detail="Fit log not found")
128    return log
129 
130@app.post("/fit-log")
131def create_fit_log(req: FitLogRequest, authorization: Optional[str] = Header(None)):
132    user_id = get_current_user(authorization)
133    global next_fit_log_id
134    log_id = next_fit_log_id
135    next_fit_log_id += 1
136    log = {
137        "id": log_id,
138        "user_id": user_id,
139        "shoe_size": req.shoe_size,
140        "brand": req.brand
141    }
142    fit_logs[log_id] = log
143    return log

1fastapi
2uvicorn
3pydantic