From 6bf9eff604bb3bfd617a82b5102d975d1e090dd2 Mon Sep 17 00:00:00 2001 From: CI Bot Date: Mon, 31 Aug 2026 13:27:03 +0000 Subject: [PATCH] style: auto-format with black + isort + prettier [skip ci-format-check] --- ...grate_template_segments_to_clip_configs.py | 4 +- .../sqlalchemy_impl/template_repository.py | 18 +++--- tests/unit/test_unify_template_segments.py | 62 +++++++++++-------- 3 files changed, 49 insertions(+), 35 deletions(-) diff --git a/alembic/versions/060_migrate_template_segments_to_clip_configs.py b/alembic/versions/060_migrate_template_segments_to_clip_configs.py index 146c601db..5df89d2cf 100644 --- a/alembic/versions/060_migrate_template_segments_to_clip_configs.py +++ b/alembic/versions/060_migrate_template_segments_to_clip_configs.py @@ -6,6 +6,7 @@ Create Date: 2026-08-31 """ import sqlalchemy as sa + from alembic import op revision = "060_migrate_segments" @@ -41,8 +42,7 @@ def upgrade() -> None: "s.id, s.template_id, 'main', s.segment_order, " "s.duration_min, s.duration_max, " "'', " + empty_json + ", " - "'cut', " - + config_expr + ", " + "'cut', " + config_expr + ", " "s.created_at, s.updated_at " "FROM template_segments s " "WHERE NOT EXISTS (" diff --git a/packages/adapters/sqlalchemy_impl/template_repository.py b/packages/adapters/sqlalchemy_impl/template_repository.py index 743999b2e..275d6bc3d 100755 --- a/packages/adapters/sqlalchemy_impl/template_repository.py +++ b/packages/adapters/sqlalchemy_impl/template_repository.py @@ -212,14 +212,16 @@ class SQLAlchemyTemplateRepository: # 复用 create_segments 写入 template_clip_configs new_segments: List[TemplateSegment] = [] for seg in source.segments: - new_segments.append(TemplateSegment( - id=str(uuid.uuid4()), - template_id=created.id, - segment_order=seg.segment_order, - duration_min=seg.duration_min, - duration_max=seg.duration_max, - material_type=seg.material_type, - )) + new_segments.append( + TemplateSegment( + id=str(uuid.uuid4()), + template_id=created.id, + segment_order=seg.segment_order, + duration_min=seg.duration_min, + duration_max=seg.duration_max, + material_type=seg.material_type, + ) + ) if new_segments: self.create_segments(new_segments) else: diff --git a/tests/unit/test_unify_template_segments.py b/tests/unit/test_unify_template_segments.py index 5c9557275..1f6e2d4af 100644 --- a/tests/unit/test_unify_template_segments.py +++ b/tests/unit/test_unify_template_segments.py @@ -48,15 +48,27 @@ def repo(session): def _make_template(template_id=None, user_id="u1", name="测试模板", mode="one_take"): tid = template_id or str(uuid.uuid4()) return Template( - id=tid, user_id=user_id, name=name, mode=mode, - category="", tags=[], estimated_duration=30.0, is_active=True, segments=[]) + id=tid, + user_id=user_id, + name=name, + mode=mode, + category="", + tags=[], + estimated_duration=30.0, + is_active=True, + segments=[], + ) def _make_segment(template_id, order=1, material_type=None): return TemplateSegment( - id=str(uuid.uuid4()), template_id=template_id, - segment_order=order, duration_min=5.0, duration_max=10.0, - material_type=material_type) + id=str(uuid.uuid4()), + template_id=template_id, + segment_order=order, + duration_min=5.0, + duration_max=10.0, + material_type=material_type, + ) class TestCreateSegments: @@ -65,8 +77,7 @@ class TestCreateSegments: repo.create(tpl) seg = _make_segment(tpl.id, order=1) repo.create_segments([seg]) - clips = session.query(TemplateClipConfigModel).filter( - TemplateClipConfigModel.template_id == tpl.id).all() + clips = session.query(TemplateClipConfigModel).filter(TemplateClipConfigModel.template_id == tpl.id).all() assert len(clips) == 1 assert clips[0].clip_type == "main" assert clips[0].order == 1 @@ -77,8 +88,7 @@ class TestCreateSegments: repo.create(tpl) seg = _make_segment(tpl.id, order=1, material_type="voiceover") repo.create_segments([seg]) - clip = session.query(TemplateClipConfigModel).filter( - TemplateClipConfigModel.template_id == tpl.id).first() + clip = session.query(TemplateClipConfigModel).filter(TemplateClipConfigModel.template_id == tpl.id).first() assert clip.config["material_type"] == "voiceover" @@ -96,8 +106,13 @@ class TestListSegments: tpl = _make_template() repo.create(tpl) old = TemplateSegmentModel( - id=str(uuid.uuid4()), template_id=tpl.id, - segment_order=1, duration_min=3.0, duration_max=8.0, material_type="场景") + id=str(uuid.uuid4()), + template_id=tpl.id, + segment_order=1, + duration_min=3.0, + duration_max=8.0, + material_type="场景", + ) session.add(old) session.commit() result = repo.list_segments(tpl.id) @@ -110,8 +125,8 @@ class TestListSegments: seg = _make_segment(tpl.id, order=1) repo.create_segments([seg]) old = TemplateSegmentModel( - id=str(uuid.uuid4()), template_id=tpl.id, - segment_order=1, duration_min=1.0, duration_max=2.0) + id=str(uuid.uuid4()), template_id=tpl.id, segment_order=1, duration_min=1.0, duration_max=2.0 + ) session.add(old) session.commit() result = repo.list_segments(tpl.id) @@ -134,8 +149,8 @@ class TestListByUser: tpl = _make_template() repo.create(tpl) old = TemplateSegmentModel( - id=str(uuid.uuid4()), template_id=tpl.id, - segment_order=1, duration_min=2.0, duration_max=6.0) + id=str(uuid.uuid4()), template_id=tpl.id, segment_order=1, duration_min=2.0, duration_max=6.0 + ) session.add(old) session.commit() result = repo.list_by_user("u1") @@ -152,8 +167,7 @@ class TestCopyTemplate: repo.create_segments([seg]) copied = repo.copy_template(tpl.id, "u1", "副本模板") assert copied.id != tpl.id - clips = session.query(TemplateClipConfigModel).filter( - TemplateClipConfigModel.template_id == copied.id).all() + clips = session.query(TemplateClipConfigModel).filter(TemplateClipConfigModel.template_id == copied.id).all() assert len(clips) == 1 assert clips[0].config["material_type"] == "voiceover" @@ -171,15 +185,13 @@ class TestDelete: seg = _make_segment(tpl.id, order=1) repo.create_segments([seg]) old = TemplateSegmentModel( - id=str(uuid.uuid4()), template_id=tpl.id, - segment_order=1, duration_min=1.0, duration_max=2.0) + id=str(uuid.uuid4()), template_id=tpl.id, segment_order=1, duration_min=1.0, duration_max=2.0 + ) session.add(old) session.commit() repo.delete(tpl.id, "u1") - c1 = session.query(TemplateClipConfigModel).filter( - TemplateClipConfigModel.template_id == tpl.id).count() - c2 = session.query(TemplateSegmentModel).filter( - TemplateSegmentModel.template_id == tpl.id).count() + c1 = session.query(TemplateClipConfigModel).filter(TemplateClipConfigModel.template_id == tpl.id).count() + c2 = session.query(TemplateSegmentModel).filter(TemplateSegmentModel.template_id == tpl.id).count() assert c1 == 0 assert c2 == 0 @@ -189,8 +201,8 @@ class TestDelete: seg = _make_segment(tpl.id, order=1) repo.create_segments([seg]) old = TemplateSegmentModel( - id=str(uuid.uuid4()), template_id=tpl.id, - segment_order=2, duration_min=1.0, duration_max=2.0) + id=str(uuid.uuid4()), template_id=tpl.id, segment_order=2, duration_min=1.0, duration_max=2.0 + ) session.add(old) session.commit() count = repo.delete_segments_by_template(tpl.id) -- 2.54.0