Fix missing calls to getTSFrame.

2023-12-02 21:06:25 +01:00
parent 650724c966
commit 9a8f97a278
1 changed files with 7 additions and 3 deletions
--- 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 = []