Add basic media chooser viewset

Author: zerolab

src/wagtailmedia/views/viewsets.py

@@ -0,0 +1,63 @@
+from django.utils.translation import gettext_lazy as _
+
+from wagtail.admin.ui.tables import Column
+from wagtail.admin.views.generic.chooser import (
+    ChooseResultsView,
+    ChooseView,
+    ChosenView,
+)
+from wagtail.admin.viewsets.chooser import ChooserViewSet
+
+from ..forms import get_media_form
+from ..models import get_media_model
+
+
+Media = get_media_model()
+
+
+class MediaChooserMixin:
+    def get_object_list(self):
+        return Media.objects.select_related("collection").only(
+            "title", "file", "type", "collection__name", "created_at"
+        )
+
+    @property
+    def columns(self):
+        return super().columns + [
+            Column("file", label="File", accessor="file"),
+            Column("type", label="Type", accessor="type"),
+            Column("collection", label="Collection", accessor="collection"),
+            Column("created_at", label="Uploaded", accessor="created_at"),
+        ]
+
+
+class MediaChooseView(MediaChooserMixin, ChooseView):
+    pass
+
+
+class MediaChooseResultsView(MediaChooserMixin, ChooseResultsView):
+    pass
+
+
+class MediaChosenView(ChosenView):
+    def get_object(self, pk):
+        return Media.objects.only("title", "thumbnail").get(pk=pk)
+
+
+class MediaChooserViewSet(ChooserViewSet):
+    choose_view_class = MediaChooseView
+    chosen_view_class = MediaChosenView
+    choose_results_view_class = MediaChooseResultsView
+    creation_form_class = get_media_form(Media)
+
+    icon = "media"
+    choose_one_text = _("Choose a media item")
+    choose_another_text = _("Choose another media item")
+    edit_item_text = _("Edit this media item")
+    creation_tab_label = _("Upload")
+    create_action_label = _("Upload")
+    create_action_clicked_label = _("Uploading…")
+    form_fields = ["type", "title", "file", "collection", "thumbnail"]
+
+
+media_chooser_viewset = MediaChooserViewSet("media_chooser", model=Media)

src/wagtailmedia/wagtail_hooks.py

@@ -17,6 +17,7 @@
 from wagtailmedia.forms import GroupMediaPermissionFormSet
 from wagtailmedia.models import get_media_model
 from wagtailmedia.permissions import permission_policy
+from wagtailmedia.views.viewsets import media_chooser_viewset
 
 
 @hooks.register("register_admin_urls")
@@ -136,3 +137,8 @@ def register_icons(icons):
         "wagtailmedia/icons/wagtailmedia-audio.svg",  # {% icon "wagtailmedia-audio" %}
         "wagtailmedia/icons/wagtailmedia-video.svg",  # {% icon "wagtailmedia-video" %}
     ]
+
+
+@hooks.register("register_admin_viewset")
+def register_viewset():
+    return media_chooser_viewset