Browse the data

1import os
2import shutil
3import uuid
4from pathlib import Path
5from io import BytesIO
6 
7from fastapi import FastAPI, HTTPException, Header, UploadFile, File, Query
8from fastapi.responses import Response
9from PIL import Image
10import uvicorn
11 
12app = FastAPI()
13 
14users = {}
15tokens = {}
16properties = {}
17photos_path = Path("/var/properties")
18 
19PHOTOS_BASE = photos_path
20 
21def require_auth(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(401, "unauthorized")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(401, "unauthorized")
27    return tokens[token]
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    if username in users:
32        raise HTTPException(400, "user exists")
33    user_id = len(users) + 1
34    users[username] = {"id": user_id, "username": username, "password": password}
35    return {"id": user_id, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    if username not in users or users[username]["password"] != password:
40        raise HTTPException(401, "invalid credentials")
41    token = str(uuid.uuid4())
42    tokens[token] = users[username]["id"]
43    return {"token": token}
44 
45@app.post("/property")
46def create_property(authorization: str = Header(None)):
47    user_id = require_auth(authorization)
48    prop_id = len(properties) + 1
49    properties[prop_id] = {"id": prop_id, "owner_id": user_id, "photos": []}
50    os.makedirs(PHOTOS_BASE / str(prop_id), exist_ok=True)
51    return properties[prop_id]
52 
53@app.get("/property/{prop_id}")
54def get_property(prop_id: int, authorization: str = Header(None)):
55    require_auth(authorization)
56    if prop_id not in properties:
57        raise HTTPException(404, "not found")
58    return properties[prop_id]
59 
60@app.post("/property/{prop_id}/photos")
61def upload_photo(prop_id: int, file: UploadFile = File(...), authorization: str = Header(None)):
62    require_auth(authorization)
63    if prop_id not in properties:
64        raise HTTPException(404, "property not found")
65    dir_path = PHOTOS_BASE / str(prop_id)
66    dir_path.mkdir(parents=True, exist_ok=True)
67    file_id = len(properties[prop_id]["photos"]) + 1
68    ext = Path(file.filename).suffix if file.filename else ".jpg"
69    filename = f"{file_id}{ext}"
70    file_path = dir_path / filename
71    with open(file_path, "wb") as f:
72        shutil.copyfileobj(file.file, f)
73    properties[prop_id]["photos"].append(filename)
74    return {"file": filename, "id": file_id}
75 
76@app.get("/property/{prop_id}/photos/{file_name:path}")
77def get_photo(prop_id: int, file_name: str, size: str = Query(None), authorization: str = Header(None)):
78    require_auth(authorization)
79    if prop_id not in properties:
80        raise HTTPException(404, "property not found")
81    file_path = PHOTOS_BASE / str(prop_id) / file_name
82    if not file_path.exists():
83        raise HTTPException(404, "file not found")
84    if size == "small":
85        img = Image.open(file_path)
86        img.thumbnail((300, 300))
87        buf = BytesIO()
88        img.save(buf, format="JPEG")
89        buf.seek(0)
90        return Response(content=buf.read(), media_type="image/jpeg")
91    return Response(content=open(file_path, "rb").read(), media_type="image/jpeg")

1fastapi
2uvicorn
3Pillow
4python-multipart