build.py

# ZBUILD

import os
import json
import sys
import subprocess
from staticjinja import Site
from pathlib import Path

OUTPUT_PREFIX = "./dist"
RSYNC_COMMAND = "rsync -av {0} {1} {2}"

#  watchman-make -p 'templates/**' --run 'python3 build.py template'
#  watchman-make -p 'static/**' --run 'python3 build.py static'
def unpack_submodules():
    # get all submodules
    modules = [x for x in subprocess.run("grep path .gitmodules | sed 's/.*= //'", shell=True, stdout=subprocess.PIPE).stdout.decode("utf-8").split("\n") if x]
    for path in modules:
        # for every submodule copy it into OUTPUT_PREFIX, excluding exclude key in json
        with open(path + "/.build") as f:
            data = json.load(f)
            os.system(RSYNC_COMMAND.format(' '.join([f'--exclude=\'{e}\'' for e in data['exclude']]), path, OUTPUT_PREFIX + "/" + data["to"]))

def template():
    # Load the data directory into env
    data = {}
    for file in os.listdir("data"):
        path = os.path.join("data", file)
        if os.path.isfile(path):
            with open(path) as f:
                data[Path(path).stem] = json.load(f)

    site = Site.make_site(outpath=OUTPUT_PREFIX, env_globals=data)
    site.render(use_reloader=False)

def static():
    os.system(f"cp -r static {OUTPUT_PREFIX}")
    os.system(f"mv {OUTPUT_PREFIX}/static/favicon.ico {OUTPUT_PREFIX}")

if __name__ == "__main__":
    
    print("Executing build script")

    # create directory dist
    # assert submodules
    os.system(f"mkdir -p {OUTPUT_PREFIX}")

    if "template" in sys.argv:
        print("Templating")
        template()
    elif "static" in sys.argv:
        print("Copying static")
        static()

    elif "production" in sys.argv:
        print("Production")
        template()
        static()

        print("Updating submodules")
        os.system("git submodule update --init --recursive")
        os.system("git submodule update --remote --merge")
        print("Submodules updated")

        unpack_submodules()

    else:
        raise Exception("Invalid option")