def __new__(cls, name, bases, attrs):
# get all valid existing Fields and sort them
fields = [(field_name, attrs.pop(field_name)) for field_name, obj in \
attrs.items() if isinstance(obj, forms.Field)]
fields.sort(lambda x, y: cmp(x[1].creation_counter, y[1].creation_counter))
# get all Fields from base classes
for base in bases[::-1]:
if hasattr(base, 'base_fields'):
fields = base.base_fields.items() + fields
# add the fields as "our" base fields
attrs['base_fields'] = SortedDict(fields)
# Meta class available?
if 'Meta' in attrs and hasattr(attrs['Meta'], 'document') and \
issubclass(attrs['Meta'].document, BaseDocument):
doc_fields = SortedDict()
formfield_generator = getattr(attrs['Meta'], 'formfield_generator', \
MongoFormFieldGenerator)()
widgets = getattr(attrs["Meta"], "widgets", {})
# walk through the document fields
for field_name, field in iter_valid_fields(attrs['Meta']):
# add field and override clean method to respect mongoengine-validator
# use to get a custom widget
if hasattr(field, 'get_custom_widget'):
widget = field.get_custom_widget()
else:
widget = widgets.get(field_name, None)
if widget:
doc_fields[field_name] = formfield_generator.generate(
field, widget=widget)
else:
doc_fields[field_name] = formfield_generator.generate(
field)
if not isinstance(field, FileField):
doc_fields[field_name].clean = mongoengine_validate_wrapper(
field,
doc_fields[field_name].clean, field._validate)
# write the new document fields to base_fields
doc_fields.update(attrs['base_fields'])
attrs['base_fields'] = doc_fields
# maybe we need the Meta class later
attrs['_meta'] = attrs.get('Meta', object())
new_class = super(MongoFormMetaClass, cls).__new__(cls, name, bases, attrs)
if 'media' not in attrs:
new_class.media = media_property(new_class)
return new_class
def __new__(cls, name, bases, attrs):
# get all valid existing Fields and sort them
fields = [(field_name, attrs.pop(field_name)) for field_name, obj in \
attrs.items() if isinstance(obj, forms.Field)]
fields.sort(lambda x, y: cmp(x[1].creation_counter, y[1].creation_counter))
# get all Fields from base classes
for base in bases[::-1]:
if hasattr(base, 'base_fields'):
fields = base.base_fields.items() + fields
# add the fields as "our" base fields
attrs['base_fields'] = SortedDict(fields)
# Meta class available?
if 'Meta' in attrs and hasattr(attrs['Meta'], 'document') and \
issubclass(attrs['Meta'].document, BaseDocument):
doc_fields = SortedDict()
formfield_generator = getattr(attrs['Meta'], 'formfield_generator', \
MongoFormFieldGenerator)()
widgets = getattr(attrs["Meta"], "widgets", {})
# walk through the document fields
for field_name, field in iter_valid_fields(attrs['Meta']):
# add field and override clean method to respect mongoengine-validator
# use to get a custom widget
if hasattr(field, 'get_custom_widget'):
widget = field.get_custom_widget()
else:
widget = widgets.get(field_name, None)
if widget:
doc_fields[field_name] = formfield_generator.generate(
field, widget=widget)
else:
doc_fields[field_name] = formfield_generator.generate(
field)
doc_fields[field_name].clean = mongoengine_validate_wrapper(
field,
doc_fields[field_name].clean, field._validate)
# write the new document fields to base_fields
doc_fields.update(attrs['base_fields'])
attrs['base_fields'] = doc_fields
# maybe we need the Meta class later
attrs['_meta'] = attrs.get('Meta', object())
return super(MongoFormMetaClass, cls).__new__(cls, name, bases, attrs)
def __init__(
self,
data=None,
files=None,
auto_id="id_%s",
prefix=None,
initial=None,
error_class=forms.util.ErrorList,
label_suffix=":",
empty_permitted=False,
instance=None,
):
""" initialize the form"""
assert isinstance(instance, (types.NoneType, BaseDocument)), (
"instance must be a mongoengine document, not %s" % type(instance).__name__
)
assert hasattr(self, "Meta"), "Meta class is needed to use MongoForm"
# new instance or updating an existing one?
if instance is None:
if self._meta.document is None:
raise ValueError("MongoForm has no document class specified.")
self.instance = self._meta.document()
object_data = {}
self.instance._adding = True
else:
self.instance = instance
self.instance._adding = False
object_data = {}
# walk through the document fields
for field_name, field in iter_valid_fields(self._meta):
# add field data if needed
field_data = getattr(instance, field_name)
if isinstance(self._meta.document._fields[field_name], ReferenceField):
# field data could be None for not populated refs
field_data = field_data and str(field_data.id)
object_data[field_name] = field_data
# additional initial data available?
if initial is not None:
object_data.update(initial)
self._validate_unique = False
super(MongoForm, self).__init__(
data, files, auto_id, prefix, object_data, error_class, label_suffix, empty_permitted
)
def save(self, commit=True):
"""save the instance or create a new one.."""
# walk through the document fields
for field_name, field in iter_valid_fields(self._meta):
# FileFields need some more work to ensure the filename is unique
if isinstance(self.instance._fields[field_name], FileField):
field = save_file(self.instamce, field_name, self.cleaned_data.get(field_name))
setattr(self.instance, field_name, field)
continue
setattr(self.instance, field_name, self.cleaned_data.get(field_name))
if commit:
self.instance.save()
return self.instance
def save(self, commit=True):
"""save the instance or create a new one.."""
# walk through the document fields
for field_name, field in iter_valid_fields(self._meta):
# FileFields need some more work to ensure the filename is unique
if isinstance(self.instance._fields[field_name], FileField):
field = save_file(self.instamce, field_name, self.cleaned_data.get(field_name))
setattr(self.instance, field_name, field)
continue
setattr(self.instance, field_name, self.cleaned_data.get(field_name))
if commit:
self.instance.save()
return self.instance
def __init__(self,
data=None,
files=None,
auto_id='id_%s',
prefix=None,
initial=None,
error_class=forms.utils.ErrorList,
label_suffix=':',
empty_permitted=False,
instance=None):
""" initialize the form"""
assert isinstance(instance, (types.NoneType, BaseDocument)), \
'instance must be a mongoengine document, not %s' % \
type(instance).__name__
assert hasattr(self, 'Meta'), 'Meta class is needed to use MongoForm'
# new instance or updating an existing one?
if instance is None:
if self._meta.document is None:
raise ValueError('MongoForm has no document class specified.')
self.instance = self._meta.document()
object_data = {}
self.instance._adding = True
else:
self.instance = instance
self.instance._adding = False
object_data = {}
# walk through the document fields
for field_name, field in iter_valid_fields(self._meta):
# add field data if needed
field_data = getattr(instance, field_name)
if isinstance(self._meta.document._fields[field_name],
ReferenceField):
# field data could be None for not populated refs
field_data = field_data and str(field_data.id)
object_data[field_name] = field_data
# additional initial data available?
if initial is not None:
object_data.update(initial)
self._validate_unique = False
super(MongoForm,
self).__init__(data, files, auto_id, prefix, object_data,
error_class, label_suffix, empty_permitted)
