pub/aerich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: initialize an empty database. (#267)

Browse Source
This commit is contained in:
long2ice 2022-09-27 22:42:54 +08:00
parent ab31445fb2
commit f8e1f9ff44
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -6,6 +6,7 @@
- Fix syntax error with python3.8.10. (#265) - Fix syntax error with python3.8.10. (#265)
- Fix sql generate error. (#263) - Fix sql generate error. (#263)
- Fix initialize an empty database. (#267)
### 0.7.1rc1 ### 0.7.1rc1

11
aerich/__init__.py
View File

@ -50,7 +50,7 @@ class Command:
file_path = Path(Migrate.migrate_location, version_file) file_path = Path(Migrate.migrate_location, version_file)
m = import_py_file(file_path) m = import_py_file(file_path)
upgrade = getattr(m, "upgrade") upgrade = getattr(m, "upgrade")
await upgrade(conn) await conn.execute_script(await upgrade(conn))
await Aerich.create( await Aerich.create(
version=version_file, version=version_file,
app=self.app, app=self.app,
@ -80,10 +80,11 @@ class Command:
) as conn: ) as conn:
file_path = Path(Migrate.migrate_location, file) file_path = Path(Migrate.migrate_location, file)
m = import_py_file(file_path) m = import_py_file(file_path)
downgrade = getattr(m, "downgrade", None) downgrade = getattr(m, "downgrade")
if not downgrade: downgrade_sql = await downgrade(conn)
if not downgrade_sql.strip():
raise DowngradeError("No downgrade items found") raise DowngradeError("No downgrade items found")
await downgrade(conn) await conn.execute_script(downgrade_sql)
await version.delete() await version.delete()
if delete: if delete:
os.unlink(file_path) os.unlink(file_path)
@ -138,6 +139,6 @@ class Command:
content=get_models_describe(app), content=get_models_describe(app),
) )
version_file = Path(dirname, version) version_file = Path(dirname, version)
content = MIGRATE_TEMPLATE.format(upgrade_sql=f'"""{schema}"""', downgrade_sql="") content = MIGRATE_TEMPLATE.format(upgrade_sql=schema, downgrade_sql="")
with open(version_file, "w", encoding="utf-8") as f: with open(version_file, "w", encoding="utf-8") as f:
f.write(content) f.write(content)