Mercurial > hg > MakeItSo
diff makeitso/main.py @ 2:825922315ce6
make shebang work
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Thu, 04 Nov 2010 18:58:11 -0700 |
parents | c2f8464e0395 |
children | 8931be6d2c8a |
line wrap: on
line diff
--- a/makeitso/main.py Wed Nov 03 08:35:16 2010 -0700 +++ b/makeitso/main.py Thu Nov 04 18:58:11 2010 -0700 @@ -45,9 +45,15 @@ args = _args # get the content - content = sys.stdin.read() - template = Template(content) - print template.substitute(**variables) + if args: + for arg in args: + content = file(arg).read() + template = Template(content) + print template.substitute(**variables) + else: + content = sys.stdin.read() + template = Template(content) + print template.substitute(**variables) if __name__ == '__main__':