add aliases to test prefetch_related

2020-11-26 06:33:24 +01:00
parent 6893aa5095
commit f8dbb76965
3 changed files with 11 additions and 11 deletions
--- a/ormar/queryset/prefetch_query.py
+++ b/ormar/queryset/prefetch_query.py
@ -385,6 +385,7 @@ class PrefetchQuery:
            instance = self._populate_nested_related(
                model=instance, prefetch_dict=prefetch_dict,
            )
+            field_db_name = target_model.get_column_alias(field_name)
            self.already_extracted[target_model.get_name()].setdefault(
                field_name, dict()
-            ).setdefault(row[field_name], []).append(instance)
+            ).setdefault(row[field_db_name], []).append(instance)