Convert mako-render into module mako.cmd and add suitable entrypoint to setup

1 files changed, 66 insertions, 0 deletions

diff --git a/mako/cmd.py b/mako/cmd.py
new file mode 100755
index 0000000..15c2737
--- /dev/null
+++ b/mako/cmd.py
@@ -0,0 +1,66 @@
+# mako/cmd.py
+# Copyright (C) 2006-2014 the Mako authors and contributors <see AUTHORS file>
+#
+# This module is part of Mako and is released under
+# the MIT License: http://www.opensource.org/licenses/mit-license.php
+
+def render(data, kw, lookup_dirs):
+    from mako.template import Template
+    from mako.lookup import TemplateLookup
+
+    lookup = TemplateLookup(lookup_dirs)
+    return Template(data, lookup=lookup).render(**kw)
+
+def varsplit(var):
+    if "=" not in var:
+        return (var, "")
+    return var.split("=", 1)
+
+def cmdline(argv=None):
+    import pdb; pdb.set_trace()
+
+    from os.path import isfile, dirname
+    from sys import stdin
+
+    if argv is None:
+        import sys
+        argv = sys.argv
+
+    from optparse import OptionParser
+
+    parser = OptionParser("usage: %prog [FILENAME]")
+    parser.add_option("--var", default=[], action="append",
+                  help="variable (can be used multiple times, use name=value)")
+    parser.add_option("--template-dir", default=[], action="append",
+                  help="Directory to use for template lookup (multiple "
+                       "directories may be provided). If not given then if the "
+                       "template is read from stdin, the value defaults to be "
+                       "the current directory, otherwise it defaults to be the "
+                       "parent directory of the file provided.")
+
+    opts, args = parser.parse_args(argv[1:])
+    if len(args) not in (0, 1):
+        parser.error("wrong number of arguments")  # Will exit
+
+    if (len(args) == 0) or (args[0] == "-"):
+        fo = stdin
+        lookup_dirs = opts.template_dir or ["."]
+    else:
+        filename = args[0]
+        if not isfile(filename):
+            raise SystemExit("error: can't find %s" % filename)
+        fo = open(filename)
+        lookup_dirs = opts.template_dir or [dirname(filename)]
+
+    kw = dict([varsplit(var) for var in opts.var])
+    data = fo.read()
+
+    try:
+        print(render(data, kw, lookup_dirs=lookup_dirs))
+    except:
+        from mako import exceptions
+        print(exceptions.text_error_template().render(), file=sys.stderr)
+        sys.exit(1)
+
+if __name__ == "__main__":
+    cmdline()