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)