From 33a66b0ae3d0c96cd5eb94f4c18db739f9356c4e Mon Sep 17 00:00:00 2001 From: Jason Date: Fri, 1 Jul 2022 16:05:53 -0400 Subject: [PATCH] feat: allows the subreddit choice to begin with r/ --- reddit/subreddit.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/reddit/subreddit.py b/reddit/subreddit.py index eeae977..3419fbf 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -57,8 +57,11 @@ def get_subreddit_threads(): print_substep( f"Using subreddit: r/{getenv('SUBREDDIT')} from environment variable config" ) + subreddit_choice = getenv("SUBREDDIT") + if subreddit_choice.casefold().startswith('r/'): # removes the r/ from the input + subreddit_choice = subreddit_choice[2:] subreddit = reddit.subreddit( - getenv("SUBREDDIT") + subreddit_choice ) # Allows you to specify in .env. Done for automation purposes. if getenv("POST_ID"):