django rest_framework swagger使用案例

環境准備

　　環境要求：

　　python3

　　django2

　　pip3

　　模塊安裝：

　　pip3 install django-rest-framework

　　pip3 install django-rest-swagger

　　搭建項目：

　　搭建django項目，創建testapi app

參數配置

　　setting.py：　

　　INSTALLED_APPS中添加：rest_framework,rest_framework_swagger

　　

 

視圖編輯

　　編輯views.py

　　

# Create your views here.
# -*- coding: utf-8 -*-

from rest_framework.views import APIView

from rest_framework.permissions import AllowAny
from rest_framework.schemas import SchemaGenerator
from rest_framework.schemas.generators import LinkNode, insert_into
from rest_framework.renderers import *
from rest_framework_swagger import renderers
from rest_framework.response import Response

# from rest_framework.schemas import SchemaGenerator
class MySchemaGenerator(SchemaGenerator):

    def get_links(self, request=None):
        # from rest_framework.schemas.generators import LinkNode,
        links = LinkNode()

        paths = []
        view_endpoints = []
        for path, method, callback in self.endpoints:
            view = self.create_view(callback, method, request)
            path = self.coerce_path(path, method, view)
            paths.append(path)
            view_endpoints.append((path, method, view))

        # Only generate the path prefix for paths that will be included
        if not paths:
            return None
        prefix = self.determine_path_prefix(paths)

        for path, method, view in view_endpoints:
            if not self.has_view_permissions(path, method, view):
                continue
            link = view.schema.get_link(path, method, base_url=self.url)
            # 添加下面這一行方便在views編寫過程中自定義參數.
            link._fields += self.get_core_fields(view)

            subpath = path[len(prefix):]
            keys = self.get_keys(subpath, method, view)

            # from rest_framework.schemas.generators import LinkNode, insert_into
            insert_into(links, keys, link)

        return links

    # 從類中取出我們自定義的參數, 交給swagger 以生成接口文檔.
    def get_core_fields(self, view):
        return getattr(view, 'coreapi_fields', ())

def DocParam(name="default", location="query", required=True, description=None, type="string", *args, **kwargs):
    return coreapi.Field(name=name, location=location, required=required, description=description, type=type)


class ReturnJson(APIView):
    coreapi_fields = (  #用於swagger doc顯示方法必須字符串
        DocParam("name", description='test'),
        DocParam("nalanxiao", required=False, description='rohero'),
    )
    def get(self, request, *args, **kwargs):
        json_data = {'name': 'post', 'id': 0}
        return Response(json_data)

    def post(self, request, *args, **kwargs):
        json_data = {'name': 'post', 'id': 0}
        return Response(json_data)

 

路由設置

　　編輯urls.py

　　

from django.conf.urls import url
from .views import SwaggerSchemaView, ReturnJson, StudentsApiView

urlpatterns = [
    url(r'^api/$', ReturnJson.as_view(), name='api'),
    url(r'^api/v1/$', StudentsApiView.as_view(), name='api_v1'),
    url(r'^docs/', SwaggerSchemaView.as_view(), name='apiDocs'),
]

效果展示：

　　

github：

　　https://github.com/Roherolxh/opstest

　　覺得有幫助望給個小星星