cmsmanage/membershipworks/admin.py

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):
    def has_add_permission(self, request, obj=None):
        return False

    def has_change_permission(self, request, obj=None):
        return False

    def has_delete_permission(self, request, obj=None):
        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()
            )
            last_run_time = (
                naturaltime(last_run.started) if last_run is not None else "Never"
            )
            tool["label"] = f"Refresh Data [Last Run {last_run_time}]"
        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(BaseMembershipWorksAdmin):
    search_fields = ["^first_name", "^last_name", "^account_name"]
    inlines = [MemberFlagInline]


@admin.register(Flag)
class FlagAdmin(BaseMembershipWorksAdmin):
    inlines = [MemberFlagInline]
    list_display = ["name", "type"]
    list_filter = ["type"]
    show_facets = admin.ShowFacets.ALWAYS
    search_fields = ["name"]


@admin.register(Transaction)
class TransactionAdmin(BaseMembershipWorksAdmin):
    list_display = ["timestamp", "member", "name", "type", "sum", "note"]
    list_filter = ["type"]
    show_facets = admin.ShowFacets.ALWAYS
    search_fields = ["member", "name", "type", "note"]
    date_hierarchy = "timestamp"