new project

fastapi_admin/template.py

@@ -0,0 +1,75 @@
+import os
+import typing
+from datetime import date
+from typing import Any, List, Tuple
+from urllib.parse import urlencode
+
+from babel.support import Translations
+from jinja2 import contextfilter
+from starlette.requests import Request
+from starlette.templating import Jinja2Templates
+
+from fastapi_admin import VERSION
+from fastapi_admin.constants import BASE_DIR
+
+if typing.TYPE_CHECKING:
+    from fastapi_admin.resources import Field
+
+templates = Jinja2Templates(directory=os.path.join(BASE_DIR, "templates"))
+templates.env.globals["VERSION"] = VERSION
+templates.env.globals["NOW_YEAR"] = date.today().year
+templates.env.add_extension("jinja2.ext.i18n")
+templates.env.add_extension("jinja2.ext.autoescape")
+templates.env.add_extension("jinja2.ext.with_")
+templates.env.add_extension("jinja2.ext.do")
+
+TRANSLATIONS = {
+    "zh_CN": Translations.load(os.path.join(BASE_DIR, "locales"), locales=["zh_CN"]),
+    "en_US": Translations.load(os.path.join(BASE_DIR, "locales"), locales=["en_US"]),
+}
+
+
+@contextfilter
+def current_page_with_params(context: dict, params: dict):
+    request = context.get("request")  # type:Request
+    full_path = request.scope["raw_path"].decode()
+    query_params = dict(request.query_params)
+    for k, v in params.items():
+        query_params[k] = v
+    return full_path + "?" + urlencode(query_params)
+
+
+templates.env.filters["current_page_with_params"] = current_page_with_params
+
+
+def set_locale(locale: str):
+    translations = TRANSLATIONS.get(locale) or TRANSLATIONS.get("en_US")
+    templates.env.install_gettext_translations(translations)
+    translations.install(locale)
+
+
+def add_template_folder(*folders: str):
+    for folder in folders:
+        templates.env.loader.searchpath.append(folder)
+
+
+async def render_values(
+    fields: List["Field"], values: List[Tuple[Any]], display: bool = True
+) -> List[List[Any]]:
+    """
+    render values with template render
+    :param fields:
+    :param values:
+    :param display:
+    :return:
+    """
+    ret = []
+    for value in values:
+        item = []
+        for i, v in enumerate(value):
+            if display:
+                item.append(await fields[i].display.render(v))
+            else:
+                item.append(await fields[i].input.render(v))
+        ret.append(item)
+    return ret