Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10tutors = {}
11lessons = {}
12session_notes = {}
13next_user_id = 1
14next_tutor_id = 1
15next_lesson_id = 1
16next_note_id = 1
17 
18def get_current_user(authorization: str = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth header")
21    token = authorization.replace("Bearer ", "")
22    user_id = tokens.get(token)
23    if not user_id:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return user_id
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class TutorCreate(BaseModel):
36    name: str
37    bio: str = ""
38 
39class LessonCreate(BaseModel):
40    tutor_id: int
41    title: str
42    content: str = ""
43 
44class SessionNoteCreate(BaseModel):
45    lesson_id: int
46    student_id: int
47    note: str
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    for u in users.values():
53        if u["username"] == req.username:
54            raise HTTPException(status_code=400, detail="Username taken")
55    uid = next_user_id
56    next_user_id += 1
57    users[uid] = {"id": uid, "username": req.username, "password": req.password}
58    return {"id": uid, "username": req.username}
59 
60@app.post("/login")
61def login(req: LoginRequest):
62    for u in users.values():
63        if u["username"] == req.username and u["password"] == req.password:
64            token = secrets.token_hex(16)
65            tokens[token] = u["id"]
66            return {"token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69@app.post("/tutors")
70def create_tutor(t: TutorCreate, authorization: str = Header(None)):
71    get_current_user(authorization)
72    global next_tutor_id
73    tid = next_tutor_id
74    next_tutor_id += 1
75    tutors[tid] = {"id": tid, "name": t.name, "bio": t.bio}
76    return tutors[tid]
77 
78@app.get("/tutors/{tutor_id}")
79def get_tutor(tutor_id: int, authorization: str = Header(None)):
80    get_current_user(authorization)
81    t = tutors.get(tutor_id)
82    if not t:
83        raise HTTPException(status_code=404, detail="Tutor not found")
84    return t
85 
86@app.post("/lessons")
87def create_lesson(l: LessonCreate, authorization: str = Header(None)):
88    get_current_user(authorization)
89    if l.tutor_id not in tutors:
90        raise HTTPException(status_code=400, detail="Tutor not found")
91    global next_lesson_id
92    lid = next_lesson_id
93    next_lesson_id += 1
94    lessons[lid] = {"id": lid, "tutor_id": l.tutor_id, "title": l.title, "content": l.content}
95    return lessons[lid]
96 
97@app.get("/lessons/{lesson_id}")
98def get_lesson(lesson_id: int, authorization: str = Header(None)):
99    get_current_user(authorization)
100    l = lessons.get(lesson_id)
101    if not l:
102        raise HTTPException(status_code=404, detail="Lesson not found")
103    return l
104 
105@app.post("/session_notes")
106def create_session_note(n: SessionNoteCreate, authorization: str = Header(None)):
107    get_current_user(authorization)
108    if n.lesson_id not in lessons:
109        raise HTTPException(status_code=400, detail="Lesson not found")
110    if n.student_id not in users:
111        raise HTTPException(status_code=400, detail="Student not found")
112    global next_note_id
113    nid = next_note_id
114    next_note_id += 1
115    session_notes[nid] = {"id": nid, "lesson_id": n.lesson_id, "student_id": n.student_id, "note": n.note}
116    return session_notes[nid]
117 
118@app.get("/session_notes/{note_id}")
119def get_session_note(note_id: int, authorization: str = Header(None)):
120    get_current_user(authorization)
121    n = session_notes.get(note_id)
122    if not n:
123        raise HTTPException(status_code=404, detail="Session note not found")
124    return n

1fastapi
2uvicorn
3pydantic