diff --git a/crawler/api/app.py b/crawler/api/app.py index 688ad13..d9a1506 100644 --- a/crawler/api/app.py +++ b/crawler/api/app.py @@ -15,6 +15,7 @@ from dotenv import load_dotenv from fastapi import Depends, FastAPI, HTTPException, Query from api.auth import User from models.listing import QueryParameters +from rec import districts from repositories.listing_repository import ListingRepository from repositories.listing_repository import ListingRepository from database import engine @@ -104,3 +105,10 @@ async def get_task_status( "status": task_result.status, "result": json.dumps(result), } + + +@app.get("/api/get_districts") +async def get_task_status( + user: Annotated[User, Depends(get_current_user)], +) -> dict[str, str]: + return districts.get_districts()