예제 #1
0
        scene_stamps[show][episode_name] = None

# retrieve time_stamps and scene_stamps from disk copies (if they exist)
for show in list_of_shows:
    for episode_name in video_file_names[show]:
        time_stamps_path = DIR_STAMPS[show]+'/'+episode_name+'_timestamps.json'
        scene_stamps_path = DIR_STAMPS[show]+'/'+episode_name+'_scenestamps.json'
        try:
            with open(time_stamps_path, 'r') as fp1, open(scene_stamps_path, 'r') as fp2:
                time_stamps[show][episode_name] = json.load(fp1)
                scene_stamps[show][episode_name] = json.load(fp2)
        except IOError:
        # if no time_stamps/scene_stamps exist, fetch them and store a copy on disk
        # the get scene_stamps function gets time and scene_stamps
            episode_path = DIR_VIDS[show]+'/'+episode_name+'.mp4'
            result = preprocessor.get_scene_stamps(episode_path)
            time_stamps[show][episode_name] = result['time_stamps']
            scene_stamps[show][episode_name] = result['scene_stamps']
            # json converts tuples into lists
            with open(time_stamps_path, 'w') as fp1, open(scene_stamps_path, 'w') as fp2:
                json.dump(time_stamps[show][episode_name], fp1)
                json.dump(scene_stamps[show][episode_name], fp2)

# process and store the plot on the drive
plot_sentences = {}
for show in list_of_shows:
    plot_sentences[show] = {}

for show in list_of_shows:
    for episode_name in video_file_names[show]:
    # storing the processed part in json
예제 #2
0
    "BCS1E08",
    "BCS1E09",
    "better.call.saul.110.hdtv-lol",
]

time_stamps, scene_stamps = [], []
for vid_file in video_file_names:
    time_stamps_path = DIR_STAMPS + "/" + vid_file + "_proc_ts.json"
    scene_stamps_path = DIR_STAMPS + "/" + vid_file + "_proc_ss.json"
    try:
        with open(time_stamps_path, "r") as fp1, open(scene_stamps_path, "r") as fp2:
            time_stamps.append(json.load(fp1))
            scene_stamps.append(json.load(fp2))
    except IOError:
        with open(time_stamps_path, "w") as fp1, open(scene_stamps_path, "w") as fp2:
            t1, t2 = get_scene_stamps(DIR_VIDS + "/" + vid_file + ".mp4")
            time_stamps.append(t1)
            scene_stamps.append(t2)
            json.dump(time_stamps[-1], fp1)
            json.dump(scene_stamps[-1], fp2)

plot_sentences = []
for plot in file_names:
    # storing the processed part in json
    file_path = DIR_PLOTS + "/" + plot + "_proc_plot.json"
    try:
        with open(file_path, "r") as fp:
            plot_sentences.append(json.load(fp))
    except IOError:
        with open(file_path, "w") as fp:
            plot_sentences.append(fetch_plot_data(DIR_PLOTS + "/" + plot + "_plot.txt"))