Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uuid
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11enrollments = {}
12courses = {}
13next_user_id = 1
14next_enrollment_id = 1
15next_course_id = 1
16 
17def get_current_user(authorization: str = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth token")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25class SignupRequest(BaseModel):
26    username: str
27    password: str
28 
29class LoginRequest(BaseModel):
30    username: str
31    password: str
32 
33class EnrollmentCreate(BaseModel):
34    student_id: int
35    course_id: int
36 
37class EnrollmentUpdate(BaseModel):
38    grade: Optional[str] = None
39    payment_plan: Optional[str] = None
40    access_level: Optional[str] = None
41    custom_fields: Optional[dict] = None
42 
43class CourseCreate(BaseModel):
44    title: str
45    description: Optional[str] = None
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    global next_user_id
50    user_id = next_user_id
51    next_user_id += 1
52    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
53    return {"id": user_id, "username": req.username}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for user in users.values():
58        if user["username"] == req.username and user["password"] == req.password:
59            token = str(uuid.uuid4())
60            tokens[token] = user["id"]
61            return {"token": token}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.get("/enrollments/{enrollment_id}")
65def get_enrollment(enrollment_id: int, authorization: str = Header(None)):
66    get_current_user(authorization)
67    if enrollment_id not in enrollments:
68        raise HTTPException(status_code=404, detail="Enrollment not found")
69    return enrollments[enrollment_id]
70 
71@app.post("/enrollments")
72def create_enrollment(req: EnrollmentCreate, authorization: str = Header(None)):
73    get_current_user(authorization)
74    global next_enrollment_id
75    enrollment_id = next_enrollment_id
76    next_enrollment_id += 1
77    enrollments[enrollment_id] = {
78        "id": enrollment_id,
79        "student_id": req.student_id,
80        "course_id": req.course_id,
81        "grade": None,
82        "payment_plan": None,
83        "access_level": None,
84        "custom_fields": {}
85    }
86    return enrollments[enrollment_id]
87 
88@app.put("/enrollments/{enrollment_id}")
89def update_enrollment(enrollment_id: int, req: EnrollmentUpdate, authorization: str = Header(None)):
90    get_current_user(authorization)
91    if enrollment_id not in enrollments:
92        raise HTTPException(status_code=404, detail="Enrollment not found")
93    enrollment = enrollments[enrollment_id]
94    if req.grade is not None:
95        enrollment["grade"] = req.grade
96    if req.payment_plan is not None:
97        enrollment["payment_plan"] = req.payment_plan
98    if req.access_level is not None:
99        enrollment["access_level"] = req.access_level
100    if req.custom_fields is not None:
101        enrollment["custom_fields"] = req.custom_fields
102    return enrollment
103 
104@app.get("/courses/{course_id}")
105def get_course(course_id: int, authorization: str = Header(None)):
106    get_current_user(authorization)
107    if course_id not in courses:
108        raise HTTPException(status_code=404, detail="Course not found")
109    return courses[course_id]
110 
111@app.post("/courses")
112def create_course(req: CourseCreate, authorization: str = Header(None)):
113    get_current_user(authorization)
114    global next_course_id
115    course_id = next_course_id
116    next_course_id += 1
117    courses[course_id] = {
118        "id": course_id,
119        "title": req.title,
120        "description": req.description
121    }
122    return courses[course_id]

1fastapi
2uvicorn