Mercurial > hg > MakeItSo
view makeitso/cli.py @ 81:0d5fac58b1a8
add a template for doctests
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Sat, 08 Jan 2011 21:24:17 -0800 |
parents | 059b02808efa |
children | cb6c54b1adf3 |
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): 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