Option to set a custom path to swagger template.

prettier than owerwriting
aiohttp_swagger.helpers.builders.SWAGGER_TEMPLATE

Author: Pavol Vargovcik

aiohttp_swagger/__init__.py

@@ -46,7 +46,8 @@ def setup_swagger(app: web.Application,
                   contact: str = "",
                   swagger_home_decor: FunctionType = None,
                   swagger_def_decor: FunctionType = None,
-                  swagger_info: dict = None):
+                  swagger_info: dict = None,
+                  swagger_template_path: str = None):
     _swagger_url = ("/{}".format(swagger_url)
                     if not swagger_url.startswith("/")
                     else swagger_url)
@@ -60,7 +61,8 @@ def setup_swagger(app: web.Application,
         else:
             swagger_info = generate_doc_from_each_end_point(
                 app, api_base_url=api_base_url, description=description,
-                api_version=api_version, title=title, contact=contact
+                api_version=api_version, title=title, contact=contact,
+                template_path=swagger_template_path
             )
     else:
         swagger_info = json.dumps(swagger_info)

aiohttp_swagger/helpers/builders.py

@@ -66,7 +66,8 @@ def generate_doc_from_each_end_point(
         description: str = "Swagger API definition",
         api_version: str = "1.0.0",
         title: str = "Swagger API",
-        contact: str = ""):
+        contact: str = "",
+        template_path: str = None):
     # Clean description
     _start_desc = 0
     for i, word in enumerate(description):
@@ -76,7 +77,10 @@ def generate_doc_from_each_end_point(
     cleaned_description = "    ".join(description[_start_desc:].splitlines())
 
     # Load base Swagger template
-    with open(join(SWAGGER_TEMPLATE, "swagger.yaml"), "r") as f:
+    if template_path is None:
+        template_path = join(SWAGGER_TEMPLATE, "swagger.yaml")
+
+    with open(template_path, "r") as f:
         swagger_base = (
             Template(f.read()).render(
                 description=cleaned_description,