78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
from typing import Annotated, Optional
|
|
|
|
from fastapi import FastAPI, Depends, Request
|
|
from sqlalchemy.orm import Session
|
|
from starlette import status
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
import models
|
|
from auth import get_db, get_current_user, router as auth_router
|
|
from config import templates, ACCESS_TOKEN_EXPIRE_MINUTES
|
|
from models import Property
|
|
from properties import router as properties_router
|
|
|
|
app = FastAPI()
|
|
|
|
app.include_router(auth_router)
|
|
app.include_router(properties_router)
|
|
app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
|
|
models.Base.metadata.create_all(bind=__import__("database", fromlist=["engine"]).engine)
|
|
|
|
|
|
class RefreshTokenMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
response = await call_next(request)
|
|
new_token = getattr(request.state, "new_access_token", None)
|
|
if new_token:
|
|
response.set_cookie(
|
|
"access_token", new_token,
|
|
httponly=True, max_age=ACCESS_TOKEN_EXPIRE_MINUTES * 60, samesite="lax",
|
|
)
|
|
return response
|
|
|
|
app.add_middleware(RefreshTokenMiddleware)
|
|
|
|
|
|
class PricePrefMiddleware(BaseHTTPMiddleware):
|
|
async def dispatch(self, request: Request, call_next):
|
|
from price_pref import get_price_pref
|
|
request.state.price_pref = get_price_pref(request)
|
|
return await call_next(request)
|
|
|
|
app.add_middleware(PricePrefMiddleware)
|
|
|
|
|
|
db_dependency = Annotated[Session, Depends(get_db)]
|
|
user_dependency = Annotated[Optional[dict], Depends(get_current_user)]
|
|
|
|
|
|
@app.get("/", status_code=status.HTTP_200_OK)
|
|
async def homepage(request: Request, db: db_dependency, user: user_dependency):
|
|
# Show featured properties; if none are featured, show the latest listings
|
|
featured = (
|
|
db.query(Property)
|
|
.filter(Property.is_featured == True, Property.status == "active")
|
|
.limit(6)
|
|
.all()
|
|
)
|
|
if not featured:
|
|
featured = (
|
|
db.query(Property)
|
|
.filter(Property.status == "active")
|
|
.order_by(Property.created_at.desc())
|
|
.limit(6)
|
|
.all()
|
|
)
|
|
for prop in featured:
|
|
prop.primary_image = next(
|
|
(img for img in prop.images if img.is_primary),
|
|
prop.images[0] if prop.images else None,
|
|
)
|
|
return templates.TemplateResponse(request, "index.html", {
|
|
"user": user,
|
|
"properties": featured,
|
|
"price_pref": request.state.price_pref,
|
|
})
|