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.
pull/4/head
owengaspard 3 years ago
parent 5b3989611f
commit b2bcb28f8f

@ -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)

@ -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
Loading…
Cancel
Save