style: fix

This commit is contained in:
long2ice 2024-01-18 09:50:41 +08:00
parent 4370b5ed08
commit b63fbcc7a4
2 changed files with 5 additions and 3 deletions

View File

@ -153,14 +153,15 @@ class Migrate:
"""
builds content for diff file from template
"""
def join_lines(lines: List[str]) -> str:
if not lines:
return ""
return ";\n ".join(lines) + ";"
return MIGRATE_TEMPLATE.format(
upgrade_sql=join_lines(cls.upgrade_operators),
downgrade_sql=join_lines(cls.downgrade_operators)
upgrade_sql=join_lines(cls.upgrade_operators),
downgrade_sql=join_lines(cls.downgrade_operators),
)
@classmethod

View File

@ -970,6 +970,7 @@ def test_sort_all_version_files(mocker):
"11_datetime_update.py",
]
async def test_empty_migration(mocker) -> None:
mocker.patch("os.listdir", return_value=[])
Migrate.app = "foo"