Homec4science
Diffusion dcsr-spack-site (master) Browse

dcsr-spack-site/deploy-env.py
65202a41e414master

Last Change
Raw File

deploy-env.py
View Options

#!/usr/bin/env python3
# This script creates multiple spack environment files from a template
#
# Input sources:
# - a Jinja2 template (spack.yaml.j2)
# - a YAML file containing (meleze.yaml)
# - the environments to create
# - variables to be written per environment
#
# This file and the input live in the spack-site repo
#
import jinja2
import yaml
import datetime
# Variables for paths and the like - to be moved elsewhere!
spack_root="/dcsrsoft/spack/"
spack_release="meleze"
spack_release_version="v2.1"
spack_environment_root=spack_root + spack_release + "/" + spack_release_version + "/var/spack/environments/"
spack_env = jinja2.Environment(loader=jinja2.FileSystemLoader('./'), trim_blocks=True, lstrip_blocks=True)
spack_env_template = spack_env.get_template('spack.yaml.j2')
config = yaml.load(open('./meleze.yaml'), Loader=yaml.FullLoader)
# Create a log message to write to each file
info_message="This file was created by magic at " + datetime.datetime.now().strftime("%d/%m/%Y %H:%M:%S")
# Iterate over the config file for each environment name
for spack_environment in config["envname"]:
spack_yaml_root = spack_environment_root + spack_environment
print("Creating evironment " + spack_environment + " in " + spack_yaml_root)
# create a dictionary for each environment
customisation = {}
customisation["envname"] = spack_environment
customisation["info_message"] = info_message
for k in config.keys():
if k != 'envname':
customisation[k]=config[k].get(spack_environment, config[k]["default"])
with open(spack_yaml_root+"/spack.yaml","w+") as f:
f.write(spack_env_template.render(customisation))
print("Bye and don't forget to run spack install for each environment!")
c4science · Help