Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8# In-memory storage
9users = {}
10tokens = {}
11episodes = {}
12next_user_id = 1
13next_episode_id = 1
14 
15# Auth helpers
16def get_user_id(authorization: str = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth token")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if not user_id:
22        raise HTTPException(status_code=401, detail="Invalid auth token")
23    return user_id
24 
25# Models
26class SignupRequest(BaseModel):
27    username: str
28    password: str
29 
30class LoginRequest(BaseModel):
31    username: str
32    password: str
33 
34class EpisodeCreate(BaseModel):
35    title: str
36    description: str = ""
37    visibility: str = "public"
38    is_featured: bool = False
39    category: str = ""
40 
41class EpisodeUpdate(BaseModel):
42    title: Optional[str] = None
43    description: Optional[str] = None
44    visibility: Optional[str] = None
45    is_featured: Optional[bool] = None
46    category: Optional[str] = None
47 
48# Auth endpoints
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    user_id = next_user_id
53    next_user_id += 1
54    users[user_id] = {"username": req.username, "password": req.password}
55    return {"user_id": user_id, "message": "User created"}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    for uid, u in users.items():
60        if u["username"] == req.username and u["password"] == req.password:
61            token = secrets.token_hex(16)
62            tokens[token] = uid
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66# Episode endpoints
67@app.get("/episodes/{episode_id}")
68def get_episode(episode_id: int, authorization: str = Header(None)):
69    get_user_id(authorization)
70    episode = episodes.get(episode_id)
71    if not episode:
72        raise HTTPException(status_code=404, detail="Episode not found")
73    return episode
74 
75@app.post("/episodes")
76def create_episode(episode: EpisodeCreate, authorization: str = Header(None)):
77    user_id = get_user_id(authorization)
78    global next_episode_id
79    episode_id = next_episode_id
80    next_episode_id += 1
81    episodes[episode_id] = {
82        "id": episode_id,
83        "user_id": user_id,
84        "title": episode.title,
85        "description": episode.description,
86        "visibility": episode.visibility,
87        "is_featured": episode.is_featured,
88        "category": episode.category
89    }
90    return episodes[episode_id]
91 
92@app.patch("/episodes/{episode_id}")
93def update_episode(episode_id: int, update: EpisodeUpdate, authorization: str = Header(None)):
94    user_id = get_user_id(authorization)
95    episode = episodes.get(episode_id)
96    if not episode:
97        raise HTTPException(status_code=404, detail="Episode not found")
98    if episode["user_id"] != user_id:
99        raise HTTPException(status_code=403, detail="Not your episode")
100    if update.title is not None:
101        episode["title"] = update.title
102    if update.description is not None:
103        episode["description"] = update.description
104    if update.visibility is not None:
105        episode["visibility"] = update.visibility
106    if update.is_featured is not None:
107        episode["is_featured"] = update.is_featured
108    if update.category is not None:
109        episode["category"] = update.category
110    return episode

1fastapi
2uvicorn