owengaspard
f12f0da76d
Python was giving errors with TRUE as a bool, so I changed it to a string. Giving an undefined name returned a different error than I put in the except, so I edited it and ran some tests which were successful. I will work on implementing something to detect if the subreddit exists and if it includes anything (e.g. special characters) which cannot be in subreddit names.
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
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 subreddit 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"),
|
|
)
|
|
|
|
if os.getenv("ASK_EACH_TIME") == "TRUE":
|
|
try:
|
|
subreddit = reddit.subreddit(input("What subreddit would you like to pull from? "))
|
|
except ValueError:
|
|
subreddit = reddit.subreddit("askreddit")
|
|
print_substep("Subreddit not defined. Using AskReddit.")
|
|
else:
|
|
try:
|
|
subreddit = reddit.subreddit(os.getenv("SUBREDDIT"))
|
|
except ValueError:
|
|
subreddit = reddit.subreddit("askreddit")
|
|
print_substep("Subreddit not defined. Using AskReddit.")
|
|
|
|
threads = subreddit.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 subreddit threads Successfully.", style="bold green")
|
|
return content
|