Browse the data

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

1fastapi
2uvicorn