Coverage for core/querysets.py: 85%

1from django.apps import apps 

2from django.contrib.contenttypes.models import ContentType 

3from django.db.utils import ProgrammingError 

4 

5from utilities.querysets import RestrictedQuerySet 

6 

7__all__ = ( 

8 'ObjectChangeQuerySet', 

9) 

10 

11 

12class ObjectChangeQuerySet(RestrictedQuerySet): 

13 

14 def valid_models(self): 

15 # Exclude any change records which refer to an instance of a model that's no longer installed. This 

16 # can happen when a plugin is removed but its data remains in the database, for example. 

17 try: 

18 content_types = ContentType.objects.get_for_models(*apps.get_models()).values() 

19 except ProgrammingError: 

20 # Handle the case where the database schema has not yet been initialized 

21 content_types = ContentType.objects.none() 

22 

23 content_type_ids = set( 

24 ct.pk for ct in content_types 

25 ) 

26 return self.filter(changed_object_type_id__in=content_type_ids)