reddit-save/save.py

#!/usr/bin/env python 

import argparse
import os
from tqdm import tqdm
from utilities import *

# Get arguments
parser = argparse.ArgumentParser(description="Save reddit posts to file.")
parser.add_argument("mode", type=str, nargs=1, choices=["saved", "upvoted"], help="The file to convert.")
parser.add_argument("location", type=str, nargs=1, help="The path to save to.")
args = parser.parse_args()
mode = args.mode[0]
location = args.location[0]

# Is location specified a directory?
if not os.path.isdir(location):
    print(location, "is not a directory")

# Make a client object
client = make_client()

# Saved posts or upvoted posts?
if mode == "saved":
    html_file = "saved.html"
    get_posts = get_saved_posts
else:
    html_file = "upvoted.html"
    get_posts = get_upvoted_posts

# Make directory for media
if not os.path.exists(os.path.join(location, "media")):
    os.mkdir(os.path.join(location, "media"))

posts_html = []

posts = get_posts(client)
for post in tqdm(posts):
    post_html = get_post_html(post)
    media = save_media(post, location)
    if media:
        post_html = add_media_preview_to_html(post_html, media)
    posts_html.append(post_html)

with open(os.path.join("html", html_file)) as f:
    html = f.read()

with open(os.path.join("html", "style.css")) as f:
    html = html.replace("<style></style>", f"<style>\n{f.read()}\n</style>")

html = html.replace("<!--posts-->", "\n".join(posts_html))

with open(os.path.join(location, html_file), "w") as f:
    f.write(html)