diff options
| author | Doug Beck | 2014-11-18 01:26:23 -0500 |
|---|---|---|
| committer | Doug Beck | 2014-11-18 01:26:23 -0500 |
| commit | 67735687b297e66c7cfe61614040efcd9763f1f1 (patch) | |
| tree | a542e37d3ddcc1179bb7b9c4e1a7e6ea97369d79 /rest_framework/utils/model_meta.py | |
| parent | 0f508c58211051c873aae5a2d1c65a0c595e732a (diff) | |
| download | django-rest-framework-67735687b297e66c7cfe61614040efcd9763f1f1.tar.bz2 | |
Ensure `_resolve_model` does not return `None`
Diffstat (limited to 'rest_framework/utils/model_meta.py')
| -rw-r--r-- | rest_framework/utils/model_meta.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/rest_framework/utils/model_meta.py b/rest_framework/utils/model_meta.py index 82361edf..54f9310d 100644 --- a/rest_framework/utils/model_meta.py +++ b/rest_framework/utils/model_meta.py @@ -43,7 +43,11 @@ def _resolve_model(obj): """ if isinstance(obj, six.string_types) and len(obj.split('.')) == 2: app_name, model_name = obj.split('.') - return models.get_model(app_name, model_name) + resolved_model = models.get_model(app_name, model_name) + if not resolved_model: + raise ValueError("Django did not return a model for " + "{0}.{1}".format(app_name, model_name)) + return resolved_model elif inspect.isclass(obj) and issubclass(obj, models.Model): return obj raise ValueError("{0} is not a Django model".format(obj)) |
