# HG changeset patch # User Jeff Hammel # Date 1344896093 25200 # Node ID fcd98303a90d9bc6a59c5c5e847d80bd6c56f639 # Parent c1cbb79a6aaf7f2bc714f25e620f7db1f7bab282 provide CLI handler for web diff -r c1cbb79a6aaf -r fcd98303a90d setup.py --- a/setup.py Mon Aug 13 00:01:02 2012 -0700 +++ b/setup.py Mon Aug 13 15:14:53 2012 -0700 @@ -14,6 +14,7 @@ kw['entry_points'] = """ [console_scripts] talos-names = talosnames.main:main + talos-names-web = talosnames.web:main """ kw['install_requires'] = dependencies kw['dependency_links'] = ['http://hg.mozilla.org/build/talos/archive/tip.tar.gz#egg=talos'] diff -r c1cbb79a6aaf -r fcd98303a90d talosnames/web.py --- 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() + + +