You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
4.9 KiB
89 lines
4.9 KiB
#!/usr/bin/env python
|
|
from asyncio import run
|
|
import math
|
|
from subprocess import Popen
|
|
from os import name
|
|
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.checker import envUpdate
|
|
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 Reddit
|
|
from video_creation.voices import save_text_to_mp3
|
|
|
|
VERSION = "2.2.9"
|
|
print(
|
|
"""
|
|
██████╗ ███████╗██████╗ ██████╗ ██╗████████╗ ██╗ ██╗██╗██████╗ ███████╗ ██████╗ ███╗ ███╗ █████╗ ██╗ ██╗███████╗██████╗
|
|
██╔══██╗██╔════╝██╔══██╗██╔══██╗██║╚══██╔══╝ ██║ ██║██║██╔══██╗██╔════╝██╔═══██╗ ████╗ ████║██╔══██╗██║ ██╔╝██╔════╝██╔══██╗
|
|
██████╔╝█████╗ ██║ ██║██║ ██║██║ ██║ ██║ ██║██║██║ ██║█████╗ ██║ ██║ ██╔████╔██║███████║█████╔╝ █████╗ ██████╔╝
|
|
██╔══██╗██╔══╝ ██║ ██║██║ ██║██║ ██║ ╚██╗ ██╔╝██║██║ ██║██╔══╝ ██║ ██║ ██║╚██╔╝██║██╔══██║██╔═██╗ ██╔══╝ ██╔══██╗
|
|
██║ ██║███████╗██████╔╝██████╔╝██║ ██║ ╚████╔╝ ██║██████╔╝███████╗╚██████╔╝ ██║ ╚═╝ ██║██║ ██║██║ ██╗███████╗██║ ██║
|
|
╚═╝ ╚═╝╚══════╝╚═════╝ ╚═════╝ ╚═╝ ╚═╝ ╚═══╝ ╚═╝╚═════╝ ╚══════╝ ╚═════╝ ╚═╝ ╚═╝╚═╝ ╚═╝╚═╝ ╚═╝╚══════╝╚═╝ ╚═╝
|
|
"""
|
|
)
|
|
# Modified by JasonLovesDoggo
|
|
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. You can find solutions to many common problems in the [Documentation](https://luka-hietala.gitbook.io/documentation-for-the-reddit-bot/)"
|
|
)
|
|
print_step(f"You are using V{VERSION} of the bot")
|
|
|
|
|
|
async def main(
|
|
POST_ID=None
|
|
):
|
|
cleanup()
|
|
reddit_object = get_subreddit_threads(POST_ID)
|
|
length, number_of_comments = save_text_to_mp3(reddit_object)
|
|
length = math.ceil(length)
|
|
reddit_screenshots = Reddit(reddit_object, number_of_comments)
|
|
browser = await reddit_screenshots.get_browser()
|
|
await reddit_screenshots.download_screenshots(browser)
|
|
bg_config = get_background_config()
|
|
download_background(bg_config)
|
|
chop_background_video(bg_config, length)
|
|
make_final_video(number_of_comments, length, reddit_object, bg_config)
|
|
|
|
|
|
async 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....
|
|
await main()
|
|
Popen("cls" if name == "nt" else "clear", shell=True).wait()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
config = settings.check_toml(".config.template.toml", "config.toml")
|
|
config is False and exit()
|
|
try:
|
|
if config["settings"]["times_to_run"]:
|
|
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("+"))}'
|
|
)
|
|
run(
|
|
main(post_id)
|
|
)
|
|
Popen("cls" if name == "nt" else "clear", shell=True).wait()
|
|
else:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print_markdown("## Clearing temp files")
|
|
cleanup()
|
|
exit()
|