Rename main.py to reddit.py for clarity.

master
jowj 5 years ago
parent 97e44e5874
commit 5384007fcc

@ -1,7 +1,7 @@
import praw
import json import json
import os import os
import pdb import praw
def munge_idiot_data(reddit_dict): def munge_idiot_data(reddit_dict):
""" """
@ -22,26 +22,37 @@ def munge_idiot_data(reddit_dict):
if __name__ == "__main__": if __name__ == "__main__":
reddit = praw.Reddit(client_id=os.environ.get('REDDIT_ID'), REDDIT = praw.Reddit(client_id=os.environ.get('REDDIT_ID'),
client_secret=os.environ.get('REDDIT_SECRET'), client_secret=os.environ.get('REDDIT_SECRET'),
user_agent='/u/ pynit-tasks', user_agent='/u/ pynit-tasks',
username=os.environ.get('REDDIT_UN'), username=os.environ.get('REDDIT_UN'),
password=os.environ.get('REDDIT_PW') password=os.environ.get('REDDIT_PW')
) )
your_user = reddit.redditor(os.environ.get('REDDIT_UN')) # this line is the most cursed line in programming
saved_posts = your_user.saved(limit=None) # REDDIT.redditor,
YOUR_USER = REDDIT.redditor(os.environ.get('REDDIT_UN'))
SAVED_POSTS = YOUR_USER.saved(limit=None)
posts_to_save = [] POSTS_TO_SAVE = []
for link in saved_posts: for link in SAVED_POSTS:
if hasattr(link, 'is_self'): if hasattr(link, 'is_self'):
posts_to_save.append({'title':link.title, 'tag':link.subreddit.display_name + ' added-by-pynnit', 'description':link.selftext, 'url':link.permalink}) POSTS_TO_SAVE.append({
'title': link.title,
'tag': link.subreddit.display_name + ' added-by-pynnit',
'description': link.selftext,
'url': link.permalink
})
elif hasattr(link, 'is_root'): elif hasattr(link, 'is_root'):
posts_to_save.append({'title':link.link_title, 'tag':link.subreddit.display_name + ' added-by-pynnit', 'description':link.body, 'url':link.link_url}) POSTS_TO_SAVE.append({
'title': link.link_title,
'tag': link.subreddit.display_name + ' added-by-pynnit',
'description': link.body,
'url': link.link_url
})
else: else:
print("shit is fucked.") print("shit is fucked.")
munged_data = munge_idiot_data(posts_to_save) MUNGED_DATA = munge_idiot_data(POSTS_TO_SAVE)
with open('data.json', 'w') as outfile: with open('data.json', 'w') as outfile:
json.dump(munged_data, outfile, indent=2) json.dump(MUNGED_DATA, outfile, indent=2)
Loading…
Cancel
Save