From ef8d245c08a2c9082113d68a24184da5b08f8817 Mon Sep 17 00:00:00 2001 From: Ajmal Sharif Date: Sun, 20 Feb 2022 22:26:12 -0500 Subject: [PATCH] Add logic for when video starts in a segment --- main.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/main.py b/main.py index 6c8db1b..637084f 100644 --- a/main.py +++ b/main.py @@ -81,14 +81,16 @@ async def get_segments(vid_id, web_session, categories = ["sponsor"]): async def time_to_segment(segments, position, rc): - future_segments = [] - for i in segments: - if i[0] > position: - future_segments.append(i) - next_segement = future_segments[0] - time_to_next = next_segement[0] - position - await skip(time_to_next, next_segement[1], rc) - + for segment in segments: + if position < 2 and (position >= segment[0] and position < segment[1]): + next_segment = [position, segment[1]] + break + if segment[0] > position: + next_segment = segment + break + time_to_next = next_segment[0] - position + await skip(time_to_next, next_segment[1], rc) + async def skip(time_to, position, rc): await asyncio.sleep(time_to) await rc.set_position(position)