ac107458/python-capport
2
0
Fork 0
Code

fix DispatchingJinjaLoader to not use internal methods of parent class

Browse Source
This commit is contained in:
Stefan Bühler 2023-11-15 08:57:59 +01:00
parent 9e31c8c673
commit 29d1a3226a
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/capport/api/app_cls.py
View File

@ -21,11 +21,24 @@ class DispatchingJinjaLoader(quart.templating.DispatchingJinjaLoader):
def __init__(self, app: MyQuartApp) -> None: def __init__(self, app: MyQuartApp) -> None:
super().__init__(app) super().__init__(app)
def _loaders(self) -> typing.Generator[jinja2.BaseLoader, None, None]: def get_source(
self,
environment: typing.Any,
template: str,
) -> tuple[str, str | None, typing.Callable | None]:
if self.app.custom_loader: if self.app.custom_loader:
yield self.app.custom_loader try:
for loader in super()._loaders(): return self.app.custom_loader.get_source(environment, template)
yield loader except jinja2.TemplateNotFound:
pass
return super().get_source(environment, template)
def list_templates(self) -> list[str]:
result = set()
if self.app.custom_loader:
result.update(self.app.custom_loader.list_templates())
result.update(super().list_templates())
return list(result)
class MyQuartApp(quart_trio.QuartTrio): class MyQuartApp(quart_trio.QuartTrio):