Add Command class. (#148) (#141) (#123) (#106)

This commit is contained in:
long2ice
2021-08-03 16:18:07 +08:00
parent 1e5a83c281
commit 6f5a9ab78c
8 changed files with 238 additions and 131 deletions

View File

@@ -264,7 +264,11 @@ class Migrate:
# rename field
if (
changes[0]
== ("change", "name", (old_data_field_name, new_data_field_name),)
== (
"change",
"name",
(old_data_field_name, new_data_field_name),
)
and changes[1]
== (
"change",
@@ -302,11 +306,16 @@ class Migrate:
)
else:
cls._add_operator(
cls._rename_field(model, *changes[1][2]), upgrade,
cls._rename_field(model, *changes[1][2]),
upgrade,
)
if not is_rename:
cls._add_operator(
cls._add_field(model, new_data_field,), upgrade,
cls._add_field(
model,
new_data_field,
),
upgrade,
)
# remove fields
for old_data_field_name in set(old_data_fields_name).difference(
@@ -406,7 +415,8 @@ class Migrate:
else:
# modify column
cls._add_operator(
cls._modify_field(model, new_data_field), upgrade,
cls._modify_field(model, new_data_field),
upgrade,
)
for old_model in old_models: