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 = {}
10properties = {}
11next_user_id = 1
12next_property_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class PropertyCreate(BaseModel):
23    name: str
24    address: str
25    rent_amount: float
26    status: Optional[str] = "active"
27    manager_id: Optional[int] = None
28 
29class PropertyUpdate(BaseModel):
30    name: Optional[str] = None
31    address: Optional[str] = None
32    rent_amount: Optional[float] = None
33    status: Optional[str] = None
34    manager_id: Optional[int] = None
35 
36def get_current_user(token: str = Header(...)):
37    if token not in tokens:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return tokens[token]
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    user_id = next_user_id
45    next_user_id += 1
46    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
47    return {"id": user_id, "username": req.username}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    for uid, u in users.items():
52        if u["username"] == req.username and u["password"] == req.password:
53            token = secrets.token_hex(16)
54            tokens[token] = uid
55            return {"token": token}
56    raise HTTPException(status_code=401, detail="Invalid credentials")
57 
58@app.post("/properties")
59def create_property(req: PropertyCreate, token: str = Header(...)):
60    get_current_user(token)
61    global next_property_id
62    prop_id = next_property_id
63    next_property_id += 1
64    properties[prop_id] = {
65        "id": prop_id,
66        "name": req.name,
67        "address": req.address,
68        "rent_amount": req.rent_amount,
69        "status": req.status,
70        "manager_id": req.manager_id
71    }
72    return properties[prop_id]
73 
74@app.get("/properties/{prop_id}")
75def get_property(prop_id: int, token: str = Header(...)):
76    get_current_user(token)
77    if prop_id not in properties:
78        raise HTTPException(status_code=404, detail="Property not found")
79    return properties[prop_id]
80 
81@app.patch("/properties/{prop_id}")
82def update_property(prop_id: int, req: PropertyUpdate, token: str = Header(...)):
83    get_current_user(token)
84    if prop_id not in properties:
85        raise HTTPException(status_code=404, detail="Property not found")
86    prop = properties[prop_id]
87    if req.name is not None:
88        prop["name"] = req.name
89    if req.address is not None:
90        prop["address"] = req.address
91    if req.rent_amount is not None:
92        prop["rent_amount"] = req.rent_amount
93    if req.status is not None:
94        prop["status"] = req.status
95    if req.manager_id is not None:
96        prop["manager_id"] = req.manager_id
97    return prop

1fastapi
2uvicorn