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

Call main() logic correctly

Browse source
This commit is contained in:
George Kaklamanos 2023-06-10 19:15:49 +03:00
parent 9cdccd044e
commit 2d88461872
1 changed files with 27 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

53
main.py
View file

@ -6,32 +6,6 @@ import os
import yaml import yaml
import argparse import argparse
load_dotenv()
reddit = praw.Reddit(
username = os.environ["PRAW_USERNAME"],
password = os.environ["PRAW_PASSWORD"],
client_id = os.environ["PRAW_CLIENT_ID"],
client_secret = os.environ["PRAW_CLIENT_SECRET"],
user_agent = os.environ["PRAW_USER_AGENT"],
)
parser = argparse.ArgumentParser()
parser.add_argument('--subreddits', action="store_true", help="Save subcribed subreddits in subreddits.txt")
parser.add_argument('--multireddits', action="store_true", help="Save user's multireddits in multireddits.yaml")
parser.add_argument('--saved', action="store_true", help="Save saved comments and submissions in saved.yaml")
parser.add_argument('--redditor', action="store_true", help="Save redditor's submissions and comments in redditor.yaml")
args = parser.parse_args()
if args.subreddits:
subreddits(reddit)
if args.multireddits:
multireddits(reddit)
if args.saved:
saved(reddit)
if args.redditor:
redditor(reddit)
def subreddits(reddit): def subreddits(reddit):
with open("subreddits.txt", "w") as file: with open("subreddits.txt", "w") as file:
for subreddit in reddit.user.subreddits(limit=None): for subreddit in reddit.user.subreddits(limit=None):
@ -102,3 +76,30 @@ def redditor(reddit):
cs.append(s) cs.append(s)
yaml.dump(cs, file) yaml.dump(cs, file)
if __name__ == '__main__':
load_dotenv()
reddit = praw.Reddit(
username = os.environ["PRAW_USERNAME"],
password = os.environ["PRAW_PASSWORD"],
client_id = os.environ["PRAW_CLIENT_ID"],
client_secret = os.environ["PRAW_CLIENT_SECRET"],
user_agent = os.environ["PRAW_USER_AGENT"],
)
parser = argparse.ArgumentParser()
parser.add_argument('--subreddits', action="store_true", help="Save subcribed subreddits in subreddits.txt")
parser.add_argument('--multireddits', action="store_true", help="Save user's multireddits in multireddits.yaml")
parser.add_argument('--saved', action="store_true", help="Save saved comments and submissions in saved.yaml")
parser.add_argument('--redditor', action="store_true", help="Save redditor's submissions and comments in redditor.yaml")
args = parser.parse_args()
if args.subreddits:
subreddits(reddit)
if args.multireddits:
multireddits(reddit)
if args.saved:
saved(reddit)
if args.redditor:
redditor(reddit)