Coverage for backend/django/core/auxiliary/methods/copy_object/model_lookup.py: 100%

1"""Old-to-new row mappings used by selected-object duplication.""" 

2 

3from collections.abc import Iterable, Iterator 

4 

5from django.db.models import Model 

6 

7 

8class ModelLookup: 

9 """Map source primary keys to copied model instances.""" 

10 

11 def __init__(self, object_list: Iterable[Model]): 

12 self.model_map = {model.pk: model for model in object_list} 

13 

14 def get_model(self, pk: int) -> Model | None: 

15 """Return the copied instance for a source primary key.""" 

16 

17 return self.model_map.get(pk) 

18 

19 def __iter__(self) -> Iterator[Model]: 

20 return iter(self.model_map.values()) 

21 

22 

23ModelLookupDict = dict[type[Model], ModelLookup]