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

13
main.py
View file

@ -15,11 +15,13 @@ 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):
with open("subreddits.txt", "w") as file:
for subreddit in reddit.user.subreddits(limit=None): for subreddit in reddit.user.subreddits(limit=None):
file.write(str(subreddit) + "\n") file.write(str(subreddit) + "\n")
with open("multireddits.yaml", "w") as file: def multireddits(reddit):
with open("multireddits.yaml", "w") as file:
multireddits = {} multireddits = {}
for multireddit in reddit.user.multireddits(): for multireddit in reddit.user.multireddits():
subreddits = [] subreddits = []
@ -28,7 +30,8 @@ with open("multireddits.yaml", "w") as file:
multireddits[multireddit.name] = subreddits multireddits[multireddit.name] = subreddits
yaml.dump(multireddits, file) yaml.dump(multireddits, file)
with open("saved.yaml", "w") as file: def saved(reddit):
with open("saved.yaml", "w") as file:
saved = [] saved = []
redditor = reddit.user.me() redditor = reddit.user.me()
for item in redditor.saved(limit=None): for item in redditor.saved(limit=None):
@ -55,7 +58,8 @@ with open("saved.yaml", "w") as file:
saved.append(i) saved.append(i)
yaml.dump(saved, file) yaml.dump(saved, file)
with open("redditor.yaml", "w") as file: def saved(reddit):
with open("redditor.yaml", "w") as file:
cs = [] cs = []
redditor = reddit.user.me() redditor = reddit.user.me()
for comment in redditor.comments.new(): for comment in redditor.comments.new():
@ -79,6 +83,5 @@ with open("redditor.yaml", "w") as file:
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)