Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9books = {}
10members = {}
11borrows = {}
12book_id_counter = 1
13member_id_counter = 1
14borrow_id_counter = 1
15 
16def get_current_user(authorization: str = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    if username in users:
27        raise HTTPException(status_code=400, detail="User exists")
28    users[username] = password
29    token = secrets.token_hex(16)
30    tokens[token] = username
31    return {"token": token}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    if username not in users or users[username] != password:
36        raise HTTPException(status_code=401, detail="Invalid credentials")
37    token = secrets.token_hex(16)
38    tokens[token] = username
39    return {"token": token}
40 
41@app.get("/books")
42def get_books(authorization: str = Header(None)):
43    get_current_user(authorization)
44    return books
45 
46@app.get("/books/{book_id}")
47def get_book(book_id: int, authorization: str = Header(None)):
48    get_current_user(authorization)
49    if book_id not in books:
50        raise HTTPException(status_code=404, detail="Book not found")
51    return books[book_id]
52 
53@app.post("/books")
54def create_book(title: str, author: str, isbn: str, copies: int, shelf: str, authorization: str = Header(None)):
55    get_current_user(authorization)
56    global book_id_counter
57    book_id = book_id_counter
58    book_id_counter += 1
59    books[book_id] = {
60        "id": book_id,
61        "title": title,
62        "author": author,
63        "isbn": isbn,
64        "available_copies": copies,
65        "shelf_location": shelf
66    }
67    return books[book_id]
68 
69@app.post("/borrow")
70def borrow_book(book_id: int, member_id: int, authorization: str = Header(None)):
71    get_current_user(authorization)
72    if book_id not in books:
73        raise HTTPException(status_code=404, detail="Book not found")
74    if member_id not in members:
75        raise HTTPException(status_code=404, detail="Member not found")
76    if books[book_id]["available_copies"] <= 0:
77        raise HTTPException(status_code=400, detail="No copies available")
78    global borrow_id_counter
79    borrow_id = borrow_id_counter
80    borrow_id_counter += 1
81    due_date = datetime.now() + timedelta(days=14)
82    borrows[borrow_id] = {
83        "id": borrow_id,
84        "book_id": book_id,
85        "member_id": member_id,
86        "borrow_date": datetime.now().isoformat(),
87        "due_date": due_date.isoformat()
88    }
89    books[book_id]["available_copies"] -= 1
90    return borrows[borrow_id]
91 
92@app.post("/members")
93def create_member(name: str, email: str, authorization: str = Header(None)):
94    get_current_user(authorization)
95    global member_id_counter
96    member_id = member_id_counter
97    member_id_counter += 1
98    members[member_id] = {
99        "id": member_id,
100        "name": name,
101        "email": email
102    }
103    return members[member_id]
104 
105@app.get("/members/{member_id}")
106def get_member(member_id: int, authorization: str = Header(None)):
107    get_current_user(authorization)
108    if member_id not in members:
109        raise HTTPException(status_code=404, detail="Member not found")
110    return members[member_id]

1fastapi
2uvicorn