Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10interests_pool = ["hiking", "reading", "cooking", "gaming", "music", "travel", "photography", "coding", "yoga", "dancing"]
11next_user_id = 1
12next_suggestion_id = 1
13suggestions = {}
14 
15def generate_token():
16    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth header")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return tokens[token]
25 
26@app.post("/signup")
27def signup(username: str, interests: str = ""):
28    global next_user_id
29    if any(u["username"] == username for u in users.values()):
30        raise HTTPException(status_code=400, detail="Username taken")
31    user_id = next_user_id
32    next_user_id += 1
33    interest_list = [i.strip() for i in interests.split(",") if i.strip()] if interests else []
34    users[user_id] = {
35        "id": user_id,
36        "username": username,
37        "interests": interest_list,
38        "latitude": 0.0,
39        "longitude": 0.0
40    }
41    token = generate_token()
42    tokens[token] = user_id
43    return {"user_id": user_id, "token": token}
44 
45@app.post("/login")
46def login(username: str):
47    for uid, u in users.items():
48        if u["username"] == username:
49            token = generate_token()
50            tokens[token] = uid
51            return {"token": token}
52    raise HTTPException(status_code=404, detail="User not found")
53 
54@app.get("/users/{user_id}")
55def get_user(user_id: int):
56    if user_id not in users:
57        raise HTTPException(status_code=404, detail="User not found")
58    return users[user_id]
59 
60@app.post("/users")
61def create_user(username: str, interests: str = "", latitude: float = 0.0, longitude: float = 0.0):
62    global next_user_id
63    if any(u["username"] == username for u in users.values()):
64        raise HTTPException(status_code=400, detail="Username taken")
65    user_id = next_user_id
66    next_user_id += 1
67    interest_list = [i.strip() for i in interests.split(",") if i.strip()] if interests else []
68    users[user_id] = {
69        "id": user_id,
70        "username": username,
71        "interests": interest_list,
72        "latitude": latitude,
73        "longitude": longitude
74    }
75    return users[user_id]
76 
77@app.post("/interests")
78def add_interest(user_id: int, interest: str, authorization: Optional[str] = Header(None)):
79    current_user = get_current_user(authorization)
80    if user_id != current_user:
81        raise HTTPException(status_code=403, detail="Not your account")
82    if user_id not in users:
83        raise HTTPException(status_code=404, detail="User not found")
84    if interest not in users[user_id]["interests"]:
85        users[user_id]["interests"].append(interest)
86    return users[user_id]
87 
88@app.get("/interests/{interest_id}")
89def get_interest(interest_id: int):
90    return {"interest": interests_pool[interest_id - 1]} if 0 < interest_id <= len(interests_pool) else HTTPException(404, "Interest not found")
91 
92@app.post("/interests")
93def create_interest(interest: str):
94    if interest in interests_pool:
95        raise HTTPException(400, "Interest already exists")
96    interests_pool.append(interest)
97    return {"id": len(interests_pool), "interest": interest}
98 
99@app.get("/matches/suggestions")
100def get_suggestions(authorization: Optional[str] = Header(None)):
101    current_user = get_current_user(authorization)
102    user = users[current_user]
103    user_interests = set(user["interests"])
104    
105    nearby_users = []
106    for uid, u in users.items():
107        if uid == current_user:
108            continue
109        mutual_interests = list(user_interests & set(u["interests"]))
110        
111        lat_diff = abs(u["latitude"] - user["latitude"])
112        lon_diff = abs(u["longitude"] - user["longitude"])
113        is_nearby = lat_diff < 1.0 and lon_diff < 1.0
114        
115        if mutual_interests or is_nearby:
116            nearby_users.append({
117                "user_id": uid,
118                "username": u["username"],
119                "mutual_interests": mutual_interests,
120                "is_nearby": is_nearby
121            })
122    
123    global next_suggestion_id
124    suggestion_id = next_suggestion_id
125    next_suggestion_id += 1
126    suggestions[suggestion_id] = {
127        "id": suggestion_id,
128        "user_id": current_user,
129        "suggestions": nearby_users
130    }
131    return suggestions[suggestion_id]
132 
133@app.get("/suggestions/{suggestion_id}")
134def get_suggestion(suggestion_id: int):
135    if suggestion_id not in suggestions:
136        raise HTTPException(status_code=404, detail="Suggestion not found")
137    return suggestions[suggestion_id]

1fastapi
2uvicorn