Warning
This document is for an in-development version of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.
Source code for galaxy.selenium.scripts.dump_tour
#!/usr/bin/env python
import argparse
import datetime
import os
import sys
import time
from galaxy.selenium import cli
DESCRIPTION = "Walk a Galaxy tour and dump screenshots."
[docs]def main(argv=None):
if argv is None:
argv = sys.argv[1:]
args = _arg_parser().parse_args(argv)
driver_wrapper = cli.DriverWrapper(args)
output = args.output
assert output is not None
output = output.replace("TIMESTAMP", datetime.datetime.now().strftime("%Y%m%d%H%M%s"))
if os.path.exists(output):
raise Exception("Output directory %s exists, skipping")
else:
os.makedirs(output)
callback = DumpTourCallback(driver_wrapper, output)
driver_wrapper.run_tour(args.tour, tour_callback=callback)
[docs]class DumpTourCallback(object):
[docs] def __init__(self, driver_wrapper, output):
self.driver_wrapper = driver_wrapper
self.output = output
[docs] def handle_step(self, step, step_index):
time.sleep(.5)
self.driver_wrapper.driver.save_screenshot("%s/%i.png" % (self.output, step_index))
time.sleep(.5)
def _arg_parser():
parser = argparse.ArgumentParser(description=DESCRIPTION)
parser.add_argument('tour', metavar='TOUR', help='tour to walk')
parser.add_argument('-o', '--output', default="tour_dump_TIMESTAMP", help='directory to dump tour to')
parser = cli.add_selenium_arguments(parser)
return parser
if __name__ == "__main__":
main()