89 lines
No EOL
2.4 KiB
Python
89 lines
No EOL
2.4 KiB
Python
|
|
from .resource import Resource
|
|
from .utilities import rupdate
|
|
from apispec import APISpec
|
|
from apispec.ext.marshmallow import MarshmallowPlugin
|
|
|
|
from .fields import Field
|
|
from marshmallow import Schema as BaseSchema
|
|
|
|
from collections import Mapping
|
|
|
|
def view2path(rule: str, view: Resource, spec: APISpec):
|
|
params = {
|
|
"path": rule, # TODO: Validate this slightly (leading / etc)
|
|
"operations": view2operations(view, spec),
|
|
}
|
|
|
|
if hasattr(view, "__apispec__"):
|
|
# Recursively update params
|
|
rupdate(params, view.__apispec__)
|
|
|
|
return params
|
|
|
|
def view2operations(view: Resource, spec: APISpec):
|
|
ops = {}
|
|
for method in Resource.methods:
|
|
if hasattr(view, method):
|
|
ops[method] = {}
|
|
if hasattr(getattr(view, method), "__apispec__"):
|
|
ops[method] = doc2operation(getattr(view, method).__apispec__, spec)
|
|
|
|
return ops
|
|
|
|
|
|
def doc2operation(apispec: dict, spec: APISpec):
|
|
op = {}
|
|
if "_params" in apispec:
|
|
op["requestBody"] = {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": convert_schema(apispec.get("_params"), spec)
|
|
}
|
|
},
|
|
}
|
|
|
|
if "_schema" in apispec:
|
|
op["responses"] = {
|
|
200: {
|
|
"description": "success",
|
|
"content": {
|
|
"application/json": {
|
|
"schema": convert_schema(apispec.get("_schema"), spec)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
for key, val in apispec.items():
|
|
if not key in ["_params", "_schema"]:
|
|
op[key] = val
|
|
|
|
return op
|
|
|
|
|
|
def convert_schema(schema, spec: APISpec):
|
|
if isinstance(schema, BaseSchema):
|
|
return schema
|
|
elif isinstance(schema, Mapping):
|
|
return map2properties(schema, spec)
|
|
else:
|
|
raise TypeError("Unsupported schema type. Ensure schema is a Schema class, or dictionary of Field objects")
|
|
|
|
def map2properties(schema, spec: APISpec):
|
|
marshmallow_plugin = next(
|
|
plugin for plugin in spec.plugins
|
|
if isinstance(plugin, MarshmallowPlugin)
|
|
)
|
|
converter = marshmallow_plugin.converter
|
|
|
|
d = {}
|
|
for k, v in schema.items():
|
|
if isinstance(v, Field):
|
|
d[k] = converter.field2property(v)
|
|
elif isinstance(v, Mapping):
|
|
d[k] = map2properties(v, spec)
|
|
else:
|
|
d[k] = v
|
|
|
|
return {"properties": d} |