feat: allow to specify custom jinja template

This commit is contained in:
Georg K
2022-07-28 03:25:37 +03:00
parent 7618066b7f
commit 207204fe53
2 changed files with 4 additions and 3 deletions

View File

@@ -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