Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9landlords = {}
10tenants = {}
11properties = {}
12units = {}
13maintenance_requests = {}
14inspection_reports = {}
15 
16next_user_id = 1
17next_landlord_id = 1
18next_tenant_id = 1
19next_property_id = 1
20next_unit_id = 1
21next_request_id = 1
22next_report_id = 1
23 
24def get_current_user(authorization: Optional[str] = Header(None)):
25    if not authorization:
26        raise HTTPException(status_code=401, detail="Missing auth header")
27    token = authorization.replace("Bearer ", "")
28    for uid, t in tokens.items():
29        if t == token:
30            return uid
31    raise HTTPException(status_code=401, detail="Invalid token")
32 
33@app.post("/signup")
34def signup(username: str, password: str, role: str = "tenant"):
35    global next_user_id
36    if any(u["username"] == username for u in users.values()):
37        raise HTTPException(status_code=400, detail="User exists")
38    uid = next_user_id
39    users[uid] = {"id": uid, "username": username, "password": password, "role": role}
40    next_user_id += 1
41    return {"id": uid, "username": username, "role": role}
42 
43@app.post("/login")
44def login(username: str, password: str):
45    for uid, u in users.items():
46        if u["username"] == username and u["password"] == password:
47            token = secrets.token_hex(16)
48            tokens[uid] = token
49            return {"token": token, "user_id": uid}
50    raise HTTPException(status_code=401, detail="Invalid credentials")
51 
52@app.post("/landlords")
53def create_landlord(name: str, email: str, authorization: Optional[str] = Header(None)):
54    uid = get_current_user(authorization)
55    global next_landlord_id
56    lid = next_landlord_id
57    landlords[lid] = {"id": lid, "name": name, "email": email, "user_id": uid}
58    next_landlord_id += 1
59    return landlords[lid]
60 
61@app.get("/landlords/{landlord_id}")
62def get_landlord(landlord_id: int, authorization: Optional[str] = Header(None)):
63    get_current_user(authorization)
64    if landlord_id not in landlords:
65        raise HTTPException(status_code=404, detail="Not found")
66    return landlords[landlord_id]
67 
68@app.post("/properties")
69def create_property(name: str, address: str, landlord_id: int, authorization: Optional[str] = Header(None)):
70    get_current_user(authorization)
71    if landlord_id not in landlords:
72        raise HTTPException(status_code=400, detail="Invalid landlord")
73    global next_property_id
74    pid = next_property_id
75    properties[pid] = {"id": pid, "name": name, "address": address, "landlord_id": landlord_id}
76    next_property_id += 1
77    return properties[pid]
78 
79@app.get("/properties/{property_id}")
80def get_property(property_id: int, authorization: Optional[str] = Header(None)):
81    get_current_user(authorization)
82    if property_id not in properties:
83        raise HTTPException(status_code=404, detail="Not found")
84    return properties[property_id]
85 
86@app.post("/units")
87def create_unit(property_id: int, unit_number: str, rent_amount: float, authorization: Optional[str] = Header(None)):
88    get_current_user(authorization)
89    if property_id not in properties:
90        raise HTTPException(status_code=400, detail="Invalid property")
91    global next_unit_id
92    uid = next_unit_id
93    units[uid] = {"id": uid, "property_id": property_id, "unit_number": unit_number, "rent_amount": rent_amount}
94    next_unit_id += 1
95    return units[uid]
96 
97@app.get("/units/{unit_id}")
98def get_unit(unit_id: int, authorization: Optional[str] = Header(None)):
99    get_current_user(authorization)
100    if unit_id not in units:
101        raise HTTPException(status_code=404, detail="Not found")
102    return units[unit_id]
103 
104@app.post("/tenants")
105def create_tenant(name: str, email: str, unit_id: int, authorization: Optional[str] = Header(None)):
106    uid = get_current_user(authorization)
107    if unit_id not in units:
108        raise HTTPException(status_code=400, detail="Invalid unit")
109    global next_tenant_id
110    tid = next_tenant_id
111    tenants[tid] = {"id": tid, "name": name, "email": email, "unit_id": unit_id, "user_id": uid}
112    next_tenant_id += 1
113    return tenants[tid]
114 
115@app.get("/tenants/{tenant_id}")
116def get_tenant(tenant_id: int, authorization: Optional[str] = Header(None)):
117    get_current_user(authorization)
118    if tenant_id not in tenants:
119        raise HTTPException(status_code=404, detail="Not found")
120    return tenants[tenant_id]
121 
122@app.post("/maintenance-requests")
123def create_maintenance_request(tenant_id: int, unit_id: int, description: str, authorization: Optional[str] = Header(None)):
124    get_current_user(authorization)
125    if tenant_id not in tenants:
126        raise HTTPException(status_code=400, detail="Invalid tenant")
127    if unit_id not in units:
128        raise HTTPException(status_code=400, detail="Invalid unit")
129    global next_request_id
130    rid = next_request_id
131    maintenance_requests[rid] = {"id": rid, "tenant_id": tenant_id, "unit_id": unit_id, "description": description, "status": "open"}
132    next_request_id += 1
133    return maintenance_requests[rid]
134 
135@app.get("/maintenance-requests/{request_id}")
136def get_maintenance_request(request_id: int, authorization: Optional[str] = Header(None)):
137    get_current_user(authorization)
138    if request_id not in maintenance_requests:
139        raise HTTPException(status_code=404, detail="Not found")
140    return maintenance_requests[request_id]
141 
142@app.post("/inspection-reports")
143def create_inspection_report(property_id: int, inspector_name: str, notes: str, rating: int, authorization: Optional[str] = Header(None)):
144    get_current_user(authorization)
145    if property_id not in properties:
146        raise HTTPException(status_code=400, detail="Invalid property")
147    global next_report_id
148    rid = next_report_id
149    inspection_reports[rid] = {"id": rid, "property_id": property_id, "inspector_name": inspector_name, "notes": notes, "rating": rating}
150    next_report_id += 1
151    return inspection_reports[rid]
152 
153@app.get("/inspection-reports/{report_id}")
154def get_inspection_report(report_id: int, authorization: Optional[str] = Header(None)):
155    get_current_user(authorization)
156    if report_id not in inspection_reports:
157        raise HTTPException(status_code=404, detail="Not found")
158    return inspection_reports[report_id]
159 
160@app.get("/inspection-reports/by-property/{property_id}")
161def get_inspection_reports_by_property(property_id: int):
162    if property_id not in properties:
163        raise HTTPException(status_code=404, detail="Property not found")
164    reports = [r for r in inspection_reports.values() if r["property_id"] == property_id]
165    return reports

1fastapi
2uvicorn