Fixed #27563 -- Moved "apply limit_choices_to" code from BaseModelForm to fields_for_model().

This commit is contained in:
Jon Dufresne 2016-12-01 18:28:29 -08:00
parent 6d1394182d
commit 6abd6c598e
2 changed files with 9 additions and 7 deletions

View File

@ -170,6 +170,11 @@ def fields_for_model(model, fields=None, exclude=None, widgets=None,
formfield = formfield_callback(f, **kwargs)
if formfield:
# Apply ``limit_choices_to``.
if hasattr(formfield, 'queryset') and hasattr(formfield, 'get_limit_choices_to'):
limit_choices_to = formfield.get_limit_choices_to()
if limit_choices_to is not None:
formfield.queryset = formfield.queryset.complex_filter(limit_choices_to)
field_list.append((f.name, formfield))
else:
ignored.append(f.name)
@ -291,13 +296,6 @@ class BaseModelForm(BaseForm):
data, files, auto_id, prefix, object_data, error_class,
label_suffix, empty_permitted, use_required_attribute=use_required_attribute,
)
# Apply ``limit_choices_to`` to each field.
for field_name in self.fields:
formfield = self.fields[field_name]
if hasattr(formfield, 'queryset') and hasattr(formfield, 'get_limit_choices_to'):
limit_choices_to = formfield.get_limit_choices_to()
if limit_choices_to is not None:
formfield.queryset = formfield.queryset.complex_filter(limit_choices_to)
def _get_validation_exclusions(self):
"""

View File

@ -2854,6 +2854,10 @@ class LimitChoicesToTest(TestCase):
f = StumpJokeWithCustomFieldForm()
self.assertEqual(f.fields['custom'].queryset, 42)
def test_fields_for_model_applies_limit_choices_to(self):
fields = fields_for_model(StumpJoke, ['has_fooled_today'])
self.assertSequenceEqual(fields['has_fooled_today'].queryset, [self.threepwood])
class FormFieldCallbackTests(SimpleTestCase):