Rework executable script so that it uses a main

This commit is contained in:
Eldon 2014-01-22 02:30:12 +00:00
parent 7b219ab011
commit 428d6cb7da
2 changed files with 9 additions and 3 deletions

View File

@ -1,3 +1,6 @@
#!/usr/bin/env sh
python -m umbra.umbra $@
#!/usr/bin/env python
from umbra import umbra
if __name__=="__main__":
umbra.main()

View File

@ -76,7 +76,7 @@ class Chrome():
def __exit__(self, *args):
self.chrome_process.kill()
if __name__ == "__main__":
def main():
arg_parser = argparse.ArgumentParser(prog=os.path.basename(sys.argv[0]),
description='umbra - Browser automation tool',
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
@ -89,3 +89,6 @@ if __name__ == "__main__":
args = arg_parser.parse_args(args=sys.argv[1:])
with Chrome(args.port, args.executable, args.browser_wait):
Umbra(args.port)
if __name__ == "__main__":
main()