Remove _initialized flag from TrackingModelMixin

This commit is contained in:
Jeremy Stretch 2023-07-24 10:26:39 -04:00
parent 0ee05a6810
commit e7481f3bcc

View File

@ -11,10 +11,6 @@ class Tracker:
class TrackingModelMixin: class TrackingModelMixin:
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._initialized = True
@property @property
def tracker(self): def tracker(self):
if not hasattr(self._state, "_tracker"): if not hasattr(self._state, "_tracker"):
@ -32,18 +28,17 @@ class TrackingModelMixin:
self.tracker.changed = {} self.tracker.changed = {}
def __setattr__(self, name, value): def __setattr__(self, name, value):
if hasattr(self, "_initialized"): change_tracking_fields = registry['counter_fields'][self.__class__]
change_tracking_fields = registry['counter_fields'][self.__class__] if name in change_tracking_fields:
if name in change_tracking_fields: if name not in self.tracker.changed:
if name not in self.tracker.changed: if name in self.__dict__:
if name in self.__dict__: old_value = getattr(self, name)
old_value = getattr(self, name) if value != old_value:
if value != old_value: self.tracker.changed[name] = old_value
self.tracker.changed[name] = old_value
else:
self.tracker.changed[name] = DeferredAttribute
else: else:
if value == self.tracker.changed[name]: self.tracker.changed[name] = DeferredAttribute
self.tracker.changed.pop(name) else:
if value == self.tracker.changed[name]:
self.tracker.changed.pop(name)
super().__setattr__(name, value) super().__setattr__(name, value)