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    name: str
24    interests: List[str] = []
25    preferences: dict = {}
26    account_tier: str = "free"
27 
28class ProfileUpdate(BaseModel):
29    name: Optional[str] = None
30    interests: Optional[List[str]] = None
31    preferences: Optional[dict] = None
32    account_tier: Optional[str] = None
33 
34def get_current_user(authorization: str = Header(...)):
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global user_id_counter
43    if any(u["username"] == req.username for u in users.values()):
44        raise HTTPException(status_code=400, detail="Username already exists")
45    user_id = user_id_counter
46    user_id_counter += 1
47    users[user_id] = {"username": req.username, "password": req.password}
48    return {"id": user_id, "username": req.username}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for uid, u in users.items():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = uid
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.post("/profiles")
60def create_profile(profile: ProfileCreate, authorization: str = Header(...)):
61    global profile_id_counter
62    user_id = get_current_user(authorization)
63    profile_id = profile_id_counter
64    profile_id_counter += 1
65    profiles[profile_id] = {
66        "id": profile_id,
67        "user_id": user_id,
68        "name": profile.name,
69        "interests": profile.interests,
70        "preferences": profile.preferences,
71        "account_tier": profile.account_tier
72    }
73    return profiles[profile_id]
74 
75@app.get("/profiles/{profile_id}")
76def get_profile(profile_id: int, authorization: str = Header(...)):
77    get_current_user(authorization)
78    if profile_id not in profiles:
79        raise HTTPException(status_code=404, detail="Profile not found")
80    return profiles[profile_id]
81 
82@app.put("/profiles/{profile_id}")
83def update_profile(profile_id: int, profile: ProfileUpdate, authorization: str = Header(...)):
84    user_id = get_current_user(authorization)
85    if profile_id not in profiles:
86        raise HTTPException(status_code=404, detail="Profile not found")
87    if profiles[profile_id]["user_id"] != user_id:
88        raise HTTPException(status_code=403, detail="Not your profile")
89    p = profiles[profile_id]
90    if profile.name is not None:
91        p["name"] = profile.name
92    if profile.interests is not None:
93        p["interests"] = profile.interests
94    if profile.preferences is not None:
95        p["preferences"] = profile.preferences
96    if profile.account_tier is not None:
97        p["account_tier"] = profile.account_tier
98    return p

1fastapi
2uvicorn