Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10profiles = {}
11swipes = {}
12next_user_id = 1
13next_profile_id = 1
14next_swipe_id = 1
15next_token_id = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_current_user(authorization: str = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth token")
23    token = authorization.replace("Bearer ", "")
24    for uid, tok in tokens.items():
25        if tok == token:
26            return uid
27    raise HTTPException(status_code=401, detail="Invalid token")
28 
29@app.post("/signup")
30def signup(username: str, password: str, age: int, bio: str, interests: str):
31    global next_user_id
32    uid = next_user_id
33    next_user_id += 1
34    users[uid] = {
35        "id": uid,
36        "username": username,
37        "password": password,
38        "age": age,
39        "bio": bio,
40        "interests": interests.split(",")
41    }
42    token = generate_token()
43    tokens[uid] = token
44    return {"user_id": uid, "token": token}
45 
46@app.post("/login")
47def login(username: str, password: str):
48    for uid, u in users.items():
49        if u["username"] == username and u["password"] == password:
50            token = generate_token()
51            tokens[uid] = token
52            return {"token": token}
53    raise HTTPException(status_code=401, detail="Invalid credentials")
54 
55@app.get("/profiles")
56def get_profiles(authorization: str = Header(None)):
57    current_user = get_current_user(authorization)
58    current_user_data = users[current_user]
59    current_interests = set(current_user_data["interests"])
60    
61    result = []
62    for pid, p in profiles.items():
63        if p["user_id"] == current_user:
64            continue
65        shared = len(current_interests & set(p["interests"]))
66        total = len(current_interests | set(p["interests"]))
67        compatibility = round(shared / total * 100, 2) if total > 0 else 0
68        result.append({
69            "id": pid,
70            "age": p["age"],
71            "bio": p["bio"],
72            "interests": p["interests"],
73            "compatibility_score": compatibility
74        })
75    return result
76 
77@app.post("/profiles/swipe")
78def swipe(profile_id: int, direction: str, authorization: str = Header(None)):
79    current_user = get_current_user(authorization)
80    if profile_id not in profiles:
81        raise HTTPException(status_code=404, detail="Profile not found")
82    if direction not in ("like", "pass"):
83        raise HTTPException(status_code=400, detail="Direction must be 'like' or 'pass'")
84    
85    global next_swipe_id
86    sid = next_swipe_id
87    next_swipe_id += 1
88    swipes[sid] = {
89        "id": sid,
90        "user_id": current_user,
91        "profile_id": profile_id,
92        "direction": direction
93    }
94    
95    match = False
96    if direction == "like":
97        for s in swipes.values():
98            if s["user_id"] == profiles[profile_id]["user_id"] and s["profile_id"] == current_user and s["direction"] == "like":
99                match = True
100                break
101    
102    return {"match": match}
103 
104@app.get("/profiles/{profile_id}")
105def get_profile(profile_id: int, authorization: str = Header(None)):
106    current_user = get_current_user(authorization)
107    if profile_id not in profiles:
108        raise HTTPException(status_code=404, detail="Profile not found")
109    return profiles[profile_id]
110 
111@app.post("/profiles")
112def create_profile(age: int, bio: str, interests: str, authorization: str = Header(None)):
113    current_user = get_current_user(authorization)
114    global next_profile_id
115    pid = next_profile_id
116    next_profile_id += 1
117    profiles[pid] = {
118        "id": pid,
119        "user_id": current_user,
120        "age": age,
121        "bio": bio,
122        "interests": interests.split(",")
123    }
124    return {"profile_id": pid}

1fastapi
2uvicorn