Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9transactions = {}
10budgets = {}
11next_user_id = 1
12next_tx_id = 1
13next_budget_id = 1
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="Missing auth header")
18    token = authorization.replace("Bearer ", "")
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    global next_user_id
26    user_id = next_user_id
27    users[user_id] = {"id": user_id, "username": username, "password": password}
28    next_user_id += 1
29    return {"user_id": user_id, "username": username}
30 
31@app.post("/login")
32def login(username: str, password: str):
33    for uid, u in users.items():
34        if u["username"] == username and u["password"] == password:
35            token = secrets.token_hex(16)
36            tokens[token] = uid
37            return {"token": token}
38    raise HTTPException(status_code=401, detail="Invalid credentials")
39 
40@app.get("/transactions/{transaction_id}")
41def get_transaction(transaction_id: int, authorization: Optional[str] = Header(None)):
42    user_id = get_current_user(authorization)
43    if transaction_id not in transactions:
44        raise HTTPException(status_code=404, detail="Transaction not found")
45    return transactions[transaction_id]
46 
47@app.post("/transactions")
48def create_transaction(category: str, amount: float, merchant: str, authorization: Optional[str] = Header(None)):
49    global next_tx_id
50    user_id = get_current_user(authorization)
51    tx_id = next_tx_id
52    transactions[tx_id] = {"id": tx_id, "category": category, "amount": amount, "merchant": merchant, "user_id": user_id}
53    next_tx_id += 1
54    return transactions[tx_id]
55 
56@app.get("/budgets/{budget_id}")
57def get_budget(budget_id: int, authorization: Optional[str] = Header(None)):
58    user_id = get_current_user(authorization)
59    if budget_id not in budgets:
60        raise HTTPException(status_code=404, detail="Budget not found")
61    return budgets[budget_id]
62 
63@app.post("/budgets")
64def create_budget(category: str, monthly_limit: float, authorization: Optional[str] = Header(None)):
65    global next_budget_id
66    user_id = get_current_user(authorization)
67    budget_id = next_budget_id
68    total_spent = sum(t["amount"] for t in transactions.values() if t["category"] == category and t["user_id"] == user_id)
69    spend_pct = (total_spent / monthly_limit) * 100 if monthly_limit > 0 else 0
70    budgets[budget_id] = {
71        "id": budget_id,
72        "category": category,
73        "monthly_limit": monthly_limit,
74        "current_spend": total_spent,
75        "spend_percentage": round(spend_pct, 2),
76        "user_id": user_id
77    }
78    next_budget_id += 1
79    return budgets[budget_id]
80 
81@app.get("/transactions")
82def get_recent_transactions(limit: int = 10, authorization: Optional[str] = Header(None)):
83    user_id = get_current_user(authorization)
84    user_txs = [t for t in transactions.values() if t["user_id"] == user_id]
85    sorted_txs = sorted(user_txs, key=lambda x: x["id"], reverse=True)
86    return sorted_txs[:limit]
87 
88@app.get("/budgets")
89def get_budgets(authorization: Optional[str] = Header(None)):
90    user_id = get_current_user(authorization)
91    user_budgets = [b for b in budgets.values() if b["user_id"] == user_id]
92    for b in user_budgets:
93        total_spent = sum(t["amount"] for t in transactions.values() if t["category"] == b["category"] and t["user_id"] == user_id)
94        b["current_spend"] = total_spent
95        b["spend_percentage"] = round((total_spent / b["monthly_limit"]) * 100, 2) if b["monthly_limit"] > 0 else 0
96    return user_budgets

1fastapi
2uvicorn