Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4from datetime import datetime, timedelta
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10collections = {}
11prompts = {}
12used_prompts = {}
13next_user_id = 1
14next_token_id = 1
15next_collection_id = 1
16next_prompt_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class CreateCollectionRequest(BaseModel):
27    name: str
28    constraints: str = ""
29 
30class CreatePromptRequest(BaseModel):
31    theme: str
32    collection_id: int
33 
34def get_current_user(authorization: str = Header(None)):
35    if not authorization:
36        raise HTTPException(status_code=401, detail="No auth token")
37    token = authorization.replace("Bearer ", "")
38    user_id = tokens.get(token)
39    if not user_id:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return user_id
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    user_id = next_user_id
47    next_user_id += 1
48    users[user_id] = {"username": req.username, "password": req.password}
49    return {"user_id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    for uid, u in users.items():
54        if u["username"] == req.username and u["password"] == req.password:
55            global next_token_id
56            token = secrets.token_hex(32)
57            tokens[token] = uid
58            return {"token": token}
59    raise HTTPException(status_code=401, detail="Invalid credentials")
60 
61@app.post("/collections")
62def create_collection(req: CreateCollectionRequest, authorization: str = Header(None)):
63    user_id = get_current_user(authorization)
64    global next_collection_id
65    cid = next_collection_id
66    next_collection_id += 1
67    collections[cid] = {"id": cid, "name": req.name, "constraints": req.constraints, "owner": user_id}
68    return collections[cid]
69 
70@app.get("/collections/{collection_id}")
71def get_collection(collection_id: int, authorization: str = Header(None)):
72    user_id = get_current_user(authorization)
73    coll = collections.get(collection_id)
74    if not coll:
75        raise HTTPException(status_code=404, detail="Collection not found")
76    return coll
77 
78@app.post("/prompts")
79def create_prompt(req: CreatePromptRequest, authorization: str = Header(None)):
80    user_id = get_current_user(authorization)
81    if req.collection_id not in collections:
82        raise HTTPException(status_code=404, detail="Collection not found")
83    global next_prompt_id
84    pid = next_prompt_id
85    next_prompt_id += 1
86    prompts[pid] = {"id": pid, "theme": req.theme, "collection_id": req.collection_id, "owner": user_id}
87    return prompts[pid]
88 
89@app.get("/prompts/{prompt_id}")
90def get_prompt(prompt_id: int, authorization: str = Header(None)):
91    user_id = get_current_user(authorization)
92    prompt = prompts.get(prompt_id)
93    if not prompt:
94        raise HTTPException(status_code=404, detail="Prompt not found")
95    return prompt
96 
97@app.get("/collections/{collection_id}/random")
98def get_random_prompt(collection_id: int, authorization: str = Header(None)):
99    user_id = get_current_user(authorization)
100    coll = collections.get(collection_id)
101    if not coll:
102        raise HTTPException(status_code=404, detail="Collection not found")
103    
104    available = [p for p in prompts.values() if p["collection_id"] == collection_id and p["id"] not in used_prompts.get(collection_id, set())]
105    if not available:
106        raise HTTPException(status_code=404, detail="No unused prompts available")
107    
108    import random
109    prompt = random.choice(available)
110    if collection_id not in used_prompts:
111        used_prompts[collection_id] = set()
112    used_prompts[collection_id].add(prompt["id"])
113    return prompt

1fastapi
2uvicorn
3pydantic