Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-16 20:00:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Allow FieldTracker to work in child models

Browse source 
Fixes #57.
This commit is contained in:
Trey Hunner 2013-07-29 12:54:49 -07:00
parent 5687d9836d
commit 98f078d718
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
model_utils/tracker.py
View file

@ -70,10 +70,13 @@ class FieldTracker(object):
if self.fields is None:
self.fields = [field.attname for field in sender._meta.local_fields]
self.field_map = self.get_field_map(sender)
models.signals.post_init.connect(self.initialize_tracker, sender=sender)
models.signals.post_init.connect(self.initialize_tracker)
self.model_class = sender
setattr(sender, self.name, self)
def initialize_tracker(self, sender, instance, **kwargs):
if not isinstance(instance, self.model_class):
return # Only init instances of given model (including children)
tracker = self.tracker_class(instance, self.fields, self.field_map)
setattr(instance, self.attname, tracker)
tracker.set_saved_fields()