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 | |
| parent | 0f508c58211051c873aae5a2d1c65a0c595e732a (diff) | |
| download | django-rest-framework-67735687b297e66c7cfe61614040efcd9763f1f1.tar.bz2 | |
Ensure `_resolve_model` does not return `None`
Diffstat (limited to 'rest_framework/utils')
| -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)) | 
