def load_sites(arg):
''' Check that the sites file exists, and return its contents. '''
# Check that the file exists
path = utils.path_exists(arg)
with open(path, 'r') as sites_file:
sites = sites_file.readlines()
return sites
def load_specs(arg):
''' Check to make sure arg file exists. Return its contents. '''
spec_file = utils.path_exists(arg)
with open(spec_file, 'r') as fn:
specs = yaml.load(fn, Loader=yaml.Loader)
return specs
def load_images(arg):
''' Check that input image file exists, and that it contains the
requested section. Return a 2-list (required by argparse) of the
file path and dictionary of images to be created.
'''
# Agument is expected to be a 2-list of file name and internal
# section name.
image_file = arg[0]
image_set = arg[1]
# Check that the file exists
image_file = utils.path_exists(image_file)
# Load yaml file
with open(image_file, 'r') as fn:
images = yaml.load(fn, Loader=yaml.Loader)[image_set]
return [images.get('model'), images.get('variables')]