mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
29 lines
810 B
Python
29 lines
810 B
Python
# coding=utf-8
|
||
"""
|
||
@project: qabot
|
||
@Author：虎
|
||
@file： swagger_conf.py
|
||
@date：2023/9/5 14:01
|
||
@desc: 用于swagger 分组
|
||
"""
|
||
|
||
from drf_yasg.inspectors import SwaggerAutoSchema
|
||
|
||
tags_dict = {
|
||
'user': '用户'
|
||
}
|
||
|
||
|
||
class CustomSwaggerAutoSchema(SwaggerAutoSchema):
|
||
def get_tags(self, operation_keys=None):
|
||
tags = super().get_tags(operation_keys)
|
||
if "api" in tags and operation_keys:
|
||
return [tags_dict.get(operation_keys[1]) if operation_keys[1] in tags_dict else operation_keys[1]]
|
||
return tags
|
||
def get_schema(self, request=None, public=False):
|
||
schema = super().get_schema(request, public)
|
||
if request.is_secure():
|
||
schema.schemes = ['https']
|
||
else:
|
||
schema.schemes = ['http']
|
||
return schema |