Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import math
6from datetime import datetime
7 
8app = FastAPI()
9 
10# In-memory stores
11users = {}
12tokens = {}
13matches = {}
14feedbacks = []
15next_user_id = 1
16next_match_id = 1
17 
18# Simple token generation
19def generate_token():
20    return ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=32))
21 
22# Auth helper
23def get_user_id(authorization: str = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth header")
26    token = authorization.replace("Bearer ", "")
27    user_id = tokens.get(token)
28    if not user_id:
29        raise HTTPException(status_code=401, detail="Invalid token")
30    return user_id
31 
32class SignupRequest(BaseModel):
33    name: str
34    email: str
35    password: str
36    latitude: float = 0.0
37    longitude: float = 0.0
38    interests: list[str] = []
39 
40class LoginRequest(BaseModel):
41    email: str
42    password: str
43 
44class Match(BaseModel):
45    user_id: int
46    status: str = "pending"  # pending/accepted/rejected
47 
48class FeedbackRequest(BaseModel):
49    match_id: int
50    thumbs: str  # up or down
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    global next_user_id
55    for u in users.values():
56        if u["email"] == req.email:
57            raise HTTPException(status_code=400, detail="Email already exists")
58    user_id = next_user_id
59    next_user_id += 1
60    users[user_id] = {
61        "id": user_id,
62        "name": req.name,
63        "email": req.email,
64        "password": req.password,
65        "latitude": req.latitude,
66        "longitude": req.longitude,
67        "interests": req.interests,
68        "created_at": datetime.now().isoformat()
69    }
70    token = generate_token()
71    tokens[token] = user_id
72    return {"user_id": user_id, "token": token}
73 
74@app.post("/login")
75def login(req: LoginRequest):
76    for uid, u in users.items():
77        if u["email"] == req.email and u["password"] == req.password:
78            token = generate_token()
79            tokens[token] = uid
80            return {"token": token, "user_id": uid}
81    raise HTTPException(status_code=401, detail="Invalid credentials")
82 
83@app.get("/users/{user_id}")
84def get_user(user_id: int, authorization: str = Header(None)):
85    get_user_id(authorization)
86    if user_id not in users:
87        raise HTTPException(status_code=404, detail="User not found")
88    return users[user_id]
89 
90@app.post("/users")
91def create_user(req: SignupRequest):
92    return signup(req)
93 
94@app.get("/matches")
95def get_matches(authorization: str = Header(None)):
96    current_user_id = get_user_id(authorization)
97    current_user = users[current_user_id]
98    
99    potential_matches = []
100    for uid, u in users.items():
101        if uid == current_user_id:
102            continue
103        
104        # Compatibility score based on shared interests
105        shared = set(current_user["interests"]) & set(u["interests"])
106        compatibility = len(shared) / max(len(current_user["interests"]), 1) * 100
107        
108        # Distance calculation (simplified)
109        lat1, lon1 = current_user["latitude"], current_user["longitude"]
110        lat2, lon2 = u["latitude"], u["longitude"]
111        distance = math.sqrt((lat1 - lat2)**2 + (lon1 - lon2)**2) * 111  # rough km
112        
113        potential_matches.append({
114            "user_id": uid,
115            "name": u["name"],
116            "compatibility_score": round(compatibility, 1),
117            "distance_km": round(distance, 1),
118            "shared_interests": list(shared)
119        })
120    
121    # Sort by compatibility descending
122    potential_matches.sort(key=lambda x: x["compatibility_score"], reverse=True)
123    return {"matches": potential_matches[:20]}  # top 20
124 
125@app.post("/matches")
126def create_match(match: Match, authorization: str = Header(None)):
127    global next_match_id
128    current_user_id = get_user_id(authorization)
129    
130    if match.user_id not in users:
131        raise HTTPException(status_code=404, detail="Target user not found")
132    
133    match_id = next_match_id
134    next_match_id += 1
135    matches[match_id] = {
136        "id": match_id,
137        "user_id": current_user_id,
138        "target_user_id": match.user_id,
139        "status": match.status,
140        "created_at": datetime.now().isoformat()
141    }
142    return {"match_id": match_id}
143 
144@app.get("/matches/{match_id}")
145def get_match(match_id: int, authorization: str = Header(None)):
146    get_user_id(authorization)
147    if match_id not in matches:
148        raise HTTPException(status_code=404, detail="Match not found")
149    return matches[match_id]
150 
151@app.post("/feedback")
152def feedback(req: FeedbackRequest):
153    if req.match_id not in matches:
154        raise HTTPException(status_code=404, detail="Match not found")
155    if req.thumbs not in ["up", "down"]:
156        raise HTTPException(status_code=400, detail="Thumbs must be 'up' or 'down'")
157    
158    feedbacks.append({
159        "match_id": req.match_id,
160        "thumbs": req.thumbs,
161        "timestamp": datetime.now().isoformat()
162    })
163    return {"status": "logged"}

1fastapi
2uvicorn
3pydantic