Allow to process only specific selected webhook payloads

Author: rennerocha

thebook/webhooks/admin.py

@@ -1,22 +1,14 @@
-from django.contrib import admin
+from django.contrib import admin, messages
+from django.utils.translation import gettext as _
+from django.utils.translation import ngettext
 
 from thebook.webhooks.models import OpenPixWebhookPayload, PaypalWebhookPayload
 
 
-@admin.action(description="Process OpenPix Webhook Payload")
-def process_openpix_payload(modeladmin, request, queryset):
-    OpenPixWebhookPayload.objects.process_received_payloads()
-
-
-@admin.action(description="Process PayPal Webhook Payload")
-def process_paypal_payload(modeladmin, request, queryset):
-    PaypalWebhookPayload.objects.process_received_payloads()
-
-
 @admin.register(OpenPixWebhookPayload)
 class OpenPixWebhookPayloadAdmin(admin.ModelAdmin):
     actions = [
-        process_openpix_payload,
+        "process_payload",
     ]
     list_display = [
         "status",
@@ -26,11 +18,27 @@ class OpenPixWebhookPayloadAdmin(admin.ModelAdmin):
         "status",
     ]
 
+    @admin.action(description=_("Process selected webhook payloads"))
+    def process_payload(self, request, queryset):
+        for processed, payload in enumerate(queryset, start=1):
+            payload.process()
+
+        self.message_user(
+            request,
+            ngettext(
+                "%(processed)d webhook payload processed.",
+                "%(processed)d webhook payloads processed.",
+                processed,
+            )
+            % {"processed": processed},
+            messages.SUCCESS,
+        )
+
 
 @admin.register(PaypalWebhookPayload)
 class PaypalWebhookPayloadAdmin(admin.ModelAdmin):
     actions = [
-        process_paypal_payload,
+        "process_payload",
     ]
     list_display = [
         "status",
@@ -39,3 +47,19 @@ class PaypalWebhookPayloadAdmin(admin.ModelAdmin):
     list_filter = [
         "status",
     ]
+
+    @admin.action(description=_("Process selected webhook payloads"))
+    def process_payload(self, request, queryset):
+        for processed, payload in enumerate(queryset, start=1):
+            payload.process()
+
+        self.message_user(
+            request,
+            ngettext(
+                "%(processed)d webhook payload processed.",
+                "%(processed)d webhook payloads processed.",
+                processed,
+            )
+            % {"processed": processed},
+            messages.SUCCESS,
+        )

thebook/webhooks/managers.py

@@ -9,18 +9,26 @@ class OpenPixWebhookPayloadManager(models.Manager):
     def process_received_payloads(self):
         from thebook.webhooks.models import ProcessingStatus
 
-        for payload in self.filter(status=ProcessingStatus.RECEIVED):
+        for processed, payload in enumerate(
+            self.filter(status=ProcessingStatus.RECEIVED), start=1
+        ):
             payload.process()
 
+        return processed
+
 
 class PayPalWebhookPayloadManager(models.Manager):
 
     def process_received_payloads(self):
         from thebook.webhooks.models import ProcessingStatus
 
-        for payload in self.filter(status=ProcessingStatus.RECEIVED):
+        for processed, payload in enumerate(
+            self.filter(status=ProcessingStatus.RECEIVED), start=1
+        ):
             payload.process()
 
+        return processed
+
 
 def fetch_transactions(start_date: datetime.date, end_date: datetime.date):
     bank_account, _ = BankAccount.objects.get_or_create(name="PayPal")