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 = {}
11books = {}
12next_user_id = 1
13next_book_id = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18    access_tier: Optional[str] = "basic"
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class BookCreate(BaseModel):
25    title: str
26    author: str
27    isbn: str
28    genre: str
29    location: str
30    borrower_access_tier: str
31 
32class BookUpdate(BaseModel):
33    title: Optional[str] = None
34    author: Optional[str] = None
35    isbn: Optional[str] = None
36    genre: Optional[str] = None
37    location: Optional[str] = None
38    borrower_access_tier: Optional[str] = None
39 
40def get_current_user(authorization: str = Header(None)):
41    if not authorization:
42        raise HTTPException(401, "Missing auth header")
43    token = authorization.replace("Bearer ", "")
44    if token not in tokens:
45        raise HTTPException(401, "Invalid token")
46    return tokens[token]
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    for u in users.values():
52        if u["username"] == req.username:
53            raise HTTPException(400, "User exists")
54    user_id = next_user_id
55    next_user_id += 1
56    users[user_id] = {
57        "id": user_id,
58        "username": req.username,
59        "password": req.password,
60        "access_tier": req.access_tier
61    }
62    return {"id": user_id, "username": req.username}
63 
64@app.post("/login")
65def login(req: LoginRequest):
66    for u in users.values():
67        if u["username"] == req.username and u["password"] == req.password:
68            token = secrets.token_hex(16)
69            tokens[token] = u["id"]
70            return {"token": token}
71    raise HTTPException(401, "Bad credentials")
72 
73@app.get("/books/{book_id}")
74def get_book(book_id: int, authorization: str = Header(None)):
75    user_id = get_current_user(authorization)
76    if book_id not in books:
77        raise HTTPException(404, "Not found")
78    return books[book_id]
79 
80@app.post("/books")
81def create_book(book: BookCreate, authorization: str = Header(None)):
82    global next_book_id
83    user_id = get_current_user(authorization)
84    book_id = next_book_id
85    next_book_id += 1
86    books[book_id] = {
87        "id": book_id,
88        "title": book.title,
89        "author": book.author,
90        "isbn": book.isbn,
91        "genre": book.genre,
92        "location": book.location,
93        "borrower_access_tier": book.borrower_access_tier
94    }
95    return books[book_id]
96 
97@app.put("/books/{book_id}")
98def update_book(book_id: int, book: BookUpdate, authorization: str = Header(None)):
99    user_id = get_current_user(authorization)
100    if book_id not in books:
101        raise HTTPException(404, "Not found")
102    existing = books[book_id]
103    if book.title is not None:
104        existing["title"] = book.title
105    if book.author is not None:
106        existing["author"] = book.author
107    if book.isbn is not None:
108        existing["isbn"] = book.isbn
109    if book.genre is not None:
110        existing["genre"] = book.genre
111    if book.location is not None:
112        existing["location"] = book.location
113    if book.borrower_access_tier is not None:
114        existing["borrower_access_tier"] = book.borrower_access_tier
115    return existing
116 
117if __name__ == "__main__":
118    uvicorn.run(app, host="127.0.0.1", port=8000)

1fastapi
2uvicorn