0
0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00

Get EditHandler form class from model if not overridden

Unless overridden by passing `base_form_class=CustomFormClass` to an
EditHandler, the EditHandler now gets the base form class from the
model. People who override the EditHandler of a model no longer have to
also override the base_form_class if the model needs a custom one.

Fixes #2267
This commit is contained in:
Tim Heap 2016-03-28 02:26:50 +02:00 committed by Karl Hobley
parent dcc89ce9b1
commit 9c25d36041

View File

@ -268,7 +268,7 @@ class BaseFormEditHandler(BaseCompositeEditHandler):
# edit handler. Subclasses can override this attribute to provide a form
# with custom validation, for example. Custom forms must subclass
# WagtailAdminModelForm
base_form_class = WagtailAdminModelForm
base_form_class = None
_form_class = None
@ -279,9 +279,15 @@ class BaseFormEditHandler(BaseCompositeEditHandler):
the children of this edit handler.
"""
if cls._form_class is None:
# If a custom form class was passed to the EditHandler, use it.
# Otherwise, use the base_form_class from the model.
# If that is not defined, use WagtailAdminModelForm.
model_form_class = getattr(model, 'base_form_class', WagtailAdminModelForm)
base_form_class = cls.base_form_class or model_form_class
cls._form_class = get_form_for_model(
model,
form_class=cls.base_form_class,
form_class=base_form_class,
fields=cls.required_fields(),
formsets=cls.required_formsets(),
widgets=cls.widget_overrides())
@ -293,7 +299,7 @@ class BaseTabbedInterface(BaseFormEditHandler):
class TabbedInterface(object):
def __init__(self, children, base_form_class=BaseFormEditHandler.base_form_class):
def __init__(self, children, base_form_class=None):
self.children = children
self.base_form_class = base_form_class
@ -312,7 +318,7 @@ class BaseObjectList(BaseFormEditHandler):
class ObjectList(object):
def __init__(self, children, heading="", classname="",
base_form_class=BaseFormEditHandler.base_form_class):
base_form_class=None):
self.children = children
self.heading = heading
self.classname = classname