From 9a8f97a2784d42f69f556497a2e70e6430272ae5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Tronel?= Date: Sat, 2 Dec 2023 21:06:25 +0100 Subject: [PATCH] Fix missing calls to getTSFrame. --- removeads.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/removeads.py b/removeads.py index 492859b..28c0260 100755 --- a/removeads.py +++ b/removeads.py @@ -846,9 +846,13 @@ def main(): logger.info("Found %d frames between beginning of current part and first I-frame" % nbHeadFrames) logger.info("Found %d frames between last I-frame and end of current part" % nbTailFrames) - headIFrameTS = timedelta(seconds=float(headIFrame['pts_time'])) - tailIFrameTS = timedelta(seconds=float(tailIFrame['pts_time'])) - + headIFrameTS = getTSFrame(headIFrame) + if headIFrameTS == None: + exit(-1) + tailIFrameTS = getTSFrame(tailIFrame) + if tailIFrameTS == None: + exit(-1) + checks.append(pos+headIFrameTS-ts1) subparts = []