diff --git a/app/timer/models/utils.py b/app/timer/models/utils.py index d0dcdf3..df4fedc 100644 --- a/app/timer/models/utils.py +++ b/app/timer/models/utils.py @@ -1,12 +1,13 @@ from django.db.models.fields.related import SingleRelatedObjectDescriptor from django.db.models.query import QuerySet + class InheritanceQuerySet(QuerySet): def select_subclasses(self, *subclasses): if not subclasses: subclasses = [o for o in dir(self.model) - if isinstance(getattr(self.model, o), SingleRelatedObjectDescriptor)\ - and issubclass(getattr(self.model,o).related.model, self.model)] + if isinstance(getattr(self.model, o), SingleRelatedObjectDescriptor) \ + and issubclass(getattr(self.model, o).related.model, self.model)] new_qs = self.select_related(*subclasses) new_qs.subclasses = subclasses return new_qs