Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11authors = {}
12next_user_id = 1
13next_author_id = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class AuthorCreate(BaseModel):
24    name: str
25    bio: str = ""
26    publication_role: str = ""
27    editorial_access: bool = False
28 
29class AuthorUpdate(BaseModel):
30    name: Optional[str] = None
31    bio: Optional[str] = None
32    publication_role: Optional[str] = None
33    editorial_access: Optional[bool] = None
34 
35def get_current_user(authorization: str = Header(...)):
36    if not authorization.startswith("Bearer "):
37        raise HTTPException(status_code=401, detail="Invalid auth header")
38    token = authorization[7:]
39    if token not in tokens:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return tokens[token]
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    if req.username in [u["username"] for u in users.values()]:
47        raise HTTPException(status_code=400, detail="Username taken")
48    user_id = next_user_id
49    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
50    next_user_id += 1
51    token = secrets.token_hex(16)
52    tokens[token] = user_id
53    return {"token": token, "user_id": user_id}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for uid, u in users.items():
58        if u["username"] == req.username and u["password"] == req.password:
59            token = secrets.token_hex(16)
60            tokens[token] = uid
61            return {"token": token, "user_id": uid}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.post("/authors")
65def create_author(author: AuthorCreate, authorization: str = Header(...)):
66    global next_author_id
67    get_current_user(authorization)
68    author_id = next_author_id
69    authors[author_id] = {
70        "id": author_id,
71        "name": author.name,
72        "bio": author.bio,
73        "publication_role": author.publication_role,
74        "editorial_access": author.editorial_access
75    }
76    next_author_id += 1
77    return authors[author_id]
78 
79@app.get("/authors/{author_id}")
80def get_author(author_id: int, authorization: str = Header(...)):
81    get_current_user(authorization)
82    if author_id not in authors:
83        raise HTTPException(status_code=404, detail="Author not found")
84    return authors[author_id]
85 
86@app.patch("/authors/{author_id}")
87def update_author(author_id: int, update: AuthorUpdate, authorization: str = Header(...)):
88    get_current_user(authorization)
89    if author_id not in authors:
90        raise HTTPException(status_code=404, detail="Author not found")
91    author = authors[author_id]
92    if update.name is not None:
93        author["name"] = update.name
94    if update.bio is not None:
95        author["bio"] = update.bio
96    if update.publication_role is not None:
97        author["publication_role"] = update.publication_role
98    if update.editorial_access is not None:
99        author["editorial_access"] = update.editorial_access
100    return author

1fastapi
2uvicorn