From 8ec61b8fe24b0af64f2e94be96b02becdb01d500 Mon Sep 17 00:00:00 2001
From: owengaspard <owen@ogaspard.com>
Date: Tue, 31 May 2022 13:40:44 -0500
Subject: [PATCH] Subreddit picker - .env

Alternatively, you can specify which subreddit to pull from in the .env file.
---
 .env.template       |  3 ++-
 reddit/subreddit.py | 10 +++++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/.env.template b/.env.template
index e00c242..e2c5798 100644
--- a/.env.template
+++ b/.env.template
@@ -1,4 +1,5 @@
 REDDIT_CLIENT_ID=""
 REDDIT_CLIENT_SECRET=""
 REDDIT_USERNAME=""
-REDDIT_PASSWORD=""
\ No newline at end of file
+REDDIT_PASSWORD=""
+SUBREDDIT=""
\ No newline at end of file
diff --git a/reddit/subreddit.py b/reddit/subreddit.py
index 33ef17b..6ddc661 100644
--- a/reddit/subreddit.py
+++ b/reddit/subreddit.py
@@ -10,7 +10,7 @@ def get_subreddit_threads():
     Returns a list of threads from the AskReddit subreddit.
     """
 
-    print_step("Getting AskReddit threads...")
+    print_step("Getting subreddit threads...")
 
     content = {}
     load_dotenv()
@@ -21,7 +21,15 @@ def get_subreddit_threads():
         username=os.getenv("REDDIT_USERNAME"),
         password=os.getenv("REDDIT_PASSWORD"),
     )
+    """
+    Ask user for subreddit input
+    """
     subreddit = reddit.subreddit(input("What subreddit would you like to pull from? "))
+    
+    """
+    Allow you to specify in .env. Done for automation purposes.
+    subreddit = reddit.subreddit(os.getenv("SUBREDDIT"))
+    """
     threads = subreddit.hot(limit=25)
     submission = list(threads)[random.randrange(0, 25)]
     print_substep(f"Video will be: {submission.title} :thumbsup:")