Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5 
6app = FastAPI()
7 
8# In-memory stores
9users: Dict[int, dict] = {}
10tokens: Dict[str, int] = {}  # token -> user_id
11creators: Dict[int, dict] = {}
12next_user_id = 1
13next_creator_id = 1
14 
15# Auth helper
16def get_current_user(authorization: str = Header(...)):
17    if not authorization.startswith("Bearer "):
18        raise HTTPException(401, "Invalid auth header")
19    token = authorization[7:]
20    user_id = tokens.get(token)
21    if user_id is None:
22        raise HTTPException(401, "Invalid token")
23    return user_id
24 
25# --- Auth endpoints ---
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35@app.post("/signup")
36def signup(req: SignupRequest):
37    global next_user_id
38    for u in users.values():
39        if u["username"] == req.username:
40            raise HTTPException(400, "Username taken")
41    user_id = next_user_id
42    next_user_id += 1
43    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
44    token = secrets.token_hex(16)
45    tokens[token] = user_id
46    return {"user_id": user_id, "token": token}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    for u in users.values():
51        if u["username"] == req.username and u["password"] == req.password:
52            token = secrets.token_hex(16)
53            tokens[token] = u["id"]
54            return {"token": token}
55    raise HTTPException(401, "Invalid credentials")
56 
57# --- Creator endpoints ---
58 
59class CreatorCreate(BaseModel):
60    name: str
61    channel_description: str = ""
62    monetization_enabled: bool = False
63    content_tier: str = "free"
64 
65class CreatorUpdate(BaseModel):
66    name: Optional[str] = None
67    channel_description: Optional[str] = None
68    monetization_enabled: Optional[bool] = None
69    content_tier: Optional[str] = None
70 
71@app.post("/creators")
72def create_creator(creator: CreatorCreate, authorization: str = Header(...)):
73    global next_creator_id
74    user_id = get_current_user(authorization)
75    creator_id = next_creator_id
76    next_creator_id += 1
77    creators[creator_id] = {
78        "id": creator_id,
79        "user_id": user_id,
80        "name": creator.name,
81        "channel_description": creator.channel_description,
82        "monetization_enabled": creator.monetization_enabled,
83        "content_tier": creator.content_tier,
84    }
85    return creators[creator_id]
86 
87@app.get("/creators/{creator_id}")
88def get_creator(creator_id: int, authorization: str = Header(...)):
89    get_current_user(authorization)
90    creator = creators.get(creator_id)
91    if not creator:
92        raise HTTPException(404, "Creator not found")
93    return creator
94 
95@app.patch("/creators/{creator_id}")
96def update_creator(creator_id: int, update: CreatorUpdate, authorization: str = Header(...)):
97    user_id = get_current_user(authorization)
98    creator = creators.get(creator_id)
99    if not creator:
100        raise HTTPException(404, "Creator not found")
101    if creator["user_id"] != user_id:
102        raise HTTPException(403, "Not your creator")
103    if update.name is not None:
104        creator["name"] = update.name
105    if update.channel_description is not None:
106        creator["channel_description"] = update.channel_description
107    if update.monetization_enabled is not None:
108        creator["monetization_enabled"] = update.monetization_enabled
109    if update.content_tier is not None:
110        creator["content_tier"] = update.content_tier
111    return creator

1fastapi
2uvicorn