Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10tasks = {}
11vendors = {}
12budgets = {}
13task_id_counter = 1
14vendor_id_counter = 1
15budget_id_counter = 1
16user_id_counter = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth header")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return tokens[token]
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global user_id_counter
29    if username in users:
30        raise HTTPException(status_code=400, detail="User exists")
31    user_id = user_id_counter
32    user_id_counter += 1
33    users[username] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
34    token = secrets.token_hex(16)
35    tokens[token] = username
36    return {"user_id": user_id, "token": token}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    if username not in users:
41        raise HTTPException(status_code=400, detail="Invalid credentials")
42    if users[username]["password"] != hashlib.sha256(password.encode()).hexdigest():
43        raise HTTPException(status_code=400, detail="Invalid credentials")
44    token = secrets.token_hex(16)
45    tokens[token] = username
46    return {"token": token}
47 
48@app.get("/tasks/{task_id}")
49def get_task(task_id: int, authorization: Optional[str] = Header(None)):
50    get_current_user(authorization)
51    if task_id not in tasks:
52        raise HTTPException(status_code=404, detail="Task not found")
53    return tasks[task_id]
54 
55@app.post("/tasks")
56def create_task(name: str, category: str, authorization: Optional[str] = Header(None)):
57    global task_id_counter
58    get_current_user(authorization)
59    task_id = task_id_counter
60    task_id_counter += 1
61    tasks[task_id] = {"id": task_id, "name": name, "category": category, "vendor_id": None}
62    return tasks[task_id]
63 
64@app.get("/vendors/{vendor_id}")
65def get_vendor(vendor_id: int, authorization: Optional[str] = Header(None)):
66    get_current_user(authorization)
67    if vendor_id not in vendors:
68        raise HTTPException(status_code=404, detail="Vendor not found")
69    return vendors[vendor_id]
70 
71@app.post("/vendors")
72def create_vendor(name: str, task_id: int, authorization: Optional[str] = Header(None)):
73    global vendor_id_counter
74    get_current_user(authorization)
75    if task_id not in tasks:
76        raise HTTPException(status_code=404, detail="Task not found")
77    vendor_id = vendor_id_counter
78    vendor_id_counter += 1
79    vendors[vendor_id] = {"id": vendor_id, "name": name, "task_id": task_id}
80    tasks[task_id]["vendor_id"] = vendor_id
81    return vendors[vendor_id]
82 
83@app.get("/budgets/{budget_id}")
84def get_budget(budget_id: int, authorization: Optional[str] = Header(None)):
85    get_current_user(authorization)
86    if budget_id not in budgets:
87        raise HTTPException(status_code=404, detail="Budget not found")
88    return budgets[budget_id]
89 
90@app.post("/budgets")
91def create_budget(category: str, amount: float, authorization: Optional[str] = Header(None)):
92    global budget_id_counter
93    get_current_user(authorization)
94    budget_id = budget_id_counter
95    budget_id_counter += 1
96    budgets[budget_id] = {"id": budget_id, "category": category, "amount": amount, "spent": 0.0}
97    return budgets[budget_id]

1fastapi
2uvicorn