Browse the data

1import json
2import hashlib
3import secrets
4from fastapi import FastAPI, HTTPException, Header
5from typing import Optional
6import sqlite3
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12properties = {}
13property_id_counter = 1
14user_id_counter = 1
15 
16def hash_password(password):
17    return hashlib.sha256(password.encode()).hexdigest()
18 
19def get_current_user(authorization: Optional[str] = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="No auth header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global user_id_counter
30    if username in users:
31        raise HTTPException(status_code=400, detail="User exists")
32    users[username] = {"id": user_id_counter, "password": hash_password(password)}
33    user_id_counter += 1
34    return {"message": "User created"}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    if username not in users or users[username]["password"] != hash_password(password):
39        raise HTTPException(status_code=401, detail="Invalid credentials")
40    token = secrets.token_hex(32)
41    tokens[token] = users[username]["id"]
42    return {"token": token}
43 
44@app.get("/properties/{property_id}")
45def get_property(property_id: int, authorization: Optional[str] = Header(None)):
46    get_current_user(authorization)
47    if property_id not in properties:
48        raise HTTPException(status_code=404, detail="Property not found")
49    return properties[property_id]
50 
51@app.post("/properties")
52def create_property(price: float, sqft: float, zip_code: str, beds: int, amenities: str, days_on_market: int, authorization: Optional[str] = Header(None)):
53    global property_id_counter
54    get_current_user(authorization)
55    property_id = property_id_counter
56    property_id_counter += 1
57    properties[property_id] = {
58        "id": property_id,
59        "price": price,
60        "sqft": sqft,
61        "price_per_sqft": round(price / sqft, 2) if sqft > 0 else 0,
62        "zip_code": zip_code,
63        "beds": beds,
64        "amenities": amenities,
65        "days_on_market": days_on_market
66    }
67    return properties[property_id]
68 
69@app.get("/properties")
70def search_properties(zip: Optional[str] = None, beds_min: Optional[int] = None, amenities: Optional[str] = None, authorization: Optional[str] = Header(None)):
71    get_current_user(authorization)
72    
73    conn = sqlite3.connect(':memory:')
74    conn.execute('''CREATE TABLE properties
75                    (id INT, price REAL, sqft REAL, zip_code TEXT, beds INT, amenities TEXT, days_on_market INT)''')
76    
77    for pid, prop in properties.items():
78        conn.execute("INSERT INTO properties VALUES (?,?,?,?,?,?,?)",
79                     (prop['id'], prop['price'], prop['sqft'], prop['zip_code'], prop['beds'], prop['amenities'], prop['days_on_market']))
80    
81    query = "SELECT * FROM properties WHERE 1=1"
82    params = []
83    
84    if zip:
85        query += " AND zip_code = ?"
86        params.append(zip)
87    if beds_min:
88        query += " AND beds >= ?"
89        params.append(beds_min)
90    if amenities:
91        amenity_list = amenities.split(',')
92        for a in amenity_list:
93            query += " AND amenities LIKE ?"
94            params.append(f"%{a.strip()}%")
95    
96    rows = conn.execute(query, params).fetchall()
97    conn.close()
98    
99    results = []
100    for row in rows:
101        results.append({
102            "id": row[0],
103            "price": row[1],
104            "sqft": row[2],
105            "price_per_sqft": round(row[1]/row[2], 2) if row[2] > 0 else 0,
106            "zip_code": row[3],
107            "beds": row[4],
108            "amenities": row[5],
109            "days_on_market": row[6]
110        })
111    
112    return results

1fastapi
2uvicorn