def formfield_for_dbfield(db_field, **kwargs):
if db_field.name == 'collection':
return CollectionChoiceField(queryset=Collection.objects.all(),
empty_label=None,
**kwargs)
# For all other fields, just call its formfield() method.
return db_field.formfield(**kwargs)
def formfield_for_dbfield(db_field, **kwargs):
# Check if this is the file field
if db_field.name == 'file':
return WagtailImageField(label=capfirst(db_field.verbose_name), **kwargs)
elif db_field.name == 'collection':
return CollectionChoiceField(label=_("Collection"), queryset=Collection.objects.all(), empty_label=None, **kwargs)
# For all other fields, just call its formfield() method.
return db_field.formfield(**kwargs)
