Browse the data

1import random
2import hashlib
3import secrets
4from fastapi import FastAPI, HTTPException, Header
5from pydantic import BaseModel
6from typing import Optional
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12pull_history = {}
13lootbox_items = []
14 
15rarities = ["common", "uncommon", "rare", "legendary"]
16rarity_weights = [0.5, 0.3, 0.15, 0.05]
17value_ranges = {
18    "common": (1, 10),
19    "uncommon": (10, 50),
20    "rare": (50, 200),
21    "legendary": (200, 1000)
22}
23 
24item_names = [
25    "Shadow Blade", "Phoenix Feather", "Dragon Scale", "Elixir of Power",
26    "Mystic Orb", "Iron Shield", "Healing Salve", "Thunder Staff",
27    "Crystal Amulet", "Warp Stone", "Frost Ring", "Ember Tome",
28    "Void Shard", "Storm Bow", "Guardian Helm", "Soul Gem"
29]
30 
31for name in item_names:
32    rarity = random.choices(rarities, weights=rarity_weights)[0]
33    min_val, max_val = value_ranges[rarity]
34    value = round(random.uniform(min_val, max_val), 2)
35    lootbox_items.append({
36        "id": len(lootbox_items) + 1,
37        "name": name,
38        "rarity": rarity,
39        "value": value
40    })
41 
42class SignupRequest(BaseModel):
43    username: str
44    password: str
45 
46class LoginRequest(BaseModel):
47    username: str
48    password: str
49 
50class LootboxResponse(BaseModel):
51    item_id: int
52    name: str
53    rarity: str
54    value: float
55 
56def hash_password(password: str) -> str:
57    return hashlib.sha256(password.encode()).hexdigest()
58 
59def get_user_id_from_token(authorization: str = Header(None)):
60    if not authorization:
61        raise HTTPException(status_code=401, detail="Missing auth token")
62    token = authorization.replace("Bearer ", "")
63    if token not in tokens:
64        raise HTTPException(status_code=401, detail="Invalid token")
65    return tokens[token]
66 
67@app.post("/signup")
68def signup(req: SignupRequest):
69    if req.username in users:
70        raise HTTPException(status_code=400, detail="Username already exists")
71    user_id = len(users) + 1
72    users[req.username] = {
73        "id": user_id,
74        "username": req.username,
75        "password": hash_password(req.password)
76    }
77    token = secrets.token_hex(16)
78    tokens[token] = user_id
79    pull_history[user_id] = []
80    return {"user_id": user_id, "token": token}
81 
82@app.post("/login")
83def login(req: LoginRequest):
84    if req.username not in users:
85        raise HTTPException(status_code=401, detail="Invalid credentials")
86    user = users[req.username]
87    if user["password"] != hash_password(req.password):
88        raise HTTPException(status_code=401, detail="Invalid credentials")
89    token = secrets.token_hex(16)
90    tokens[token] = user["id"]
91    pull_history[user["id"]] = pull_history.get(user["id"], [])
92    return {"user_id": user["id"], "token": token}
93 
94@app.get("/lootbox", response_model=LootboxResponse)
95def open_lootbox(authorization: str = Header(None)):
96    user_id = get_user_id_from_token(authorization)
97    item = random.choice(lootbox_items)
98    pull_history[user_id].append(item)
99    return LootboxResponse(
100        item_id=item["id"],
101        name=item["name"],
102        rarity=item["rarity"],
103        value=item["value"]
104    )
105 
106@app.get("/history")
107def get_history(authorization: str = Header(None)):
108    user_id = get_user_id_from_token(authorization)
109    return {"pulls": pull_history.get(user_id, [])}
110 
111@app.get("/lootbox/{item_id}")
112def get_lootbox_item(item_id: int):
113    for item in lootbox_items:
114        if item["id"] == item_id:
115            return item
116    raise HTTPException(status_code=404, detail="Item not found")
117 
118@app.post("/lootbox")
119def create_lootbox_item(name: str, rarity: str = "common", value: float = 0.0):
120    if rarity not in rarities:
121        raise HTTPException(status_code=400, detail="Invalid rarity")
122    new_id = len(lootbox_items) + 1
123    new_item = {
124        "id": new_id,
125        "name": name,
126        "rarity": rarity,
127        "value": value
128    }
129    lootbox_items.append(new_item)
130    return new_item

1fastapi
2uvicorn