Question

我正在尝试将django-rest-swagger==2.1.1与使用djangorestframework==3.5.3的现有项目进行整合。

该项目有一些基于类的视图和一些基于函数的视图。在集成swagger之后，它显示用于＆＃34;基于类的视图＆＃34;（显然具有序列化器）的POST请求的输入框，但不显示基于＆＃34;功能的视图＆＃34;。这个问题已被多次询问，我尝试过以下解决方案：

Solution1 Solution2

还有其他一些人，但对我的案子不起作用。是否有任何可能的方法来为基于功能的视图做到这一点＆＃39;或者我必须将它们转换为基于类的视图？

Answer 1

在REST Swagger＆gt; = 2.0中不推荐使用YAML docstring解析器我所做的是覆盖SchemaGenerator类，以按照我自己的约定来解析视图的文档字符串。

from rest_framework import exceptions
from rest_framework.permissions import AllowAny
from rest_framework.renderers import CoreJSONRenderer
from rest_framework.response import Response
from rest_framework.schemas import SchemaGenerator
from rest_framework.views import APIView

from rest_framework_swagger import renderers

import yaml
import coreapi
import urlparse

class SchemaGenerator(SchemaGenerator):
    def get_link(self, path, method, view):
        """Custom the coreapi using the func.__doc__ .

        if __doc__ of the function exist, use the __doc__ building the coreapi. else use the default serializer.

        __doc__ in yaml format, eg:

        description: the desc of this api.
        parameters:
            - name: mobile
              desc: the mobile number
              type: string
              required: true
              location: form
            - name: promotion
              desc: the activity id
              type: int
              required: true
              location: form
        """
        fields = self.get_path_fields(path, method, view)
        yaml_doc = None
        if view and view.__doc__:
            try:
                yaml_doc = yaml.load(view.__doc__)
            except:
                yaml_doc = None

        if yaml_doc and type(yaml_doc) != str:
            _method_desc = yaml_doc.get('description', '')
            params = yaml_doc.get('parameters', [])
            for i in params:
                _name = i.get('name')
                _desc = i.get('description')
                _required = i.get('required', False)
                _type = i.get('type', 'string')
                _location = i.get('location', 'form')
                field = coreapi.Field(
                    name=_name,
                    location=_location,
                    required=_required,
                    description=_desc,
                    type=_type
                )
                fields.append(field)
        else:
            _method_desc = view.__doc__ if view and view.__doc__ else ''
            fields += self.get_serializer_fields(path, method, view)
        fields += self.get_pagination_fields(path, method, view)
        fields += self.get_filter_fields(path, method, view)

        if fields and any([field.location in ('form', 'body') for field in fields]):
            encoding = self.get_encoding(path, method, view)
        else:
            encoding = None

        if self.url and path.startswith('/'):
            path = path[1:]

        return coreapi.Link(
            url=urlparse.urljoin(self.url, path),
            action=method.lower(),
            encoding=encoding,
            fields=fields,
            description=_method_desc
        )

def get_swagger_view(title=None, url=None, patterns=None, urlconf=None):
    """
    Returns schema view which renders Swagger/OpenAPI.
    """
    class SwaggerSchemaView(APIView):
        _ignore_model_permissions = True
        exclude_from_schema = True
        permission_classes = [AllowAny]
        renderer_classes = [
            CoreJSONRenderer,
            renderers.OpenAPIRenderer,
            renderers.SwaggerUIRenderer
        ]

        def get(self, request):
            generator = SchemaGenerator(
                title=title,
                url=url,
                patterns=patterns,
                urlconf=urlconf
            )
            schema = generator.get_schema(request=request)

            if not schema:
                raise exceptions.ValidationError(
                    'The schema generator did not return a schema Document'
                )

            return Response(schema)

    return SwaggerSchemaView.as_view()

在项目结构中的任何位置创建此模块。从get_swagger_view中的此模块导入project/urls.py。然后，从get_swagger_view模块中删除django_rest_swagger方法。

参考：Comment by daimon99 in REST Swagger Issues

更新：从django-rest-framework版本3.7开始，由于以上代码无法正常工作，因此会发生重大变化，解决方案为Comment by GuillaumeCisco

Answer 2

您可以使用装饰器：

from rest_framework.decorators import api_view

然后在你的函数使用之上：

@api_view(['POST'])

Django Rest Swagger 2：到目前为止，有没有为基于FUNCTION的视图的POST请求记录参数？

2 个答案: