diff --git a/segment_coverage/segment_coverage/main.py b/segment_coverage/segment_coverage/main.py index 3578ebf..b99589c 100644 --- a/segment_coverage/segment_coverage/main.py +++ b/segment_coverage/segment_coverage/main.py @@ -74,7 +74,7 @@ class CoverageChecker(object): """Checks the segment coverage for a given channel in a a given directoy.""" def __init__(self, channel, qualities, base_dir, first_hour, last_hour, - make_page, connection_string, check_interval): + make_page, connection_string, check_interval, output_dir, run_once): """Constructor for CoverageChecker. Creates a checker for a given channel with specified qualities.""" @@ -87,6 +87,8 @@ class CoverageChecker(object): self.make_page = make_page self.db_manager = None if connection_string is None else database.DBManager(dsn=connection_string) self.check_interval = check_interval + self.output_dir = output_dir + self.run_once = run_once self.stopping = gevent.event.Event() self.logger = logging.getLogger('CoverageChecker({})'.format(channel)) @@ -172,7 +174,7 @@ class CoverageChecker(object): colours[coverage_mask] = matplotlib.colors.to_rgb('tab:blue') colours[coverage_mask & partial_mask] = matplotlib.colors.to_rgb('tab:orange') # write the pixel array to a temporary file then atomically rename it - path_prefix = os.path.join(self.base_dir, 'coverage-maps', '{}_{}'.format(self.channel, quality)) + path_prefix = os.path.join(self.base_dir, self.output_dir, '{}_{}'.format(self.channel, quality)) temp_path = '{}_{}.png'.format(path_prefix, uuid.uuid4()) final_path = '{}_coverage.png'.format(path_prefix) common.ensure_directory(temp_path) @@ -232,13 +234,13 @@ class CoverageChecker(object): for node in sorted(nodes.keys()): html += """