Subreddit chooser (TEST)
Allows user to choose a subreddit to pull from by taking user input. I will work on adding it to the .env file.
This commit is contained in:
1 parent
5b3989611f
commit
b2bcb28f8f
4 files changed
+48
-2
No files matched your search
@@ -1,6 +1,6 @@
|
||||
from utils.console import print_markdown
|
||||
import time
|
||||
from reddit.askreddit import get_askreddit_threads
|
||||
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
|
||||
@@ -13,7 +13,7 @@ print_markdown(
|
||||
time.sleep(3)
|
||||
|
||||
|
||||
reddit_object = get_askreddit_threads()
|
||||
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)
|
||||
|
||||
Binary file not shown.
@@ -0,0 +1,46 @@
|
||||
from utils.console import print_markdown, print_step, print_substep
|
||||
import praw
|
||||
import random
|
||||
from dotenv import load_dotenv
|
||||
import os
|
||||
|
||||
|
||||
def get_subreddit_threads():
|
||||
"""
|
||||
Returns a list of threads from the AskReddit subreddit.
|
||||
"""
|
||||
|
||||
print_step("Getting AskReddit threads...")
|
||||
|
||||
content = {}
|
||||
load_dotenv()
|
||||
reddit = praw.Reddit(
|
||||
client_id=os.getenv("REDDIT_CLIENT_ID"),
|
||||
client_secret=os.getenv("REDDIT_CLIENT_SECRET"),
|
||||
user_agent="Accessing AskReddit threads",
|
||||
username=os.getenv("REDDIT_USERNAME"),
|
||||
password=os.getenv("REDDIT_PASSWORD"),
|
||||
)
|
||||
subreddit = reddit.subreddit(input("What subreddit would you like to pull from? "))
|
||||
threads = subreddot.hot(limit=25)
|
||||
submission = list(threads)[random.randrange(0, 25)]
|
||||
print_substep(f"Video will be: {submission.title} :thumbsup:")
|
||||
try:
|
||||
|
||||
content["thread_url"] = submission.url
|
||||
content["thread_title"] = submission.title
|
||||
content["comments"] = []
|
||||
|
||||
for top_level_comment in submission.comments:
|
||||
content["comments"].append(
|
||||
{
|
||||
"comment_body": top_level_comment.body,
|
||||
"comment_url": top_level_comment.permalink,
|
||||
"comment_id": top_level_comment.id,
|
||||
}
|
||||
)
|
||||
|
||||
except AttributeError as e:
|
||||
pass
|
||||
print_substep("Received subreddot threads Successfully.", style="bold green")
|
||||
return content
|
||||
Binary file not shown.
Reference in new issue
Block a user