Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import json, os, random, string, hashlib
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9teams = {}
10 
11AUTH_TOKEN_LENGTH = 32
12 
13def generate_token():
14    return ''.join(random.choices(string.ascii_letters + string.digits, k=AUTH_TOKEN_LENGTH))
15 
16def hash_password(password: str) -> str:
17    return hashlib.sha256(password.encode()).hexdigest()
18 
19def get_current_user(authorization: str = Header(None)) -> int:
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing Authorization header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class TeamCreate(BaseModel):
36    name: str
37    sport: str
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    if req.username in users:
42        raise HTTPException(status_code=400, detail="User already exists")
43    user_id = len(users) + 1
44    users[req.username] = {"id": user_id, "password": hash_password(req.password)}
45    token = generate_token()
46    tokens[token] = user_id
47    return {"user_id": user_id, "token": token}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    if req.username not in users:
52        raise HTTPException(status_code=400, detail="Invalid credentials")
53    if users[req.username]["password"] != hash_password(req.password):
54        raise HTTPException(status_code=400, detail="Invalid credentials")
55    token = generate_token()
56    tokens[token] = users[req.username]["id"]
57    return {"token": token}
58 
59@app.get("/teams/{team_id}")
60def get_team(team_id: int, authorization: str = Header(None)):
61    get_current_user(authorization)
62    if team_id not in teams:
63        raise HTTPException(status_code=404, detail="Team not found")
64    return teams[team_id]
65 
66@app.post("/teams")
67def create_team(team: TeamCreate, authorization: str = Header(None)):
68    get_current_user(authorization)
69    team_id = len(teams) + 1
70    teams[team_id] = {"id": team_id, "name": team.name, "sport": team.sport}
71    return teams[team_id]
72 
73@app.get("/team/{team_id}/stats/{file:path}")
74def get_team_stats(team_id: int, file: str, authorization: str = Header(None)):
75    get_current_user(authorization)
76    base_path = f"/var/sports/{team_id}"
77    file_path = os.path.join(base_path, file)
78    if not os.path.exists(file_path):
79        raise HTTPException(status_code=404, detail="File not found")
80    with open(file_path, "r") as f:
81        content = f.read()
82    try:
83        return json.loads(content)
84    except json.JSONDecodeError:
85        return {"content": content}
86 
87@app.get("/team/{team_id}/compare/{season1}/{season2}")
88def compare_seasons(team_id: int, season1: str, season2: str, authorization: str = Header(None)):
89    get_current_user(authorization)
90    base_path = f"/var/sports/{team_id}"
91    s1_path = os.path.join(base_path, season1, "stats.json")
92    s2_path = os.path.join(base_path, season2, "stats.json")
93    result = {}
94    for label, path in [("season1", s1_path), ("season2", s2_path)]:
95        if not os.path.exists(path):
96            result[label] = None
97        else:
98            with open(path, "r") as f:
99                result[label] = json.load(f)
100    return result

1fastapi
2uvicorn