更新drf-yasg的试用

Author: aeasringnar

base_django_api/settings.py

@@ -243,6 +243,8 @@
     # 使用这个时需要使用django-rest的admin 也就是需要配置 url(r'^api-auth/', include('rest_framework.urls', namespace='rest_framework')),
     # 'LOGIN_URL': 'rest_framework:login',
     # 'LOGOUT_URL': 'rest_framework:logout',
+    # 自定义swagger的路由tag
+    'DEFAULT_GENERATOR_CLASS': 'config.swagger.BaseOpenAPISchemaGenerator',
     'USE_SESSION_AUTH': False,
     # 'SHOW_EXTENSIONS': False,
     'DOC_EXPANSION': 'none',

config/swagger.py

@@ -0,0 +1,36 @@
+from drf_yasg.generators import OpenAPISchemaGenerator
+
+
+class BaseOpenAPISchemaGenerator(OpenAPISchemaGenerator):
+    '''重写 OpenAPISchemaGenerator 手动为每个路由添加 tag'''
+
+    def get_schema(self, request=None, public=False):
+        '''重写父类方法'''
+        swagger = super().get_schema(request, public)
+        swagger.tags = [
+            {
+                "name": "adminlogin",
+                "description": "核心功能",
+            },
+            {
+                'name': 'auth',
+                'description': '权限管理'
+            },
+            {
+                'name': 'celery',
+                'description': '异步任务测试'
+            },
+            {
+                'name': 'confdict',
+                'description': '系统字典'
+            },
+            {
+                'name': 'getcode',
+                'description': '发送短信验证码'
+            },
+            {
+                'name': 'member',
+                'description': '会员管理'
+            }
+        ]
+        return swagger