apidocs html render

Vladimir Volkov · Vladimir Volkov · commit adba9a2ef02e · 2021-04-15T17:55:52.000+03:00
diff --git a/README.md b/README.md
@@ -121,4 +121,4 @@ if __name__ == '__main__':
     uvicorn.run(app, port=8080)
 ```
 
-Open [http://localhost:8080/apidocs.json](http://localhost:8080/apidocs.json) to view generated openapi schema.
+Open [http://localhost:8080/apidocs](http://localhost:8080/apidocs) to view generated openapi schema.
diff --git a/setup.py b/setup.py
@@ -30,7 +30,7 @@ def get_packages(package):
         'apispec<4',
         'python-multipart'
     ],
-    version='0.0.16',
+    version='0.0.17',
     url='https://github.com/slv0/start_resty',
     license='BSD',
     description='The web framework',
diff --git a/star_resty/apidocs/setup.py b/star_resty/apidocs/setup.py
@@ -5,10 +5,10 @@
 from apispec.ext.marshmallow import MarshmallowPlugin
 from starlette.applications import Starlette
 from starlette.requests import Request
-from starlette.responses import JSONResponse
+from starlette.responses import JSONResponse, HTMLResponse
 
 from .route import setup_routes
-from .utils import resolve_schema_name
+from .utils import resolve_schema_name, apispec_json_to_html
 
 logger = logging.getLogger(__name__)
 
@@ -20,8 +20,9 @@ def setup_spec(app: Starlette, title: str,
                openapi_version='2.0',
                schemes=None,
                base_path='/',
-               route: str = '/apidocs.json',
+               route: str = '/apidocs',
                add_head_methods: bool = False,
+               render_to_html: bool = True,
                options: Optional[Mapping] = None,
                **kwargs):
     if options is None:
@@ -46,8 +47,10 @@ def generate_api_docs(_: Request):
             setup_routes(app.routes, spec, version=get_open_api_version(openapi_version)
                          , add_head_methods=add_head_methods)
             api_spec = spec.to_dict()
-
-        return JSONResponse(api_spec)
+        if render_to_html:
+            return HTMLResponse(apispec_json_to_html(api_spec))
+        else:
+            return JSONResponse(api_spec)
 
 
 def get_open_api_version(version: str) -> int:
diff --git a/star_resty/apidocs/template.html b/star_resty/apidocs/template.html
@@ -0,0 +1,65 @@
+<!--<script src="https://gist.github.com/oseiskar/dbd51a3727fc96dcf5ed189fca491fb3.js"></script>-->
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>Swagger UI</title>
+  <link href="https://fonts.googleapis.com/css?family=Open+Sans:400,700|Source+Code+Pro:300,600|Titillium+Web:400,600,700" rel="stylesheet">
+  <link rel="stylesheet" type="text/css" href="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui.css" >
+  <style>
+    html
+    {
+      box-sizing: border-box;
+      overflow: -moz-scrollbars-vertical;
+      overflow-y: scroll;
+    }
+    *,
+    *:before,
+    *:after
+    {
+      box-sizing: inherit;
+    }
+
+    body {
+      margin:0;
+      background: #fafafa;
+    }
+  </style>
+</head>
+<body>
+
+<div id="swagger-ui"></div>
+
+<script src="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui-bundle.js"> </script>
+<script src="https://cdnjs.cloudflare.com/ajax/libs/swagger-ui/3.24.2/swagger-ui-standalone-preset.js"> </script>
+<script>
+window.onload = function() {
+
+  var spec = %s;
+
+  // Build a system
+  const ui = SwaggerUIBundle({
+    spec: spec,
+    dom_id: '#swagger-ui',
+    deepLinking: true,
+    presets: [
+      SwaggerUIBundle.presets.apis,
+      SwaggerUIStandalonePreset
+    ],
+    plugins: [
+      SwaggerUIBundle.plugins.DownloadUrl
+    ],
+    layout: "StandaloneLayout"
+  })
+
+  window.ui = ui
+  var elements = document.getElementsByClassName("download-url-wrapper");
+  while(elements.length > 0){
+      elements[0].parentNode.removeChild(elements[0]);
+  }
+
+}
+</script>
+</body>
+
+</html>
diff --git a/star_resty/apidocs/utils.py b/star_resty/apidocs/utils.py
@@ -1,8 +1,10 @@
 import inspect
+import os
 import re
+import json
 from typing import Any
 
-__all__ = ('resolve_schema_name', 'convert_path')
+__all__ = ('resolve_schema_name', 'convert_path', 'apispec_json_to_html')
 
 
 def resolve_schema_name(schema: Any) -> str:
@@ -17,3 +19,9 @@ def resolve_schema_name(schema: Any) -> str:
 
 def convert_path(path: str) -> str:
     return re.sub(r'{([^:]+).*}', r'{\1}', path)
+
+
+def apispec_json_to_html(apispec_json: dict) -> str:
+    template_path = os.path.join(os.path.dirname(__file__), 'template.html')
+    template = open(template_path, 'r').read()
+    return template % json.dumps(apispec_json)
