From 8f22dae7b0d0f036e4c91ad6d0b4d885368e4b2b Mon Sep 17 00:00:00 2001 From: Sage Abdullah Date: Wed, 12 Apr 2023 12:44:32 +0100 Subject: [PATCH] Move snippet_viewset assignment to SnippetViewSet.on_register() --- wagtail/snippets/models.py | 1 - wagtail/snippets/views/snippets.py | 3 +++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/wagtail/snippets/models.py b/wagtail/snippets/models.py index a16d01296f..1102fa6fdc 100644 --- a/wagtail/snippets/models.py +++ b/wagtail/snippets/models.py @@ -97,7 +97,6 @@ def _register_snippet_immediately(model, viewset=None): admin_viewset = viewset(model) viewsets.register(admin_viewset) - model.snippet_viewset = admin_viewset SNIPPET_MODELS.append(model) SNIPPET_MODELS.sort(key=lambda x: x._meta.verbose_name) diff --git a/wagtail/snippets/views/snippets.py b/wagtail/snippets/views/snippets.py index 2952e9c342..ab873dcb37 100644 --- a/wagtail/snippets/views/snippets.py +++ b/wagtail/snippets/views/snippets.py @@ -1371,5 +1371,8 @@ class SnippetViewSet(ModelViewSet): def on_register(self): super().on_register() + # For convenience, attach viewset to the model class to allow accessing + # the configuration of a given model. + self.model.snippet_viewset = self viewsets.register(self.chooser_viewset) self.register_model_check()