Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import random
6import string
7 
8app = FastAPI()
9 
10users = {}
11players = {}
12tokens = {}
13next_user_id = 1
14next_player_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_current_user(authorization: str = Header(None)):
20    if not authorization or not authorization.startswith("Bearer "):
21        raise HTTPException(status_code=401, detail="Invalid auth")
22    token = authorization[7:]
23    user_id = tokens.get(token)
24    if user_id is None:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return user_id
27 
28class SignupRequest(BaseModel):
29    username: str
30    password: str
31 
32class LoginRequest(BaseModel):
33    username: str
34    password: str
35 
36class PlayerCreate(BaseModel):
37    display_name: str
38    score: int = 0
39    rank: int = 0
40    tier: str = "bronze"
41    admin: bool = False
42 
43class PlayerUpdate(BaseModel):
44    display_name: Optional[str] = None
45    score: Optional[int] = None
46    rank: Optional[int] = None
47    tier: Optional[str] = None
48    admin: Optional[bool] = None
49 
50@app.post("/signup")
51def signup(req: SignupRequest):
52    global next_user_id
53    for u in users.values():
54        if u["username"] == req.username:
55            raise HTTPException(status_code=400, detail="Username taken")
56    user_id = next_user_id
57    next_user_id += 1
58    users[user_id] = {
59        "id": user_id,
60        "username": req.username,
61        "password": hashlib.sha256(req.password.encode()).hexdigest()
62    }
63    token = generate_token()
64    tokens[token] = user_id
65    return {"user_id": user_id, "token": token}
66 
67@app.post("/login")
68def login(req: LoginRequest):
69    for u in users.values():
70        if u["username"] == req.username and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
71            token = generate_token()
72            tokens[token] = u["id"]
73            return {"user_id": u["id"], "token": token}
74    raise HTTPException(status_code=401, detail="Invalid credentials")
75 
76@app.get("/players/{player_id}")
77def get_player(player_id: int, authorization: str = Header(None)):
78    get_current_user(authorization)
79    player = players.get(player_id)
80    if not player:
81        raise HTTPException(status_code=404, detail="Player not found")
82    return player
83 
84@app.post("/players")
85def create_player(player: PlayerCreate, authorization: str = Header(None)):
86    global next_player_id
87    get_current_user(authorization)
88    player_id = next_player_id
89    next_player_id += 1
90    players[player_id] = {
91        "id": player_id,
92        "display_name": player.display_name,
93        "score": player.score,
94        "rank": player.rank,
95        "tier": player.tier,
96        "admin": player.admin
97    }
98    return players[player_id]
99 
100@app.patch("/players/{player_id}")
101def update_player(player_id: int, update: PlayerUpdate, authorization: str = Header(None)):
102    get_current_user(authorization)
103    player = players.get(player_id)
104    if not player:
105        raise HTTPException(status_code=404, detail="Player not found")
106    update_data = update.dict(exclude_unset=True)
107    for key, value in update_data.items():
108        player[key] = value
109    return player

1fastapi
2uvicorn