From ef702047e44c018316e0f4a056a2ae9b9ff372e7 Mon Sep 17 00:00:00 2001 From: Cizz22 Date: Thu, 19 Feb 2026 14:47:57 +0700 Subject: [PATCH] feat: Return the cleaned filename in the import project service response. --- src/aeros_project/router.py | 4 ++-- src/aeros_project/service.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/aeros_project/router.py b/src/aeros_project/router.py index 6590867..25dd97a 100644 --- a/src/aeros_project/router.py +++ b/src/aeros_project/router.py @@ -18,7 +18,7 @@ from .service import import_aro_project, fetch_aro_record, reset_project, create router = APIRouter() -@router.post("", response_model=StandardResponse[None]) +@router.post("", response_model=StandardResponse[str]) async def import_aro( db_session: DbSession, current_user: CurrentUser, @@ -38,7 +38,7 @@ async def import_aro( result = await import_aro_project(db_session=db_session, aeros_project_in=aeros_project_input) - return {"data": None, "status": "success", "message": "Success"} + return {"data": result, "status": "success", "message": "Success"} @router.get("/download") diff --git a/src/aeros_project/service.py b/src/aeros_project/service.py index 8d97047..ffbb42c 100644 --- a/src/aeros_project/service.py +++ b/src/aeros_project/service.py @@ -41,7 +41,7 @@ async def import_aro_project(*, db_session: DbSession, aeros_project_in: AerosPr status_code=400, detail=f"Invalid filename: {str(e)}" ) - + # Check if mime type is application/octet-stream if file.content_type != "application/octet-stream": raise HTTPException( @@ -65,6 +65,7 @@ async def import_aro_project(*, db_session: DbSession, aeros_project_in: AerosPr print("read file") + # Read and check file size content = await file.read() if len(content) > MAX_FILE_SIZE: @@ -98,6 +99,8 @@ async def import_aro_project(*, db_session: DbSession, aeros_project_in: AerosPr ) + return clean_filename + # Project name hardcode # project_name = "trialapi"