Register as snippet

praekeltfoundation · Feb 7, 2024 · cc9136e · cc9136e
1 parent 0d14a06
commit cc9136e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/home/models.py b/home/models.py
@@ -983,9 +983,10 @@ def update_embedding(sender, instance, *args, **kwargs):
 
     instance.embedding = embedding
 
-
 class OrderedContentSet(DraftStateMixin, RevisionMixin, index.Indexed, models.Model):
-    revisions = GenericRelation("wagtailcore.Revision", related_query_name="orderedcontentset")
+    revisions = GenericRelation(
+        "wagtailcore.Revision", related_query_name="orderedcontentset"
+    )
     name = models.CharField(
         max_length=255, help_text="The name of the ordered content set."
     )

diff --git a/home/wagtail_hooks.py b/home/wagtail_hooks.py
@@ -4,6 +4,7 @@
 from wagtail.admin import widgets as wagtailadmin_widgets
 from wagtail.admin.menu import AdminOnlyMenuItem
 from wagtail.contrib.modeladmin.options import ModelAdmin, modeladmin_register
+from wagtail.snippets.models import register_snippet
 
 from .models import ContentPage, OrderedContentSet
 
@@ -15,6 +16,7 @@
     ContentUploadView,
 )
 
+register_snippet(OrderedContentSet)
 
 @hooks.register("register_admin_urls")
 def register_import_urls():