Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import datetime
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11activities = {}
12user_id_counter = 1
13 
14def generate_token():
15    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
16 
17def get_current_user(authorization: str = Header(None)):
18    if not authorization:
19        raise HTTPException(401, "No auth header")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(401, "Invalid token")
23    user_id = tokens[token]
24    if user_id not in users:
25        raise HTTPException(401, "User not found")
26    return users[user_id], user_id
27 
28@app.post("/signup")
29def signup(name: str, bio: str = "", age: int = 18, preferences: str = ""):
30    global user_id_counter
31    user_id = user_id_counter
32    user_id_counter += 1
33    user = {
34        "id": user_id,
35        "name": name,
36        "bio": bio,
37        "age": age,
38        "preferences": preferences,
39        "is_admin": False,
40        "is_moderator": False,
41        "created_at": datetime.now().isoformat()
42    }
43    users[user_id] = user
44    return {"id": user_id, "message": "User created"}
45 
46@app.post("/login")
47def login(user_id: int):
48    if user_id not in users:
49        raise HTTPException(404, "User not found")
50    token = generate_token()
51    tokens[token] = user_id
52    return {"token": token}
53 
54@app.get("/users/{user_id}")
55def get_user(user_id: int):
56    if user_id not in users:
57        raise HTTPException(404, "User not found")
58    return users[user_id]
59 
60@app.post("/users")
61def create_user(name: str, bio: str = "", age: int = 18, preferences: str = ""):
62    return signup(name, bio, age, preferences)
63 
64@app.patch("/users/{user_id}")
65def update_profile(
66    user_id: int,
67    name: Optional[str] = None,
68    bio: Optional[str] = None,
69    age: Optional[int] = None,
70    preferences: Optional[str] = None,
71    is_admin: Optional[bool] = None,
72    is_moderator: Optional[bool] = None,
73    authorization: str = Header(None)
74):
75    current_user, current_user_id = get_current_user(authorization)
76    
77    if user_id not in users:
78        raise HTTPException(404, "User not found")
79    
80    target_user = users[user_id]
81    
82    if is_admin is not None or is_moderator is not None:
83        if not current_user.get("is_admin"):
84            raise HTTPException(403, "Only admins can set roles")
85    
86    if name is not None:
87        target_user["name"] = name
88    if bio is not None:
89        target_user["bio"] = bio
90    if age is not None:
91        target_user["age"] = age
92    if preferences is not None:
93        target_user["preferences"] = preferences
94    if is_admin is not None:
95        target_user["is_admin"] = is_admin
96    if is_moderator is not None:
97        target_user["is_moderator"] = is_moderator
98    
99    activity_id = len(activities) + 1
100    activities[activity_id] = {
101        "id": activity_id,
102        "user_id": current_user_id,
103        "target_user_id": user_id,
104        "action": "update_profile",
105        "timestamp": datetime.now().isoformat(),
106        "changes": {
107            "name": name,
108            "bio": bio,
109            "age": age,
110            "preferences": preferences,
111            "is_admin": is_admin,
112            "is_moderator": is_moderator
113        }
114    }
115    
116    return {"message": "Profile updated", "user": target_user}
117 
118@app.get("/activities/{activity_id}")
119def get_activity(activity_id: int):
120    if activity_id not in activities:
121        raise HTTPException(404, "Activity not found")
122    return activities[activity_id]
123 
124@app.post("/activities")
125def create_activity(user_id: int, target_user_id: int, action: str):
126    activity_id = len(activities) + 1
127    activities[activity_id] = {
128        "id": activity_id,
129        "user_id": user_id,
130        "target_user_id": target_user_id,
131        "action": action,
132        "timestamp": datetime.now().isoformat()
133    }
134    return {"id": activity_id, "message": "Activity created"}

1fastapi
2uvicorn