Use UrlPathVersioning for schema view

5monkeys · Apr 4, 2019 · 6e619a3 · 6e619a3
1 parent 42939cc
commit 6e619a3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bananas/admin/api/schemas/yasg.py b/bananas/admin/api/schemas/yasg.py
@@ -9,6 +9,7 @@
 from rest_framework.authentication import SessionAuthentication
 from rest_framework.routers import SimpleRouter
 from rest_framework.schemas.generators import is_custom_action
+from rest_framework.versioning import URLPathVersioning
 
 from ..versioning import BananasVersioning
 from .base import BananasBaseRouter
@@ -104,7 +105,7 @@ def is_navigation(self):
 
 class BananasSimpleRouter(BananasBaseRouter, SimpleRouter):
     def get_schema_view(self):
-        return get_schema_view(
+        view = get_schema_view(
             openapi.Info(
                 title="Django Bananas Admin API Schema",
                 default_version=BananasVersioning.default_version,
@@ -119,3 +120,5 @@ def get_schema_view(self):
             permission_classes=(permissions.AllowAny,),
             patterns=self.urls,
         )
+        view.versioning_class = URLPathVersioning
+        return view