def check_frames()

in video.py [0:0]

• 20 lines of code
• 9 McCabe index (conditional complexity)

    def check_frames(self, frame_dir, extension, frames=None):
        if not os.path.isdir(frame_dir):
            return False

        files = os.listdir(frame_dir)
        files = [n for n in files if n.endswith(extension)]
        if len(files) == 0:
            return False

        if frames is None:
            frames = range(self.frame_count)

        if len(files) != len(frames):
            sys.exit(
                "ERROR: expected to find %d files but found %d in '%s'"
                % (self.frame_count, len(files), frame_dir)
            )
        for i in frames:
            frame_file = "%s/frame_%06d.%s" % (frame_dir, i, extension)
            if not os.path.exists(frame_file):
                sys.exit("ERROR: did not find expected file '%s'" % frame_file)
        print("Frames found, checked OK.")

        return True