Browse the data

1import json
2import os
3import secrets
4from pathlib import Path
5 
6from fastapi import FastAPI, HTTPException, Header, File, UploadFile, Form
7from fastapi.responses import FileResponse, JSONResponse
8from pydantic import BaseModel
9 
10app = FastAPI()
11 
12users = {}
13tokens = {}
14properties = {}
15next_property_id = 1
16PROPERTY_ASSETS_DIR = Path("/property_assets")
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class PropertyCreate(BaseModel):
27    price: float
28    address: str
29 
30def get_current_user(authorization: str = Header(None)):
31    if not authorization:
32        raise HTTPException(status_code=401, detail="Missing Authorization header")
33    token = authorization.replace("Bearer ", "")
34    if token not in tokens:
35        raise HTTPException(status_code=401, detail="Invalid token")
36    return tokens[token]
37 
38@app.post("/signup")
39def signup(req: SignupRequest):
40    if req.username in users:
41        raise HTTPException(status_code=400, detail="User already exists")
42    users[req.username] = req.password
43    return {"message": "User created"}
44 
45@app.post("/login")
46def login(req: LoginRequest):
47    if req.username not in users or users[req.username] != req.password:
48        raise HTTPException(status_code=401, detail="Invalid credentials")
49    token = secrets.token_hex(16)
50    tokens[token] = req.username
51    return {"token": token}
52 
53@app.post("/properties")
54def create_property(price: float = Form(...), address: str = Form(...), files: list[UploadFile] = File(...), authorization: str = Header(None)):
55    user = get_current_user(authorization)
56    global next_property_id
57    prop_id = next_property_id
58    next_property_id += 1
59    photo_names = []
60    prop_dir = PROPERTY_ASSETS_DIR / str(prop_id)
61    prop_dir.mkdir(parents=True, exist_ok=True)
62    for file in files:
63        file_path = prop_dir / file.filename
64        with open(file_path, "wb") as f:
65            f.write(file.file.read())
66        photo_names.append(file.filename)
67    properties[prop_id] = {
68        "id": prop_id,
69        "price": price,
70        "address": address,
71        "photos": photo_names
72    }
73    return {"id": prop_id, "price": price, "address": address, "photos": photo_names}
74 
75@app.get("/properties/{prop_id}/details")
76def get_property_details(prop_id: int, authorization: str = Header(None)):
77    user = get_current_user(authorization)
78    prop = properties.get(prop_id)
79    if not prop:
80        raise HTTPException(status_code=404, detail="Property not found")
81    return {
82        "price": prop["price"],
83        "address": prop["address"],
84        "photos": prop["photos"]
85    }
86 
87@app.get("/properties/{prop_id}/photos/{photo_name}")
88def get_property_photo(prop_id: int, photo_name: str, authorization: str = Header(None)):
89    user = get_current_user(authorization)
90    prop = properties.get(prop_id)
91    if not prop:
92        raise HTTPException(status_code=404, detail="Property not found")
93    if photo_name not in prop["photos"]:
94        raise HTTPException(status_code=404, detail="Photo not found")
95    file_path = PROPERTY_ASSETS_DIR / str(prop_id) / photo_name
96    if not file_path.exists():
97        raise HTTPException(status_code=404, detail="File not found")
98    return FileResponse(str(file_path))
99 
100@app.get("/properties/{prop_id}")
101def get_property(prop_id: int, authorization: str = Header(None)):
102    user = get_current_user(authorization)
103    prop = properties.get(prop_id)
104    if not prop:
105        raise HTTPException(status_code=404, detail="Property not found")
106    return prop

1fastapi
2uvicorn
3python-multipart