add aliases to test prefetch_related
This commit is contained in:
@ -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)
|
||||
|
||||
@ -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)
|
||||
|
||||
Reference in New Issue
Block a user