Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import random
4import string
5import json
6from typing import Optional
7 
8app = FastAPI()
9 
10users = {}
11profiles = {}
12tokens = {}
13next_user_id = 1
14next_profile_id = 1
15next_token_id = 1
16 
17def generate_token():
18    global next_token_id
19    token = f"tok_{next_token_id}_{''.join(random.choices(string.ascii_letters + string.digits, k=16))}"
20    next_token_id += 1
21    return token
22 
23def get_current_user(authorization: str = Header(...)):
24    if not authorization.startswith("Bearer "):
25        raise HTTPException(status_code=401, detail="Invalid auth header")
26    token = authorization[7:]
27    if token not in tokens:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return tokens[token]
30 
31@app.post("/signup")
32def signup(username: str, password: str):
33    global next_user_id
34    user_id = next_user_id
35    next_user_id += 1
36    users[user_id] = {"id": user_id, "username": username, "password": password}
37    token = generate_token()
38    tokens[token] = user_id
39    return {"user_id": user_id, "token": token}
40 
41@app.post("/login")
42def login(username: str, password: str):
43    for uid, u in users.items():
44        if u["username"] == username and u["password"] == password:
45            token = generate_token()
46            tokens[token] = uid
47            return {"token": token}
48    raise HTTPException(status_code=401, detail="Invalid credentials")
49 
50@app.get("/profiles/{profile_id}")
51def get_profile(profile_id: int, authorization: str = Header(...)):
52    get_current_user(authorization)
53    if profile_id not in profiles:
54        raise HTTPException(status_code=404, detail="Profile not found")
55    return profiles[profile_id]
56 
57@app.post("/profiles")
58def create_profile(
59    name: str,
60    age: int,
61    interests: str,
62    authorization: str = Header(...)
63):
64    global next_profile_id
65    user_id = get_current_user(authorization)
66    profile_id = next_profile_id
67    next_profile_id += 1
68    profiles[profile_id] = {
69        "id": profile_id,
70        "user_id": user_id,
71        "name": name,
72        "age": age,
73        "interests": interests,
74        "last_active": datetime.utcnow().isoformat()
75    }
76    return profiles[profile_id]
77 
78@app.get("/profiles")
79def search_profiles(
80    age_min: Optional[int] = None,
81    interests: Optional[str] = None,
82    authorization: str = Header(...)
83):
84    get_current_user(authorization)
85    results = []
86    for pid, p in profiles.items():
87        if age_min is not None and p["age"] < age_min:
88            continue
89        if interests is not None:
90            profile_interests = json.loads(p["interests"]) if isinstance(p["interests"], str) else p.get("interests", [])
91            if isinstance(profile_interests, str):
92                profile_interests = [profile_interests]
93            query_interests = json.loads(interests) if isinstance(interests, str) else [interests]
94            if not any(i in profile_interests for i in query_interests):
95                continue
96        compatibility = random.randint(50, 100)
97        results.append({
98            "profile": p,
99            "compatibility_score": compatibility,
100            "last_active": p["last_active"]
101        })
102    results.sort(key=lambda x: x["compatibility_score"], reverse=True)
103    return results

1fastapi
2uvicorn