1

我正在尝试django-rest-swagger==2.1.1与使用djangorestframework==3.5.3.

该项目有一些基于类的视图和一些基于函数的视图。集成 swagger 后,它显示“基于类的视图”(显然有序列化器)的 POST 请求的输入框,但不显示“基于函数的视图”。这个问题被问了好几次,我尝试了以下解决方案:

解决 方案1 ​​解决方案2

其他人也很少,但对我的情况不起作用。对于“基于函数的视图”是否有任何可能的方法,或者我必须将它们转换为基于类的视图?

4

2 回答 2

1

YAML 文档字符串解析器在 REST Swagger>=2.0 中已弃用
我所做的是重写 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

参考:由 daimon99 在 REST Swagger 问题中发表的评论

更新:从django-rest-framework3.7 版开始,由于上述代码无法正常工作,存在重大更改,解决方案是GuillaumeCisco 的评论

于 2017-09-26T08:18:07.450 回答
-2

您可以使用装饰器:

from rest_framework.decorators import api_view

然后在您的功能上方使用:

@api_view(['POST'])
于 2017-09-26T08:19:43.227 回答