frame.py 文件源码

def fix_episode(episode_path):
    try:
        episode = load_episode(episode_path)
    except EOFError:
        print("Error reading: {}".format(episode_path))
        os.remove(episode_path)
        return
    if episode.version == 2:
        print("Version 2 already: {}".format(episode_path))
        return

    old_frames = episode.frames
    episode.frames = []
    for i in range(len(old_frames) - 1):
        f = old_frames[i]
        f.action = old_frames[i + 1].action
        episode.frames.append(f)

    episode.version = 2

    s = pickle.dumps(episode)
    with gzip.open(episode_path, "wb") as f:
        f.write(s)

        # pickler = pickle.Pickler(f)
        # pickler.dump(episode)

    # save_episode(episode_path, episode)