gkaklas/reddit_export_data
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move all functionality in functions

Browse source
This commit is contained in:
George Kaklamanos 2023-06-10 18:38:25 +03:00
parent 149b39ff6c
commit 85c35b829e
1 changed files with 61 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

119
main.py
View file

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