Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import json
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11profiles = {}
12profile_id_counter = 1
13 
14class UserCreate(BaseModel):
15    username: str
16    password: str
17 
18class UserLogin(BaseModel):
19    username: str
20    password: str
21 
22class ProfileCreate(BaseModel):
23    name: str
24    city: str
25    age: int
26    interests: Optional[List[str]] = None
27    bio: Optional[str] = None
28 
29class ProfileUpdate(BaseModel):
30    name: Optional[str] = None
31    city: Optional[str] = None
32    age: Optional[int] = None
33    interests: Optional[List[str]] = None
34    bio: Optional[str] = None
35 
36def get_current_user(authorization: str = Header(None)):
37    if not authorization:
38        raise HTTPException(status_code=401, detail="Missing authorization header")
39    token = authorization.replace("Bearer ", "")
40    if token not in tokens:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return tokens[token]
43 
44@app.post("/signup")
45def signup(user: UserCreate):
46    if user.username in users:
47        raise HTTPException(status_code=400, detail="Username already exists")
48    users[user.username] = {"username": user.username, "password": user.password}
49    token = secrets.token_hex(16)
50    tokens[token] = user.username
51    return {"token": token}
52 
53@app.post("/login")
54def login(user: UserLogin):
55    if user.username not in users or users[user.username]["password"] != user.password:
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    token = secrets.token_hex(16)
58    tokens[token] = user.username
59    return {"token": token}
60 
61@app.get("/profiles/{profile_id}")
62def get_profile(profile_id: int, authorization: str = Header(None)):
63    user = get_current_user(authorization)
64    if profile_id not in profiles:
65        raise HTTPException(status_code=404, detail="Profile not found")
66    return profiles[profile_id]
67 
68@app.post("/profiles")
69def create_profile(profile: ProfileCreate, authorization: str = Header(None)):
70    global profile_id_counter
71    user = get_current_user(authorization)
72    profile_data = profile.dict()
73    profile_data["id"] = profile_id_counter
74    profile_data["owner"] = user
75    if profile_data.get("interests"):
76        profile_data["interests"] = json.dumps(profile_data["interests"])
77    profiles[profile_id_counter] = profile_data
78    profile_id_counter += 1
79    return profile_data
80 
81@app.get("/profiles")
82def search_profiles(
83    city: str = None,
84    min_age: int = None,
85    max_age: int = None,
86    interest: str = None,
87    authorization: str = Header(None)
88):
89    user = get_current_user(authorization)
90    results = []
91    for pid, profile in profiles.items():
92        if city and profile["city"] != city:
93            continue
94        if min_age and profile["age"] < min_age:
95            continue
96        if max_age and profile["age"] > max_age:
97            continue
98        if interest:
99            interests = json.loads(profile["interests"]) if profile.get("interests") else []
100            if interest not in interests:
101                continue
102        results.append(profile)
103    return results

1fastapi
2uvicorn