nikdoof/limetime
1
0
Fork 0
mirror of https://github.com/nikdoof/limetime.git synced 2025-12-13 09:42:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

PEP8 the utils

Browse Source
This commit is contained in:
Andrew Williams
2013-05-05 19:12:29 +01:00
parent 6a7fbd496a
commit b0f833ab68
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/timer/models/utils.py
View File

@@ -1,12 +1,13 @@
from django.db.models.fields.related import SingleRelatedObjectDescriptor from django.db.models.fields.related import SingleRelatedObjectDescriptor
from django.db.models.query import QuerySet from django.db.models.query import QuerySet
class InheritanceQuerySet(QuerySet): class InheritanceQuerySet(QuerySet):
def select_subclasses(self, *subclasses): def select_subclasses(self, *subclasses):
if not subclasses: if not subclasses:
subclasses = [o for o in dir(self.model) subclasses = [o for o in dir(self.model)
if isinstance(getattr(self.model, o), SingleRelatedObjectDescriptor)\ if isinstance(getattr(self.model, o), SingleRelatedObjectDescriptor) \
and issubclass(getattr(self.model,o).related.model, self.model)] and issubclass(getattr(self.model, o).related.model, self.model)]
new_qs = self.select_related(*subclasses) new_qs = self.select_related(*subclasses)
new_qs.subclasses = subclasses new_qs.subclasses = subclasses
return new_qs return new_qs