Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6from datetime import datetime, timedelta
7 
8app = FastAPI()
9 
10users = {}
11profiles = {}
12matches = {}
13swap_requests = {}
14tokens = {}
15next_user_id = 1
16next_profile_id = 1
17next_match_id = 1
18next_swap_id = 1
19 
20class SignupRequest(BaseModel):
21    username: str
22    password: str
23 
24class LoginRequest(BaseModel):
25    username: str
26    password: str
27 
28class ProfileRequest(BaseModel):
29    skills_offered: List[str]
30    skills_wanted: List[str]
31 
32class SwapRequest(BaseModel):
33    topic: str
34    time: str
35    match_id: int
36 
37def generate_token():
38    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
39 
40def get_user_from_token(authorization: str = Header(None)):
41    if not authorization:
42        raise HTTPException(status_code=401, detail="No auth header")
43    token = authorization.replace("Bearer ", "")
44    if token not in tokens:
45        raise HTTPException(status_code=401, detail="Invalid token")
46    return tokens[token]
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    for uid, u in users.items():
52        if u["username"] == req.username:
53            raise HTTPException(status_code=400, detail="Username exists")
54    user_id = next_user_id
55    next_user_id += 1
56    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
57    return {"id": user_id, "username": req.username}
58 
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, u in users.items():
62        if u["username"] == req.username and u["password"] == req.password:
63            token = generate_token()
64            tokens[token] = uid
65            return {"token": token, "user_id": uid}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.post("/profile")
69def create_profile(req: ProfileRequest, authorization: str = Header(None)):
70    global next_profile_id
71    user_id = get_user_from_token(authorization)
72    profile_id = next_profile_id
73    next_profile_id += 1
74    profiles[profile_id] = {
75        "id": profile_id,
76        "user_id": user_id,
77        "skills_offered": req.skills_offered,
78        "skills_wanted": req.skills_wanted
79    }
80    return profiles[profile_id]
81 
82@app.get("/profile/{profile_id}")
83def get_profile(profile_id: int, authorization: str = Header(None)):
84    get_user_from_token(authorization)
85    if profile_id not in profiles:
86        raise HTTPException(status_code=404, detail="Profile not found")
87    return profiles[profile_id]
88 
89@app.get("/matches")
90def get_matches(authorization: str = Header(None)):
91    user_id = get_user_from_token(authorization)
92    user_profile = None
93    for pid, p in profiles.items():
94        if p["user_id"] == user_id:
95            user_profile = p
96            break
97    if not user_profile:
98        raise HTTPException(status_code=400, detail="Create a profile first")
99    
100    result = []
101    for pid, p in profiles.items():
102        if p["user_id"] == user_id:
103            continue
104        offered_overlap = set(p["skills_offered"]) & set(user_profile["skills_wanted"])
105        wanted_overlap = set(p["skills_wanted"]) & set(user_profile["skills_offered"])
106        if offered_overlap or wanted_overlap:
107            result.append({
108                "profile_id": pid,
109                "user_id": p["user_id"],
110                "matching_skills_offered": list(offered_overlap),
111                "matching_skills_wanted": list(wanted_overlap)
112            })
113    return result
114 
115@app.post("/swap/request")
116def create_swap_request(req: SwapRequest, authorization: str = Header(None)):
117    global next_swap_id
118    user_id = get_user_from_token(authorization)
119    swap_id = next_swap_id
120    next_swap_id += 1
121    swap_requests[swap_id] = {
122        "id": swap_id,
123        "requester_id": user_id,
124        "match_id": req.match_id,
125        "topic": req.topic,
126        "time": req.time,
127        "status": "pending"
128    }
129    return swap_requests[swap_id]
130 
131@app.get("/swap/request/{swap_id}")
132def get_swap_request(swap_id: int, authorization: str = Header(None)):
133    get_user_from_token(authorization)
134    if swap_id not in swap_requests:
135        raise HTTPException(status_code=404, detail="Swap request not found")
136    return swap_requests[swap_id]

1fastapi
2uvicorn