From 0990a24580fff2cee6f01f79b391ccb795712275 Mon Sep 17 00:00:00 2001 From: Jacob Chambers Date: Sat, 4 Jun 2022 01:21:54 +0100 Subject: [PATCH] Loops functionality and custom names for output --- main.py | 28 +++++++++++++++++++++------- 1 file changed, 21 insertions(+), 7 deletions(-) diff --git a/main.py b/main.py index 8cc3c9a..3293e08 100644 --- a/main.py +++ b/main.py @@ -8,6 +8,8 @@ from video_creation.screenshot_downloader import download_screenshots_of_reddit_ from video_creation.final_video import make_final_video from dotenv import load_dotenv import os +import re +import math print_markdown( "### Thanks for using this tool! [Feel free to contribute to this project on GitHub!](https://lewismenelaws.com) If you have any questions, feel free to reach out to me on Twitter or submit a GitHub issue." @@ -15,12 +17,24 @@ print_markdown( time.sleep(3) +while True: + reddit_object = get_subreddit_threads() -reddit_object = get_subreddit_threads() + thread_title = str.strip(reddit_object["thread_title"]).replace(" ", "_") + thread_title = re.sub(r"[^_\w]", "", thread_title) + file_name = f"{os.getenv('SUBREDDIT')}-{thread_title}.mp4" + file_exists = os.path.isfile(f"output/{file_name}") -load_dotenv() -length, number_of_comments = save_text_to_mp3(reddit_object) -download_screenshots_of_reddit_posts(reddit_object, number_of_comments, os.getenv("THEME")) -download_background() -chop_background_video(length) -final_video = make_final_video(number_of_comments) + if file_exists is False: + load_dotenv() + length, number_of_comments = save_text_to_mp3(reddit_object) + print(f"Video is {math.ceil(length)}s long with {number_of_comments} comments.") + download_screenshots_of_reddit_posts(reddit_object, number_of_comments, os.getenv("THEME")) + download_background() + chop_background_video(length) + final_video = make_final_video(number_of_comments, name=f"{os.getenv('SUBREDDIT')}-{thread_title}") + else: + print(f"Video for thread already exists! Trying again...") + + if os.getenv("LOOP") != "true": + break