Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import hashlib
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10exhibits = {}
11next_user_id = 1
12next_exhibit_id = 1
13 
14def get_current_user(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing Authorization header")
17    token = authorization.replace("Bearer ", "")
18    user_id = tokens.get(token)
19    if user_id is None:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return user_id
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    global next_user_id
26    if username in users:
27        raise HTTPException(status_code=400, detail="Username already exists")
28    user_id = next_user_id
29    next_user_id += 1
30    users[username] = {"id": user_id, "password": password}
31    return {"id": user_id, "username": username}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    user = users.get(username)
36    if not user or user["password"] != password:
37        raise HTTPException(status_code=401, detail="Invalid credentials")
38    token = secrets.token_hex(16)
39    tokens[token] = user["id"]
40    return {"token": token}
41 
42@app.post("/exhibits")
43def create_exhibit(name: str, category: str, date: str, ticket_price: float, remaining_capacity: int, guided_tour: bool = False, authorization: str = Header(None)):
44    get_current_user(authorization)
45    global next_exhibit_id
46    exhibit_id = next_exhibit_id
47    next_exhibit_id += 1
48    exhibits[exhibit_id] = {
49        "id": exhibit_id,
50        "name": name,
51        "category": category,
52        "date": date,
53        "ticket_price": ticket_price,
54        "remaining_capacity": remaining_capacity,
55        "guided_tour": guided_tour
56    }
57    return exhibits[exhibit_id]
58 
59@app.get("/exhibits/{exhibit_id}")
60def get_exhibit(exhibit_id: int, authorization: str = Header(None)):
61    get_current_user(authorization)
62    exhibit = exhibits.get(exhibit_id)
63    if not exhibit:
64        raise HTTPException(status_code=404, detail="Exhibit not found")
65    return exhibit
66 
67@app.get("/exhibits")
68def search_exhibits(category: str = None, date: str = None, guided_tour: bool = None, authorization: str = Header(None)):
69    get_current_user(authorization)
70    results = []
71    for exhibit in exhibits.values():
72        if category and exhibit["category"] != category:
73            continue
74        if date:
75            if date == "tomorrow":
76                tomorrow = (datetime.now() + timedelta(days=1)).strftime("%Y-%m-%d")
77                if exhibit["date"] != tomorrow:
78                    continue
79            else:
80                if exhibit["date"] != date:
81                    continue
82        if guided_tour is not None and exhibit["guided_tour"] != guided_tour:
83            continue
84        results.append({
85            "id": exhibit["id"],
86            "name": exhibit["name"],
87            "category": exhibit["category"],
88            "date": exhibit["date"],
89            "ticket_price": exhibit["ticket_price"],
90            "remaining_capacity": exhibit["remaining_capacity"],
91            "guided_tour": exhibit["guided_tour"]
92        })
93    return results

1fastapi
2uvicorn