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 = {}
10vinyls = {}
11offers = {}
12vinyl_id_counter = 1
13offer_id_counter = 1
14user_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class VinylCreate(BaseModel):
25    genre: str
26    year: int
27    condition: str
28    stock: int = 1
29 
30class OfferCreate(BaseModel):
31    vinyl_id: int
32    amount: float
33 
34def get_current_user(authorization: str = Header(None)):
35    if not authorization:
36        raise HTTPException(status_code=401, detail="Missing auth header")
37    token = authorization.replace("Bearer ", "")
38    if token not in tokens:
39        raise HTTPException(status_code=401, detail="Invalid token")
40    return tokens[token]
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global user_id_counter
45    if req.username in users:
46        raise HTTPException(status_code=400, detail="Username already exists")
47    user_id = user_id_counter
48    user_id_counter += 1
49    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
50    return {"id": user_id, "username": req.username}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    user = users.get(req.username)
55    if not user or user["password"] != req.password:
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    token = secrets.token_hex(16)
58    tokens[token] = user["id"]
59    return {"token": token}
60 
61@app.get("/vinyls")
62def list_vinyls():
63    return list(vinyls.values())
64 
65@app.get("/vinyls/{vinyl_id}")
66def get_vinyl(vinyl_id: int):
67    vinyl = vinyls.get(vinyl_id)
68    if not vinyl:
69        raise HTTPException(status_code=404, detail="Vinyl not found")
70    return vinyl
71 
72@app.post("/vinyls")
73def create_vinyl(vinyl: VinylCreate, user_id: int = None):
74    global vinyl_id_counter
75    new_id = vinyl_id_counter
76    vinyl_id_counter += 1
77    vinyls[new_id] = {
78        "id": new_id,
79        "genre": vinyl.genre,
80        "year": vinyl.year,
81        "condition": vinyl.condition,
82        "stock": vinyl.stock
83    }
84    return vinyls[new_id]
85 
86@app.get("/offers/{offer_id}")
87def get_offer(offer_id: int):
88    offer = offers.get(offer_id)
89    if not offer:
90        raise HTTPException(status_code=404, detail="Offer not found")
91    return offer
92 
93@app.post("/offer")
94def create_offer(offer: OfferCreate, authorization: str = Header(None)):
95    current_user = get_current_user(authorization)
96    global offer_id_counter
97    if offer.vinyl_id not in vinyls:
98        raise HTTPException(status_code=404, detail="Vinyl not found")
99    new_id = offer_id_counter
100    offer_id_counter += 1
101    offers[new_id] = {
102        "id": new_id,
103        "vinyl_id": offer.vinyl_id,
104        "user_id": current_user,
105        "amount": offer.amount
106    }
107    return offers[new_id]
108 
109@app.get("/low-stock")
110def low_stock():
111    low = [v for v in vinyls.values() if v["stock"] < 5]
112    return low

1fastapi
2uvicorn