From ed3a9449fd5428bb9d76ecec2a66125cb717f9ba Mon Sep 17 00:00:00 2001
From: Pim Vernooij <pim.vernooij@gmail.com>
Date: Sun, 11 Dec 2016 16:00:35 +0100
Subject: [PATCH] adding personalisable StructBlock for in-page personalisation

---
 src/personalisation/blocks.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 src/personalisation/blocks.py

diff --git a/src/personalisation/blocks.py b/src/personalisation/blocks.py
new file mode 100644
index 0000000..105fa08
--- /dev/null
+++ b/src/personalisation/blocks.py
@@ -0,0 +1,12 @@
+from wagtail.wagtailcore import blocks
+from personalisation.models import Segment
+
+class PersonalisedStructBlock(blocks.StructBlock):
+
+    if Segment.objects.count() > 0:
+    	segment = blocks.ChoiceBlock(choices=[
+            (segment.pk, "{} ({})".format(segment.name, segment.status) ) \
+                for segment in Segment.objects.all()
+        ], required=False, label="Personalisation segment",
+        help_text="Only show this content block for users in this segment")
+