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

Merge ModelTracker and descriptor classes

Browse source
This commit is contained in:
Trey Hunner 2013-02-18 16:36:05 -08:00
parent a68649c9e7
commit 1ae2effc22
1 changed files with 8 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
model_utils/tracker.py
View file

@ -8,22 +8,16 @@ class ModelTracker(object):
def contribute_to_class(self, cls, name):
self.name = name
models.signals.class_prepared.connect(self.finalize, sender=cls)
def finalize(self, sender, **kwargs):
descriptor = ModelTrackerDescriptor(sender, self.name, self.fields)
setattr(sender, self.name, descriptor)
class ModelTrackerDescriptor(object):
def __init__(self, cls, name, fields):
self.attname = '_%s' % name
self.fields = fields
if self.fields is None:
self.fields = [field.attname for field in cls._meta.local_fields]
models.signals.post_init.connect(self.initialize, sender=cls)
models.signals.class_prepared.connect(self.finalize_class, sender=cls)
def initialize(self, sender, instance, **kwargs):
def finalize_class(self, sender, **kwargs):
if self.fields is None:
self.fields = [field.attname for field in sender._meta.local_fields]
models.signals.post_init.connect(self.initialize_tracker, sender=sender)
setattr(sender, self.name, self)
def initialize_tracker(self, sender, instance, **kwargs):
tracker = ModelInstanceTracker(instance, self.fields)
setattr(instance, self.attname, tracker)
tracker.set_saved_fields()