From 7afcc1f9e08d61620c5a3e1a0f1c3102982e0041 Mon Sep 17 00:00:00 2001
From: Adam Goldsmith <adam@adamgoldsmith.name>
Date: Thu, 21 Dec 2023 14:56:38 -0500
Subject: [PATCH] membershipworks: Add "refresh data" changelist actions to
 admin

---
 membershipworks/admin.py | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/membershipworks/admin.py b/membershipworks/admin.py
index 2435475..be82d98 100644
--- a/membershipworks/admin.py
+++ b/membershipworks/admin.py
@@ -1,6 +1,12 @@
 from django.contrib import admin
+from django.contrib.humanize.templatetags.humanize import naturaltime
+
+from django_object_actions import DjangoObjectActions, action
+from django_q.tasks import async_task
+from django_q.models import Task
 
 from .models import Member, Flag, Transaction
+from .tasks.scrape import scrape_membershipworks
 
 
 class ReadOnlyAdmin(admin.ModelAdmin):
@@ -14,18 +20,42 @@ class ReadOnlyAdmin(admin.ModelAdmin):
         return False
 
 
+class BaseMembershipWorksAdmin(DjangoObjectActions, ReadOnlyAdmin):
+    changelist_actions = ("refresh_membershipworks_data",)
+
+    # internal method from DjangoObjectActions
+    def _get_tool_dict(self, tool_name):
+        tool = super(DjangoObjectActions, self)._get_tool_dict(tool_name)
+        if tool_name == "refresh_membershipworks_data":
+            last_run = (
+                Task.objects.filter(group="Scrape Data from MembershipWorks")
+                .order_by("started")
+                .last()
+            )
+            tool["label"] = f"Refresh Data [Last Run {naturaltime(last_run.started)}]"
+        return tool
+
+    @action
+    def refresh_membershipworks_data(self, request, obj):
+        async_task(scrape_membershipworks, group="Scrape Data from MembershipWorks")
+        self.message_user(
+            request,
+            "Queued refresh, please wait a few seconds/minutes then refresh the page",
+        )
+
+
 class MemberFlagInline(admin.TabularInline):
     model = Member.flags.through
 
 
 @admin.register(Member)
-class MemberAdmin(ReadOnlyAdmin):
+class MemberAdmin(BaseMembershipWorksAdmin):
     search_fields = ["^first_name", "^last_name", "^account_name"]
     inlines = [MemberFlagInline]
 
 
 @admin.register(Flag)
-class FlagAdmin(ReadOnlyAdmin):
+class FlagAdmin(BaseMembershipWorksAdmin):
     inlines = [MemberFlagInline]
     list_display = ["name", "type"]
     list_filter = ["type"]
@@ -34,7 +64,7 @@ class FlagAdmin(ReadOnlyAdmin):
 
 
 @admin.register(Transaction)
-class TransactionAdmin(ReadOnlyAdmin):
+class TransactionAdmin(BaseMembershipWorksAdmin):
     list_display = ["timestamp", "member", "name", "type", "sum", "note"]
     list_filter = ["type"]
     show_facets = admin.ShowFacets.ALWAYS