mirror of
https://github.com/correl/openapi-core.git
synced 2024-11-29 03:00:12 +00:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""OpenAPI core operations module"""
|
|
import logging
|
|
|
|
from six import iteritems
|
|
|
|
from openapi_core.parameters import ParametersGenerator
|
|
from openapi_core.request_bodies import RequestBodyFactory
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Operation(object):
|
|
"""Represents an OpenAPI Operation."""
|
|
|
|
def __init__(
|
|
self, http_method, path_name, parameters, request_body=None,
|
|
deprecated=False, operation_id=None):
|
|
self.http_method = http_method
|
|
self.path_name = path_name
|
|
self.parameters = dict(parameters)
|
|
self.request_body = request_body
|
|
self.deprecated = deprecated
|
|
self.operation_id = operation_id
|
|
|
|
def __getitem__(self, name):
|
|
return self.parameters[name]
|
|
|
|
|
|
class OperationsGenerator(object):
|
|
"""Represents an OpenAPI Operation in a service."""
|
|
|
|
def __init__(self, dereferencer):
|
|
self.dereferencer = dereferencer
|
|
|
|
def generate(self, path_name, path):
|
|
path_deref = self.dereferencer.dereference(path)
|
|
for http_method, operation in iteritems(path_deref):
|
|
if http_method.startswith('x-') or http_method == 'parameters':
|
|
continue
|
|
|
|
operation_deref = self.dereferencer.dereference(operation)
|
|
deprecated = operation_deref.get('deprecated', False)
|
|
parameters_list = operation_deref.get('parameters', [])
|
|
parameters = self._generate_parameters(parameters_list)
|
|
|
|
request_body = None
|
|
if 'requestBody' in operation_deref:
|
|
request_body_spec = operation_deref.get('requestBody')
|
|
request_body = self._create_request_body(request_body_spec)
|
|
|
|
yield (
|
|
http_method,
|
|
Operation(
|
|
http_method, path_name, list(parameters),
|
|
request_body=request_body, deprecated=deprecated,
|
|
),
|
|
)
|
|
|
|
def _generate_parameters(self, parameters):
|
|
return ParametersGenerator(self.dereferencer).generate(parameters)
|
|
|
|
def _create_request_body(self, request_body_spec):
|
|
return RequestBodyFactory(self.dereferencer).create(request_body_spec)
|