Fix: migrate doesn't use source_field in unique_together

2021-07-28 17:17:28 +05:00
parent 79b7ae343a
commit aa9f40ae27
5 changed files with 18 additions and 9 deletions
--- a/aerich/migrate.py
+++ b/aerich/migrate.py
@@ -447,7 +447,10 @@ class Migrate:
    def _resolve_fk_fields_name(cls, model: Type[Model], fields_name: Tuple[str]):
        ret = []
        for field_name in fields_name:
-            if field_name in model._meta.fk_fields:
+            field = model._meta.fields_map[field_name]
+            if field.source_field:
+                ret.append(field.source_field)
+            elif field_name in model._meta.fk_fields:
                ret.append(field_name + "_id")
            else:
                ret.append(field_name)