# -*- coding: utf-8 -*-
|
"""
|
@File : swagger.py
|
@Time : 2023/1/13 18:22
|
@Author : geekbing
|
@LastEditTime : -
|
@LastEditors : -
|
@Description : -
|
"""
|
|
from drf_yasg.inspectors import SwaggerAutoSchema
|
|
|
class CustomSwaggerAutoSchema(SwaggerAutoSchema):
|
def get_tags(self, operation_keys=None):
|
tags = super().get_tags(operation_keys)
|
|
if "api" in tags and len(operation_keys) >= 3:
|
# `operation_keys` 内容像这样 ['v1', 'prize_join_log', 'create']
|
if operation_keys[2].startswith('run'):
|
tags[0] = 'run'
|
else:
|
tags[0] = operation_keys[2]
|
|
return tags
|
