diff --git a/apps/worker/worker_app/tasks/generation.py b/apps/worker/worker_app/tasks/generation.py index bb3b52574..a089726af 100644 --- a/apps/worker/worker_app/tasks/generation.py +++ b/apps/worker/worker_app/tasks/generation.py @@ -1686,9 +1686,12 @@ def generate_video(self, task_id: str) -> dict: from packages.adapters.sqlalchemy_impl.models import ( GenerationTaskModel, ) - _m = _meta_session.query(GenerationTaskModel).filter( - GenerationTaskModel.id == task_id - ).first() + + _m = ( + _meta_session.query(GenerationTaskModel) + .filter(GenerationTaskModel.id == task_id) + .first() + ) if _m: existing = dict(_m.extra_meta or {}) existing["asset_analyses"] = asset_analyses diff --git a/tests/unit/test_worker_cover_meta_and_status.py b/tests/unit/test_worker_cover_meta_and_status.py index f93a98b78..9d6fa1ed3 100644 --- a/tests/unit/test_worker_cover_meta_and_status.py +++ b/tests/unit/test_worker_cover_meta_and_status.py @@ -2,20 +2,13 @@ - Bug 1: gen_task 过期内存对象 _repo.update() 覆盖 DB status 为 pending - Bug 2: 封面模型误用 .metadata(SQLAlchemy 保留属性),应为 .extra_meta """ + import ast from pathlib import Path import pytest - -GENERATION_FILE = ( - Path(__file__).resolve().parents[2] - / "apps" - / "worker" - / "worker_app" - / "tasks" - / "generation.py" -) +GENERATION_FILE = Path(__file__).resolve().parents[2] / "apps" / "worker" / "worker_app" / "tasks" / "generation.py" def _read_source() -> str: