From 712b4114c7478c9144d4e67b1706786d16389e8f Mon Sep 17 00:00:00 2001 From: yx#5700 <107807189+o78@users.noreply.github.com> Date: Tue, 21 Jun 2022 23:50:18 +0200 Subject: [PATCH] multi subreddit support when entering multiple subreddits split by " " choosing random one. --- reddit/subreddit.py | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/reddit/subreddit.py b/reddit/subreddit.py index 7c5db91..2cfb71e 100644 --- a/reddit/subreddit.py +++ b/reddit/subreddit.py @@ -62,10 +62,23 @@ def get_subreddit_threads(): subreddit = reddit.subreddit("askreddit") print_substep("Subreddit not defined. Using AskReddit.") else: - print_substep(f"Using subreddit: r/{getenv('SUBREDDIT')} from environment variable config") - subreddit = reddit.subreddit( - getenv("SUBREDDIT") - ) # Allows you to specify in .env. Done for automation purposes. + subreddit1 = reddit.subreddit(getenv("SUBREDDIT")) + subreddits_list=subreddit1.split(" ") + + formatted_subreddits_list="" + for subreddit in subreddits_list: + formatted_subreddits_list=formatted_subreddits_list+f"r/{subreddit}\n" + + if len(subreddits_list)>2: # check if multiple subreddits + print_substep(f"Choosing random subreddit from below:", style="bold green") + print_substep(formatted_subreddits_list, style="bold green") + + randomsubreddit=random.choice(subreddits_list) + subreddit=reddit.subreddit(randomsubreddit) + print_substep(f"Chose r/{subreddit}.", style="bold yellow") + else: # single sunreddit + subreddit=reddit.subreddit(getenv("SUBREDDIT")) + print_substep(f"Using r/{subreddit}", style="bold yellow") if getenv("POST_ID"): submission = reddit.submission(id=getenv("POST_ID"))