From fe94ddaaeb2cd0ed22e9b4a1e82868f423c0c8d1 Mon Sep 17 00:00:00 2001 From: Likalikali Date: Fri, 7 Aug 2026 20:40:44 +0800 Subject: [PATCH] chore: dep inject for model in workflow run archive download (#40157) --- api/controllers/console/workflow_run_archive.py | 9 +++++---- .../controllers/console/test_workflow_run_archive.py | 2 +- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/api/controllers/console/workflow_run_archive.py b/api/controllers/console/workflow_run_archive.py index e15b897d5b5..83f0ece1b6f 100644 --- a/api/controllers/console/workflow_run_archive.py +++ b/api/controllers/console/workflow_run_archive.py @@ -13,6 +13,7 @@ from controllers.console.wraps import ( account_initialization_required, cloud_edition_billing_enabled, cloud_edition_billing_paid_plan_required, + model_validate, only_edition_cloud, setup_required, ) @@ -144,16 +145,16 @@ class WorkflowRunArchiveDownloadsApi(Resource): @only_edition_cloud @cloud_edition_billing_enabled @cloud_edition_billing_paid_plan_required - def post(self): + @model_validate(WorkflowRunArchiveDownloadPayload) + def post(self, req_data: WorkflowRunArchiveDownloadPayload): tenant_id, account_id = _current_owner_or_admin_ids() - payload = WorkflowRunArchiveDownloadPayload.model_validate(console_ns.payload or {}) try: task = create_workflow_run_archive_download_task( db.session(), tenant_id=tenant_id, requested_by=account_id, - year=payload.year, - month=payload.month, + year=req_data.year, + month=req_data.month, ) except WorkflowRunArchiveNotFoundError as exc: raise NotFound(str(exc)) from exc diff --git a/api/tests/unit_tests/controllers/console/test_workflow_run_archive.py b/api/tests/unit_tests/controllers/console/test_workflow_run_archive.py index bff41cb97b1..4ecc1e8584e 100644 --- a/api/tests/unit_tests/controllers/console/test_workflow_run_archive.py +++ b/api/tests/unit_tests/controllers/console/test_workflow_run_archive.py @@ -54,7 +54,7 @@ def test_current_owner_or_admin_ids_returns_current_ids_for_manager( ("method", "args"), [ (WorkflowRunArchivesApi.get, ()), - (WorkflowRunArchiveDownloadsApi.post, ()), + (WorkflowRunArchiveDownloadsApi.post, (None,)), (WorkflowRunArchiveDownloadApi.get, ("download-1",)), (WorkflowRunArchiveDownloadFileApi.get, ("download-1",)), ],