WesternFriend/WF-website

from django.views.generic import DetailView
import django_filters
from wagtail.admin.filters import DateRangePickerWidget
from wagtail.admin.viewsets.base import ViewSetGroup
from wagtail.admin.viewsets.pages import PageListingViewSet
from wagtail.admin.ui.tables import DateColumn
from wagtail.admin.ui.tables.pages import (
    BulkActionsColumn,
    PageTitleColumn,
    PageStatusColumn,
    NavigateToChildrenColumn,
)

from subscription.views import SubscriptionViewSet
from .models import (
    ArchiveIssue,
    MagazineDepartment,
    MagazineIssue,
)


class MagazineDepartmentDetail(DetailView):
    model = MagazineDepartment
    context_object_name = "department"

    template_name = "magazine/magazine_department_detail.html"


class MagazineIssueFilterSet(PageListingViewSet.filterset_class):
    publication_date = django_filters.DateFromToRangeFilter(
        label="Publication Date",
        widget=DateRangePickerWidget,
    )

    class Meta:
        model = MagazineIssue
        fields = [
            "publication_date",
        ]


class ArchiveIssueViewSet(PageListingViewSet):
    model = ArchiveIssue
    menu_label = "Archive Issues"
    name = "archive_issues"
    icon = "doc-full"
    add_to_admin_menu = False
    columns = [
        PageTitleColumn(
            "title",
            label="Title",
            sort_key="title",
            classname="title",
        ),
        DateColumn(
            "publication_date",
            label="Publication Date",
            sort_key="publication_date",
        ),
        PageStatusColumn(
            "live",
            label="Live",
            sort_key="live",
        ),
        BulkActionsColumn(
            "bulk_actions",
            label="Bulk Actions",
        ),
    ]
    search_fields = (
        "title",
        "internet_archive_identifier",
    )


class MagazineDepartmentViewSet(PageListingViewSet):
    model = MagazineDepartment
    menu_label = "Departments"
    name = "magazine_departments"
    icon = "tag"
    add_to_admin_menu = False
    columns = [
        PageTitleColumn(
            "title",
            label="Title",
            sort_key="title",
            classname="title",
        ),
        BulkActionsColumn(
            "bulk_actions",
            label="Bulk Actions",
        ),
    ]
    search_fields = ("title",)


class MagazineIssueViewSet(PageListingViewSet):
    model = MagazineIssue
    menu_label = "Issues"
    name = "magazine_issues"
    icon = "doc-full-inverse"
    add_to_admin_menu = False
    search_fields = ("title",)
    filterset_class = MagazineIssueFilterSet
    columns = [
        PageTitleColumn(
            "title",
            label="Title",
            sort_key="title",
            classname="title",
        ),
        DateColumn(
            "publication_date",
            label="Publication Date",
            sort_key="publication_date",
        ),
        PageStatusColumn(
            "live",
            label="Live",
            sort_key="live",
        ),
        NavigateToChildrenColumn(
            "navigate_to_children",
            label="Navigate to Children",
        ),
    ]


class MagazineViewSetGroup(ViewSetGroup):
    menu_label = "Magazine"
    menu_icon = "tablet-alt"
    menu_order = 102
    items = (
        ArchiveIssueViewSet,
        MagazineDepartmentViewSet,
        MagazineIssueViewSet,
        SubscriptionViewSet,
    )