From 5beef765d79c8bbe3ba504d1687ac98bf1962b57 Mon Sep 17 00:00:00 2001 From: Jason Date: Wed, 8 Jun 2022 17:33:52 -0400 Subject: [PATCH] chore: reformatted feat:adds tiny delay closes #40 --- reddit/subreddit.py | 8 +++++--- utils/subreddit.py | 6 +++++- video_creation/TTSwrapper.py | 6 ++---- video_creation/screenshot_downloader.py | 8 +++++--- video_creation/voices.py | 6 +++++- 5 files changed, 22 insertions(+), 12 deletions(-) diff --git a/reddit/subreddit.py b/reddit/subreddit.py index bf745b5..6a2c273 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -43,7 +43,9 @@ def get_subreddit_threads(): Ask user for subreddit input """ print_step("Getting subreddit threads...") - if not getenv("SUBREDDIT"): # note to self. you can have multiple subreddits via reddit.subreddit("redditdev+learnpython") + if not getenv( + "SUBREDDIT" + ): # note to self. you can have multiple subreddits via reddit.subreddit("redditdev+learnpython") subreddit = reddit.subreddit( input("What subreddit would you like to pull from? ") ) # if the env isnt set, ask user @@ -60,7 +62,7 @@ def get_subreddit_threads(): else: threads = subreddit.hot(limit=25) submission = get_subreddit_undone(threads, subreddit) - submission = check_done(submission) # double checking + submission = check_done(submission) # double checking if submission is None: return get_subreddit_threads() # submission already done. rerun upvotes = submission.score @@ -80,7 +82,7 @@ def get_subreddit_threads(): try: content["thread_url"] = submission.url content["thread_title"] = submission.title - #ontent["thread_content"] = submission.content + # ontent["thread_content"] = submission.content content["comments"] = [] for top_level_comment in submission.comments: diff --git a/utils/subreddit.py b/utils/subreddit.py index 4b82f2c..3fc1613 100644 --- a/utils/subreddit.py +++ b/utils/subreddit.py @@ -1,6 +1,7 @@ from typing import List import json + def get_subreddit_undone(submissions: List, subreddit): """ recursively checks if the top submission in the list was already done. @@ -11,7 +12,10 @@ def get_subreddit_undone(submissions: List, subreddit): if already_done(done_videos, submission): continue return submission - return get_subreddit_undone(subreddit.top(time_filter="hour"), subreddit) # all of the videos in hot have already been done + return get_subreddit_undone( + subreddit.top(time_filter="hour"), subreddit + ) # all of the videos in hot have already been done + def already_done(done_videos: list, submission): diff --git a/video_creation/TTSwrapper.py b/video_creation/TTSwrapper.py index 2ebdb28..62201cb 100644 --- a/video_creation/TTSwrapper.py +++ b/video_creation/TTSwrapper.py @@ -116,11 +116,9 @@ class TTTTSWrapper: # TikTok Text-to-Speech Wrapper chunkId = chunkId + 1 - if(len(audio_clips) > 1): + if len(audio_clips) > 1: cbn.convert(samplerate=44100, n_channels=2) - cbn.build( - audio_clips, filename, 'concatenate' - ) + cbn.build(audio_clips, filename, "concatenate") else: os.rename(audio_clips[0], filename) diff --git a/video_creation/screenshot_downloader.py b/video_creation/screenshot_downloader.py index 8d169f6..89d6728 100644 --- a/video_creation/screenshot_downloader.py +++ b/video_creation/screenshot_downloader.py @@ -2,7 +2,7 @@ import json from os import getenv from pathlib import Path -#from playwright.async_api import async_playwright +# from playwright.async_api import async_playwright from playwright.sync_api import sync_playwright, ViewportSize from rich.progress import track @@ -55,10 +55,12 @@ def download_screenshots_of_reddit_posts(reddit_object, screenshot_num): path="assets/temp/png/title.png" ) if storymode: - page.locator('[data-click-id="text"]').screenshot(path="assets/temp/png/story_content.png") + page.locator('[data-click-id="text"]').screenshot( + path="assets/temp/png/story_content.png" + ) else: for idx, comment in track( - enumerate(reddit_object["comments"]), "Downloading screenshots..." + enumerate(reddit_object["comments"]), "Downloading screenshots..." ): # Stop if we have reached the screenshot_num diff --git a/video_creation/voices.py b/video_creation/voices.py index 955f160..be7bc7d 100644 --- a/video_creation/voices.py +++ b/video_creation/voices.py @@ -36,7 +36,11 @@ def save_text_to_mp3(reddit_obj): except HeaderNotFoundError: # note to self AudioFileClip length += sox.file_info.duration(f"assets/temp/mp3/title.mp3") if getenv("STORYMODE").casefold() == "true": - ttttsw.tts(sanitize_text(reddit_obj["thread_content"]), filename=f"assets/temp/mp3/story_content.mp3", random_speaker=False) + ttttsw.tts( + sanitize_text(reddit_obj["thread_content"]), + filename=f"assets/temp/mp3/story_content.mp3", + random_speaker=False, + ) #'story_content' com = 0 for comment in track((reddit_obj["comments"]), "Saving..."):