include the subreddit parameter

pull/280/head
iaacornus 3 years ago
parent 0944d3b38a
commit 558d9bd531
No known key found for this signature in database
GPG Key ID: 281739AE7252598C

@ -7,8 +7,12 @@ from dotenv import load_dotenv
from utils.console import print_step, print_substep
def get_subreddit_threads():
def get_subreddit_threads(subreddit_):
"""
Takes subreddit_ as parameter which defaults to None, but in this
case since it is None, it would raise ValueError, thus defaulting
to AskReddit.
Returns a list of threads from the AskReddit subreddit.
"""
@ -27,7 +31,6 @@ def get_subreddit_threads():
passkey = os.getenv("REDDIT_PASSWORD")
content = {}
reddit = praw.Reddit(
client_id=os.getenv("REDDIT_CLIENT_ID"),
client_secret=os.getenv("REDDIT_CLIENT_SECRET"),
@ -37,9 +40,10 @@ def get_subreddit_threads():
)
try:
subreddit = reddit.subreddit(
input("What subreddit would you like to pull from? ")
)
if subreddit_ is None:
raise ValueError
subreddit = reddit.subreddit(subreddit_)
except ValueError:
subreddit = reddit.subreddit("askreddit")
print_substep("Subreddit not defined. Using AskReddit.")

Loading…
Cancel
Save