Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import hashlib
6import time
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12flagged_items = {}
13flag_id_counter = 1
14auth_token_counter = 1000
15 
16ADMIN_TOKEN = "admin-secret-token-123"
17 
18def generate_token():
19    global auth_token_counter
20    auth_token_counter += 1
21    return f"tok_{auth_token_counter}_{int(time.time())}"
22 
23def get_user_from_token(authorization: str = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth header")
26    token = authorization.replace("Bearer ", "")
27    for uid, tok in tokens.items():
28        if tok == token:
29            return uid
30    raise HTTPException(status_code=401, detail="Invalid token")
31 
32class SignupRequest(BaseModel):
33    username: str
34    password: str
35 
36class LoginRequest(BaseModel):
37    username: str
38    password: str
39 
40class FlagRequest(BaseModel):
41    text: Optional[str] = None
42    image_url: Optional[str] = None
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="User exists")
48    uid = len(users) + 1
49    users[req.username] = {"id": uid, "password": req.password}
50    token = generate_token()
51    tokens[uid] = token
52    return {"user_id": uid, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    user = users.get(req.username)
57    if not user or user["password"] != req.password:
58        raise HTTPException(status_code=401, detail="Invalid credentials")
59    token = generate_token()
60    tokens[user["id"]] = token
61    return {"token": token}
62 
63@app.post("/moderate/flag")
64def flag_content(req: FlagRequest, authorization: str = Header(None)):
65    user_id = get_user_from_token(authorization)
66    global flag_id_counter
67    toxicity_score = random.uniform(0, 1)
68    categories = []
69    if toxicity_score > 0.7:
70        categories.append("hate_speech")
71    if toxicity_score > 0.5:
72        categories.append("harassment")
73    if toxicity_score > 0.3:
74        categories.append("nsfw")
75    item = {
76        "id": flag_id_counter,
77        "user_id": user_id,
78        "text": req.text,
79        "image_url": req.image_url,
80        "toxicity_score": round(toxicity_score, 2),
81        "flagged_categories": categories,
82        "resolved": False,
83        "created_at": int(time.time())
84    }
85    flagged_items[flag_id_counter] = item
86    flag_id_counter += 1
87    return item
88 
89@app.get("/moderate/reports")
90def get_reports(authorization: str = Header(None)):
91    user_id = get_user_from_token(authorization)
92    return {"reports": [v for v in flagged_items.values()]}
93 
94@app.get("/moderate/reports/{report_id}")
95def get_report(report_id: int, authorization: str = Header(None)):
96    user_id = get_user_from_token(authorization)
97    item = flagged_items.get(report_id)
98    if not item:
99        raise HTTPException(status_code=404, detail="Report not found")
100    return item
101 
102@app.post("/moderate/reports/{report_id}/resolve")
103def resolve_report(report_id: int, authorization: str = Header(None)):
104    user_id = get_user_from_token(authorization)
105    item = flagged_items.get(report_id)
106    if not item:
107        raise HTTPException(status_code=404, detail="Report not found")
108    item["resolved"] = True
109    return item

1fastapi
2uvicorn