From 33cf0217da5a753836863c53993da9f14e3c8fba Mon Sep 17 00:00:00 2001 From: blurrah Date: Tue, 30 May 2017 17:36:32 +0200 Subject: [PATCH] fixes page crashing when visiting a page that doesnt inherit PersonalisablePage --- src/wagtail_personalisation/wagtail_hooks.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py index 34f7439..47843c3 100644 --- a/src/wagtail_personalisation/wagtail_hooks.py +++ b/src/wagtail_personalisation/wagtail_hooks.py @@ -141,14 +141,13 @@ def _check_for_variations(segments, page): """ for segment in segments: page_class = page.__class__ - if not any(item == PersonalisablePage for item in page_class.__bases__): - page_class = PersonalisablePage + if any(item == PersonalisablePage for item in page_class.__bases__): - variation = page_class.objects.filter( - canonical_page=page, segment=segment) + variation = page_class.objects.filter( + canonical_page=page, segment=segment) - if variation: - return variation + if variation: + return variation return None