split subreddits and users config

Author: Tetrax-10

backup.py

@@ -5,7 +5,8 @@
 # ---------- CONFIG ----------
 
 # value: True | False
-BACKUP_SUBSCRIPTIONS = True
+BACKUP_SUBSCRIBED_SUBREDDITS = True
+BACKUP_FOLLOWED_USERS = True
 BACKUP_MULTIREDDITS = True
 BACKUP_SAVED_POSTS = True
 BACKUP_SAVED_COMMENTS = True
@@ -34,16 +35,20 @@
 
 
 # Backup subreddits
-if BACKUP_SUBSCRIPTIONS:
+if BACKUP_SUBSCRIBED_SUBREDDITS or BACKUP_FOLLOWED_USERS:
     for subreddit in reddit.user.subreddits(limit=None):
         subreddit_name = subreddit.display_name
         if subreddit_name.startswith("u_"):
-            backup["users"].append(subreddit_name)
+            if BACKUP_FOLLOWED_USERS:
+                backup["users"].append(subreddit_name)
         else:
-            backup["subreddits"].append(subreddit_name)
+            if BACKUP_SUBSCRIBED_SUBREDDITS:
+                backup["subreddits"].append(subreddit_name)
 
-    print(f"Backed up {len(backup['subreddits'])} subscribed subreddits.")
-    print(f"Backed up {len(backup['users'])} followed users.")
+    if BACKUP_SUBSCRIBED_SUBREDDITS:
+        print(f"Backed up {len(backup['subreddits'])} subscribed subreddits.")
+    if BACKUP_FOLLOWED_USERS:
+        print(f"Backed up {len(backup['users'])} followed users.")
 elif cache_backup:
     if "subreddits" in cache_backup:
         backup["subreddits"] = cache_backup["subreddits"]
@@ -76,6 +81,7 @@
     else:
         backup["multireddits"] = {}
 
+
 me = reddit.user.me()
 
 # Backup saved posts
@@ -87,6 +93,7 @@
         else:
             if BACKUP_SAVED_COMMENTS:
                 backup["saved_comments"].append(saved_item.id)
+
     if BACKUP_SAVED_POSTS:
         print(f"Backed up {len(backup['saved_posts'])} saved posts.")
     if BACKUP_SAVED_COMMENTS:

clear.py

@@ -4,7 +4,8 @@
 # ---------- CONFIG ----------
 
 # value: True | False
-CLEAR_SUBSCRIPTIONS = True
+CLEAR_SUBSCRIBED_SUBREDDITS = True
+CLEAR_FOLLOWED_USERS = True
 CLEAR_MULTIREDDITS = True
 CLEAR_SAVED_POSTS = True
 CLEAR_SAVED_COMMENTS = True
@@ -29,13 +30,23 @@
 print(f"Clearing {username}...\n")
 
 # Unsubscribe all subreddits and unfollow all users
-if CLEAR_SUBSCRIPTIONS:
+if CLEAR_SUBSCRIBED_SUBREDDITS or CLEAR_FOLLOWED_USERS:
     for subreddit in reddit.user.subreddits(limit=None):
+        subreddit_name = subreddit.display_name
         try:
-            subreddit.unsubscribe()
+            if subreddit_name.startswith("u_"):
+                if CLEAR_FOLLOWED_USERS:
+                    subreddit.unsubscribe()
+            else:
+                if CLEAR_SUBSCRIBED_SUBREDDITS:
+                    subreddit.unsubscribe()
         except:
             print("Can't unsubscribe subreddit", subreddit.display_name)
-    print("Unsubscribed all subreddits and unfollowed all users.")
+    if CLEAR_SUBSCRIBED_SUBREDDITS:
+        print("Unsubscribed all subreddits.")
+    if CLEAR_FOLLOWED_USERS:
+        print("Unfollowed all users.")
+
 
 # Delete all multireddits
 if CLEAR_MULTIREDDITS:
@@ -46,6 +57,7 @@
             print("Can't delete multireddit", multireddit.display_name)
     print("Deleted all multireddits.")
 
+
 me = reddit.user.me()
 
 # Unsave all saved posts

restore.py

@@ -5,7 +5,8 @@
 # ---------- CONFIG ----------
 
 # value: True | False
-RESTORE_SUBSCRIPTIONS = True
+RESTORE_SUBSCRIBED_SUBREDDITS = True
+RESTORE_FOLLOWED_USERS = True
 RESTORE_MULTIREDDITS = True
 RESTORE_SAVED_POSTS = True
 RESTORE_SAVED_COMMENTS = True
@@ -37,15 +38,22 @@
 
 
 # Subscribe to subreddits and follow users
-if RESTORE_SUBSCRIPTIONS and "subreddits" in backup and "users" in backup:
-    subreddit_to_subscribe = backup["subreddits"] + backup["users"]
+if (RESTORE_SUBSCRIBED_SUBREDDITS and "subreddits" in backup) or (RESTORE_FOLLOWED_USERS and "users" in backup):
+    subreddit_to_subscribe = []
+    if RESTORE_SUBSCRIBED_SUBREDDITS:
+        subreddit_to_subscribe += backup["subreddits"]
+    if RESTORE_FOLLOWED_USERS:
+        subreddit_to_subscribe += backup["users"]
 
     for subreddit in subreddit_to_subscribe:
         try:
             reddit.subreddit(subreddit).subscribe()
         except:
             print("Can't subscribe to", subreddit)
-    print("Restored subscribed subreddits and followed users.")
+    if RESTORE_SUBSCRIBED_SUBREDDITS:
+        print("Restored subscribed subreddits.")
+    if RESTORE_FOLLOWED_USERS:
+        print("Restored followed users.")
 
 
 # Create multireddits and add subreddits and users