Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import time
6from datetime import datetime, timedelta
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12next_user_id = 1
13next_token_id = 1
14 
15estimates = {}
16history_data = {}
17next_estimate_id = 1
18 
19API_TOKEN = "supersecret123"
20 
21def verify_token(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing Authorization header")
24    token = authorization.replace("Bearer ", "")
25    if token != API_TOKEN and token not in tokens.values():
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return token
28 
29class UserSignup(BaseModel):
30    username: str
31    password: str
32 
33class UserLogin(BaseModel):
34    username: str
35    password: str
36 
37class EstimateRequest(BaseModel):
38    zip_code: str
39 
40class EstimateResponse(BaseModel):
41    id: int
42    zip_code: str
43    median_rent: float
44    price_per_sqft: float
45    vacancy_rate: float
46 
47@app.post("/signup")
48def signup(user: UserSignup):
49    global next_user_id
50    user_id = next_user_id
51    next_user_id += 1
52    users[user_id] = {"username": user.username, "password": user.password}
53    return {"id": user_id, "username": user.username}
54 
55@app.post("/login")
56def login(user: UserLogin):
57    for uid, u in users.items():
58        if u["username"] == user.username and u["password"] == user.password:
59            global next_token_id
60            token = f"token_{next_token_id}"
61            next_token_id += 1
62            tokens[uid] = token
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.post("/estimate", response_model=EstimateResponse)
67def create_estimate(req: EstimateRequest, authorization: str = Header(None)):
68    verify_token(authorization)
69    global next_estimate_id
70    # Simulate real estate data
71    random.seed(hash(req.zip_code) % (2**31))
72    median_rent = round(random.uniform(800, 4000), 2)
73    price_per_sqft = round(random.uniform(1.5, 5.0), 2)
74    vacancy_rate = round(random.uniform(0.01, 0.15), 4)
75    est_id = next_estimate_id
76    next_estimate_id += 1
77    estimates[est_id] = {
78        "id": est_id,
79        "zip_code": req.zip_code,
80        "median_rent": median_rent,
81        "price_per_sqft": price_per_sqft,
82        "vacancy_rate": vacancy_rate
83    }
84    # Generate history data (6 months of simulated trends)
85    hist = []
86    base_rent = median_rent - random.uniform(100, 300)
87    base_pps = price_per_sqft - random.uniform(0.3, 0.8)
88    base_vac = vacancy_rate - random.uniform(0.01, 0.03)
89    for i in range(6):
90        month_date = (datetime.now() - timedelta(days=30*(5-i))).strftime("%Y-%m")
91        hist.append({
92            "month": month_date,
93            "median_rent": round(base_rent + random.uniform(-50, 50) + i*20, 2),
94            "price_per_sqft": round(base_pps + random.uniform(-0.2, 0.2) + i*0.1, 2),
95            "vacancy_rate": round(max(0, base_vac + random.uniform(-0.01, 0.01) - i*0.005), 4)
96        })
97    history_data[est_id] = hist
98    return estimates[est_id]
99 
100@app.get("/estimate/{estimate_id}", response_model=EstimateResponse)
101def get_estimate(estimate_id: int, authorization: str = Header(None)):
102    verify_token(authorization)
103    if estimate_id not in estimates:
104        raise HTTPException(status_code=404, detail="Estimate not found")
105    return estimates[estimate_id]
106 
107@app.get("/history/{estimate_id}")
108def get_history(estimate_id: int, authorization: str = Header(None)):
109    verify_token(authorization)
110    if estimate_id not in history_data:
111        raise HTTPException(status_code=404, detail="History not found")
112    return history_data[estimate_id]

1fastapi
2uvicorn
3pydantic