From 554f1582f6cd4e494e455deda6af4ec836ff9dc8 Mon Sep 17 00:00:00 2001
From: George Kaklamanos <gkaklas@gkaklas.gr>
Date: Sat, 10 Jun 2023 18:46:10 +0300
Subject: [PATCH] Call functions according to provided cli arguments

---
 main.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/main.py b/main.py
index 62ad097..2b72cdf 100644
--- a/main.py
+++ b/main.py
@@ -4,6 +4,7 @@ import praw
 from dotenv import load_dotenv
 import os
 import yaml
+import argparse
 
 load_dotenv()
 
@@ -15,6 +16,22 @@ reddit = praw.Reddit(
     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):