Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3from datetime import datetime, date
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11transactions = {}
12budgets = {}
13next_user_id = 1
14next_transaction_id = 1
15next_budget_id = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth token")
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28@app.post("/signup")
29def signup(username: str, password: str):
30    global next_user_id
31    for user in users.values():
32        if user["username"] == username:
33            raise HTTPException(status_code=400, detail="Username already exists")
34    user_id = next_user_id
35    next_user_id += 1
36    users[user_id] = {"id": user_id, "username": username, "password": password}
37    return {"id": user_id, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for user in users.values():
42        if user["username"] == username and user["password"] == password:
43            token = generate_token()
44            tokens[token] = user["id"]
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.get("/transactions/{id}")
49def get_transaction(id: int, authorization: Optional[str] = Header(None)):
50    get_current_user(authorization)
51    if id not in transactions:
52        raise HTTPException(status_code=404, detail="Transaction not found")
53    return transactions[id]
54 
55@app.get("/transactions")
56def get_transactions(authorization: Optional[str] = Header(None)):
57    get_current_user(authorization)
58    return list(transactions.values())
59 
60@app.post("/transactions")
61def create_transaction(amount: float, category: str, date: str, authorization: Optional[str] = Header(None)):
62    global next_transaction_id
63    user_id = get_current_user(authorization)
64    tid = next_transaction_id
65    next_transaction_id += 1
66    transactions[tid] = {
67        "id": tid,
68        "user_id": user_id,
69        "amount": amount,
70        "category": category,
71        "date": date
72    }
73    return transactions[tid]
74 
75@app.get("/budgets/{id}")
76def get_budget(id: int, authorization: Optional[str] = Header(None)):
77    get_current_user(authorization)
78    if id not in budgets:
79        raise HTTPException(status_code=404, detail="Budget not found")
80    return budgets[id]
81 
82@app.get("/budgets")
83def get_budgets(authorization: Optional[str] = Header(None)):
84    get_current_user(authorization)
85    return list(budgets.values())
86 
87@app.post("/budgets")
88def create_budget(month: str, category: str, budget_amount: float, authorization: Optional[str] = Header(None)):
89    global next_budget_id
90    user_id = get_current_user(authorization)
91    bid = next_budget_id
92    next_budget_id += 1
93    
94    total_spent = sum(t["amount"] for t in transactions.values() if t["user_id"] == user_id and t["category"] == category and t["date"].startswith(month))
95    
96    budgets[bid] = {
97        "id": bid,
98        "user_id": user_id,
99        "month": month,
100        "category": category,
101        "budget_amount": budget_amount,
102        "spent": total_spent,
103        "remaining": budget_amount - total_spent
104    }
105    return budgets[bid]

1fastapi
2uvicorn