From c5e21658deb87afbc30b1699771fac75acfac259 Mon Sep 17 00:00:00 2001
From: Todd Dembrey <todd.dembrey@torchbox.com>
Date: Wed, 24 Jan 2018 14:11:21 +0000
Subject: [PATCH] Allow customisation of the form class

---
 opentech/apply/stream_forms/models.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/opentech/apply/stream_forms/models.py b/opentech/apply/stream_forms/models.py
index 239d5c00c..699714c23 100644
--- a/opentech/apply/stream_forms/models.py
+++ b/opentech/apply/stream_forms/models.py
@@ -8,6 +8,8 @@ from .forms import BlockFieldWrapper, StreamBaseForm
 
 
 class AbstractStreamForm(AbstractForm):
+    page_form_class = StreamBaseForm
+
     class Meta:
         abstract = True
 
@@ -27,4 +29,4 @@ class AbstractStreamForm(AbstractForm):
         return form_fields
 
     def get_form_class(self):
-        return type('WagtailStreamForm', (StreamBaseForm,), self.get_form_fields())
+        return type('WagtailStreamForm', (self.page_form_class,), self.get_form_fields())
-- 
GitLab