Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9profiles = {}
10tokens = {}
11user_id_counter = 1
12profile_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class ProfileCreate(BaseModel):
23    bio: Optional[str] = ""
24    interests: Optional[List[str]] = []
25    photos: Optional[List[str]] = []
26    privacy_settings: Optional[dict] = {}
27    verified_status: Optional[str] = ""
28    tier: Optional[str] = "free"
29 
30class ProfileUpdate(BaseModel):
31    bio: Optional[str] = None
32    interests: Optional[List[str]] = None
33    photos: Optional[List[str]] = None
34    privacy_settings: Optional[dict] = None
35    verified_status: Optional[str] = None
36    tier: Optional[str] = None
37 
38def get_user_id_from_token(authorization: str = Header(None)):
39    if not authorization:
40        raise HTTPException(status_code=401, detail="No auth header")
41    token = authorization.replace("Bearer ", "")
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_id_counter
49    user_id = user_id_counter
50    user_id_counter += 1
51    users[user_id] = {"username": req.username, "password": req.password}
52    token = secrets.token_hex(32)
53    tokens[token] = user_id
54    return {"user_id": user_id, "token": token}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(32)
61            tokens[token] = uid
62            return {"user_id": uid, "token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/profiles")
66def create_profile(profile: ProfileCreate, authorization: str = Header(None)):
67    user_id = get_user_id_from_token(authorization)
68    global profile_id_counter
69    profile_id = profile_id_counter
70    profile_id_counter += 1
71    profiles[profile_id] = {
72        "id": profile_id,
73        "user_id": user_id,
74        "bio": profile.bio,
75        "interests": profile.interests,
76        "photos": profile.photos,
77        "privacy_settings": profile.privacy_settings,
78        "verified_status": profile.verified_status,
79        "tier": profile.tier
80    }
81    return profiles[profile_id]
82 
83@app.get("/profiles/{profile_id}")
84def get_profile(profile_id: int, authorization: str = Header(None)):
85    user_id = get_user_id_from_token(authorization)
86    if profile_id not in profiles:
87        raise HTTPException(status_code=404, detail="Profile not found")
88    return profiles[profile_id]
89 
90@app.patch("/profiles/{profile_id}")
91def update_profile(profile_id: int, update: ProfileUpdate, authorization: str = Header(None)):
92    user_id = get_user_id_from_token(authorization)
93    if profile_id not in profiles:
94        raise HTTPException(status_code=404, detail="Profile not found")
95    profile = profiles[profile_id]
96    if update.bio is not None:
97        profile["bio"] = update.bio
98    if update.interests is not None:
99        profile["interests"] = update.interests
100    if update.photos is not None:
101        profile["photos"] = update.photos
102    if update.privacy_settings is not None:
103        profile["privacy_settings"] = update.privacy_settings
104    if update.verified_status is not None:
105        profile["verified_status"] = update.verified_status
106    if update.tier is not None:
107        profile["tier"] = update.tier
108    return profile

1fastapi
2uvicorn