Skip to content

Commit

Permalink
Expand jinja environment (#26)
Browse files Browse the repository at this point in the history
  • Loading branch information
DavidHuber-NOAA authored May 3, 2024
1 parent 2c48c56 commit 71f6b10
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions src/wxflow/jinja.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,11 @@ def get_set_env(self, loader: jinja2.BaseLoader, filters: Dict[str, callable] =
to_timedelta: convert a string to a timedelta object
add_to_datetime: add time to a datetime, return new datetime object
The Expression Statement extension "jinja2.ext.do", which enables
{% do ... %} statements. These are useful for appending to lists.
e.g. {{ bar.append(foo) }} would print "None" to the parsed jinja
template, but {% do bar.append(foo) %} would not.
Parameters
----------
loader: jinja2.BaseLoader
Expand All @@ -124,6 +129,9 @@ def get_set_env(self, loader: jinja2.BaseLoader, filters: Dict[str, callable] =
"""

env = jinja2.Environment(loader=loader, undefined=self.undefined)

env.add_extension("jinja2.ext.do")

env.filters["strftime"] = lambda dt, fmt: strftime(dt, fmt)
env.filters["to_isotime"] = lambda dt: to_isotime(dt) if not isinstance(dt, SilentUndefined) else dt
env.filters["to_fv3time"] = lambda dt: to_fv3time(dt) if not isinstance(dt, SilentUndefined) else dt
Expand All @@ -133,8 +141,10 @@ def get_set_env(self, loader: jinja2.BaseLoader, filters: Dict[str, callable] =
env.filters["to_f90bool"] = lambda bool: ".true." if bool else ".false."
env.filters['getenv'] = lambda name, default='UNDEFINED': os.environ.get(name, default)
env.filters["relpath"] = lambda pathname, start=os.curdir: os.path.relpath(pathname, start)
env.filters["add_to_datetime"] = (lambda dt, delta: add_to_datetime(dt, delta)
if not (isinstance(dt, SilentUndefined) or isinstance(delta, SilentUndefined)) else dt)
env.filters["add_to_datetime"] = (
lambda dt, delta: add_to_datetime(dt, delta)
if not (isinstance(dt, SilentUndefined) or isinstance(delta, SilentUndefined))
else dt if isinstance(dt, SilentUndefined) else delta)
env.filters["to_timedelta"] = lambda delta_str: to_timedelta(delta_str) if not isinstance(delta_str, SilentUndefined) else delta_str

# Add any additional filters
Expand Down

0 comments on commit 71f6b10

Please sign in to comment.