Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import time
5from datetime import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11exhibits = {}
12audio_files = {}
13dwell_times = {}
14next_user_id = 1
15next_exhibit_id = 1
16next_audio_id = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth token")
21    token = authorization.replace("Bearer ", "")
22    user_id = tokens.get(token)
23    if not user_id:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return user_id
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global next_user_id
30    for u in users.values():
31        if u["username"] == username:
32            raise HTTPException(status_code=400, detail="Username already exists")
33    user_id = next_user_id
34    next_user_id += 1
35    users[user_id] = {"id": user_id, "username": username, "password": password}
36    return {"user_id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    for u in users.values():
41        if u["username"] == username and u["password"] == password:
42            token = str(uuid.uuid4())
43            tokens[token] = u["id"]
44            return {"token": token}
45    raise HTTPException(status_code=401, detail="Invalid credentials")
46 
47@app.post("/exhibits")
48def create_exhibit(name: str, description: str = "", authorization: Optional[str] = Header(None)):
49    get_current_user(authorization)
50    global next_exhibit_id
51    exhibit_id = next_exhibit_id
52    next_exhibit_id += 1
53    exhibits[exhibit_id] = {
54        "id": exhibit_id,
55        "name": name,
56        "description": description,
57        "audio_files": [],
58        "created_at": datetime.now().isoformat()
59    }
60    return exhibits[exhibit_id]
61 
62@app.get("/exhibits/{exhibit_id}")
63def get_exhibit(exhibit_id: int, authorization: Optional[str] = Header(None)):
64    get_current_user(authorization)
65    exhibit = exhibits.get(exhibit_id)
66    if not exhibit:
67        raise HTTPException(status_code=404, detail="Exhibit not found")
68    return exhibit
69 
70@app.post("/exhibits/{exhibit_id}/audio")
71def add_audio_file(exhibit_id: int, file_name: str, audio_url: str, authorization: Optional[str] = Header(None)):
72    get_current_user(authorization)
73    if exhibit_id not in exhibits:
74        raise HTTPException(status_code=404, detail="Exhibit not found")
75    global next_audio_id
76    audio_id = next_audio_id
77    next_audio_id += 1
78    audio_files[audio_id] = {
79        "id": audio_id,
80        "exhibit_id": exhibit_id,
81        "file_name": file_name,
82        "audio_url": audio_url,
83        "created_at": datetime.now().isoformat()
84    }
85    exhibits[exhibit_id]["audio_files"].append(audio_id)
86    return audio_files[audio_id]
87 
88@app.get("/exhibits/{exhibit_id}/audio")
89def get_exhibit_audio(exhibit_id: int, authorization: Optional[str] = Header(None)):
90    get_current_user(authorization)
91    if exhibit_id not in exhibits:
92        raise HTTPException(status_code=404, detail="Exhibit not found")
93    return [audio_files[aid] for aid in exhibits[exhibit_id]["audio_files"] if aid in audio_files]
94 
95@app.post("/visits")
96def start_visit(exhibit_id: int, authorization: Optional[str] = Header(None)):
97    user_id = get_current_user(authorization)
98    if exhibit_id not in exhibits:
99        raise HTTPException(status_code=404, detail="Exhibit not found")
100    visit_id = len(dwell_times) + 1
101    dwell_times[visit_id] = {
102        "id": visit_id,
103        "user_id": user_id,
104        "exhibit_id": exhibit_id,
105        "start_time": time.time(),
106        "end_time": None,
107        "duration_seconds": None
108    }
109    return dwell_times[visit_id]
110 
111@app.post("/visits/{visit_id}/end")
112def end_visit(visit_id: int, authorization: Optional[str] = Header(None)):
113    get_current_user(authorization)
114    if visit_id not in dwell_times:
115        raise HTTPException(status_code=404, detail="Visit not found")
116    if dwell_times[visit_id]["end_time"] is not None:
117        raise HTTPException(status_code=400, detail="Visit already ended")
118    dwell_times[visit_id]["end_time"] = time.time()
119    dwell_times[visit_id]["duration_seconds"] = dwell_times[visit_id]["end_time"] - dwell_times[visit_id]["start_time"]
120    return dwell_times[visit_id]
121 
122@app.get("/visits/{visit_id}")
123def get_visit(visit_id: int, authorization: Optional[str] = Header(None)):
124    get_current_user(authorization)
125    visit = dwell_times.get(visit_id)
126    if not visit:
127        raise HTTPException(status_code=404, detail="Visit not found")
128    return visit

1fastapi
2uvicorn