Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import sqlite3
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11products = {}
12product_id_counter = 1
13 
14# In-memory SQLite for fast product lookup
15conn = sqlite3.connect(":memory:", check_same_thread=False)
16conn.execute("CREATE TABLE products (id INTEGER PRIMARY KEY, name TEXT, season TEXT, vendor TEXT, price_per_lb REAL, in_season INTEGER, vendor_rating REAL, booth_number TEXT)")
17conn.execute("CREATE INDEX idx_products_season_vendor ON products(season, vendor)")
18 
19def hash_password(password: str) -> str:
20    return hashlib.sha256(password.encode()).hexdigest()
21 
22def generate_token() -> str:
23    return secrets.token_hex(32)
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    if username in users:
28        raise HTTPException(400, "User exists")
29    users[username] = hash_password(password)
30    token = generate_token()
31    tokens[token] = username
32    return {"token": token}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    if username not in users or users[username] != hash_password(password):
37        raise HTTPException(401, "Invalid credentials")
38    token = generate_token()
39    tokens[token] = username
40    return {"token": token}
41 
42def get_current_user(authorization: Optional[str] = Header(None)):
43    if not authorization or not authorization.startswith("Bearer "):
44        raise HTTPException(401, "Missing or invalid token")
45    token = authorization.split(" ")[1]
46    if token not in tokens:
47        raise HTTPException(401, "Invalid token")
48    return tokens[token]
49 
50@app.get("/products/{product_id}")
51def get_product(product_id: int, authorization: Optional[str] = Header(None)):
52    get_current_user(authorization)
53    if product_id not in products:
54        raise HTTPException(404, "Product not found")
55    return products[product_id]
56 
57@app.post("/products")
58def create_product(name: str, season: str, vendor: str, price_per_lb: float, in_season: bool, vendor_rating: float, booth_number: str, authorization: Optional[str] = Header(None)):
59    get_current_user(authorization)
60    global product_id_counter
61    product = {
62        "id": product_id_counter,
63        "name": name,
64        "season": season,
65        "vendor": vendor,
66        "price_per_lb": price_per_lb,
67        "in_season": int(in_season),
68        "vendor_rating": vendor_rating,
69        "booth_number": booth_number
70    }
71    products[product_id_counter] = product
72    conn.execute("INSERT INTO products VALUES (?,?,?,?,?,?,?,?)",
73                 (product_id_counter, name, season, vendor, price_per_lb, int(in_season), vendor_rating, booth_number))
74    conn.commit()
75    product_id_counter += 1
76    return product
77 
78@app.get("/products")
79def list_products(season: Optional[str] = None, vendor: Optional[str] = None, max_price: Optional[float] = None, in_season: Optional[bool] = None, authorization: Optional[str] = Header(None)):
80    get_current_user(authorization)
81    query = "SELECT * FROM products WHERE 1=1"
82    params = []
83    if season:
84        query += " AND season = ?"
85        params.append(season)
86    if vendor:
87        query += " AND vendor = ?"
88        params.append(vendor)
89    if max_price is not None:
90        query += " AND price_per_lb <= ?"
91        params.append(max_price)
92    if in_season is not None:
93        query += " AND in_season = ?"
94        params.append(int(in_season))
95    cursor = conn.execute(query, params)
96    rows = cursor.fetchall()
97    results = []
98    for row in rows:
99        results.append({
100            "id": row[0],
101            "name": row[1],
102            "season": row[2],
103            "vendor": row[3],
104            "price_per_lb": row[4],
105            "in_season": bool(row[5]),
106            "vendor_rating": row[6],
107            "booth_number": row[7]
108        })
109    return {"products": results}

1fastapi
2uvicorn