⚡ Adds methods to process_db_object_saved of FCMDevice and related setting.

Author: codekaust

fcm_django/models.py

@@ -1,12 +1,12 @@
-from copy import copy
+import importlib
+from copy import copy, deepcopy
 from itertools import repeat
 from typing import List, NamedTuple, Optional, Sequence, Union
 
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 from firebase_admin import messaging
 from firebase_admin.exceptions import FirebaseError, InvalidArgumentError
-
 from fcm_django.settings import FCM_DJANGO_SETTINGS as SETTINGS
 
 # Set by Firebase. Adjust when they adjust; developers can override too if we don't
@@ -346,6 +346,49 @@ def deactivate_devices_with_error_result(
 
 
 class FCMDevice(AbstractFCMDevice):
+
+    # https://github.com/Tournafest/django_backend/blob/aab3250c70fad0fdd25dad7bf472d2d1c188989d/utils/models.py#L131
+    # Followed this strategy to overwrite the save method
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+        self.original_object = deepcopy(self)
+        self.ALLOW_RUN_POST_SAVE_TRIGGERS = True
+        self.COUNT_RUN_SAVE = 0
+        self.IGNORE_COUNT_RUN_SAVE = False
+
+    def save(self, *args, **kwargs):
+
+        self.COUNT_RUN_SAVE += 1
+        if "ALLOW_RUN_POST_SAVE_TRIGGERS" in kwargs:
+            _ALLOW_RUN_POST_SAVE_TRIGGERS = kwargs.pop("ALLOW_RUN_POST_SAVE_TRIGGERS")
+        else:
+            _ALLOW_RUN_POST_SAVE_TRIGGERS = self.ALLOW_RUN_POST_SAVE_TRIGGERS
+
+        if "IGNORE_COUNT_RUN_SAVE" in kwargs:
+            _IGNORE_COUNT_RUN_SAVE = kwargs.pop("IGNORE_COUNT_RUN_SAVE")
+        else:
+            _IGNORE_COUNT_RUN_SAVE = self.IGNORE_COUNT_RUN_SAVE
+
+        # Save the object
+        super().save(*args, **kwargs)
+
+        # Process changes only for the first time the object is saved on an object
+        if _ALLOW_RUN_POST_SAVE_TRIGGERS and (
+            _IGNORE_COUNT_RUN_SAVE or (self.COUNT_RUN_SAVE == 1)
+        ):
+            # Post save processing
+            self.process_db_object_saved()
+
+    def process_db_object_saved(self):
+        if SETTINGS["FCM_POST_SAVE_FUNCTION"]:
+            module_name, function_name = SETTINGS["FCM_POST_SAVE_FUNCTION"].rsplit(
+                ".", 1
+            )
+            module = importlib.import_module(module_name)
+            function = getattr(module, function_name)
+            function(self.original_object, self)
+
     class Meta:
         verbose_name = _("FCM device")
         verbose_name_plural = _("FCM devices")

fcm_django/settings.py

@@ -12,6 +12,10 @@
 # User model
 FCM_DJANGO_SETTINGS.setdefault("USER_MODEL", settings.AUTH_USER_MODEL)
 
+# Post FCM Save Function
+# Input params: original_object, updated_object
+FCM_DJANGO_SETTINGS.setdefault("FCM_POST_SAVE_FUNCTION", "")
+
 FCM_DJANGO_SETTINGS.setdefault(
     "ERRORS",
     {