Mercurial > hg > MakeItSo
view makeitso/main.py @ 3:8931be6d2c8a
stub for getting missing variables
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Thu, 04 Nov 2010 19:12:56 -0700 |
parents | 825922315ce6 |
children |
line wrap: on
line source
#!/usr/bin/env python """ filesystem template interpreter """ import os import subprocess import sys from optparse import OptionParser from tempita import Template def call(command, *args, **kw): code = subprocess.call(command, *args, **kw) if code: if isinstance(command, basestring): cmdstr = command else: cmdstr = ' '.join(command) raise SystemExit("Command `%s` exited with code %d" % (cmdstr, code)) def get_missing(name_error): """ This is a horrible hack because python doesn't do the proper thing via eval and return the name of the variable; instead, it just gives you a message: >>> try: ... eval('2*foo') ... except Exception, e: ... pass """ message = name_error.args[0] varname = message.split("'")[1] return varname def missing_variables(template, variables): """return additional variables needed""" vars = variables.copy() missing = set([]) while True: try: template.substitute(**missing) return missing except NameError, e: missed = get_missing(e) missing.add(missed) vars[missed] = '' def template_variables(template): """return the variables needed for a template""" return missing_variables(template, {}) def main(args=sys.argv[1:]): # create option parser usage = '%prog [options]' parser = OptionParser(usage, description=__doc__) parser.add_option('--variables', dest='variables', action='store_true', help='print the variables in a template') options, args = parser.parse_args(args) if options.variables: variables = template_variables() # TODO: pass template return # template variables variables = {} _vars = [] _args = [] for arg in args: if '=' in arg: key, value = arg.split('=') variables[key] = value else: _args.append(arg) args = _args # get the content 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__': main()