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 = {}
11inquiries = {}
12next_user_id = 1
13next_property_id = 1
14next_inquiry_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class PropertyCreate(BaseModel):
25    price: float
26    bedrooms: int
27    location: str
28 
29class InquiryCreate(BaseModel):
30    message: str
31    property_id: int
32 
33def get_current_user(authorization: Optional[str] = Header(None)):
34    if not authorization:
35        raise HTTPException(status_code=401, detail="Missing auth header")
36    token = authorization.replace("Bearer ", "")
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    if req.username in users:
45        raise HTTPException(status_code=400, detail="Username already exists")
46    user_id = next_user_id
47    next_user_id += 1
48    users[req.username] = {"id": user_id, "password": req.password, "username": req.username}
49    return {"id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    if req.username not in users or users[req.username]["password"] != req.password:
54        raise HTTPException(status_code=401, detail="Invalid credentials")
55    token = secrets.token_hex(16)
56    tokens[token] = req.username
57    return {"token": token}
58 
59@app.get("/properties")
60def get_properties():
61    result = []
62    for pid, prop in properties.items():
63        result.append({
64            "id": pid,
65            "price": prop["price"],
66            "bedrooms": prop["bedrooms"],
67            "location": prop["location"]
68        })
69    return result
70 
71@app.post("/properties")
72def create_property(prop: PropertyCreate, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_property_id
75    pid = next_property_id
76    next_property_id += 1
77    properties[pid] = {
78        "price": prop.price,
79        "bedrooms": prop.bedrooms,
80        "location": prop.location
81    }
82    return {"id": pid}
83 
84@app.get("/properties/{property_id}")
85def get_property(property_id: int):
86    if property_id not in properties:
87        raise HTTPException(status_code=404, detail="Property not found")
88    prop = properties[property_id]
89    return {
90        "id": property_id,
91        "price": prop["price"],
92        "bedrooms": prop["bedrooms"],
93        "location": prop["location"]
94    }
95 
96@app.post("/inquiry")
97def create_inquiry(inq: InquiryCreate):
98    if inq.property_id not in properties:
99        raise HTTPException(status_code=404, detail="Property not found")
100    global next_inquiry_id
101    iid = next_inquiry_id
102    next_inquiry_id += 1
103    inquiries[iid] = {
104        "message": inq.message,
105        "property_id": inq.property_id
106    }
107    return {"id": iid}
108 
109@app.get("/inquiry/{inquiry_id}")
110def get_inquiry(inquiry_id: int):
111    if inquiry_id not in inquiries:
112        raise HTTPException(status_code=404, detail="Inquiry not found")
113    inq = inquiries[inquiry_id]
114    return {
115        "id": inquiry_id,
116        "message": inq["message"],
117        "property_id": inq["property_id"]
118    }

1fastapi
2uvicorn