From 85c35b829e369ef2579355d85d935766c38f2b3e Mon Sep 17 00:00:00 2001
From: George Kaklamanos <gkaklas@gkaklas.gr>
Date: Sat, 10 Jun 2023 18:38:25 +0300
Subject: [PATCH] Move all functionality in functions

---
 main.py | 119 +++++++++++++++++++++++++++++---------------------------
 1 file changed, 61 insertions(+), 58 deletions(-)

diff --git a/main.py b/main.py
index 25db85a..62ad097 100644
--- a/main.py
+++ b/main.py
@@ -15,70 +15,73 @@ reddit = praw.Reddit(
     user_agent = os.environ["PRAW_USER_AGENT"],
 )
 
-with open("subreddits.txt", "w") as file:
-    for subreddit in reddit.user.subreddits(limit=None):
-        file.write(str(subreddit) + "\n")
+def subreddits(reddit):
+    with open("subreddits.txt", "w") as file:
+        for subreddit in reddit.user.subreddits(limit=None):
+            file.write(str(subreddit) + "\n")
 
-with open("multireddits.yaml", "w") as file:
-    multireddits = {}
-    for multireddit in reddit.user.multireddits():
-        subreddits = []
-        for subreddit in multireddit.subreddits:
-            subreddits.append(str(subreddit))
-        multireddits[multireddit.name] = subreddits
-    yaml.dump(multireddits, file)
+def multireddits(reddit):
+    with open("multireddits.yaml", "w") as file:
+        multireddits = {}
+        for multireddit in reddit.user.multireddits():
+            subreddits = []
+            for subreddit in multireddit.subreddits:
+                subreddits.append(str(subreddit))
+            multireddits[multireddit.name] = subreddits
+        yaml.dump(multireddits, file)
 
-with open("saved.yaml", "w") as file:
-    saved = []
-    redditor = reddit.user.me()
-    for item in redditor.saved(limit=None):
-        i= {}
-        i["type"] = "comment"
-        if isinstance(item, praw.models.Submission):
-            i["type"] = "submission"
+def saved(reddit):
+    with open("saved.yaml", "w") as file:
+        saved = []
+        redditor = reddit.user.me()
+        for item in redditor.saved(limit=None):
+            i= {}
+            i["type"] = "comment"
+            if isinstance(item, praw.models.Submission):
+                i["type"] = "submission"
 
-            i["url"] = item.url
-            i["title"] = item.title
-            i["upvote_ratio"] = item.upvote_ratio
-            i["is_self"] = item.is_self
-            i["external_link"] = False
+                i["url"] = item.url
+                i["title"] = item.title
+                i["upvote_ratio"] = item.upvote_ratio
+                i["is_self"] = item.is_self
+                i["external_link"] = False
 
-            domains = ["redd.it","i.imgur.com","reddit.com"]
-            is_media = any(domain in item.url for domain in domains)
-            if not is_media and not item.is_self:
-                i["external_link"] = True
+                domains = ["redd.it","i.imgur.com","reddit.com"]
+                is_media = any(domain in item.url for domain in domains)
+                if not is_media and not item.is_self:
+                    i["external_link"] = True
 
-        i["id"] = item.id
-        i["created_utc"] = item.created_utc
-        i["permalink"] = item.permalink
-        i["score"] = item.score
-        saved.append(i)
-    yaml.dump(saved, file)
+            i["id"] = item.id
+            i["created_utc"] = item.created_utc
+            i["permalink"] = item.permalink
+            i["score"] = item.score
+            saved.append(i)
+        yaml.dump(saved, file)
 
-with open("redditor.yaml", "w") as file:
-    cs = []
-    redditor = reddit.user.me()
-    for comment in redditor.comments.new():
-        c= {}
-        c["type"] = "comment"
-        c["id"] = comment.id
-        c["created_utc"] = comment.created_utc
-        c["permalink"] = comment.permalink
-        c["score"] = comment.score
-        cs.append(c)
-    for submission in redditor.submissions.new():
-        s= {}
-        s["type"] = "submission"
-        s["id"] = submission.id
-        s["created_utc"] = submission.created_utc
-        s["permalink"] = submission.permalink
-        s["score"] = submission.score
+def saved(reddit):
+    with open("redditor.yaml", "w") as file:
+        cs = []
+        redditor = reddit.user.me()
+        for comment in redditor.comments.new():
+            c= {}
+            c["type"] = "comment"
+            c["id"] = comment.id
+            c["created_utc"] = comment.created_utc
+            c["permalink"] = comment.permalink
+            c["score"] = comment.score
+            cs.append(c)
+        for submission in redditor.submissions.new():
+            s= {}
+            s["type"] = "submission"
+            s["id"] = submission.id
+            s["created_utc"] = submission.created_utc
+            s["permalink"] = submission.permalink
+            s["score"] = submission.score
 
-        s["url"] = submission.url
-        s["title"] = submission.title
-        s["upvote_ratio"] = submission.upvote_ratio
+            s["url"] = submission.url
+            s["title"] = submission.title
+            s["upvote_ratio"] = submission.upvote_ratio
 
-        cs.append(s)
-
-    yaml.dump(cs, file)
+            cs.append(s)
+        yaml.dump(cs, file)