Reverted back to default template titles and references.

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)