django - drf-yasg：如何隐藏 Django rest 框架模式？

Question

我将drf_yasg swagger用于我的 Django API。我想知道如何轻松禁用模式和模型。 截屏

这是我的代码：

from .models import Articles
from .serializers import ArticlesSerializer
from rest_framework import viewsets
from rest_framework.response import Response
from rest_framework import status
from rest_framework.authentication import SessionAuthentication,TokenAuthentication, BasicAuthentication
from rest_framework.views import APIView
from rest_framework.permissions import IsAuthenticated
from rest_framework.parsers import JSONParser
from django.utils.decorators import method_decorator
from django.contrib.auth import authenticate, login, logout
from rest_framework.decorators import api_view

from drf_yasg.utils import swagger_auto_schema
from drf_yasg import openapi


@swagger_auto_schema(methods=['get'], operation_description="description", manual_parameters=[
    openapi.Parameter('category', openapi.IN_QUERY, "category1, category2, category3", type=openapi.TYPE_STRING),
    openapi.Parameter('name', openapi.IN_QUERY, "full name", type=openapi.TYPE_STRING),
], responses={
    200: openapi.Response('Response', ArticlesSerializer),
}, tags=['Articles'])

# desactivate POST methode on swagger
@swagger_auto_schema(method='POST', auto_schema=None)

@api_view(['GET','POST'])   
def articles(request):
    """
    List all articles.
    """
    if request.user.is_authenticated:
        if request.method == 'GET':
            articles = Articles.objects.all()
            serializer = ArticlesSerializer(Articles, many=True)
            return JsonResponse(serializer.data, safe=False)

        elif request.method == 'POST':
            data = JSONParser().parse(request)
            serializer = ArticlesSerializer(data=data)
            if serializer.is_valid():
                serializer.save()
                return JsonResponse(serializer.data, status=201)
            return JsonResponse(serializer.errors, status=400)
    return JsonResponse({"status":"403", "message":"User not authenticated"})

如果我添加这个

class UserList(APIView):
swagger_schema = None

我得到了错误：

AssertionError: `method` or `methods` can only be specified on @action or @api_view views

代码编辑：文章功能非常简单，与 API 无关，只有 Python 代码。

这里的views 类也很简单。

类视图：

from django.db import models

class Articles(models.Model):
STATUS = (
   (1, 'PENDING'),
   (2, 'COMPLETED'),
   (3, 'DECLINED'),
   (0, 'BANNED'),
)
name = models.CharField(max_length=100)
...
status = models.PositiveSmallIntegerField(
   choices = STATUS,
   default = 1,
)

标签： djangodjango-rest-frameworkdrf-yasg

Answer 1

我只有一半的答案，为了禁用模型，我将它添加到我的setting.py

SWAGGER_SETTINGS = {
    'DEFAULT_FIELD_INSPECTORS': [
        'drf_yasg.inspectors.CamelCaseJSONFilter',
        'drf_yasg.inspectors.InlineSerializerInspector',
        'drf_yasg.inspectors.RelatedFieldInspector',
        'drf_yasg.inspectors.ChoiceFieldInspector',
        'drf_yasg.inspectors.FileFieldInspector',
        'drf_yasg.inspectors.DictFieldInspector',
        'drf_yasg.inspectors.SimpleFieldInspector',
        'drf_yasg.inspectors.StringDefaultFieldInspector',
    ],
}

归功于这些家伙，这里是有关更多详细信息的文档。