from utils.console import print_markdown, print_step, print_substep import praw import random from dotenv import load_dotenv import os def get_askreddit_threads(): """ Returns a list of threads from the AskReddit subreddit. """ load_dotenv() print_step("Getting AskReddit threads...") print(os.getenv("REDDIT_2FA")) if os.getenv("REDDIT_2FA").lower() == "yes": print( "\nEnter your two-factor authentication code from your authenticator app.\n" ) code = input("> ") print() pw = os.getenv("REDDIT_PASSWORD") passkey = f"{pw}:{code}" else: passkey = os.getenv("REDDIT_PASSWORD") content = {} 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=passkey, ) askreddit = reddit.subreddit("askreddit") threads = askreddit.hot(limit=25) submission = list(threads)[random.randrange(0, 25)] print_substep(f"Video will be: {submission.title}") 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 AskReddit threads successfully.", style="bold green") return content