Merge pull request #241 from elebumm/default-env-values

Add default values for environment variables

main.py

@@ -1,13 +1,13 @@
 from utils.console import print_markdown
-import time
-
 from reddit.subreddit import get_subreddit_threads
 from video_creation.background import download_background, chop_background_video
 from video_creation.voices import save_text_to_mp3
 from video_creation.screenshot_downloader import download_screenshots_of_reddit_posts
 from video_creation.final_video import make_final_video
 from dotenv import load_dotenv
-import os
+import os, time, shutil
+
+REQUIRED_VALUES = ["REDDIT_CLIENT_ID","REDDIT_CLIENT_SECRET","REDDIT_USERNAME","REDDIT_PASSWORD"]
 
 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 +15,23 @@ print_markdown(
 
 time.sleep(3)
 
+load_dotenv()
+
+configured = True
 
-reddit_object = get_subreddit_threads()
+if not os.path.exists(".env"):
+    shutil.copy(".env.template", ".env")
+    configured = False
 
-load_dotenv()
+for val in REQUIRED_VALUES:
-length, number_of_comments = save_text_to_mp3(reddit_object)
+    if val not in os.environ or not os.getenv(val):
-download_screenshots_of_reddit_posts(reddit_object, number_of_comments, os.getenv("THEME"))
+        print(f"Please set the variable \"{val}\" in your .env file.")
-download_background()
+        configured = False
-chop_background_video(length)
+
-final_video = make_final_video(number_of_comments)
+if configured:
+    reddit_object = get_subreddit_threads()
+    length, number_of_comments = save_text_to_mp3(reddit_object)
+    download_screenshots_of_reddit_posts(reddit_object, number_of_comments, os.getenv("THEME", "light"))
+    download_background()
+    chop_background_video(length)
+    final_video = make_final_video(number_of_comments)

reddit/subreddit.py

@@ -13,7 +13,7 @@ def get_subreddit_threads():
 
     print_step("Getting AskReddit threads...")
 
-    if os.getenv("REDDIT_2FA").lower() == "yes":
+    if os.getenv("REDDIT_2FA", default="no").casefold() == "yes":
         print(
             "\nEnter your two-factor authentication code from your authenticator app.\n"
         )