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:
[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(0.5) self.driver_wrapper.driver.save_screenshot("%s/%i.png" % (self.output, step_index)) time.sleep(0.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()