diff --git a/.env.template b/.env.template index 68044bf..dab64d3 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" + SUBREDDIT="" \ No newline at end of file diff --git a/reddit/subreddit.py b/reddit/subreddit.py index 5814697..27e21c2 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -22,16 +22,14 @@ def get_subreddit_threads(): 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.") + if os.getenv("SUBREDDIT"): + subreddit = reddit.subreddit(os.getenv("SUBREDDIT")) else: + # ! Prompt the user to enter a subreddit try: - subreddit = reddit.subreddit(os.getenv("SUBREDDIT")) - print_substep(f"Using subreddit {os.getenv('SUBREDDIT')}") + 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.")