From 207204fe533b1a528726e91574cfbd3b9df36a67 Mon Sep 17 00:00:00 2001 From: Georg K Date: Thu, 28 Jul 2022 03:25:37 +0300 Subject: [PATCH] feat: allow to specify custom jinja template --- aiohttp_pydantic/oas/__init__.py | 5 +++-- aiohttp_pydantic/oas/index.j2 | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/aiohttp_pydantic/oas/__init__.py b/aiohttp_pydantic/oas/__init__.py index d269f94..fbc1f90 100644 --- a/aiohttp_pydantic/oas/__init__.py +++ b/aiohttp_pydantic/oas/__init__.py @@ -14,12 +14,13 @@ def setup( url_prefix: str = "/oas", enable: bool = True, version_spec: Optional[str] = None, - title_spec: Optional[str] = None + title_spec: Optional[str] = None, + custom_template: Optional[jinja2.Template] = None ): if enable: oas_app = web.Application() oas_app["apps to expose"] = tuple(apps_to_expose) or (app,) - oas_app["index template"] = jinja2.Template( + oas_app["index template"] = custom_template or jinja2.Template( resources.read_text("aiohttp_pydantic.oas", "index.j2") ) oas_app["version_spec"] = version_spec diff --git a/aiohttp_pydantic/oas/index.j2 b/aiohttp_pydantic/oas/index.j2 index fe18e0c..c1b0745 100644 --- a/aiohttp_pydantic/oas/index.j2 +++ b/aiohttp_pydantic/oas/index.j2 @@ -5,7 +5,7 @@ {{ title | default('Swagger UI') }} - +