Use constants for enabled/disabling status on Segment

2017-05-31 14:38:58 +02:00
parent 3d920d8ed8
commit 31f8a329f2
6 changed files with 22 additions and 18 deletions
--- a/src/wagtail_personalisation/wagtail_hooks.py
+++ b/src/wagtail_personalisation/wagtail_hooks.py
@ -76,8 +76,8 @@ def serve_variation(page, request, serve_args, serve_kwargs):

    for segment in adapter.get_all_segments():
        try:
-            user_segment = Segment.objects.get(pk=segment['id'],
-                                               status='enabled')
+            user_segment = Segment.objects.get(
+                pk=segment['id'], status=Segment.STATUS_ENABLED)
        except Segment.DoesNotExist:
            user_segment = None
        if user_segment: