Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9scores = {}
10tokens = {}
11weekly_scores = {}
12weekly_last_reset = datetime.now()
13 
14def generate_token():
15    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
16 
17def get_current_user(authorization: str = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing auth header")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25def reset_weekly():
26    global weekly_last_reset
27    now = datetime.now()
28    if weekly_last_reset.weekday() != 0 or weekly_last_reset.date() != now.date():
29        if now.weekday() == 0:
30            weekly_scores.clear()
31            weekly_last_reset = now
32 
33@app.post("/signup")
34def signup(username: str, password: str):
35    user_id = len(users) + 1
36    users[user_id] = {"username": username, "password": password}
37    scores[user_id] = {"score": 0, "games_played": 0}
38    return {"id": user_id, "username": username}
39 
40@app.post("/login")
41def login(username: str, password: str):
42    for uid, u in users.items():
43        if u["username"] == username and u["password"] == password:
44            token = generate_token()
45            tokens[token] = uid
46            return {"token": token}
47    raise HTTPException(status_code=401, detail="Invalid credentials")
48 
49@app.get("/leaderboard")
50def get_leaderboard():
51    sorted_users = sorted(scores.items(), key=lambda x: x[1]["score"], reverse=True)
52    result = []
53    for rank, (uid, data) in enumerate(sorted_users, 1):
54        result.append({
55            "rank": rank,
56            "user_id": uid,
57            "username": users[uid]["username"],
58            "score": data["score"],
59            "games_played": data["games_played"]
60        })
61    return result
62 
63@app.get("/leaderboard/weekly")
64def get_weekly_leaderboard():
65    reset_weekly()
66    sorted_users = sorted(weekly_scores.items(), key=lambda x: x[1]["score"], reverse=True)
67    result = []
68    for rank, (uid, data) in enumerate(sorted_users, 1):
69        result.append({
70            "rank": rank,
71            "user_id": uid,
72            "username": users[uid]["username"],
73            "score": data["score"],
74            "games_played": data["games_played"]
75        })
76    return result
77 
78@app.post("/score")
79def submit_score(score: int, authorization: str = Header(None)):
80    user_id = get_current_user(authorization)
81    scores[user_id]["score"] += score
82    scores[user_id]["games_played"] += 1
83    reset_weekly()
84    if user_id not in weekly_scores:
85        weekly_scores[user_id] = {"score": 0, "games_played": 0}
86    weekly_scores[user_id]["score"] += score
87    weekly_scores[user_id]["games_played"] += 1
88    return {"user_id": user_id, "total_score": scores[user_id]["score"], "games_played": scores[user_id]["games_played"]}
89 
90@app.get("/user/{user_id}")
91def get_user(user_id: int):
92    if user_id not in users:
93        raise HTTPException(status_code=404, detail="User not found")
94    return {"id": user_id, "username": users[user_id]["username"], "score": scores[user_id]["score"], "games_played": scores[user_id]["games_played"]}

1fastapi
2uvicorn