Mercurial > mozilla > hg > talosnames
diff talosnames/web.py @ 46:fcd98303a90d
provide CLI handler for web
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Mon, 13 Aug 2012 15:14:53 -0700 |
parents | 10e59c3ae847 |
children | 11249e75ced6 |
line wrap: on
line diff
--- a/talosnames/web.py Mon Aug 13 00:01:02 2012 -0700 +++ b/talosnames/web.py Mon Aug 13 15:14:53 2012 -0700 @@ -7,6 +7,7 @@ import optparse import os import pprint +import sys import talos.test import tempita from api import TalosNames @@ -38,7 +39,6 @@ 'tbpl': dict([(suite, self.api.tbpl_name(suite)) for suite in suites]), 'tests': tests, - 'pprint': pprint.pformat, 'list2cmdline': list2cmdline } @@ -125,8 +125,7 @@ _template = tempita.HTMLTemplate(contents) return _template.substitute(data) -if __name__ == '__main__': - +def main(args=sys.argv[1:]): parser = optparse.OptionParser() parser.add_option('-o', '--output', dest='output', help="file to output to") @@ -136,6 +135,7 @@ options, args = parser.parse_args() app = Handler() + print "Done creating handler." if options.output: f = file(options.output, 'w') @@ -147,3 +147,8 @@ server.serve_forever() +if __name__ == '__main__': + main() + + +