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')]