Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10tenants = {}
11next_user_id = 1
12next_tenant_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class TenantCreate(BaseModel):
23    name: str
24    email: str
25    phone: str = ""
26    lease_info: str = ""
27    role: str = "tenant"
28    tier: str = "standard"
29 
30class TenantUpdate(BaseModel):
31    name: Optional[str] = None
32    email: Optional[str] = None
33    phone: Optional[str] = None
34    lease_info: Optional[str] = None
35    role: Optional[str] = None
36    tier: Optional[str] = None
37 
38def get_current_user(authorization: str = Header(...)):
39    if not authorization.startswith("Bearer "):
40        raise HTTPException(status_code=401, detail="Invalid auth header")
41    token = authorization[7:]
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global next_user_id
49    for user in users.values():
50        if user["username"] == req.username:
51            raise HTTPException(status_code=400, detail="User exists")
52    user_id = next_user_id
53    next_user_id += 1
54    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
55    return {"id": user_id, "username": req.username}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    for user in users.values():
60        if user["username"] == req.username and user["password"] == req.password:
61            token = secrets.token_hex(16)
62            tokens[token] = user["id"]
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.get("/tenants/{tenant_id}")
67def get_tenant(tenant_id: int, authorization: str = Header(...)):
68    get_current_user(authorization)
69    if tenant_id not in tenants:
70        raise HTTPException(status_code=404, detail="Tenant not found")
71    return tenants[tenant_id]
72 
73@app.post("/tenants")
74def create_tenant(req: TenantCreate, authorization: str = Header(...)):
75    global next_tenant_id
76    get_current_user(authorization)
77    tenant_id = next_tenant_id
78    next_tenant_id += 1
79    tenants[tenant_id] = {
80        "id": tenant_id,
81        "name": req.name,
82        "email": req.email,
83        "phone": req.phone,
84        "lease_info": req.lease_info,
85        "role": req.role,
86        "tier": req.tier
87    }
88    return tenants[tenant_id]
89 
90@app.patch("/tenants/{tenant_id}/profile")
91def update_tenant_profile(tenant_id: int, req: TenantUpdate, authorization: str = Header(...)):
92    get_current_user(authorization)
93    if tenant_id not in tenants:
94        raise HTTPException(status_code=404, detail="Tenant not found")
95    tenant = tenants[tenant_id]
96    if req.name is not None:
97        tenant["name"] = req.name
98    if req.email is not None:
99        tenant["email"] = req.email
100    if req.phone is not None:
101        tenant["phone"] = req.phone
102    if req.lease_info is not None:
103        tenant["lease_info"] = req.lease_info
104    if req.role is not None:
105        tenant["role"] = req.role
106    if req.tier is not None:
107        tenant["tier"] = req.tier
108    return tenant

1fastapi
2uvicorn