Mercurial > hg > MakeItSo
view makeitso/cli.py @ 95:e74baa8e6df4
fix CLI interface a bit....write a test for it
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Tue, 11 Jan 2011 09:06:25 -0800 |
parents | cb6c54b1adf3 |
children | 37f92ae8f999 |
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' 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) and description is not None: description += ' [DEFAULT: %s]' % variable.default parser.add_option('--%s' % variable.name, dest=variable.name, default=variable.default, help=description) return parser def parse(self, args=None, parser=None, options=None): # parse the command line if not parser or not options: parser = self.parser() options, args = parser.parse_args(args=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('_')]) # return variables, args[0] def __call__(self, *args): variables, output = self.parse(list(args)) template = self.template_class(variables=variables) template.substitute({}, output=output)