From 2d884618721a3659d5c04de786720e7758e09758 Mon Sep 17 00:00:00 2001
From: George Kaklamanos <gkaklas@gkaklas.gr>
Date: Sat, 10 Jun 2023 19:15:49 +0300
Subject: [PATCH] Call main() logic correctly

---
 main.py | 53 +++++++++++++++++++++++++++--------------------------
 1 file changed, 27 insertions(+), 26 deletions(-)

diff --git a/main.py b/main.py
index d5c44b5..b37223d 100644
--- a/main.py
+++ b/main.py
@@ -6,32 +6,6 @@ import os
 import yaml
 import argparse
 
-load_dotenv()
-
-reddit = praw.Reddit(
-    username = os.environ["PRAW_USERNAME"],
-    password = os.environ["PRAW_PASSWORD"],
-    client_id = os.environ["PRAW_CLIENT_ID"],
-    client_secret = os.environ["PRAW_CLIENT_SECRET"],
-    user_agent = os.environ["PRAW_USER_AGENT"],
-)
-
-parser = argparse.ArgumentParser()
-parser.add_argument('--subreddits', action="store_true", help="Save subcribed subreddits in subreddits.txt")
-parser.add_argument('--multireddits', action="store_true", help="Save user's multireddits in multireddits.yaml")
-parser.add_argument('--saved', action="store_true", help="Save saved comments and submissions in saved.yaml")
-parser.add_argument('--redditor', action="store_true", help="Save redditor's submissions and comments in redditor.yaml")
-
-args = parser.parse_args()
-if args.subreddits:
-    subreddits(reddit)
-if args.multireddits:
-    multireddits(reddit)
-if args.saved:
-    saved(reddit)
-if args.redditor:
-    redditor(reddit)
-
 def subreddits(reddit):
     with open("subreddits.txt", "w") as file:
         for subreddit in reddit.user.subreddits(limit=None):
@@ -102,3 +76,30 @@ def redditor(reddit):
             cs.append(s)
         yaml.dump(cs, file)
 
+if __name__ == '__main__':
+    load_dotenv()
+
+    reddit = praw.Reddit(
+        username = os.environ["PRAW_USERNAME"],
+        password = os.environ["PRAW_PASSWORD"],
+        client_id = os.environ["PRAW_CLIENT_ID"],
+        client_secret = os.environ["PRAW_CLIENT_SECRET"],
+        user_agent = os.environ["PRAW_USER_AGENT"],
+    )
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--subreddits', action="store_true", help="Save subcribed subreddits in subreddits.txt")
+    parser.add_argument('--multireddits', action="store_true", help="Save user's multireddits in multireddits.yaml")
+    parser.add_argument('--saved', action="store_true", help="Save saved comments and submissions in saved.yaml")
+    parser.add_argument('--redditor', action="store_true", help="Save redditor's submissions and comments in redditor.yaml")
+
+    args = parser.parse_args()
+    if args.subreddits:
+        subreddits(reddit)
+    if args.multireddits:
+        multireddits(reddit)
+    if args.saved:
+        saved(reddit)
+    if args.redditor:
+        redditor(reddit)
+