23 lines
559 B
Python
23 lines
559 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
|