From 0e039402b2899b4ae1e92c2a14eadf9777f4b7a8 Mon Sep 17 00:00:00 2001 From: Mohit Mittal Date: Wed, 5 Oct 2016 00:57:01 +0000 Subject: [PATCH] change router to take base name from model along with queryset --- rest_framework/routers.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/rest_framework/routers.py b/rest_framework/routers.py index 4eec70bda..d813376bb 100644 --- a/rest_framework/routers.py +++ b/rest_framework/routers.py @@ -133,12 +133,14 @@ class SimpleRouter(BaseRouter): it from the viewset. """ queryset = getattr(viewset, 'queryset', None) - - assert queryset is not None, '`base_name` argument not specified, and could ' \ + model = getattr(viewset, 'model', None) + + assert ((queryset is not None) or (model is not None)), '`base_name` argument not specified, and could ' \ 'not automatically determine the name from the viewset, as ' \ 'it does not have a `.queryset` attribute.' - return queryset.model._meta.object_name.lower() + model = model if model is not None else queryset.model + return model._meta.object_name.lower() def get_routes(self, viewset): """