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
parent
5b3989611f
commit
b2bcb28f8f
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.
Loading…
Reference in new issue