def __init__(self, *args, **kwargs):
try:
self.type_label = kwargs.pop('type_label')
except KeyError:
raise ValueError("CustomModelField requires 'type_label' kwarg")
try:
model_field = kwargs.pop('model_field')
except KeyError:
raise ValueError("CustomModelField requires 'model_field' kwarg")
related_model = _resolve_model(model_field.rel.to)
kwargs = {
'queryset': related_model._default_manager
#,
#'many': False
}
if model_field:
kwargs['required'] = not(model_field.null or model_field.blank)
super(CustomModelField, self).__init__(**kwargs)
def __init__(self, *args, **kwargs):
try:
self.type_label = kwargs.pop('type_label')
except KeyError:
raise ValueError("CustomModelField requires 'type_label' kwarg")
try:
model_field = kwargs.pop('model_field')
except KeyError:
raise ValueError("CustomModelField requires 'model_field' kwarg")
related_model = _resolve_model(model_field.rel.to)
kwargs = {
'queryset': related_model._default_manager
#,
#'many': False
}
if model_field:
kwargs['required'] = not (model_field.null or model_field.blank)
super(CustomModelField, self).__init__(**kwargs)
def test_blows_up_if_model_does_not_resolve(self):
with self.assertRaises(ImproperlyConfigured):
_resolve_model('tests.BasicModel')
def test_resolve_improper_string_representation(self):
with self.assertRaises(ValueError):
_resolve_model('BasicModel')
def test_resolve_non_django_model(self):
with self.assertRaises(ValueError):
_resolve_model(TestCase)
def test_resolve_unicode_representation(self):
resolved_model = _resolve_model(six.text_type('tests.BasicModel'))
self.assertEqual(resolved_model, BasicModel)
def test_resolve_string_representation(self):
resolved_model = _resolve_model('tests.BasicModel')
self.assertEqual(resolved_model, BasicModel)
def test_resolve_django_model(self):
resolved_model = _resolve_model(BasicModel)
self.assertEqual(resolved_model, BasicModel)
def test_blows_up_if_model_does_not_resolve(self):
with self.assertRaises(ImproperlyConfigured):
_resolve_model('tests.BasicModel')
def test_resolve_improper_string_representation(self):
with self.assertRaises(ValueError):
_resolve_model('BasicModel')
def test_resolve_non_django_model(self):
with self.assertRaises(ValueError):
_resolve_model(TestCase)
def test_resolve_unicode_representation(self):
resolved_model = _resolve_model(six.text_type('tests.BasicModel'))
self.assertEqual(resolved_model, BasicModel)
def test_resolve_string_representation(self):
resolved_model = _resolve_model('tests.BasicModel')
self.assertEqual(resolved_model, BasicModel)
def test_resolve_django_model(self):
resolved_model = _resolve_model(BasicModel)
self.assertEqual(resolved_model, BasicModel)
