|
|
|
@ -4,15 +4,13 @@ import os, random, praw, re
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_subreddit_threads():
|
|
|
|
|
|
|
|
|
|
global submission
|
|
|
|
|
"""
|
|
|
|
|
Returns a list of threads from the AskReddit subreddit.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
load_dotenv()
|
|
|
|
|
|
|
|
|
|
print_step("Getting AskReddit threads...")
|
|
|
|
|
|
|
|
|
|
if os.getenv("REDDIT_2FA", default="no").casefold() == "yes":
|
|
|
|
|
print(
|
|
|
|
|
"\nEnter your two-factor authentication code from your authenticator app.\n"
|
|
|
|
@ -33,14 +31,25 @@ def get_subreddit_threads():
|
|
|
|
|
username=os.getenv("REDDIT_USERNAME"),
|
|
|
|
|
password=passkey,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# If the user specifies that he doesnt want a random thread, or if he doesn't insert the "RANDOM_THREAD" variable at all, ask the thread link
|
|
|
|
|
if not os.getenv("RANDOM_THREAD") or os.getenv("RANDOM_THREAD") == "no":
|
|
|
|
|
print_substep("Insert the full thread link:", style="bold green")
|
|
|
|
|
thread_link = input()
|
|
|
|
|
print_step(f"Getting the inserted thread...")
|
|
|
|
|
submission = reddit.submission(url=thread_link)
|
|
|
|
|
else:
|
|
|
|
|
# Otherwise, picks a random thread from the inserted subreddit
|
|
|
|
|
if os.getenv("SUBREDDIT"):
|
|
|
|
|
subreddit = reddit.subreddit(re.sub(r"r\/", "", os.getenv("SUBREDDIT")))
|
|
|
|
|
else:
|
|
|
|
|
# ! Prompt the user to enter a subreddit
|
|
|
|
|
try:
|
|
|
|
|
subreddit = reddit.subreddit(
|
|
|
|
|
re.sub(r"r\/", "", input("What subreddit would you like to pull from? "))
|
|
|
|
|
re.sub(
|
|
|
|
|
r"r\/",
|
|
|
|
|
"",
|
|
|
|
|
input("What subreddit would you like to pull from? "),
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
except ValueError:
|
|
|
|
|
subreddit = reddit.subreddit("askreddit")
|
|
|
|
@ -48,14 +57,15 @@ def get_subreddit_threads():
|
|
|
|
|
|
|
|
|
|
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:
|
|
|
|
|
if not top_level_comment.stickied:
|
|
|
|
|
content["comments"].append(
|
|
|
|
|
{
|
|
|
|
|
"comment_body": top_level_comment.body,
|
|
|
|
|