浅谈django框架集成swagger以及自定义参数问题
作者:长春小汪汪 发布时间:2022-01-09 20:25:10
介绍
我们在实际的开发工作中需要将django框架与swagger进行集成,用于生成API文档。网上也有一些关于django集成swagger的例子,但由于每个项目使用的依赖版本不一样,因此可能有些例子并不适合我们。我也是在实际集成过程中遇到了一些问题,例如如何自定义参数等问题,最终成功集成,并将结果分享给大家。
开发版本
我开发使用的依赖版本,我所使用的都是截止发稿日期为止最新的版本:
Django 2.2.7
django-rest-swagger 2.2.0
djangorestframework 3.10.3
修改settings.py
1、项目引入rest_framework_swagger依赖
INSTALLED_APPS = [
......
'rest_framework_swagger',
......
]
2、设置DEFAULT_SCHEMA_CLASS,此处不设置后续会报错。
REST_FRAMEWORK = {
......
'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema',
......
}
在app下面创建schema_view.py
在此文件中,我们要继承coreapi中的SchemaGenerator类,并重写get_links方法,重写的目的就是实现我们自定义参数,并且能在页面上展示。此处直接复制过去使用即可。
from rest_framework.schemas import SchemaGenerator
from rest_framework.schemas.coreapi import LinkNode, insert_into
from rest_framework.renderers import *
from rest_framework_swagger import renderers
from rest_framework.response import Response
from rest_framework.decorators import APIView
from rest_framework.permissions import AllowAny,IsAuthenticated,IsAuthenticatedOrReadOnly
from django.http import JsonResponse
class MySchemaGenerator(SchemaGenerator):
def get_links(self, request=None):
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', ())
class SwaggerSchemaView(APIView):
_ignore_model_permissions = True
exclude_from_schema = True
#permission_classes = [AllowAny]
# 此处涉及最终展示页面权限问题,如果不需要认证,则使用AllowAny,这里需要权限认证,因此使用IsAuthenticated
permission_classes = [IsAuthenticated]
# from rest_framework.renderers import *
renderer_classes = [
CoreJSONRenderer,
renderers.OpenAPIRenderer,
renderers.SwaggerUIRenderer
]
def get(self, request):
# 此处的titile和description属性是最终页面最上端展示的标题和描述
generator = MySchemaGenerator(title='API说明文档',description='''接口测试、说明文档''')
schema = generator.get_schema(request=request)
# from rest_framework.response import Response
return Response(schema)
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)
实际应用
在你的应用中定义一个接口,并发布。我这里使用一个测试接口进行验证。
注意
1、所有的接口必须采用calss的方式定义,因为要继承APIView。
2、class下方的注释post,是用来描述post方法的作用,会在页面上进行展示。
3、coreapi_fields 中定义的属性name是参数名称,location是传值方式,我这里一个采用query查询,一个采用header,因为我们进行身份认证,必须将token放在header中,如果你没有,去掉就好了,这里的参数根据你实际项目需要进行定义。
4、最后定义post方法,也可以是get、put等等,根据实际情况定义。
# 这里是之前在schema_view.py中定义好的通用方法,引入进来
from app.schema_view import DocParam
'''
测试
'''
class CustomView(APIView):
'''
post:
测试测试测试
'''
coreapi_fields = (
DocParam(name="id",location='query',description='测试接口'),
DocParam(name="AUTHORIZATION", location='header', description='token'),
)
def post(self, request):
print(request.query_params.get('id'));
return JsonResponse({'message':'成功!'})
5、接收参数这块一定要注意,我定义了一个公用的方法,这里不做过多阐述,如实际过程遇到应用接口与swagger调用接口的传值问题,可参考如下代码。
def getparam(attr,request):
obj = request.POST.get(attr);
if obj is None:
obj = request.query_params.get(attr);
return obj;
修改url.py
针对上一步中定义的测试接口,我们做如下配置。
from django.contrib import admin
from rest_framework import routers
from django.conf.urls import url,include
# 下面是刚才自定义的schema
from app.schema_view import SwaggerSchemaView
# 自定义接口
from app.recommend import CustomView
router = routers.DefaultRouter()
urlpatterns = [
# swagger接口文档路由
url(r"^docs/$", SwaggerSchemaView.as_view()),
url(r'^admin/', admin.site.urls),
url(r'^', include(router.urls)),
# drf登录
url(r'^api-auth/', include('rest_framework.urls', namespace='rest_framework'))
# 测试接口
url(r'^test1', CustomView.as_view(), name='test1'),
]
效果展示
访问地址:http://localhost:8001/docs/
来源:https://blog.csdn.net/u011449784/article/details/103292801
猜你喜欢
- hello,我是李华同学,最近开始学习爬虫,下面是我实现的一个得到弹幕的代码找一个的URL想要得到一个网站的内容,首先要找到你想要内容的具体
- 下面看下python调用函数加括号和不加括号的区别,具体代码如下所示; def bracket(data):return dat
- 本文实例讲述了Python实用库 PrettyTable。分享给大家供大家参考,具体如下:PrettyTable安装使用pip即可十分方便的
- 针对border边框属性在浏览器中的渲染方式很早以前就开始在QQ群中看到大家在讨论,而我也一直以border:0 none;的方式处理。其中
- 通常,当一个页面有太多信息要显示,而一页塞又不下所有信。为了请求速度、美观以及其他的各种理由,分页就会被我们请过来。让我们的用户可以选择是否
- 简介Python下有许多款不同的 Web 框架。Django是重量级选手中最有代表性的一位。许多成功的网站和APP都基于Django。Dja
- 最近为数据库服务器增加了内存,达到了最大支持的8G,数据库用的是mssql 2005 ,之前内存一直是4G的,不存在内存大和32位操作系统冲
- 在Mac上按照官网教程安装成功tensor flow后,但在程序中导入时,仍然报错,包括但不限于以下两个错误。对于这种错误,原因主要在于Ma
- 下面发一个简单的在线调试服务端js代码的asp源码。并可以提示代码具体错误信息。<%@language="javascrip
- 本文实例讲述了Python实现的随机森林算法。分享给大家供大家参考,具体如下:随机森林是数据挖掘中非常常用的分类预测算法,以分类或回归的决策
- 我们知道,在js中,当object作为参数传递到函数中进行处理后,实际上是修改了传入的对象本身(或者说是对象的引用),但很多时候我们并不希望
- 本文实例讲述了thinkPHP中配置的读取与C方法。分享给大家供大家参考,具体如下:1.项目公共配置Conf/config.php内容如下&
- 当我们开始精通编程语言时,我们不仅希望实现最终目标,而且希望使我们的程序高效。在这个教程中,我们将学习一些Ipython的命令,这些命令可以
- 无论是公司的同事还是外界的程序员朋友们,大部分人对JavaScript的高级应用不甚了解,已有的知识架构里会认为JavaScript仅仅是一
- 本文实例讲述了Python面向对象class类属性及子类用法。分享给大家供大家参考,具体如下:class类属性class Foo(objec
- 实例如下:/** * 数字格式转换成千分位 *@param{Object}num */function commafy(num){ &nbs
- 随机生成10位数密码,字母和数字组合import string>>> import random>>>
- <html><head><meta http-equiv="Content-T
- 在使用javascript编程时浏览器中经常会遇到的两个麻烦,下面提供两个函数解决方式做参考,并提供一个demo让你更直观地了解我所说的这两
- 内容摘要:本文介绍了使用js来实现下拉伸缩导航菜单的功能,并带有渐显的效果,值得收藏。正好这几天公司不忙,学校又没有事情,所以想抽空架一个个