mirror of
https://github.com/correl/openapi-core.git
synced 2024-11-25 11:09:53 +00:00
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""OpenAPI core specs module"""
|
|
import logging
|
|
from functools import partialmethod
|
|
|
|
from openapi_spec_validator import openapi_v3_spec_validator
|
|
|
|
from openapi_core.paths import PathsGenerator
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Spec(object):
|
|
"""Represents an OpenAPI Specification for a service."""
|
|
|
|
def __init__(self, servers=None, paths=None):
|
|
self.servers = servers or []
|
|
self.paths = paths and dict(paths) or {}
|
|
|
|
def __getitem__(self, path_name):
|
|
return self.paths[path_name]
|
|
|
|
def get_server_url(self, index=0):
|
|
return self.servers[index]['url']
|
|
|
|
def get_operation(self, path_pattern, http_method):
|
|
return self.paths[path_pattern].operations[http_method]
|
|
|
|
# operations shortcuts
|
|
|
|
get = partialmethod(get_operation, http_method='get')
|
|
put = partialmethod(get_operation, http_method='put')
|
|
post = partialmethod(get_operation, http_method='post')
|
|
delete = partialmethod(get_operation, http_method='delete')
|
|
options = partialmethod(get_operation, http_method='options')
|
|
head = partialmethod(get_operation, http_method='head')
|
|
patch = partialmethod(get_operation, http_method='patch')
|
|
|
|
|
|
class SpecFactory(object):
|
|
|
|
def __init__(self, dereferencer, config=None):
|
|
self.dereferencer = dereferencer
|
|
self.config = config or {}
|
|
|
|
def create(self, spec_dict, spec_url=''):
|
|
if self.config.get('validate_spec', True):
|
|
openapi_v3_spec_validator.validate(spec_dict, spec_url=spec_url)
|
|
|
|
spec_dict_deref = self.dereferencer.dereference(spec_dict)
|
|
|
|
servers = spec_dict_deref.get('servers', [])
|
|
|
|
paths = spec_dict_deref.get('paths', [])
|
|
paths = self._generate_paths(paths)
|
|
return Spec(servers=servers, paths=list(paths))
|
|
|
|
def _generate_paths(self, paths):
|
|
return PathsGenerator(self.dereferencer).generate(paths)
|