diff --git a/cut_races.py b/cut_races.py index 2664521..b1005eb 100644 --- a/cut_races.py +++ b/cut_races.py @@ -54,6 +54,8 @@ class NoSegments(Exception): def cut_to_file(filename, base_dir, stream, start, end, variant='source'): + if os.path.exists(filename): + return logging.info("Cutting {}".format(filename)) segments = get_best_segments( os.path.join(base_dir, stream, variant).lower(), @@ -68,7 +70,7 @@ def cut_to_file(filename, base_dir, stream, start, end, variant='source'): f.write(chunk) -def main(host='condor.host', user='necrobot-read', password=None, database='season_8', base_dir='.', output_dir='.'): +def main(host='condor.host', user='necrobot-read', password=None, database='season_8', base_dir='.', output_dir='.', find=None): logging.basicConfig(level=logging.INFO) if password is None: @@ -77,6 +79,9 @@ def main(host='condor.host', user='necrobot-read', password=None, database='seas host=host, user=user, password=password, database=database, ) + if find: + find = tuple(find.split('-')) + cur = conn.cursor() cur.execute(INFO_QUERY) @@ -89,6 +94,9 @@ def main(host='condor.host', user='necrobot-read', password=None, database='seas logging.info("Got info on {} races".format(len(data))) for racer1, racer2, cawmentator, match_id, race_number, start, duration in data: + if find and (racer1.lower(), racer2.lower()) != find: + continue + end = start + datetime.timedelta(seconds=duration/100.) base_name = "-".join(map(str, [racer1, racer2, match_id, race_number]))