Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9tenants = {}
10listings = {}
11 
12user_id_counter = 1
13tenant_id_counter = 1
14listing_id_counter = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if not user_id:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return user_id
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global user_id_counter
28    user_id = user_id_counter
29    user_id_counter += 1
30    users[user_id] = {"username": username, "password": password, "role": "user"}
31    return {"user_id": user_id}
32 
33@app.post("/login")
34def login(username: str, password: str):
35    for uid, u in users.items():
36        if u["username"] == username and u["password"] == password:
37            token = secrets.token_hex(16)
38            tokens[token] = uid
39            return {"token": token}
40    raise HTTPException(status_code=401, detail="Invalid credentials")
41 
42@app.post("/tenants")
43def create_tenant(name: str, authorization: Optional[str] = Header(None)):
44    user_id = get_current_user(authorization)
45    global tenant_id_counter
46    tenant_id = tenant_id_counter
47    tenant_id_counter += 1
48    tenants[tenant_id] = {"name": name, "admin_id": user_id, "managers": []}
49    return {"tenant_id": tenant_id}
50 
51@app.get("/tenants/{tenant_id}")
52def get_tenant(tenant_id: int, authorization: Optional[str] = Header(None)):
53    user_id = get_current_user(authorization)
54    tenant = tenants.get(tenant_id)
55    if not tenant:
56        raise HTTPException(status_code=404, detail="Tenant not found")
57    return tenant
58 
59@app.post("/tenants/{tenant_id}/promote")
60def promote_to_manager(tenant_id: int, user_id: int, authorization: Optional[str] = Header(None)):
61    current_user = get_current_user(authorization)
62    tenant = tenants.get(tenant_id)
63    if not tenant:
64        raise HTTPException(status_code=404, detail="Tenant not found")
65    if tenant["admin_id"] != current_user:
66        raise HTTPException(status_code=403, detail="Only tenant admin can promote")
67    if user_id not in users:
68        raise HTTPException(status_code=404, detail="User not found")
69    if user_id not in tenant["managers"]:
70        tenant["managers"].append(user_id)
71    return {"message": "User promoted to manager"}
72 
73@app.post("/listings")
74def create_listing(address: str, price: float, square_feet: float, tenant_id: int, authorization: Optional[str] = Header(None)):
75    user_id = get_current_user(authorization)
76    tenant = tenants.get(tenant_id)
77    if not tenant:
78        raise HTTPException(status_code=404, detail="Tenant not found")
79    if tenant["admin_id"] != user_id and user_id not in tenant["managers"]:
80        raise HTTPException(status_code=403, detail="Only admin or manager can create listings")
81    global listing_id_counter
82    listing_id = listing_id_counter
83    listing_id_counter += 1
84    listings[listing_id] = {"address": address, "price": price, "square_feet": square_feet, "tenant_id": tenant_id, "created_by": user_id}
85    return {"listing_id": listing_id}
86 
87@app.get("/listings/{listing_id}")
88def get_listing(listing_id: int, authorization: Optional[str] = Header(None)):
89    user_id = get_current_user(authorization)
90    listing = listings.get(listing_id)
91    if not listing:
92        raise HTTPException(status_code=404, detail="Listing not found")
93    return listing

1fastapi
2uvicorn