Reverted back to default template titles and references.

src/personalisation/admin_urls.py

@@ -4,5 +4,5 @@ from django.conf.urls import url
 from personalisation import views
 
 urlpatterns = [
-    url(r'^overview/$', views.overview, name='overview'),
+    url(r'^segment/$', views.overview, name='overview'),
 ]

src/personalisation/models.py

@@ -5,6 +5,7 @@ from django.db import models
 from django.utils.encoding import python_2_unicode_compatible
 
 from modelcluster.models import ClusterableModel
+from wagtail.wagtailadmin.edit_handlers import FieldPanel
 
 
 """

src/personalisation/views.py

@@ -6,5 +6,5 @@ from django.shortcuts import render
 """
 Segments overview
 """
-def overview():
+def overview(request):
-    return render(request, 'wagtailadmin/segment_overview.html')
+    return render(request, 'wagtailadmin/segment.html')

src/personalisation/wagtail_hooks.py

@@ -3,6 +3,10 @@ from django.conf.urls import include, url
 from django.core.urlresolvers import reverse
 from wagtail.wagtailadmin import widgets
 from wagtail.wagtailadmin.menu import MenuItem
+
+from personalisation import admin_urls
+from wagtail.contrib.modeladmin.helpers import ButtonHelper
+from wagtail.contrib.modeladmin.options import ModelAdmin, modeladmin_register
 from wagtail.wagtailcore import hooks
 
 from .models import Segment
@@ -11,5 +15,22 @@ from .models import Segment
 @hooks.register('register_admin_urls')
 def register_admin_urls():
     return [
-        url(r'^segments/', include(admin_urls, app_name='segments', namespace='segments')),
+        url(r'^personalisation/', include(admin_urls, app_name='personalisation', namespace='personalisation')),
     ]
+
+
+class SegmentButtonHelper(ButtonHelper):
+    def get_buttons_for_obj(self, obj, exclude=[], classnames_add=[], classnames_exclude=[]):
+        return {
+            'url': reverse('personalisation:overview'),
+            'label': _('Segments'),
+            'title': _('Report for this %s') % self.verbose_name,
+        }
+
+
+class SegmentModelAdmin(ModelAdmin):
+    model = Segment
+    add_to_settings_menu = False
+    button_helper_class = SegmentButtonHelper
+
+modeladmin_register(SegmentModelAdmin)