Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4import time
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10templates = {}
11captions = {}
12creations = {}
13share_counts = {}
14next_user_id = 1
15next_template_id = 1
16next_caption_id = 1
17next_creation_id = 1
18 
19class SignupRequest(BaseModel):
20    username: str
21    password: str
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27class TemplateCreate(BaseModel):
28    name: str
29    image_url: str
30 
31class CaptionCreate(BaseModel):
32    meme_id: int
33    text: str
34 
35class CreationCreate(BaseModel):
36    meme_id: int
37    caption_id: int
38 
39def get_current_user(authorization: str = Header(...)):
40    if not authorization.startswith("Bearer "):
41        raise HTTPException(status_code=401, detail="Invalid auth header")
42    token = authorization[7:]
43    if token not in tokens:
44        raise HTTPException(status_code=401, detail="Invalid token")
45    return tokens[token]
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    global next_user_id
50    for user in users.values():
51        if user["username"] == req.username:
52            raise HTTPException(status_code=400, detail="Username already exists")
53    user_id = next_user_id
54    next_user_id += 1
55    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
56    return {"id": user_id}
57 
58@app.post("/login")
59def login(req: LoginRequest):
60    for user in users.values():
61        if user["username"] == req.username and user["password"] == req.password:
62            token = secrets.token_hex(32)
63            tokens[token] = user["id"]
64            return {"token": token}
65    raise HTTPException(status_code=401, detail="Invalid credentials")
66 
67@app.get("/templates/{template_id}")
68def get_template(template_id: int, authorization: str = Header(...)):
69    get_current_user(authorization)
70    if template_id not in templates:
71        raise HTTPException(status_code=404, detail="Template not found")
72    return templates[template_id]
73 
74@app.post("/templates")
75def create_template(req: TemplateCreate, authorization: str = Header(...)):
76    global next_template_id
77    user_id = get_current_user(authorization)
78    template_id = next_template_id
79    next_template_id += 1
80    templates[template_id] = {"id": template_id, "name": req.name, "image_url": req.image_url, "owner_id": user_id}
81    return templates[template_id]
82 
83@app.get("/captions/{caption_id}")
84def get_caption(caption_id: int, authorization: str = Header(...)):
85    get_current_user(authorization)
86    if caption_id not in captions:
87        raise HTTPException(status_code=404, detail="Caption not found")
88    return captions[caption_id]
89 
90@app.post("/captions")
91def create_caption(req: CaptionCreate, authorization: str = Header(...)):
92    global next_caption_id
93    user_id = get_current_user(authorization)
94    if req.meme_id not in templates:
95        raise HTTPException(status_code=404, detail="Template not found")
96    caption_id = next_caption_id
97    next_caption_id += 1
98    captions[caption_id] = {"id": caption_id, "meme_id": req.meme_id, "text": req.text, "user_id": user_id}
99    return captions[caption_id]
100 
101@app.get("/creations/{creation_id}")
102def get_creation(creation_id: int, authorization: str = Header(...)):
103    get_current_user(authorization)
104    if creation_id not in creations:
105        raise HTTPException(status_code=404, detail="Creation not found")
106    return creations[creation_id]
107 
108@app.post("/creations")
109def create_creation(req: CreationCreate, authorization: str = Header(...)):
110    global next_creation_id
111    user_id = get_current_user(authorization)
112    if req.meme_id not in templates:
113        raise HTTPException(status_code=404, detail="Template not found")
114    if req.caption_id not in captions:
115        raise HTTPException(status_code=404, detail="Caption not found")
116    creation_id = next_creation_id
117    next_creation_id += 1
118    creations[creation_id] = {"id": creation_id, "meme_id": req.meme_id, "caption_id": req.caption_id, "user_id": user_id}
119    share_counts[creation_id] = 0
120    return creations[creation_id]
121 
122@app.post("/creations/{creation_id}/share")
123def share_creation(creation_id: int, authorization: str = Header(...)):
124    get_current_user(authorization)
125    if creation_id not in creations:
126        raise HTTPException(status_code=404, detail="Creation not found")
127    share_counts[creation_id] += 1
128    return {"shares": share_counts[creation_id]}
129 
130@app.get("/creations/{creation_id}/shares")
131def get_share_count(creation_id: int, authorization: str = Header(...)):
132    get_current_user(authorization)
133    if creation_id not in creations:
134        raise HTTPException(status_code=404, detail="Creation not found")
135    return {"shares": share_counts.get(creation_id, 0)}

1fastapi
2uvicorn