Move all functionality in functions

This commit is contained in:
George Kaklamanos 2023-06-10 18:38:25 +03:00
parent 149b39ff6c
commit 85c35b829e

119
main.py
View file

@ -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)