Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6import math
7from datetime import datetime
8 
9app = FastAPI()
10 
11users = {}
12tokens = {}
13profiles = {}
14swipes = {}
15matches = {}
16next_user_id = 1
17next_profile_id = 1
18next_match_id = 1
19 
20def generate_token():
21    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
22 
23def get_current_user(authorization: str = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="No auth token")
26    token = authorization.replace("Bearer ", "")
27    if token not in tokens:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return tokens[token]
30 
31class SignupRequest(BaseModel):
32    username: str
33    password: str
34    age: int
35    bio: str
36    interests: list[str]
37    latitude: float
38    longitude: float
39 
40class LoginRequest(BaseModel):
41    username: str
42    password: str
43 
44class SwipeRequest(BaseModel):
45    profile_id: int
46    direction: str  # "like" or "pass"
47 
48class ProfileCreate(BaseModel):
49    user_id: int
50    age: int
51    bio: str
52    interests: list[str]
53    latitude: float
54    longitude: float
55 
56@app.post("/signup")
57def signup(req: SignupRequest):
58    global next_user_id
59    user_id = next_user_id
60    next_user_id += 1
61    users[user_id] = {
62        "id": user_id,
63        "username": req.username,
64        "password": req.password,
65        "age": req.age,
66        "bio": req.bio,
67        "interests": req.interests,
68        "latitude": req.latitude,
69        "longitude": req.longitude
70    }
71    return {"user_id": user_id}
72 
73@app.post("/login")
74def login(req: LoginRequest):
75    for uid, u in users.items():
76        if u["username"] == req.username and u["password"] == req.password:
77            token = generate_token()
78            tokens[token] = uid
79            return {"token": token}
80    raise HTTPException(status_code=401, detail="Invalid credentials")
81 
82@app.get("/profiles")
83def get_profiles(authorization: str = Header(None)):
84    current_user_id = get_current_user(authorization)
85    current_user = users[current_user_id]
86    nearby = []
87    for pid, p in profiles.items():
88        if p["user_id"] == current_user_id:
89            continue
90        dist = haversine(current_user["latitude"], current_user["longitude"], p["latitude"], p["longitude"])
91        if dist < 50:
92            nearby.append({
93                "id": pid,
94                "age": p["age"],
95                "bio": p["bio"],
96                "interests": p["interests"],
97                "distance_km": round(dist, 2)
98            })
99    return nearby
100 
101def haversine(lat1, lon1, lat2, lon2):
102    R = 6371
103    dlat = math.radians(lat2 - lat1)
104    dlon = math.radians(lon2 - lon1)
105    a = math.sin(dlat/2)**2 + math.cos(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.sin(dlon/2)**2
106    c = 2 * math.atan2(math.sqrt(a), math.sqrt(1-a))
107    return R * c
108 
109@app.post("/swipe")
110def swipe(req: SwipeRequest, authorization: str = Header(None)):
111    current_user_id = get_current_user(authorization)
112    if req.profile_id not in profiles:
113        raise HTTPException(status_code=404, detail="Profile not found")
114    if req.direction not in ["like", "pass"]:
115        raise HTTPException(status_code=400, detail="Direction must be like or pass")
116    
117    swiped_profile = profiles[req.profile_id]
118    if current_user_id not in swipes:
119        swipes[current_user_id] = {}
120    swipes[current_user_id][req.profile_id] = req.direction
121 
122    if req.direction == "like":
123        other_user_id = swiped_profile["user_id"]
124        if other_user_id in swipes and current_user_id in swipes[other_user_id] and swipes[other_user_id][current_user_id] == "like":
125            global next_match_id
126            match_id = next_match_id
127            next_match_id += 1
128            user1_interests = set(users[current_user_id]["interests"])
129            user2_interests = set(users[other_user_id]["interests"])
130            if len(user1_interests) + len(user2_interests) > 0:
131                compatibility = round(len(user1_interests & user2_interests) / len(user1_interests | user2_interests) * 100, 2)
132            else:
133                compatibility = 0
134            matches[match_id] = {
135                "id": match_id,
136                "user1_id": current_user_id,
137                "user2_id": other_user_id,
138                "compatibility": compatibility,
139                "created_at": datetime.now().isoformat()
140            }
141            return {"match_id": match_id, "compatibility": compatibility}
142 
143    return {"status": "ok"}
144 
145@app.get("/matches")
146def get_matches(authorization: str = Header(None)):
147    current_user_id = get_current_user(authorization)
148    user_matches = []
149    for mid, m in matches.items():
150        if m["user1_id"] == current_user_id or m["user2_id"] == current_user_id:
151            user_matches.append(m)
152    return user_matches
153 
154@app.get("/profile/{profile_id}")
155def get_profile(profile_id: int, authorization: str = Header(None)):
156    current_user_id = get_current_user(authorization)
157    if profile_id not in profiles:
158        raise HTTPException(status_code=404, detail="Profile not found")
159    return profiles[profile_id]
160 
161@app.post("/profile")
162def create_profile(req: ProfileCreate, authorization: str = Header(None)):
163    current_user_id = get_current_user(authorization)
164    global next_profile_id
165    profile_id = next_profile_id
166    next_profile_id += 1
167    profiles[profile_id] = {
168        "id": profile_id,
169        "user_id": req.user_id,
170        "age": req.age,
171        "bio": req.bio,
172        "interests": req.interests,
173        "latitude": req.latitude,
174        "longitude": req.longitude
175    }
176    return {"profile_id": profile_id}
177 
178@app.get("/user/{user_id}")
179def get_user(user_id: int, authorization: str = Header(None)):
180    current_user_id = get_current_user(authorization)
181    if user_id not in users:
182        raise HTTPException(status_code=404, detail="User not found")
183    u = users[user_id]
184    return {"id": u["id"], "username": u["username"], "age": u["age"], "bio": u["bio"], "interests": u["interests"]}
185 
186@app.post("/user")
187def create_user(req: SignupRequest):
188    global next_user_id
189    user_id = next_user_id
190    next_user_id += 1
191    users[user_id] = {
192        "id": user_id,
193        "username": req.username,
194        "password": req.password,
195        "age": req.age,
196        "bio": req.bio,
197        "interests": req.interests,
198        "latitude": req.latitude,
199        "longitude": req.longitude
200    }
201    return {"user_id": user_id}

1fastapi
2uvicorn