From 3d00858bbfb71c725c0ee696853f16e010185d2a Mon Sep 17 00:00:00 2001 From: vladsaveliev Date: Thu, 22 Feb 2024 22:10:22 +0100 Subject: [PATCH] Endpoint to remove visits.csv --- app/main.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/app/main.py b/app/main.py index d7ea116..7b3ee11 100644 --- a/app/main.py +++ b/app/main.py @@ -330,6 +330,22 @@ async def update_downloads_endpoint(background_tasks: BackgroundTasks): msg = f"Failed to update the download stats: {e}" raise HTTPException(status_code=http.HTTPStatus.INTERNAL_SERVER_ERROR, detail=msg) + @app.get("/remove_visits_csv") + async def remove_visits_csv(): + try: + if CSV_FILE_PATH.exists(): + CSV_FILE_PATH.unlink() + msg = f"Removed {CSV_FILE_PATH}" + logger.info(msg) + return PlainTextResponse(content=msg) + else: + msg = f"File {CSV_FILE_PATH} doesn't exist" + logger.info(msg) + return PlainTextResponse(content=msg) + except Exception as e: + msg = f"Failed to remove {CSV_FILE_PATH}: {e}" + raise HTTPException(status_code=http.HTTPStatus.INTERNAL_SERVER_ERROR, detail=msg) + @app.get("/version.php", response_class=PlainTextResponse) async def version_legacy(background_tasks: BackgroundTasks, v: str | None = None):