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 = {}
10profiles = {}
11profile_id_counter = 1
12 
13def get_current_user(authorization: str = Header(None)):
14    if not authorization:
15        raise HTTPException(status_code=401, detail="Missing auth header")
16    token = authorization.replace("Bearer ", "")
17    if token not in tokens:
18        raise HTTPException(status_code=401, detail="Invalid token")
19    return tokens[token]
20 
21class SignupRequest(BaseModel):
22    username: str
23    password: str
24 
25class LoginRequest(BaseModel):
26    username: str
27    password: str
28 
29class ProfileUpdate(BaseModel):
30    bio: Optional[str] = None
31    interests: Optional[str] = None
32    age_range_min: Optional[int] = None
33    age_range_max: Optional[int] = None
34    tier: Optional[str] = None
35    is_verified: Optional[bool] = None
36 
37class ProfileCreate(BaseModel):
38    display_name: str
39    bio: Optional[str] = None
40    interests: Optional[str] = None
41    age_range_min: Optional[int] = None
42    age_range_max: Optional[int] = None
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="Username taken")
48    users[req.username] = req.password
49    token = secrets.token_hex(32)
50    tokens[token] = req.username
51    return {"token": token}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    if req.username not in users or users[req.username] != req.password:
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    token = secrets.token_hex(32)
58    tokens[token] = req.username
59    return {"token": token}
60 
61@app.get("/profiles/{profile_id}")
62def get_profile(profile_id: int):
63    if profile_id not in profiles:
64        raise HTTPException(status_code=404, detail="Profile not found")
65    return profiles[profile_id]
66 
67@app.post("/profiles")
68def create_profile(profile: ProfileCreate, authorization: str = Header(None)):
69    user = get_current_user(authorization)
70    global profile_id_counter
71    profile_id = profile_id_counter
72    profile_id_counter += 1
73    profiles[profile_id] = {
74        "id": profile_id,
75        "owner": user,
76        "display_name": profile.display_name,
77        "bio": profile.bio,
78        "interests": profile.interests,
79        "age_range_min": profile.age_range_min,
80        "age_range_max": profile.age_range_max,
81        "tier": "free",
82        "is_verified": False
83    }
84    return profiles[profile_id]
85 
86@app.put("/profiles/{profile_id}")
87def update_profile(profile_id: int, update: ProfileUpdate, authorization: str = Header(None)):
88    user = get_current_user(authorization)
89    if profile_id not in profiles:
90        raise HTTPException(status_code=404, detail="Profile not found")
91    if profiles[profile_id]["owner"] != user:
92        raise HTTPException(status_code=403, detail="Not your profile")
93    profile = profiles[profile_id]
94    if update.bio is not None:
95        profile["bio"] = update.bio
96    if update.interests is not None:
97        profile["interests"] = update.interests
98    if update.age_range_min is not None:
99        profile["age_range_min"] = update.age_range_min
100    if update.age_range_max is not None:
101        profile["age_range_max"] = update.age_range_max
102    if update.tier is not None:
103        profile["tier"] = update.tier
104    if update.is_verified is not None:
105        profile["is_verified"] = update.is_verified
106    return profile

1fastapi
2uvicorn