Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import hashlib
4import secrets
5import threading
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11comics = {}
12conditions = {}
13value_estimates = {}
14next_user_id = 1
15next_comic_id = 1
16next_condition_id = 1
17next_value_id = 1
18lock = threading.Lock()
19 
20WEEKLY_UPDATE_INTERVAL = timedelta(days=7)
21 
22def generate_token():
23    return secrets.token_hex(32)
24 
25def hash_password(password):
26    return hashlib.sha256(password.encode()).hexdigest()
27 
28def get_current_user(authorization: str = Header(None)):
29    if not authorization:
30        raise HTTPException(status_code=401, detail="Missing authorization header")
31    token = authorization.replace("Bearer ", "")
32    if token not in tokens:
33        raise HTTPException(status_code=401, detail="Invalid token")
34    return tokens[token]
35 
36@app.post("/signup")
37def signup(username: str, password: str):
38    global next_user_id
39    with lock:
40        for user in users.values():
41            if user["username"] == username:
42                raise HTTPException(status_code=400, detail="Username already exists")
43        user_id = next_user_id
44        next_user_id += 1
45        users[user_id] = {
46            "id": user_id,
47            "username": username,
48            "password_hash": hash_password(password)
49        }
50        return {"id": user_id, "username": username}
51 
52@app.post("/login")
53def login(username: str, password: str):
54    for user in users.values():
55        if user["username"] == username and user["password_hash"] == hash_password(password):
56            token = generate_token()
57            tokens[token] = user["id"]
58            return {"token": token}
59    raise HTTPException(status_code=401, detail="Invalid credentials")
60 
61@app.get("/comic/{comic_id}")
62def get_comic(comic_id: int, authorization: str = Header(None)):
63    user_id = get_current_user(authorization)
64    if comic_id not in comics:
65        raise HTTPException(status_code=404, detail="Comic not found")
66    return comics[comic_id]
67 
68@app.post("/comic")
69def create_comic(title: str, issue_number: int, authorization: str = Header(None)):
70    user_id = get_current_user(authorization)
71    global next_comic_id
72    with lock:
73        comic_id = next_comic_id
74        next_comic_id += 1
75        comics[comic_id] = {
76            "id": comic_id,
77            "title": title,
78            "issue_number": issue_number,
79            "owner_id": user_id,
80            "created_at": datetime.now().isoformat()
81        }
82        return comics[comic_id]
83 
84@app.get("/condition/{condition_id}")
85def get_condition(condition_id: int, authorization: str = Header(None)):
86    user_id = get_current_user(authorization)
87    if condition_id not in conditions:
88        raise HTTPException(status_code=404, detail="Condition not found")
89    return conditions[condition_id]
90 
91@app.post("/condition")
92def create_condition(comic_id: int, grade: str, notes: str = "", authorization: str = Header(None)):
93    user_id = get_current_user(authorization)
94    if comic_id not in comics:
95        raise HTTPException(status_code=404, detail="Comic not found")
96    global next_condition_id
97    with lock:
98        condition_id = next_condition_id
99        next_condition_id += 1
100        conditions[condition_id] = {
101            "id": condition_id,
102            "comic_id": comic_id,
103            "grade": grade,
104            "notes": notes,
105            "recorded_by": user_id,
106            "recorded_at": datetime.now().isoformat()
107        }
108        return conditions[condition_id]
109 
110@app.get("/value/{value_id}")
111def get_value(value_id: int, authorization: str = Header(None)):
112    user_id = get_current_user(authorization)
113    if value_id not in value_estimates:
114        raise HTTPException(status_code=404, detail="Value estimate not found")
115    return value_estimates[value_id]
116 
117@app.post("/value")
118def create_value(comic_id: int, estimated_value: float, authorization: str = Header(None)):
119    user_id = get_current_user(authorization)
120    if comic_id not in comics:
121        raise HTTPException(status_code=404, detail="Comic not found")
122    global next_value_id
123    with lock:
124        value_id = next_value_id
125        next_value_id += 1
126        value_estimates[value_id] = {
127            "id": value_id,
128            "comic_id": comic_id,
129            "estimated_value": estimated_value,
130            "estimated_by": user_id,
131            "estimated_at": datetime.now().isoformat(),
132            "next_update": (datetime.now() + WEEKLY_UPDATE_INTERVAL).isoformat()
133        }
134        return value_estimates[value_id]
135 
136@app.get("/comic/{comic_id}/condition")
137def get_comic_condition(comic_id: int, authorization: str = Header(None)):
138    user_id = get_current_user(authorization)
139    if comic_id not in comics:
140        raise HTTPException(status_code=404, detail="Comic not found")
141    comic_conditions = [c for c in conditions.values() if c["comic_id"] == comic_id]
142    return comic_conditions
143 
144@app.get("/comic/{comic_id}/value")
145def get_comic_value(comic_id: int, authorization: str = Header(None)):
146    user_id = get_current_user(authorization)
147    if comic_id not in comics:
148        raise HTTPException(status_code=404, detail="Comic not found")
149    comic_values = [v for v in value_estimates.values() if v["comic_id"] == comic_id]
150    return comic_values

1fastapi
2uvicorn