config.py

import pathlib
import json
import logging


class ConfigTool:
    def folder(self, folder: str) -> None:
        dn = pathlib.Path(folder)
        if not dn.is_dir():
            try:
                dn.mkdir(parents=True, exist_ok=True)
            except IOError:
                exit(f"It looks like {folder} is not a valid folder path")
        d = {"dataFolder": folder}
        logging.info(f"Data folder set: '{dn}'")
        LOCAL_CONFIG_FP.write_text(json.dumps(d, indent=4))

    def auth(self, id: str, secret: str, password: str, agent: str, name: str) -> None:
        d = json.loads(LOCAL_CONFIG_FP.read_text())
        d["auth"] = {
            "clientId": id,
            "clientSecret": secret,
            "password": password,
            "userAgent": agent,
            "userName": name,
        }
        logging.info(f"Reddit bot auth data saved")
        LOCAL_CONFIG_FP.write_text(json.dumps(d, indent=4))


LOCAL_CONFIG_FP = pathlib.Path(__file__).parent.resolve() / "local_config.json"
if LOCAL_CONFIG_FP.is_file():
    d = json.loads(LOCAL_CONFIG_FP.read_text())
    DATA_DIR = pathlib.Path(d["dataFolder"])
else:
    default_dir = pathlib.Path.home() / "thisdayinreddit"
    d = {"dataFolder": str(default_dir)}
    logging.info(f"Data folder set: '{default_dir}'")
    LOCAL_CONFIG_FP.write_text(json.dumps(d, indent=4))
    DATA_DIR = default_dir

for sub_dn in ["compressed", "extracted"]:
    dn = DATA_DIR / sub_dn
    dn.mkdir(parents=True, exist_ok=True)