diff --git a/amarillo/plugins/gtfs_export/router.py b/amarillo/plugins/gtfs_export/router.py index c792442..1d7583c 100644 --- a/amarillo/plugins/gtfs_export/router.py +++ b/amarillo/plugins/gtfs_export/router.py @@ -62,8 +62,9 @@ async def get_file(region_id: str, requesting_user: User = Depends(get_current_u # logger.info("Returning new response") response = requests.get(f"{config.generator_url}/region/{region_id}/gtfs/") # cache response - with open(file_path, "wb") as file: - file.write(response.content) + if response.status_code == 200: + with open(file_path, "wb") as file: + file.write(response.content) return Response(content=response.content, media_type="application/zip") @@ -88,13 +89,14 @@ async def get_file(region_id: str, format: str = 'protobuf', requesting_user: Us raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=message) if is_cached_1m(file_path): - logger.info("Returning cached response") + # logger.info("Returning cached response") return FileResponse(file_path) - logger.info("Returning new response") + # logger.info("Returning new response") response = requests.get(f"{config.generator_url}/region/{region_id}/gtfs-rt/?format={format}") # cache response - with open(file_path, "wb") as file: - file.write(response.content) + if response.status_code == 200: + with open(file_path, "wb") as file: + file.write(response.content) return Response(content=response.content) \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index 25ca68f..ecbdbe4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "amarillo-gtfs-exporter" -version = "0.0.2" +version = "0.0.4" dependencies = [ "amarillo" ]