python/django-maraudes
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

clean up

Browse Source
This commit is contained in:
Arthur Gerbaud
2016-08-06 18:08:06 +02:00
parent e2456a4ae0
commit af5c8d61e9
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
notes/models.py
View File

@@ -33,28 +33,28 @@ class Note(models.Model):
self.created_date = self.cast().get_date() self.created_date = self.cast().get_date()
return super().save(*args, **kwargs) return super().save(*args, **kwargs)
def _get_child_and_subclass(self): def _get_child_class_and_instance(self):
self._child_instance = self self._child_instance = self
self._subclass = self.__class__ self._child_class = self.__class__
for f in self._meta.get_fields(): for f in self._meta.get_fields():
if f.is_relation and f.one_to_one: if f.is_relation and f.one_to_one:
self._child_instance = getattr(self, f.name) self._child_instance = getattr(self, f.name)
self._subclass = self._child_instance.__class__ self._child_class = self._child_instance.__class__
return return
@property @property
def subclass(self): def subclass(self):
if not hasattr(self, '_subclass'): if not hasattr(self, '_child_class'):
self._get_child_and_subclass() self._get_child_class_and_instance()
return self._subclass return self._child_class
def cast(self): def cast(self):
if not hasattr(self, '_child_instance'): if not hasattr(self, '_child_instance'):
self._get_child_and_subclass() self._get_child_class_and_instance()
return self._child_instance return self._child_instance
def __str__(self): def __str__(self):
return "%s of %s" % (self.subclass.__qualname__, self.created_by) return "%s of %s" % (self.child_class.__qualname__, self.created_by)
## Attributes used by 'notes' template tags ## Attributes used by 'notes' template tags
# bg_color : background color of header # bg_color : background color of header