Warning

This document is for an old release 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()