From d4ee67b77824240b82f5d320c4ca5be884e63d9f Mon Sep 17 00:00:00 2001 From: Jasper Berghoef Date: Fri, 2 Jun 2017 15:08:33 +0200 Subject: [PATCH] Remove unnecessary query --- src/wagtail_personalisation/wagtail_hooks.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/wagtail_personalisation/wagtail_hooks.py b/src/wagtail_personalisation/wagtail_hooks.py index 447ca72..712e871 100644 --- a/src/wagtail_personalisation/wagtail_hooks.py +++ b/src/wagtail_personalisation/wagtail_hooks.py @@ -120,8 +120,7 @@ def page_listing_more_buttons(page, page_perms, is_parent=False): metadata = page.personalisation_metadata for variation in metadata.variations: - segment = Segment.objects.get(pk=variation.segment_id) - yield Button('%s variation' % (segment.name), + yield Button('%s variation' % (variation.segment.name), reverse('wagtailadmin_pages:edit', args=[variation.variant_id]), attrs={"title": _('Edit this variant')}, classes=("icon", "icon-fa-pencil"),