diff --git a/utils/videos.py b/utils/videos.py index 9f627ec..b4149ca 100755 --- a/utils/videos.py +++ b/utils/videos.py @@ -34,11 +34,12 @@ def check_done( return redditobj -def save_data(filename: str, reddit_title: str, reddit_id: str, credit: str, NSFW: bool): +def save_data(subreddit: str, filename: str, reddit_title: str, reddit_id: str, credit: str, NSFW: bool): """Saves the videos that have already been generated to a JSON file in video_creation/data/videos.json Args: filename (str): The finished video title name + @param subreddit: @param filename: @param reddit_id: @param reddit_title: @@ -49,6 +50,7 @@ def save_data(filename: str, reddit_title: str, reddit_id: str, credit: str, NSF if reddit_id in [video["id"] for video in done_vids]: return # video already done but was specified to continue anyway in the config file payload = { + "subreddit": subreddit, "id": reddit_id, "time": str(int(time.time())), "background_credit": credit, diff --git a/video_creation/final_video.py b/video_creation/final_video.py index 0f01b6c..b866b27 100755 --- a/video_creation/final_video.py +++ b/video_creation/final_video.py @@ -119,7 +119,7 @@ def make_final_video( filename = f"{name_normalize(title)}.mp4" subreddit = settings.config["reddit"]["thread"]["subreddit"] - save_data(filename, title, idx, background_config[2], nsfw) + save_data(subreddit, filename, title, idx, background_config[2], nsfw) if not exists(f"./results/{subreddit}"): print_substep("The results folder didn't exist so I made it")