Merge pull request #169 from p1c2u/feature/operations-more-fields

Operations fields

Author: p1c2u

openapi_core/schema/external_docs/__init__.py

@@ -0,0 +1,14 @@
+"""OpenAPI core external docs factories module"""
+from openapi_core.schema.external_docs.models import ExternalDocumentation
+
+
+class ExternalDocumentationFactory(object):
+
+    def __init__(self, dereferencer):
+        self.dereferencer = dereferencer
+
+    def create(self, external_doc_spec):
+        url = external_doc_spec['url']
+        description = external_doc_spec.get('description')
+
+        return ExternalDocumentation(url, description=description)

openapi_core/schema/external_docs/factories.py

@@ -0,0 +1,9 @@
+"""OpenAPI core external docs models module"""
+
+
+class ExternalDocumentation(object):
+    """Represents an OpenAPI External Documentation."""
+
+    def __init__(self, url, description=None):
+        self.url = url
+        self.description = description

openapi_core/schema/external_docs/models.py

@@ -4,10 +4,15 @@
 from openapi_spec_validator.validators import PathItemValidator
 
 from openapi_core.compat import lru_cache
+from openapi_core.schema.external_docs.factories import (
+    ExternalDocumentationFactory,
+)
 from openapi_core.schema.operations.models import Operation
 from openapi_core.schema.parameters.generators import ParametersGenerator
 from openapi_core.schema.request_bodies.factories import RequestBodyFactory
 from openapi_core.schema.responses.generators import ResponsesGenerator
+from openapi_core.schema.security.factories import SecurityRequirementFactory
+from openapi_core.schema.servers.generators import ServersGenerator
 
 
 class OperationsGenerator(object):
@@ -32,6 +37,24 @@ def generate(self, path_name, path):
                 parameters_list)
             operation_id = operation_deref.get('operationId')
             tags_list = operation_deref.get('tags', [])
+            summary = operation_deref.get('summary')
+            description = operation_deref.get('description')
+            security_requirements_list = operation_deref.get('security', [])
+            servers_spec = operation_deref.get('servers', [])
+
+            servers = self.servers_generator.generate(servers_spec)
+
+            security = None
+            if security_requirements_list:
+                security = list(map(
+                    self.security_requirement_factory.create,
+                    security_requirements_list))
+
+            external_docs = None
+            if 'externalDocs' in operation_deref:
+                external_docs_spec = operation_deref.get('externalDocs')
+                external_docs = self.external_docs_factory.create(
+                    external_docs_spec)
 
             request_body = None
             if 'requestBody' in operation_deref:
@@ -43,8 +66,11 @@ def generate(self, path_name, path):
                 http_method,
                 Operation(
                     http_method, path_name, responses, list(parameters),
+                    summary=summary, description=description,
+                    external_docs=external_docs, security=security,
                     request_body=request_body, deprecated=deprecated,
-                    operation_id=operation_id, tags=list(tags_list)
+                    operation_id=operation_id, tags=list(tags_list),
+                    servers=servers,
                 ),
             )
 
@@ -58,7 +84,22 @@ def responses_generator(self):
     def parameters_generator(self):
         return ParametersGenerator(self.dereferencer, self.schemas_registry)
 
+    @property
+    @lru_cache()
+    def external_docs_factory(self):
+        return ExternalDocumentationFactory(self.dereferencer)
+
     @property
     @lru_cache()
     def request_body_factory(self):
         return RequestBodyFactory(self.dereferencer, self.schemas_registry)
+
+    @property
+    @lru_cache()
+    def security_requirement_factory(self):
+        return SecurityRequirementFactory(self.dereferencer)
+
+    @property
+    @lru_cache()
+    def servers_generator(self):
+        return ServersGenerator(self.dereferencer)

openapi_core/schema/operations/generators.py

@@ -8,15 +8,22 @@ class Operation(object):
 
     def __init__(
             self, http_method, path_name, responses, parameters,
-            request_body=None, deprecated=False, operation_id=None, tags=None):
+            summary=None, description=None, external_docs=None, security=None,
+            request_body=None, deprecated=False, operation_id=None, tags=None,
+            servers=None):
         self.http_method = http_method
         self.path_name = path_name
         self.responses = dict(responses)
         self.parameters = dict(parameters)
+        self.summary = summary
+        self.description = description
+        self.external_docs = external_docs
+        self.security = security
         self.request_body = request_body
         self.deprecated = deprecated
         self.operation_id = operation_id
         self.tags = tags
+        self.servers = servers
 
     def __getitem__(self, name):
         return self.parameters[name]

openapi_core/schema/operations/models.py

@@ -0,0 +1,14 @@
+"""OpenAPI core security factories module"""
+from openapi_core.schema.security.models import SecurityRequirement
+
+
+class SecurityRequirementFactory(object):
+
+    def __init__(self, dereferencer):
+        self.dereferencer = dereferencer
+
+    def create(self, security_requirement_spec):
+        name = next(iter(security_requirement_spec))
+        scope_names = security_requirement_spec[name]
+
+        return SecurityRequirement(name, scope_names=scope_names)

openapi_core/schema/security/__init__.py

@@ -0,0 +1,9 @@
+"""OpenAPI core security models module"""
+
+
+class SecurityRequirement(object):
+    """Represents an OpenAPI Security Requirement."""
+
+    def __init__(self, name, scope_names=None):
+        self.name = name
+        self.scope_names = scope_names or []

openapi_core/schema/security/factories.py

@@ -12,18 +12,16 @@ def __init__(self, dereferencer):
 
     def generate(self, servers_spec):
         servers_deref = self.dereferencer.dereference(servers_spec)
-        if not servers_deref:
-            yield Server('/')
-            return
         for server_spec in servers_deref:
             url = server_spec['url']
             variables_spec = server_spec.get('variables', {})
+            description = server_spec.get('description')
 
             variables = None
             if variables_spec:
                 variables = self.variables_generator.generate(variables_spec)
 
-            yield Server(url, variables=variables)
+            yield Server(url, variables=variables, description=description)
 
     @property
     @lru_cache()

openapi_core/schema/security/models.py

@@ -4,9 +4,10 @@
 
 class Server(object):
 
-    def __init__(self, url, variables=None):
+    def __init__(self, url, variables=None, description=None):
         self.url = url
         self.variables = variables and dict(variables) or {}
+        self.description = description
 
     @property
     def default_url(self):