Import sorting and other code cleaning

src/personalisation/wagtail_hooks.py

@@ -1,7 +1,4 @@
-from django.conf import settings
 from django.conf.urls import include, url
-from django.core.urlresolvers import reverse
-
 from wagtail.contrib.modeladmin.options import ModelAdmin, modeladmin_register
 from wagtail.wagtailcore import hooks
 
@@ -11,15 +8,17 @@ from personalisation.models import Segment
 
 @hooks.register('register_admin_urls')
 def register_admin_urls():
+    """Adds the administration urls for the personalisation apps."""
     return [
-        url(r'^personalisation/', include(admin_urls, app_name='personalisation', namespace='personalisation')),
+        url(r'^personalisation/', include(
+            admin_urls,
+            app_name='personalisation',
+            namespace='personalisation')),
     ]
 
 
-"""
-The base model for the Segments administration interface
-"""
 class SegmentModelAdmin(ModelAdmin):
+    """The base model for the Segments administration interface."""
     model = Segment
     menu_icon = 'group'
     add_to_settings_menu = False
@@ -31,12 +30,14 @@ class SegmentModelAdmin(ModelAdmin):
 modeladmin_register(SegmentModelAdmin)
 
 
-"""
-Update the users visit count before each page visit
-"""
 @hooks.register('before_serve_page')
 def set_visit_count(page, request, serve_args, serve_kwargs):
+    """Update the users visit count before each page visit."""
     if request.session.get('visit_count'):
         request.session['visit_count'] = request.session.get('visit_count') + 1
     else:
-        request.session['visit_count'] = 1
+        request.session['visit_count'] = 1
+
+    print("User {} visited {} times.".format(
+        request.session.session_key,
+        request.session['visit_count']))