Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9properties = {}
10tenants = {}
11next_user_id = 1
12next_property_id = 1
13next_tenant_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 token")
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    if any(u["username"] == username for u in users.values()):
27        raise HTTPException(status_code=400, detail="Username taken")
28    user_id = next_user_id
29    next_user_id += 1
30    users[user_id] = {"id": user_id, "username": username, "password": password}
31    return {"id": user_id, "username": username}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    for u in users.values():
36        if u["username"] == username and u["password"] == password:
37            token = secrets.token_hex(16)
38            tokens[token] = u["id"]
39            return {"token": token}
40    raise HTTPException(status_code=401, detail="Invalid credentials")
41 
42@app.get("/properties/{property_id}")
43def get_property(property_id: int, authorization: Optional[str] = Header(None)):
44    user_id = get_current_user(authorization)
45    if property_id not in properties:
46        raise HTTPException(status_code=404, detail="Property not found")
47    return properties[property_id]
48 
49@app.post("/properties")
50def create_property(rent_amount: float, lease_end_date: str, maintenance_request_count: int = 0, authorization: Optional[str] = Header(None)):
51    user_id = get_current_user(authorization)
52    global next_property_id
53    prop_id = next_property_id
54    next_property_id += 1
55    properties[prop_id] = {
56        "id": prop_id,
57        "rent_amount": rent_amount,
58        "lease_end_date": lease_end_date,
59        "maintenance_request_count": maintenance_request_count
60    }
61    return properties[prop_id]
62 
63@app.get("/properties")
64def get_all_properties(authorization: Optional[str] = Header(None)):
65    user_id = get_current_user(authorization)
66    return list(properties.values())
67 
68@app.get("/tenants/{tenant_id}")
69def get_tenant(tenant_id: int, authorization: Optional[str] = Header(None)):
70    user_id = get_current_user(authorization)
71    if tenant_id not in tenants:
72        raise HTTPException(status_code=404, detail="Tenant not found")
73    return tenants[tenant_id]
74 
75@app.post("/tenants")
76def create_tenant(name: str, lease_start: str, last_payment_date: str, authorization: Optional[str] = Header(None)):
77    user_id = get_current_user(authorization)
78    global next_tenant_id
79    tenant_id = next_tenant_id
80    next_tenant_id += 1
81    tenants[tenant_id] = {
82        "id": tenant_id,
83        "name": name,
84        "lease_start": lease_start,
85        "last_payment_date": last_payment_date
86    }
87    return tenants[tenant_id]
88 
89@app.get("/tenants")
90def get_all_tenants(authorization: Optional[str] = Header(None)):
91    user_id = get_current_user(authorization)
92    return list(tenants.values())

1fastapi
2uvicorn