openapi-core/openapi_core/parameters.py

72 lines
2.2 KiB
Python
Raw Normal View History

2017-09-21 11:51:37 +00:00
"""OpenAPI core parameters module"""
import logging
import warnings
from openapi_core.exceptions import EmptyValue
2017-09-21 11:51:37 +00:00
log = logging.getLogger(__name__)
class Parameter(object):
"""Represents an OpenAPI operation Parameter."""
def __init__(
self, name, location, schema=None, required=False,
deprecated=False, allow_empty_value=False,
2017-09-21 11:51:37 +00:00
items=None, collection_format=None):
self.name = name
self.location = location
self.schema = schema
self.required = True if self.location == "path" else required
2017-09-21 11:51:37 +00:00
self.deprecated = deprecated
self.allow_empty_value = (
allow_empty_value if self.location == "query" else False
)
2017-09-21 11:51:37 +00:00
self.items = items
self.collection_format = collection_format
def unmarshal(self, value):
if self.deprecated:
warnings.warn(
"{0} parameter is deprecated".format(self.name),
DeprecationWarning,
)
if (self.location == "query" and value == "" and
not self.allow_empty_value):
raise EmptyValue(
"Value of {0} parameter cannot be empty.".format(self.name))
2017-09-21 11:51:37 +00:00
if not self.schema:
return value
return self.schema.unmarshal(value)
class ParametersGenerator(object):
2017-09-22 08:54:37 +00:00
def __init__(self, dereferencer, schemas_registry):
2017-09-21 11:51:37 +00:00
self.dereferencer = dereferencer
2017-09-22 08:54:37 +00:00
self.schemas_registry = schemas_registry
2017-09-21 11:51:37 +00:00
def generate(self, paramters):
for parameter in paramters:
parameter_deref = self.dereferencer.dereference(parameter)
allow_empty_value = parameter_deref.get('allowEmptyValue')
2017-09-21 11:51:37 +00:00
required = parameter_deref.get('required', False)
schema_spec = parameter_deref.get('schema', None)
schema = None
if schema_spec:
2017-09-22 08:54:37 +00:00
schema, _ = self.schemas_registry.get_or_create(schema_spec)
2017-09-21 11:51:37 +00:00
yield (
parameter_deref['name'],
Parameter(
parameter_deref['name'], parameter_deref['in'],
schema=schema, required=required,
allow_empty_value=allow_empty_value,
2017-09-21 11:51:37 +00:00
),
)