Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9groups = {}
10group_id_counter = 1
11user_id_counter = 1
12 
13def authenticate(authorization: Optional[str] = Header(None)):
14    if not authorization or not authorization.startswith("Bearer "):
15        raise HTTPException(401, "Invalid auth")
16    token = authorization[7:]
17    if token not in tokens:
18        raise HTTPException(401, "Invalid token")
19    return tokens[token]
20 
21@app.post("/signup")
22def signup(username: str, password: str):
23    global user_id_counter
24    if any(u["username"] == username for u in users.values()):
25        raise HTTPException(400, "Username taken")
26    uid = user_id_counter
27    user_id_counter += 1
28    users[uid] = {"id": uid, "username": username, "password": password}
29    return {"id": uid, "username": username}
30 
31@app.post("/login")
32def login(username: str, password: str):
33    for uid, u in users.items():
34        if u["username"] == username and u["password"] == password:
35            token = secrets.token_hex(16)
36            tokens[token] = uid
37            return {"token": token}
38    raise HTTPException(401, "Invalid credentials")
39 
40@app.get("/users/{user_id}")
41def get_user(user_id: int, authorization: Optional[str] = Header(None)):
42    authenticate(authorization)
43    if user_id not in users:
44        raise HTTPException(404)
45    return users[user_id]
46 
47@app.get("/groups/{group_id}")
48def get_group(group_id: int, authorization: Optional[str] = Header(None)):
49    authenticate(authorization)
50    if group_id not in groups:
51        raise HTTPException(404)
52    return groups[group_id]
53 
54@app.post("/groups")
55def create_group(name: str, book_title: str, authorization: Optional[str] = Header(None)):
56    global group_id_counter
57    uid = authenticate(authorization)
58    gid = group_id_counter
59    group_id_counter += 1
60    groups[gid] = {
61        "id": gid,
62        "name": name,
63        "book_title": book_title,
64        "creator_id": uid,
65        "members": [uid],
66        "discussion_leaders": [],
67        "reading_schedule": {},
68        "discussion_notes": {}
69    }
70    return groups[gid]
71 
72@app.post("/groups/{group_id}/join")
73def join_group(group_id: int, authorization: Optional[str] = Header(None)):
74    uid = authenticate(authorization)
75    if group_id not in groups:
76        raise HTTPException(404)
77    if uid in groups[group_id]["members"]:
78        raise HTTPException(400, "Already a member")
79    groups[group_id]["members"].append(uid)
80    return {"status": "joined"}
81 
82@app.post("/groups/{group_id}/promote")
83def promote_member(group_id: int, member_id: int, authorization: Optional[str] = Header(None)):
84    uid = authenticate(authorization)
85    if group_id not in groups:
86        raise HTTPException(404)
87    g = groups[group_id]
88    if g["creator_id"] != uid:
89        raise HTTPException(403, "Only group creator can promote")
90    if member_id not in g["members"]:
91        raise HTTPException(400, "Member not in group")
92    if member_id in g["discussion_leaders"]:
93        raise HTTPException(400, "Already a discussion leader")
94    g["discussion_leaders"].append(member_id)
95    return {"status": "promoted", "member_id": member_id}
96 
97@app.post("/groups/{group_id}/schedule")
98def set_schedule(group_id: int, chapter: int, date: str, authorization: Optional[str] = Header(None)):
99    uid = authenticate(authorization)
100    if group_id not in groups:
101        raise HTTPException(404)
102    g = groups[group_id]
103    if uid not in g["discussion_leaders"] and uid != g["creator_id"]:
104        raise HTTPException(403, "Only discussion leader or creator can set schedule")
105    g["reading_schedule"][chapter] = date
106    return {"chapter": chapter, "date": date}
107 
108@app.post("/groups/{group_id}/notes")
109def add_note(group_id: int, chapter: int, note: str, authorization: Optional[str] = Header(None)):
110    uid = authenticate(authorization)
111    if group_id not in groups:
112        raise HTTPException(404)
113    g = groups[group_id]
114    if uid not in g["discussion_leaders"] and uid != g["creator_id"]:
115        raise HTTPException(403, "Only discussion leader or creator can add notes")
116    if chapter not in g["discussion_notes"]:
117        g["discussion_notes"][chapter] = []
118    g["discussion_notes"][chapter].append({"user_id": uid, "note": note})
119    return {"chapter": chapter, "note": note}

1fastapi
2uvicorn