ADDED: check issubclass NestedViewSetMixin for nested routes

seebass · Aug 19, 2015 · cf5e799 · cf5e799
1 parent cae2a7a
commit cf5e799
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/drf_nested_routing/routers.py b/drf_nested_routing/routers.py
@@ -5,6 +5,7 @@
 
 from drf_nested_routing import add_parent_query_lookups
 import drf_nested_routing
+from drf_nested_routing.views import NestedViewSetMixin
 
 
 class NestedRegistryItem(object):
@@ -14,6 +15,8 @@ def __init__(self, router, parent_prefix, parent_item=None):
         self.parent_item = parent_item
 
     def register(self, prefix, viewset, parent_query_lookups=None, base_name=None):
+        if not issubclass(viewset, NestedViewSetMixin):
+            raise ValueError("ViewSets of nested routes must subclass NestedViewSetMixin")
         if not base_name:
             base_name = viewset.queryset.model.__name__.lower()
         if parent_query_lookups:

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name='drf-nested-routing',
-    version="0.9.1",
+    version="0.9.2",
     url='https://github.com/seebass/drf-nested-routing',
     license='MIT',
     description='Nested routing extension for Django REST Framework 3',