Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5import hashlib
6 
7app = FastAPI()
8 
9users = {}
10members = {}
11tokens = {}
12next_user_id = 1
13next_member_id = 1
14 
15def hash_password(password: str) -> str:
16    return hashlib.sha256(password.encode()).hexdigest()
17 
18def get_current_user(authorization: str = Header(...)):
19    if not authorization or not authorization.startswith("Bearer "):
20        raise HTTPException(status_code=401, detail="Invalid auth")
21    token = authorization.split(" ")[1]
22    user_id = tokens.get(token)
23    if user_id is None:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return user_id
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class MemberUpdate(BaseModel):
36    name: Optional[str] = None
37    points_balance: Optional[int] = None
38    tier_level: Optional[str] = None
39    referral_code: Optional[str] = None
40    admin_override: Optional[bool] = None
41 
42class MemberCreate(BaseModel):
43    name: str
44    points_balance: int = 0
45    tier_level: str = "Bronze"
46    referral_code: Optional[str] = None
47    admin_override: bool = False
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    user_id = next_user_id
53    next_user_id += 1
54    users[user_id] = {"username": req.username, "password": hash_password(req.password)}
55    return {"user_id": user_id, "message": "User created"}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    for uid, u in users.items():
60        if u["username"] == req.username and u["password"] == hash_password(req.password):
61            token = str(uuid.uuid4())
62            tokens[token] = uid
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.get("/members/{member_id}")
67def get_member(member_id: int, authorization: str = Header(...)):
68    get_current_user(authorization)
69    if member_id not in members:
70        raise HTTPException(status_code=404, detail="Member not found")
71    return members[member_id]
72 
73@app.post("/members")
74def create_member(req: MemberCreate, authorization: str = Header(...)):
75    global next_member_id
76    get_current_user(authorization)
77    member_id = next_member_id
78    next_member_id += 1
79    members[member_id] = {
80        "id": member_id,
81        "name": req.name,
82        "points_balance": req.points_balance,
83        "tier_level": req.tier_level,
84        "referral_code": req.referral_code,
85        "admin_override": req.admin_override
86    }
87    return members[member_id]
88 
89@app.patch("/members/{member_id}")
90def update_member(member_id: int, req: MemberUpdate, authorization: str = Header(...)):
91    get_current_user(authorization)
92    if member_id not in members:
93        raise HTTPException(status_code=404, detail="Member not found")
94    member = members[member_id]
95    if req.name is not None:
96        member["name"] = req.name
97    if req.points_balance is not None:
98        member["points_balance"] = req.points_balance
99    if req.tier_level is not None:
100        member["tier_level"] = req.tier_level
101    if req.referral_code is not None:
102        member["referral_code"] = req.referral_code
103    if req.admin_override is not None:
104        member["admin_override"] = req.admin_override
105    return member

1fastapi
2uvicorn