File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 6
6
Usage: `get_field_info(model)` returns a `FieldInfo` instance.
7
7
"""
8
8
from collections import namedtuple
9
+ from django .core .exceptions import ImproperlyConfigured
9
10
from django .db import models
10
11
from django .utils import six
11
12
from rest_framework .compat import OrderedDict
@@ -44,9 +45,9 @@ def _resolve_model(obj):
44
45
if isinstance (obj , six .string_types ) and len (obj .split ('.' )) == 2 :
45
46
app_name , model_name = obj .split ('.' )
46
47
resolved_model = models .get_model (app_name , model_name )
47
- if not resolved_model :
48
- raise ValueError ( "Django did not return a model for "
49
- "{0}.{1}" .format (app_name , model_name ))
48
+ if resolved_model is None :
49
+ msg = "Django did not return a model for {0}.{1} "
50
+ raise ImproperlyConfigured ( msg .format (app_name , model_name ))
50
51
return resolved_model
51
52
elif inspect .isclass (obj ) and issubclass (obj , models .Model ):
52
53
return obj
Original file line number Diff line number Diff line change 1
1
from __future__ import unicode_literals
2
+ from django .core .exceptions import ImproperlyConfigured
2
3
from django .conf .urls import patterns , url
3
4
from django .test import TestCase
4
5
from django .utils import six
@@ -161,5 +162,5 @@ def tearDown(self):
161
162
rest_framework .utils .model_meta .models .get_model = self .get_model
162
163
163
164
def test_blows_up_if_model_does_not_resolve (self ):
164
- with self .assertRaises (ValueError ):
165
+ with self .assertRaises (ImproperlyConfigured ):
165
166
_resolve_model ('tests.BasicModel' )
You can’t perform that action at this time.
0 commit comments