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

Support rename table. (#139)

Browse Source
This commit is contained in:
long2ice
2021-03-25 21:21:49 +08:00
parent f8aa7a8f34
commit 3f52ac348b
6 changed files with 26 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
aerich/migrate.py
View File

@@ -176,7 +176,11 @@ class Migrate:
pass
else:
old_model_describe = old_models.get(new_model_str)
# rename table
new_table = new_model_describe.get("table")
old_table = old_model_describe.get("table")
if new_table != old_table:
cls._add_operator(cls.rename_table(model, old_table, new_table), upgrade)
old_unique_together = set(
map(lambda x: tuple(x), old_model_describe.get("unique_together"))
)
@@ -406,6 +410,10 @@ class Migrate:
if old_model not in new_models.keys():
cls._add_operator(cls.drop_model(old_models.get(old_model).get("table")), upgrade)
@classmethod
def rename_table(cls, model: Type[Model], old_table_name: str, new_table_name: str):
return cls.ddl.rename_table(model, old_table_name, new_table_name)
@classmethod
def add_model(cls, model: Type[Model]):
return cls.ddl.create_table(model)