fixed the erroneous code blocks, and instead of console.log(), console.print() were used instead, since it would give much more pleasant

the codebase were also cleaned
pull/280/head
iaacornus 3 years ago
commit b158a2f241
No known key found for this signature in database
GPG Key ID: 281739AE7252598C

@ -2,17 +2,27 @@ import os
import shutil import shutil
from dotenv import load_dotenv from dotenv import load_dotenv
from rich.console import Console
from reddit.subreddit import get_subreddit_threads from reddit.subreddit import get_subreddit_threads
from video_creation.background import download_background, chop_background_video from video_creation.background import download_background, chop_background_video
from video_creation.voices import save_text_to_mp3 from video_creation.voices import save_text_to_mp3
from video_creation.screenshot_downloader import download_screenshots_of_reddit_posts from video_creation.screenshot_downloader import download_screenshots_of_reddit_posts
from video_creation.final_video import make_final_video from video_creation.final_video import make_final_video
from utils.console import print_markdown, print_substep from utils.console import print_markdown, print_substep
def main(subreddit_=None, background=None, filename=None): def main(subreddit_=None, background=None, filename=None):
"""
Load .env file if exists. If it doesnt exist, print a warning and
launch the setup wizard. If there is a .env file, check if the
required variables are set. If not, print a warning and launch the
setup wizard.
"""
console = Console()
load_dotenv()
REQUIRED_VALUES = [ REQUIRED_VALUES = [
"REDDIT_CLIENT_ID", "REDDIT_CLIENT_ID",
"REDDIT_CLIENT_SECRET", "REDDIT_CLIENT_SECRET",
@ -30,9 +40,9 @@ def main(subreddit_=None, background=None, filename=None):
configured = True configured = True
if not os.path.exists(".env"): if not os.path.exists(".env"):
shutil.copy(".env.template", ".env") shutil.copy(".env.template", ".env")
configured = False console.print("[red] Your .env file is invalid, or was never created. Standby.[/red]")
load_dotenv() console.print("[bold green]Checking environment variables...")
for val in REQUIRED_VALUES: for val in REQUIRED_VALUES:
if not os.getenv(val): if not os.getenv(val):
@ -40,12 +50,32 @@ def main(subreddit_=None, background=None, filename=None):
configured = False configured = False
if configured: if configured:
try:
float(os.getenv("OPACITY"))
except:
console.print(
f"[red]Please ensure that OPACITY is set between 0 and 1 in your .env file"
)
configured = False
exit()
console.log("[bold green]Enviroment Variables are set! Continuing...[/bold green]")
reddit_object = get_subreddit_threads(subreddit_) reddit_object = get_subreddit_threads(subreddit_)
length, number_of_comments = save_text_to_mp3(reddit_object) length, number_of_comments = save_text_to_mp3(reddit_object)
download_screenshots_of_reddit_posts( download_screenshots_of_reddit_posts(
reddit_object, number_of_comments, reddit_object,
os.getenv("THEME") number_of_comments,
os.getenv("THEME", "light")
) )
download_background(background) download_background(background)
chop_background_video(length) chop_background_video(length)
final_video = make_final_video(snumber_of_comments, filename) final_video = make_final_video(number_of_comments, filename)
else:
console.print(
"[red]Looks like you need to set your Reddit credentials in the .env file. "
+ "Please follow the instructions in the README.md file to set them up.[/red]"
)
setup_ask = input("\033[1mLaunch setup wizard? [y/N] > \033[0m")
if setup_ask in ["y", "Y"]:
os.system("python3 setup.py")

Loading…
Cancel
Save