wehub-resource-sync
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import pathlib
|
|
from typing import List, Union
|
|
|
|
from ray.util.annotations import DeveloperAPI
|
|
|
|
|
|
@DeveloperAPI
|
|
class Template:
|
|
"""Class which provides basic HTML templating."""
|
|
|
|
def __init__(self, file: str):
|
|
with open(pathlib.Path(__file__).parent / "templates" / file, "r") as f:
|
|
self.template = f.read()
|
|
|
|
def render(self, **kwargs: Union[str, List]) -> str:
|
|
"""Render an HTML template with the given data.
|
|
|
|
This is done by replacing instances of `{{ key }}` with `value`
|
|
from the keyword arguments.
|
|
|
|
Args:
|
|
**kwargs: Mapping of template variable names to their substitution
|
|
values. Each value is either a string or a list of strings;
|
|
lists are joined together before substitution.
|
|
|
|
Returns:
|
|
HTML template with the keys of the kwargs replaced with corresponding
|
|
values.
|
|
"""
|
|
rendered = self.template
|
|
for key, value in kwargs.items():
|
|
if isinstance(value, List):
|
|
value = "".join(value)
|
|
rendered = rendered.replace("{{ " + key + " }}", value if value else "")
|
|
return rendered
|
|
|
|
@staticmethod
|
|
def list_templates() -> List[pathlib.Path]:
|
|
"""List the available HTML templates.
|
|
|
|
Returns:
|
|
A list of files with .html.j2 extensions inside ../templates/
|
|
"""
|
|
return (pathlib.Path(__file__).parent / "templates").glob("*.html.j2")
|