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
7 
8app = FastAPI()
9 
10# In-memory storage
11users = {}
12tokens = {}
13groups = {}
14group_id_counter = 1
15 
16# Simple token generation
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20# Auth dependency
21def get_user_from_token(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing authorization header")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29# Models
30class SignupRequest(BaseModel):
31    username: str
32    password: str
33 
34class LoginRequest(BaseModel):
35    username: str
36    password: str
37 
38class GroupCreate(BaseModel):
39    subject: str
40    level: str
41    meeting_time: str
42    max_members: int
43    current_count: int = 0
44 
45class GroupUpdate(BaseModel):
46    subject: Optional[str] = None
47    level: Optional[str] = None
48    meeting_time: Optional[str] = None
49    max_members: Optional[int] = None
50    current_count: Optional[int] = None
51 
52# Auth endpoints
53@app.post("/signup")
54def signup(req: SignupRequest):
55    if req.username in users:
56        raise HTTPException(status_code=400, detail="User already exists")
57    users[req.username] = {"username": req.username, "password": req.password}
58    token = generate_token()
59    tokens[token] = req.username
60    return {"token": token}
61 
62@app.post("/login")
63def login(req: LoginRequest):
64    if req.username not in users or users[req.username]["password"] != req.password:
65        raise HTTPException(status_code=401, detail="Invalid credentials")
66    token = generate_token()
67    tokens[token] = req.username
68    return {"token": token}
69 
70# Groups endpoints
71@app.get("/groups/{group_id}")
72def get_group(group_id: int, authorization: Optional[str] = Header(None)):
73    get_user_from_token(authorization)
74    if group_id not in groups:
75        raise HTTPException(status_code=404, detail="Group not found")
76    return groups[group_id]
77 
78@app.post("/groups")
79def create_group(group: GroupCreate, authorization: Optional[str] = Header(None)):
80    get_user_from_token(authorization)
81    global group_id_counter
82    gid = group_id_counter
83    group_id_counter += 1
84    groups[gid] = {
85        "id": gid,
86        "subject": group.subject,
87        "level": group.level,
88        "meeting_time": group.meeting_time,
89        "max_members": group.max_members,
90        "current_count": group.current_count
91    }
92    return groups[gid]
93 
94@app.get("/groups")
95def list_groups(
96    subject: Optional[str] = None,
97    level: Optional[str] = None,
98    day_of_week: Optional[str] = None,
99    available_spots: Optional[bool] = None,
100    authorization: Optional[str] = Header(None)
101):
102    get_user_from_token(authorization)
103    result = []
104    for gid, group in groups.items():
105        if subject and group["subject"].lower() != subject.lower():
106            continue
107        if level and group["level"].lower() != level.lower():
108            continue
109        if day_of_week:
110            try:
111                dt = datetime.strptime(group["meeting_time"], "%Y-%m-%d %H:%M")
112                day_names = ["monday", "tuesday", "wednesday", "thursday", "friday", "saturday", "sunday"]
113                if day_names[dt.weekday()] != day_of_week.lower():
114                    continue
115            except:
116                continue
117        if available_spots is not None and available_spots:
118            if group["current_count"] >= group["max_members"]:
119                continue
120        result.append(group)
121    return result
122 
123@app.put("/groups/{group_id}")
124def update_group(group_id: int, group_update: GroupUpdate, authorization: Optional[str] = Header(None)):
125    get_user_from_token(authorization)
126    if group_id not in groups:
127        raise HTTPException(status_code=404, detail="Group not found")
128    group = groups[group_id]
129    if group_update.subject is not None:
130        group["subject"] = group_update.subject
131    if group_update.level is not None:
132        group["level"] = group_update.level
133    if group_update.meeting_time is not None:
134        group["meeting_time"] = group_update.meeting_time
135    if group_update.max_members is not None:
136        group["max_members"] = group_update.max_members
137    if group_update.current_count is not None:
138        group["current_count"] = group_update.current_count
139    return group
140 
141@app.delete("/groups/{group_id}")
142def delete_group(group_id: int, authorization: Optional[str] = Header(None)):
143    get_user_from_token(authorization)
144    if group_id not in groups:
145        raise HTTPException(status_code=404, detail="Group not found")
146    del groups[group_id]
147    return {"ok": True}

1fastapi
2uvicorn