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 = {}
11agents = {}
12next_user_id = 1
13next_agent_id = 1
14next_property_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19    role: str = "buyer"
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class AgentCreate(BaseModel):
26    name: str
27    email: str
28    phone: str
29 
30class PropertyCreate(BaseModel):
31    title: str
32    description: str
33    price: float
34    location: str
35    agent_id: int
36 
37def get_current_user(token: str = Header(None)):
38    if not token:
39        raise HTTPException(status_code=401, detail="No token provided")
40    user_id = tokens.get(token)
41    if not user_id:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return user_id
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    for u in users.values():
49        if u["username"] == req.username:
50            raise HTTPException(status_code=400, detail="Username already exists")
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {"id": user_id, "username": req.username, "password": req.password, "role": req.role}
54    return {"id": user_id, "username": req.username, "role": req.role}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for u in users.values():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = u["id"]
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.post("/agents")
66def create_agent(req: AgentCreate, token: str = Header(None)):
67    user_id = get_current_user(token)
68    user = users[user_id]
69    if user["role"] != "agent":
70        raise HTTPException(status_code=403, detail="Only agents can create agent profiles")
71    global next_agent_id
72    agent_id = next_agent_id
73    next_agent_id += 1
74    agents[agent_id] = {"id": agent_id, "name": req.name, "email": req.email, "phone": req.phone, "user_id": user_id}
75    return agents[agent_id]
76 
77@app.get("/agents/{agent_id}")
78def get_agent(agent_id: int):
79    agent = agents.get(agent_id)
80    if not agent:
81        raise HTTPException(status_code=404, detail="Agent not found")
82    return agent
83 
84@app.post("/properties")
85def create_property(req: PropertyCreate, token: str = Header(None)):
86    user_id = get_current_user(token)
87    user = users[user_id]
88    if user["role"] != "agent":
89        raise HTTPException(status_code=403, detail="Only agents can list properties")
90    if req.agent_id not in agents:
91        raise HTTPException(status_code=400, detail="Agent not found")
92    global next_property_id
93    property_id = next_property_id
94    next_property_id += 1
95    properties[property_id] = {
96        "id": property_id,
97        "title": req.title,
98        "description": req.description,
99        "price": req.price,
100        "location": req.location,
101        "agent_id": req.agent_id
102    }
103    return properties[property_id]
104 
105@app.get("/properties/{property_id}")
106def get_property(property_id: int):
107    prop = properties.get(property_id)
108    if not prop:
109        raise HTTPException(status_code=404, detail="Property not found")
110    return prop

1fastapi
2uvicorn