Override label_from_instance for PageChoiceField

jazzband · Jul 9, 2019 · f2c4ffb · f2c4ffb
1 parent 8a0d7b7
commit f2c4ffb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/wagtailmenus/api/form_fields.py b/wagtailmenus/api/form_fields.py
@@ -100,8 +100,18 @@ class PageChoiceField(forms.ModelChoiceField):
     def __init__(self, *args, **kwargs):
         if 'queryset' not in 'kwargs':
             kwargs['queryset'] = Page.objects.none()
+        self.indent_choice_labels = kwargs.pop('indent_choice_labels', True)
         super().__init__(*args, **kwargs)
 
+    def label_from_instance(self, obj):
+        if self.indent_choice_labels:
+            if obj.depth > 1:
+                indent = ''.join('    - ' for i in range(obj.depth-2))
+            else:
+                indent = ''
+            return '{indent}{page}'.format(indent=indent, page=obj)
+        return str(obj)
+
 
 class SiteChoiceField(forms.ModelChoiceField):
 

diff --git a/wagtailmenus/api/v1/forms.py b/wagtailmenus/api/v1/forms.py
@@ -259,6 +259,7 @@ class SectionMenuGeneratorArgumentForm(BaseMenuGeneratorArgumentForm):
     section_root_page = api_form_fields.PageChoiceField(
         label=_("Section root page"),
         required=False,
+        indent_choice_labels=False,
         help_text=_(
             "The root page for the 'section', whose children and other "
             "decendents you want show menu items for. If not supplied, the "