base16-to-lua.py

import os
import yaml

input_folder = "base16/base16-schemes"
output_folder = "base16/base16-lua"
lua_header = "-- This file was automatically generated from a YAML file.\n\n"

if not os.path.isdir(input_folder):
    os.system(f"git clone https://github.com/tinted-theming/base16-schemes {input_folder}")

if not os.path.exists(output_folder):
    os.makedirs(output_folder)

for filename in os.listdir(input_folder):
    if filename.endswith(".yaml"):
        with open(os.path.join(input_folder, filename), "r") as f:
            data = yaml.safe_load(f)

        if isinstance(data, dict):
            lua_table = lua_header
            lua_table += "return {\n"
            for key, value in data.items():
                lua_table += f"\t{key} = \"{value}\",\n"
            lua_table += "}\n"

            with open(os.path.join(output_folder, filename[:-5] + ".lua"), "w") as f:
                f.write(lua_table)
        else:
            print(f"Error: {filename} is not a valid YAML file.")

print("Lua conversion successful!")