NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-25 08:46:10 -06:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Revert "Remove _initialized flag from TrackingModelMixin"

Browse Source 
This reverts commit e7481f3bcc.
This commit is contained in:
Jeremy Stretch 2023-07-24 12:25:26 -04:00
parent 296e62579f
commit c2ef15607e
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
netbox/utilities/mixins.py
View File

@ -11,6 +11,10 @@ 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"):
@ -28,17 +32,18 @@ class TrackingModelMixin:
self.tracker.changed = {} self.tracker.changed = {}
def __setattr__(self, name, value): def __setattr__(self, name, value):
change_tracking_fields = registry['counter_fields'][self.__class__] if hasattr(self, "_initialized"):
if name in change_tracking_fields: change_tracking_fields = registry['counter_fields'][self.__class__]
if name not in self.tracker.changed: if name in change_tracking_fields:
if name in self.__dict__: if name not in self.tracker.changed:
old_value = getattr(self, name) if name in self.__dict__:
if value != old_value: old_value = getattr(self, name)
self.tracker.changed[name] = old_value if value != old_value:
self.tracker.changed[name] = old_value
else:
self.tracker.changed[name] = DeferredAttribute
else: else:
self.tracker.changed[name] = DeferredAttribute if value == self.tracker.changed[name]:
else: self.tracker.changed.pop(name)
if value == self.tracker.changed[name]:
self.tracker.changed.pop(name)
super().__setattr__(name, value) super().__setattr__(name, value)