38 lines
1,007 B
Python
38 lines
1,007 B
Python
|
|
from spaceapi_server.template import _env_init
|
|
|
|
|
|
def template_function(fn):
|
|
"""
|
|
Register the decorated function as a callable template function.
|
|
:param fn: The function to register.
|
|
"""
|
|
# Make sure the Jinja2 environment is initialized
|
|
env = _env_init()
|
|
# Add the function to the environment's globals
|
|
env.globals[fn.__name__] = fn
|
|
return fn
|
|
|
|
|
|
def template_filter(fn):
|
|
"""
|
|
Register the decorated function as a template filter.
|
|
:param fn: The function to register.
|
|
"""
|
|
# Make sure the Jinja2 environment is initialized
|
|
env = _env_init()
|
|
# Add the function to the environment's filters
|
|
env.filters[fn.__name__] = fn
|
|
return fn
|
|
|
|
|
|
def template_test(fn):
|
|
"""
|
|
Register the decorated function as a template test.
|
|
:param fn: The function to register.
|
|
"""
|
|
# Make sure the Jinja2 environment is initialized
|
|
env = _env_init()
|
|
# Add the function to the environment's tests
|
|
env.tests[fn.__name__] = fn
|
|
return fn
|