add aliases to test prefetch_related

This commit is contained in:
collerek
2020-11-26 06:33:24 +01:00
parent 6893aa5095
commit f8dbb76965
3 changed files with 11 additions and 11 deletions

View File

@ -236,9 +236,7 @@ class ModelTableProxy:
@staticmethod
def _populate_pk_column(
model: Type["Model"],
columns: List[str],
use_alias: bool = False,
model: Type["Model"], columns: List[str], use_alias: bool = False,
) -> List[str]:
pk_alias = (
model.get_column_alias(model.Meta.pkname)

View File

@ -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)