amarillo/amarillo-grfs-export
5
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

/export endpoint

Browse source
This commit is contained in:
Csaba 2024-02-09 12:34:22 +01:00
parent e39c0a76c9
commit 296a1cc386
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
amarillo/plugins/gtfs_export/router.py
View file

@ -10,10 +10,13 @@ from fastapi.responses import FileResponse
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
router = APIRouter( router = APIRouter()
prefix="/region",
tags=["region"] @router.post("/export")
) async def post_agency_conf(admin_api_key: str = Depends(verify_admin_api_key)):
#import is here to avoid circular import
from amarillo.plugins.gtfs_export.gtfs_generator import generate_gtfs
generate_gtfs()
#TODO: move to amarillo/utils? #TODO: move to amarillo/utils?
def _assert_region_exists(region_id: str) -> Region: def _assert_region_exists(region_id: str) -> Region:
@ -29,7 +32,7 @@ def _assert_region_exists(region_id: str) -> Region:
return region return region
@router.get("/{region_id}/gtfs", @router.get("region/{region_id}/gtfs",
summary="Return GTFS Feed for this region", summary="Return GTFS Feed for this region",
response_description="GTFS-Feed (zip-file)", response_description="GTFS-Feed (zip-file)",
response_class=FileResponse, response_class=FileResponse,
@ -41,7 +44,7 @@ async def get_file(region_id: str, user: str = Depends(verify_admin_api_key)):
_assert_region_exists(region_id) _assert_region_exists(region_id)
return FileResponse(f'data/gtfs/amarillo.{region_id}.gtfs.zip') return FileResponse(f'data/gtfs/amarillo.{region_id}.gtfs.zip')
@router.get("/{region_id}/gtfs-rt", @router.get("region/{region_id}/gtfs-rt",
summary="Return GTFS-RT Feed for this region", summary="Return GTFS-RT Feed for this region",
response_description="GTFS-RT-Feed", response_description="GTFS-RT-Feed",
response_class=FileResponse, response_class=FileResponse,