dotfiles/vscode/generate_install_scripts.py

66 lines
1.8 KiB
Python
Raw Normal View History

2021-12-08 09:38:24 -06:00
import yaml
from pathlib import Path
FRONTMATTER = """
VS Code Extensions install script
Author: Daylin Morgan
DO NOT EDIT MANUALLY!
Regenerate with generate_install_scripts.py
"""
FILE_DIR = Path(__file__).parent
def shell_script_writer(extensions_dict, filename):
with Path(filename).open("w") as f:
f.write("#!/bin/bash\n")
f.write(FRONTMATTER.replace("\n", "\n# ")[:-2])
f.write('\n\necho "Installing VS Code Extensions"\n')
for section, extensions in extensions_dict.items():
f.write(f"\n# {section} extensions")
f.write("\n# " + "-" * len(section) + "-" * 11)
for extension in extensions:
f.write(f"\ncode --install-extension {extension} --force")
f.write("\n")
f.write('\necho "Extension installation compeleted!"')
def batch_script_writer(extensions_dict, filename):
with Path(filename).open("w") as f:
f.write("echo off\n")
f.write(FRONTMATTER.replace("\n", "\n:: ")[:-3])
f.write('\n\necho "Installing VS Code Extensions"\necho.\n')
for section, extensions in extensions_dict.items():
f.write(f"\n:: {section} extensions")
f.write("\n:: " + "-" * len(section) + "-" * 11)
for extension in extensions:
f.write(f"\ncall code --install-extension {extension} --force")
f.write("\n")
f.write('\necho "Extension installation compeleted!"')
def main():
print("Generating shell and batch scripts with extensions.yml")
extensions_yml = FILE_DIR / "extensions.yml"
with extensions_yml.open("r") as f:
extensions_dict = yaml.load(f, Loader=yaml.FullLoader)
shell_script_writer(extensions_dict, "install-extensions.sh")
batch_script_writer(extensions_dict, "install-extensions.bat")
if __name__ == "__main__":
main()