use regex to replace r/, that it would not cause errors

3 years ago · f75b59d5a2
parent 016349ea21
commit f75b59d5a2
1 changed files with 3 additions and 1 deletions
--- a/reddit/subreddit.py
+++ b/reddit/subreddit.py
@ -75,7 +75,9 @@ def get_subreddit_threads(subreddit_, thread_link_, number_of_comments):
                if subreddit_ is None:
                    raise ValueError

-                subreddit = reddit.subreddit(subreddit_)
+                subreddit = reddit.subreddit(
+                    re.sub(r"r\/", "", subreddit_.strip())
+                )
            except ValueError:
                if os.getenv("SUBREDDIT"):
                    subreddit = reddit.subreddit(