theme.park/themes.py

from os import listdir
from os.path import isdir, isfile, join
from json import dump,dumps, loads
import subprocess

def get_shas(output):
    """Returns a dict of CSS files and SHAs"""
    output_lines = output.splitlines()
    sha_dict = {}
    for line in output_lines:
        line = line.decode('utf-8').replace("\t","").split(" ")
        sha = line[1]
        css_file = [file for file in line[2].split("/") if "css" in file][0]
        sha_dict.update({css_file: sha})
    return(sha_dict)

def create_addons_json():
    addon_shas = subprocess.check_output(["git", "ls-files", "-s", "./CSS/addons/*.css"])
    SHAS = get_shas(addon_shas)
    ADDONS = {"addons":{}}
    addon_root = './CSS/addons'
    addon_folders = [name for name in listdir(addon_root) if isdir(join(addon_root, name))]
    for app in addon_folders:
        app_addons = [addon for addon in listdir(f"{addon_root}/{app}")]
        ADDONS["addons"].update({
                app: {
                    addon: {} for addon in app_addons
                }
        })
        for addon in app_addons:
            files = [file for file in listdir(f"{addon_root}/{app}/{addon}") if isfile(join(f"{addon_root}/{app}/{addon}", file))]
            if len([f for f in files if f.endswith('.css')]) > 1:
                ADDONS["addons"][app][addon].update({
                        "css":  [f"https://{DOMAIN}/CSS/addons/{app}/{addon}/{file}?sha={SHAS.get(file)}" for file in files if file.split(".")[1] == "css"]
                            }
                )
            else:
                ADDONS["addons"][app].update({
                        addon:  f"https://{DOMAIN}/CSS/addons/{app}/{addon}/{file}?sha={SHAS.get(file)}" for file in files if file.split(".")[1] == "css" 
                            } 
                )
            extra_dirs = [dir for dir in listdir(
                f"{addon_root}/{app}/{addon}") if isdir(join(f"{addon_root}/{app}/{addon}", dir))]
            if extra_dirs:
                for dir in extra_dirs:
                    extra_dir_files = [file for file in listdir(
                        f"{addon_root}/{app}/{addon}/{dir}") if isfile(join(f"{addon_root}/{app}/{addon}/{dir}", file))]
                    ADDONS["addons"][app][addon].update({
                                dir: {
                                        "css": [f"https://{DOMAIN}/CSS/addons/{app}/{addon}/{dir}/{extra_file}?sha={SHAS.get(extra_file)}" for extra_file in extra_dir_files if extra_file.split(".")[1] == "css"]
                                    }
                                }
                            )
    return dumps(ADDONS)

def create_json(app_folders:list=None,themes:list=None,no_sub_folders=False):
    if no_sub_folders:
        THEMES = {}
        theme_shas = subprocess.check_output(["git", "ls-files", "-s", "./CSS/variables/*.css"])
        SHAS = get_shas(theme_shas)
        for theme in themes:
            THEMES.update({
                "themes": {
                    theme.split(".")[0].capitalize(): {
                        "url": f"https://{DOMAIN}/CSS/variables/{theme}?sha={SHAS.get(theme)}" 
                    }for theme in themes
                }
            })
        return dumps(THEMES)
    else:
        ADDONS = loads(create_addons_json())
        APPS = {"applications":{}}
        app_shas = subprocess.check_output(["git", "ls-files", "-s", "./CSS/themes/*base.css"])
        SHAS = get_shas(app_shas)
        for app in app_folders:
            APPS.update({
                "applications":{
                    app: {
                        "base_css": f"https://{DOMAIN}/CSS/themes/{app}/{app}-base.css?sha={SHAS.get(f'{app}-base.css')}",
                        "addons": ADDONS["addons"][app] if app in ADDONS["addons"] else {}
                        } for app in app_folders
                    }
                }
            )
        THEMES = loads(create_json(themes=themes,no_sub_folders=True))
        APPS.update(ADDONS)
        APPS.update(THEMES)
        return dumps(APPS)

if __name__== "__main__":
    app_folders = [name for name in listdir('./CSS/themes') if isdir(join('./CSS/themes', name))]
    themes = [name for name in listdir('./CSS/variables') if isfile(join('./CSS/variables', name))]
    DOMAIN = open("CNAME","rt",closefd=True).readline()
    apps = loads(create_json(app_folders,themes))

    with open("themes.json", "w") as outfile:
        dump(apps, outfile,indent=2)