Browse the data

1import secrets
2from fastapi import FastAPI, HTTPException, Header
3from pydantic import BaseModel
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9courses = {}
10modules = {}
11files = {}
12user_counter = 1
13course_counter = 1
14module_counter = 1
15file_counter = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class CourseCreate(BaseModel):
26    title: str
27 
28class ModuleCreate(BaseModel):
29    title: str
30 
31class FileCreate(BaseModel):
32    content: str
33 
34def get_current_user(authorization: str = Header(None)):
35    if not authorization or not authorization.startswith("Bearer "):
36        raise HTTPException(status_code=401, detail="Invalid token")
37    token = authorization[7:]
38    user_id = tokens.get(token)
39    if user_id is None:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return user_id
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global user_counter
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="User already exists")
48    user_id = user_counter
49    user_counter += 1
50    users[req.username] = {"id": user_id, "password": req.password}
51    return {"id": user_id, "username": req.username}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    user = users.get(req.username)
56    if not user or user["password"] != req.password:
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = secrets.token_hex(16)
59    tokens[token] = user["id"]
60    return {"token": token}
61 
62@app.post("/courses")
63def create_course(req: CourseCreate, authorization: str = Header(None)):
64    get_current_user(authorization)
65    global course_counter
66    course_id = course_counter
67    course_counter += 1
68    courses[course_id] = {"id": course_id, "title": req.title, "modules": {}}
69    return {"id": course_id, "title": req.title}
70 
71@app.get("/courses/{course_id}")
72def get_course(course_id: int, authorization: str = Header(None)):
73    get_current_user(authorization)
74    course = courses.get(course_id)
75    if not course:
76        raise HTTPException(status_code=404, detail="Course not found")
77    return course
78 
79@app.post("/courses/{course_id}/modules")
80def create_module(course_id: int, req: ModuleCreate, authorization: str = Header(None)):
81    get_current_user(authorization)
82    course = courses.get(course_id)
83    if not course:
84        raise HTTPException(status_code=404, detail="Course not found")
85    global module_counter
86    module_id = module_counter
87    module_counter += 1
88    module = {"id": module_id, "title": req.title, "files": {}}
89    course["modules"][module_id] = module
90    modules[module_id] = module
91    return {"id": module_id, "title": req.title}
92 
93@app.get("/courses/{course_id}/modules/{module_id}")
94def get_module(course_id: int, module_id: int, authorization: str = Header(None)):
95    get_current_user(authorization)
96    course = courses.get(course_id)
97    if not course:
98        raise HTTPException(status_code=404, detail="Course not found")
99    module = course["modules"].get(module_id)
100    if not module:
101        raise HTTPException(status_code=404, detail="Module not found")
102    return module
103 
104@app.post("/courses/{course_id}/modules/{module_id}/files")
105def create_file(course_id: int, module_id: int, req: FileCreate, authorization: str = Header(None)):
106    get_current_user(authorization)
107    course = courses.get(course_id)
108    if not course:
109        raise HTTPException(status_code=404, detail="Course not found")
110    module = course["modules"].get(module_id)
111    if not module:
112        raise HTTPException(status_code=404, detail="Module not found")
113    global file_counter
114    file_id = file_counter
115    file_counter += 1
116    file_obj = {"id": file_id, "content": req.content}
117    module["files"][file_id] = file_obj
118    files[file_id] = file_obj
119    return {"id": file_id}
120 
121@app.get("/courses/{course_id}/modules/{module_id}/files/{file_id}")
122def get_file(course_id: int, module_id: int, file_id: int, authorization: str = Header(None)):
123    get_current_user(authorization)
124    course = courses.get(course_id)
125    if not course:
126        raise HTTPException(status_code=404, detail="Course not found")
127    module = course["modules"].get(module_id)
128    if not module:
129        raise HTTPException(status_code=404, detail="Module not found")
130    file_obj = module["files"].get(file_id)
131    if not file_obj:
132        raise HTTPException(status_code=404, detail="File not found")
133    return file_obj

1fastapi
2uvicorn