mirror of
https://github.com/sys-nyx/red-arch.git
synced 2025-05-05 16:25:31 -04:00
added dumps.py script which will take open json files containing subreddit content, organize them in a way expected by the generate_html func in write_html.py, and then call it
This commit is contained in:
parent
9cdd43363b
commit
874bd8ee06
1 changed files with 76 additions and 0 deletions
76
dumps.py
Normal file
76
dumps.py
Normal file
|
@ -0,0 +1,76 @@
|
|||
|
||||
|
||||
import os
|
||||
import json
|
||||
import argparse
|
||||
import configparser
|
||||
from urllib.parse import urlparse
|
||||
from write_html import generate_html
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('config', type=str, help='Path to configuration file.')
|
||||
args = parser.parse_args()
|
||||
if not args.config:
|
||||
print("No config file found")
|
||||
exit()
|
||||
|
||||
config = configparser.ConfigParser()
|
||||
config.read(args.config)
|
||||
|
||||
subreddits = {}
|
||||
|
||||
for s in config.sections():
|
||||
|
||||
posts_path = config[s]['posts']
|
||||
comments_path = config[s]['comments']
|
||||
|
||||
links = []
|
||||
|
||||
with open(posts_path, 'r') as f:
|
||||
print("loading posts")
|
||||
raw_posts = json.loads(f.read())
|
||||
print('done')
|
||||
with open(comments_path, 'r') as f:
|
||||
print('loading comments')
|
||||
raw_comments = json.loads(f.read())
|
||||
print('done')
|
||||
|
||||
missing_perm = []
|
||||
comments = {}
|
||||
for c in raw_comments:
|
||||
if 'permalink' not in c.keys():
|
||||
missing_perm.append(c)
|
||||
continue
|
||||
|
||||
parent_url = '/'.join(urlparse(c['permalink']).path.split('/')[:6])
|
||||
if parent_url.endswith('/'):
|
||||
parent_url = parent_url[:-1]
|
||||
if parent_url not in comments.keys():
|
||||
comments[parent_url] = []
|
||||
comments[parent_url].append(c)
|
||||
|
||||
|
||||
complete_threads = []
|
||||
|
||||
for p in raw_posts:
|
||||
p['comments'] = []
|
||||
postp = urlparse(p['permalink']).path
|
||||
if postp.endswith('/'):
|
||||
postp = postp[:-1]
|
||||
if postp in comments.keys():
|
||||
p['comments'] = comments[postp]
|
||||
|
||||
complete_threads.append(p)
|
||||
subreddits[s.lower()] = complete_threads
|
||||
print("Total threads: ",len(raw_posts))
|
||||
print("Total comments: ", len(raw_comments))
|
||||
print("Comments missing permalinks: ", len(missing_perm))
|
||||
print("Comment chains found: ", len(comments))
|
||||
print("Threads rebuilt: ", len(complete_threads))
|
||||
|
||||
|
||||
generate_html([s.lower() for s in config.sections()], subreddits)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Add table
Add a link
Reference in a new issue