76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
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"
|