Source code for molten.openapi.handlers

# This file is a part of molten.
# Copyright (C) 2018 CLEARTYPE SRL <[email protected]>
# molten is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# molten is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# License for more details.
# You should have received a copy of the GNU Lesser General Public License
# along with this program.  If not, see <>.

from typing import Any, Dict, List, Optional

import pkg_resources

from import BaseApp
from ..http import HTTP_200, Request, Response
from .documents import Metadata, SecurityScheme, generate_openapi_document

[docs]class OpenAPIHandler: """Dynamically generates and serves OpenAPI v3 documents based on the current application object. Once generated, the document is subsequently served from cache. Examples: >>> get_schema = OpenAPIHandler( ... metadata=Metadata(title="Pet Store", version="0.1.0"), ... security_schemes=[HTTPSecurityScheme("Bearer", "bearer")], ... default_security_scheme="Bearer", ... ) Parameters: metadata: Various meta-information about the current API. security_schemes: A list of security schemes used throughout the API. """ def __init__( self, metadata: Metadata, security_schemes: Optional[List[SecurityScheme]] = None, default_security_scheme: Optional[str] = None, ) -> None: self.metadata = metadata self.security_schemes = security_schemes or [] self.default_security_scheme = default_security_scheme self.document: Optional[Dict[str, Any]] = None @property def __name__(self) -> str: return type(self).__name__ # type: ignore def __call__(self, app: BaseApp) -> Optional[Dict[str, Any]]: """Generates an OpenAPI v3 document. """ if not self.document: self.document = generate_openapi_document( app, self.metadata, self.security_schemes, self.default_security_scheme, ) return self.document
[docs]class OpenAPIUIHandler: """Renders the `Swagger UI`_. Parameters: schema_route_name: The name of the route that exposes the schema. The actual path to the schema is looked up whenever the handler is called. .. _Swagger UI: """ def __init__(self, schema_route_name: str = "OpenAPIHandler") -> None: self.schema_route_name = schema_route_name self.template = pkg_resources.resource_string(f"molten.openapi.templates", "index.html").decode("utf-8") @property def __name__(self) -> str: return type(self).__name__ # type: ignore def __call__(self, app: BaseApp, request: Request) -> Response: """Renders the Swagger UI. """ schema_uri = app.reverse_uri(self.schema_route_name) rendered_template = self.template % {"schema_uri": schema_uri} return Response(HTTP_200, content=rendered_template, headers={ "content-type": "text/html", })