Mercurial > hg > MakeItSo
view makeitso/cli.py @ 66:7821c82772f5
determine the location in a better way
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Fri, 07 Jan 2011 10:53:12 -0800 |
parents | 30100690ad3f |
children | cce0da329f59 |
line wrap: on
line source
""" command line parser for MakeItSo """ from template import Undefined from optparse import OptionParser class MakeItSoCLI(object): """command line interface to a makeitso template""" def __init__(self, template_class): self.template_class = template_class def parser(self): """ return a command line parser for the template """ usage = '%prog [options] output [var1=value1] [var2=value2] [...]' description = getattr(self.template_class, 'description', None) parser = OptionParser(usage=usage, description=description) # add the variables as options for variable in self.template_class.vars: description = variable.description if (variable.default is not None) and (variable.default is not Undefined): description += ' [DEFAULT: %s]' % variable.default parser.add_option('--%s' % variable.name, dest=variable.name, default=variable.default, help=description) return parser def parse(self): parser = self.parser() options, args = parser.parse_args() # ensure output is given if len(args) != 1: parser.error("Please specify a single output destination") # template variables variables = dict([(key, value) for key, value in options.__dict__.items() if not key.startswith('_')]) # instantiate the template template = self.template_class(output=args[0], variables=variables) return template