Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10clubs = {}
11club_members = {}
12club_moderators = {}
13reading_progress = {}
14discussion_threads = {}
15next_user_id = 1
16next_club_id = 1
17next_thread_id = 1
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global next_user_id
30    for u in users.values():
31        if u["username"] == username:
32            raise HTTPException(status_code=400, detail="Username taken")
33    user_id = next_user_id
34    next_user_id += 1
35    users[user_id] = {"id": user_id, "username": username, "password": password}
36    return {"id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for u in users.values():
41        if u["username"] == username and u["password"] == password:
42            token = secrets.token_hex(16)
43            tokens[token] = u["id"]
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.get("/user/{user_id}")
48def get_user(user_id: int):
49    if user_id not in users:
50        raise HTTPException(status_code=404, detail="User not found")
51    return users[user_id]
52 
53@app.post("/club")
54def create_club(name: str, description: str = "", authorization: Optional[str] = Header(None)):
55    global next_club_id
56    user_id = get_current_user(authorization)
57    club_id = next_club_id
58    next_club_id += 1
59    clubs[club_id] = {"id": club_id, "name": name, "description": description, "created_by": user_id}
60    club_members[club_id] = [user_id]
61    club_moderators[club_id] = [user_id]
62    reading_progress[club_id] = {}
63    discussion_threads[club_id] = []
64    return clubs[club_id]
65 
66@app.get("/club/{club_id}")
67def get_club(club_id: int):
68    if club_id not in clubs:
69        raise HTTPException(status_code=404, detail="Club not found")
70    return clubs[club_id]
71 
72@app.post("/club/{club_id}/join")
73def join_club(club_id: int, authorization: Optional[str] = Header(None)):
74    user_id = get_current_user(authorization)
75    if club_id not in clubs:
76        raise HTTPException(status_code=404, detail="Club not found")
77    if user_id in club_members.get(club_id, []):
78        raise HTTPException(status_code=400, detail="Already a member")
79    club_members[club_id].append(user_id)
80    reading_progress[club_id][user_id] = 0
81    return {"message": "Joined club"}
82 
83@app.get("/club/{club_id}/members")
84def get_club_members(club_id: int):
85    if club_id not in clubs:
86        raise HTTPException(status_code=404, detail="Club not found")
87    return {"members": club_members.get(club_id, [])}
88 
89@app.post("/club/{club_id}/promote")
90def promote_member(club_id: int, user_id: int, authorization: Optional[str] = Header(None)):
91    requester = get_current_user(authorization)
92    if club_id not in clubs:
93        raise HTTPException(status_code=404, detail="Club not found")
94    if requester not in club_moderators.get(club_id, []):
95        raise HTTPException(status_code=403, detail="Not a moderator")
96    if user_id not in club_members.get(club_id, []):
97        raise HTTPException(status_code=400, detail="User not a member")
98    if user_id in club_moderators[club_id]:
99        raise HTTPException(status_code=400, detail="Already a moderator")
100    club_moderators[club_id].append(user_id)
101    return {"message": "Promoted to co-mod"}
102 
103@app.post("/club/{club_id}/progress")
104def update_progress(club_id: int, page: int, authorization: Optional[str] = Header(None)):
105    user_id = get_current_user(authorization)
106    if club_id not in clubs:
107        raise HTTPException(status_code=404, detail="Club not found")
108    if user_id not in club_members.get(club_id, []):
109        raise HTTPException(status_code=403, detail="Not a member")
110    reading_progress[club_id][user_id] = page
111    return {"message": "Progress updated"}
112 
113@app.get("/club/{club_id}/progress")
114def get_progress(club_id: int):
115    if club_id not in clubs:
116        raise HTTPException(status_code=404, detail="Club not found")
117    return {"progress": reading_progress.get(club_id, {})}
118 
119@app.post("/club/{club_id}/thread")
120def create_thread(club_id: int, title: str, content: str, authorization: Optional[str] = Header(None)):
121    global next_thread_id
122    user_id = get_current_user(authorization)
123    if club_id not in clubs:
124        raise HTTPException(status_code=404, detail="Club not found")
125    if user_id not in club_members.get(club_id, []):
126        raise HTTPException(status_code=403, detail="Not a member")
127    thread_id = next_thread_id
128    next_thread_id += 1
129    thread = {
130        "id": thread_id,
131        "club_id": club_id,
132        "title": title,
133        "content": content,
134        "author_id": user_id,
135        "created_at": datetime.datetime.utcnow().isoformat(),
136        "replies": []
137    }
138    discussion_threads[club_id].append(thread)
139    return thread
140 
141@app.get("/club/{club_id}/threads")
142def get_threads(club_id: int):
143    if club_id not in clubs:
144        raise HTTPException(status_code=404, detail="Club not found")
145    return {"threads": discussion_threads.get(club_id, [])}
146 
147@app.post("/club/{club_id}/thread/{thread_id}/reply")
148def reply_to_thread(club_id: int, thread_id: int, content: str, authorization: Optional[str] = Header(None)):
149    user_id = get_current_user(authorization)
150    if club_id not in clubs:
151        raise HTTPException(status_code=404, detail="Club not found")
152    if user_id not in club_members.get(club_id, []):
153        raise HTTPException(status_code=403, detail="Not a member")
154    threads = discussion_threads.get(club_id, [])
155    thread = next((t for t in threads if t["id"] == thread_id), None)
156    if not thread:
157        raise HTTPException(status_code=404, detail="Thread not found")
158    reply = {
159        "author_id": user_id,
160        "content": content,
161        "created_at": datetime.datetime.utcnow().isoformat()
162    }
163    thread["replies"].append(reply)
164    return reply

1fastapi
2uvicorn