initial

lib/decorators.py

@@ -0,0 +1,83 @@
+import logging
+from functools import wraps
+
+from django.db.models import Model
+
+from django_proxmox_mikrotik.configs import MIKROTIK_READONLY, MikrotikConfig, PROXMOX_READONLY, ProxmoxConfig
+
+
+def readonly(func):
+    """Decorator to temporarily enable READONLY for both Proxmox and Mikrotik"""
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        pm_initial = ProxmoxConfig.READONLY
+        mk_initial = MikrotikConfig.READONLY
+
+        ProxmoxConfig.READONLY = True
+        MikrotikConfig.READONLY = True
+        logging.debug(f"READONLY: Setting ProxmoxConfig.READONLY from {pm_initial} to {ProxmoxConfig.READONLY} "
+                      f"and MikrotikConfig.READONLY from {mk_initial} to {MikrotikConfig.READONLY} for {func.__name__}")
+
+        try:
+            return func(*args, **kwargs)
+        finally:
+            logging.debug(f"READONLY: Resetting ProxmoxConfig.READONLY from {pm_initial} to {ProxmoxConfig.READONLY} "
+                          f"and MikrotikConfig.READONLY from {mk_initial} to {MikrotikConfig.READONLY} for {func.__name__}")
+            ProxmoxConfig.READONLY = PROXMOX_READONLY
+            MikrotikConfig.READONLY = MIKROTIK_READONLY
+    
+    return wrapper
+
+def force_write(func):
+    """Decorator to temporarily disable READONLY for both Proxmox and Mikrotik"""
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        pm_initial = ProxmoxConfig.READONLY
+        mk_initial = MikrotikConfig.READONLY
+        
+        ProxmoxConfig.READONLY = False
+        MikrotikConfig.READONLY = False
+
+        logging.debug(f"FORCE WRITE: Setting ProxmoxConfig.READONLY from {pm_initial} to {ProxmoxConfig.READONLY} "
+                      f"and MikrotikConfig.READONLY from {mk_initial} to {MikrotikConfig.READONLY} for {func.__name__}")
+        try:
+            return func(*args, **kwargs)
+        finally:
+            logging.debug(f"FORCE WRITE: Resetting ProxmoxConfig.READONLY from {pm_initial} to {ProxmoxConfig.READONLY} "
+                          f"and MikrotikConfig.READONLY from {mk_initial} to {MikrotikConfig.READONLY} for {func.__name__}")
+            ProxmoxConfig.READONLY = PROXMOX_READONLY
+            MikrotikConfig.READONLY = MIKROTIK_READONLY
+    
+    return wrapper
+
+
+def skip_signal(signaltype='post_save', **kwargs):
+    """This should be used as decorator for signal handlers to prevent recursion.
+    Mostly used for post_save and post_delete.
+    The signaltype is just for logging."""
+
+    def _decorator(signal_handler):
+        @wraps(signal_handler)
+        def _wrapper(sender, instance: Model, **kwargs):
+
+            if getattr(instance, '_skip_signal', False):
+                logging.debug(
+                    f'Skip signal handler for {signaltype} : {signal_handler.__name__} - {sender.__name__} - {instance.__class__.__name__}')
+                return
+
+            instance._skip_signal = True
+
+            try:
+                return signal_handler(sender, instance, **kwargs)
+            finally:
+                try:
+                    del instance._skip_signal
+                except AttributeError:
+                    logging.debug(
+                        f'{instance.__class__.__name__} instance has no attribute "_skip_signal" - could not delete it.')
+                except Exception as e:
+                    logging.exception('WTF????', str(e))
+
+        return _wrapper
+
+    return _decorator