1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
def response_schema(schema, code=200, required=False, description=None):
"""
Add response info into the swagger spec
Usage:
.. code-block:: python
from aiohttp import web
from marshmallow import Schema, fields
class ResponseSchema(Schema):
msg = fields.Str()
data = fields.Dict()
@response_schema(ResponseSchema(), 200)
async def index(request):
return web.json_response({'msg': 'done', 'data': {}})
:param str description: response description
:param bool required:
:param schema: :class:`Schema <marshmallow.Schema>` class or instance
:param int code: HTTP response code
"""
if callable(schema):
schema = schema()
def wrapper(func):
if not hasattr(func, "__apispec__"):
func.__apispec__ = {"schemas": [], "responses": {}, "parameters": []}
func.__schemas__ = []
func.__apispec__["responses"]["%s" % code] = {
"schema": schema,
"required": required,
"description": description or "",
}
return func
return wrapper
# For backward compatibility
marshal_with = response_schema
|
