mirror of
https://github.com/internetarchive/brozzler.git
synced 2025-04-19 23:35:54 -04:00
Rework executable script so that it uses a main
This commit is contained in:
parent
7b219ab011
commit
428d6cb7da
@ -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()
|
||||
|
@ -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()
|
||||
|
Loading…
x
Reference in New Issue
Block a user