diff --git a/.env.template b/.env.template index c598b49..68044bf 100644 --- a/.env.template +++ b/.env.template @@ -2,5 +2,5 @@ REDDIT_CLIENT_ID="" REDDIT_CLIENT_SECRET="" REDDIT_USERNAME="" REDDIT_PASSWORD="" -ASK_EACH_TIME=TRUE +ASK_EACH_TIME="TRUE" SUBREDDIT="" \ No newline at end of file diff --git a/reddit/subreddit.py b/reddit/subreddit.py index c88220a..53477d5 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -22,16 +22,16 @@ def get_subreddit_threads(): password=os.getenv("REDDIT_PASSWORD"), ) - if os.getenv("ASK_EACH_TIME") == TRUE: + if os.getenv("ASK_EACH_TIME") == "TRUE": try: subreddit = reddit.subreddit(input("What subreddit would you like to pull from? ")) - except NameError: + except ValueError: subreddit = reddit.subreddit("askreddit") print_substep("Subreddit not defined. Using AskReddit.") else: try: subreddit = reddit.subreddit(os.getenv("SUBREDDIT")) - except NameError: + except ValueError: subreddit = reddit.subreddit("askreddit") print_substep("Subreddit not defined. Using AskReddit.")