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 = {}
10videos = {}
11video_id_counter = 1
12user_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class VideoCreate(BaseModel):
23    title: str
24    description: Optional[str] = None
25 
26class VideoSettingsUpdate(BaseModel):
27    visibility: Optional[str] = None
28    age_restriction: Optional[int] = None
29    moderator_id: Optional[int] = None
30    is_admin: Optional[bool] = None
31    tier: Optional[str] = None
32 
33def get_current_user(authorization: str = Header(None)):
34    if not authorization:
35        raise HTTPException(status_code=401, detail="Missing Authorization header")
36    token = authorization.replace("Bearer ", "")
37    if token not in tokens:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    user_id = tokens[token]
40    return users[user_id]
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global user_id_counter
45    user_id = user_id_counter
46    user_id_counter += 1
47    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
48    return {"id": user_id, "username": req.username}
49 
50@app.post("/login")
51def login(req: LoginRequest):
52    for user in users.values():
53        if user["username"] == req.username and user["password"] == req.password:
54            token = secrets.token_hex(16)
55            tokens[token] = user["id"]
56            return {"token": token}
57    raise HTTPException(status_code=401, detail="Invalid credentials")
58 
59@app.get("/videos/{video_id}")
60def get_video(video_id: int):
61    if video_id not in videos:
62        raise HTTPException(status_code=404, detail="Video not found")
63    return videos[video_id]
64 
65@app.post("/videos")
66def create_video(req: VideoCreate, authorization: str = Header(None)):
67    user = get_current_user(authorization)
68    global video_id_counter
69    video_id = video_id_counter
70    video_id_counter += 1
71    videos[video_id] = {
72        "id": video_id,
73        "title": req.title,
74        "description": req.description,
75        "visibility": "public",
76        "age_restriction": 0,
77        "moderator_id": None,
78        "is_admin": False,
79        "tier": "free",
80        "user_id": user["id"]
81    }
82    return videos[video_id]
83 
84@app.put("/videos/{video_id}/settings")
85def update_video_settings(video_id: int, req: VideoSettingsUpdate, authorization: str = Header(None)):
86    user = get_current_user(authorization)
87    if video_id not in videos:
88        raise HTTPException(status_code=404, detail="Video not found")
89    video = videos[video_id]
90    if req.visibility is not None:
91        video["visibility"] = req.visibility
92    if req.age_restriction is not None:
93        video["age_restriction"] = req.age_restriction
94    if req.moderator_id is not None:
95        video["moderator_id"] = req.moderator_id
96    if req.is_admin is not None:
97        video["is_admin"] = req.is_admin
98    if req.tier is not None:
99        video["tier"] = req.tier
100    return video

1fastapi
2uvicorn