Add the ability to import filters

Fixes mattrobenolt#85

Author: mattrobenolt

jinja2cli/cli.py

@@ -242,7 +242,7 @@ def _load_json5():
 }
 
 
-def render(template_path, data, extensions, strict=False):
+def render(template_path, data, extensions, filters, strict=False):
     from jinja2 import (
         __version__ as jinja_version,
         Environment,
@@ -273,6 +273,13 @@ def render(template_path, data, extensions, strict=False):
     env.globals["environ"] = lambda key: force_text(os.environ.get(key))
     env.globals["get_context"] = lambda: data
 
+    if filters:
+        from jinja2.utils import import_string
+
+        for filter in set(filters):
+            filter = import_string(filter)
+            env.filters[filter.__name__] = filter
+
     return env.get_template(os.path.basename(template_path)).render(data)
 
 
@@ -362,7 +369,7 @@ def cli(opts, args):
 
         out = codecs.getwriter("utf8")(out)
 
-    out.write(render(template_path, data, extensions, opts.strict))
+    out.write(render(template_path, data, extensions, opts.filters, opts.strict))
     out.flush()
     return 0
 
@@ -430,6 +437,14 @@ def main():
         action="append",
         default=["do", "loopcontrols"],
     )
+    parser.add_option(
+        "-f",
+        "--filter",
+        help="extra jinja2 filters to load",
+        dest="filters",
+        action="append",
+        default=[],
+    )
     parser.add_option(
         "-D",
         help="Define template variable in the form of key=value",

tests/test_jinja2cli.py

@@ -10,7 +10,7 @@ def test_relative_path():
     path = "./files/template.j2"
 
     title = b"\xc3\xb8".decode("utf8")
-    output = cli.render(path, {"title": title}, [])
+    output = cli.render(path, {"title": title}, [], [])
     assert output == title
     assert type(output) == cli.text_type
 
@@ -20,6 +20,6 @@ def test_absolute_path():
     path = os.path.join(absolute_base_path, "files", "template.j2")
 
     title = b"\xc3\xb8".decode("utf8")
-    output = cli.render(path, {"title": title}, [])
+    output = cli.render(path, {"title": title}, [], [])
     assert output == title
     assert type(output) == cli.text_type