main.py

import math
from subprocess import Popen
from os import name

from prawcore import ResponseException

from reddit.subreddit import get_subreddit_threads
from utils.cleanup import cleanup
from utils.console import print_markdown, print_step
from utils import settings
from utils.id import id
from utils.version import check_version

from video_creation.background import (
  download_background,
  chop_background_video,
  get_background_config
)
from video_creation.final_video import make_final_video
from video_creation.screenshot_downloader import download_screenshots_of_reddit_posts
from video_creation.voices import save_text_to_mp3

__VERSION__ = "1.0.0"

print(
    """
██████╗ ███████╗██████╗ ██████╗ ██╗████████╗    ██╗   ██╗██╗██████╗ ███████╗ ██████╗     ███╗   ███╗ █████╗ ██╗  ██╗███████╗██████╗
██╔══██╗██╔════╝██╔══██╗██╔══██╗██║╚══██╔══╝    ██║   ██║██║██╔══██╗██╔════╝██╔═══██╗    ████╗ ████║██╔══██╗██║ ██╔╝██╔════╝██╔══██╗
██████╔╝█████╗  ██║  ██║██║  ██║██║   ██║       ██║   ██║██║██║  ██║█████╗  ██║   ██║    ██╔████╔██║███████║█████╔╝ █████╗  ██████╔╝
██╔══██╗██╔══╝  ██║  ██║██║  ██║██║   ██║       ╚██╗ ██╔╝██║██║  ██║██╔══╝  ██║   ██║    ██║╚██╔╝██║██╔══██║██╔═██╗ ██╔══╝  ██╔══██╗
██║  ██║███████╗██████╔╝██████╔╝██║   ██║        ╚████╔╝ ██║██████╔╝███████╗╚██████╔╝    ██║ ╚═╝ ██║██║  ██║██║  ██╗███████╗██║  ██║
╚═╝  ╚═╝╚══════╝╚═════╝ ╚═════╝ ╚═╝   ╚═╝         ╚═══╝  ╚═╝╚═════╝ ╚══════╝ ╚═════╝     ╚═╝     ╚═╝╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚═╝  ╚═╝
"""
)

print_markdown(
  "### Thanks for using this tool!"
)
check_version(__VERSION__)

def main(POST_ID=None):
  reddit_object = get_subreddit_threads(POST_ID)
  global redditid
  redditid = id(reddit_object)
  length, number_of_comments = save_text_to_mp3(reddit_object)
  length = math.ceil(length)
  download_screenshots_of_reddit_posts(reddit_object, number_of_comments)
  bg_config = get_background_config()
  download_background(bg_config)
  chop_background_video(bg_config, length, reddit_object)
  make_final_video(number_of_comments, length, reddit_object, bg_config)

def run_many(times):
  for x in range(1, times + 1):
    print_step(
      f'on the {x}{("th", "st", "nd", "rd", "th", "th", "th", "th", "th", "th")[x % 10]} iteration of {times}'
    )  # correct 1st 2nd 3rd 4th 5th....
    main()
    Popen("cls" if name == "nt" else "clear", shell=True).wait()

def shutdown():
  print_markdown("## Clearing temp files")
  try:
    redditid
  
  except NameError:
    print("Exiting...")
    exit()
  
  else:
    cleanup(redditid)
    print("Exiting...")
    exit()

if __name__ == "__main__":
  config = settings.check_toml("utils/.config.template.toml", "config.toml")
  config is False and exit()
  try:
    if config["settings"]["times_to_run"]:
      run_many(config["settings"]["times_to_run"])

    elif len(config["reddit"]["thread"]["post_id"].split("+")) > 1:
      for index, post_id in enumerate(config["reddit"]["thread"]["post_id"].split("+")):
        index += 1
        print_step(
          f'on the {index}{("st" if index % 10 == 1 else ("nd" if index % 10 == 2 else ("rd" if index % 10 == 3 else "th")))} post of {len(config["reddit"]["thread"]["post_id"].split("+"))}'
        )
        main(post_id)
        Popen("cls" if name == "nt" else "clear", shell=True).wait()
    else:
      main()
  
  except KeyboardInterrupt:
    shutdown()
  
  except ResponseException:
    # error for invalid credentials
    print_markdown("## Invalid credentials")
    print_markdown("Please check your credentials in the config.toml file")

    shutdown()