def __init__(self, output_dir, augmented_site_level_config, execution_id, component): self.component_props = get_lightweight_component( augmented_site_level_config, execution_id)['supplemental_config'][component] with open( '{output_dir}/supplemental_mapfile'.format( output_dir=output_dir), 'a') as mapfile: if component == 'htcondor': output_file = '{output_dir}/99_problems_condor.conf'.format( output_dir=output_dir) print('99_problems_condor.conf' + ':/etc/condor/config.d/99_problems.conf', file=mapfile) elif component == 'htcondor-ce': output_file = '{output_dir}/99_problems_condor_ce.conf'.format( output_dir=output_dir) print('99_problems_condor_ce.conf' + ':/etc/condor-ce/config.d/99_problems.conf', file=mapfile) else: output_file = '{output_dir}/{component}'.format( output_dir=output_dir, component=component.replace('/', '_')) print(component.replace('/', '_') + ':' + component, file=mapfile) ConfigFile.__init__(self, output_file, augmented_site_level_config, execution_id)
def __init__(self, output_file, augmented_site_level_config_file, execution_id): ConfigFile.__init__(self, output_file, augmented_site_level_config_file, execution_id)
def get_wn_list_file_categories(data): advanced = wn_list_advanced.get(data) return advanced def get_hosts_container_file_advanced_category(data, id): advanced = hosts_container_advanced.get(data, id) return advanced if __name__ == "__main__": args = parse_args() id = args['execution_id'] site_config_filename = args['augmented_site_level_config_file'] site_config = open(site_config_filename, 'r') data = yaml.load(site_config) output_dir = args['output_dir'] cream_info_file = ConfigFile(output_dir +'/cream-info.def', data) users_file = ConfigFile(output_dir + '/users.conf', data) groups_file = ConfigFile(output_dir + '/groups.conf', data) wn_list_file = ConfigFile(output_dir + '/wn-list.conf', data) edgusers_file = ConfigFile(output_dir + '/edgusers.conf', data) hosts_container_file = ConfigFile(output_dir + '/hosts-container.conf', data) cream_info_file.add_categories(get_cream_info_file_categories(data, id)) cream_info_file.generate_output_file() users_file.add_categories(get_users_conf_file_categories(data, id)) users_file.generate_output_file() groups_file.add_categories(get_groups_conf_file_categories(data, id)) groups_file.generate_output_file()
def get_wn_list_file_categories(data): advanced = wn_list_advanced.get(data) return advanced if __name__ == "__main__": args = parse_args() execution_id = args['execution_id'] augmented_site_level_config_file = args['augmented_site_level_config_file'] output_dir = args['output_dir'] site_config_filename = args['augmented_site_level_config_file'] site_config = open(site_config_filename, 'r') data = yaml.load(site_config) wn_info_file = ConfigFile(output_dir + '/wn-info.def', data) groups_conf_file = ConfigFile(output_dir + '/groups.conf', data) users_conf_file = ConfigFile(output_dir + '/users.conf', data) wn_list_file = ConfigFile(output_dir + '/wn-list.conf', data) wn_info_file.add_categories(get_wn_info_file_categories( data, execution_id)) groups_conf_file.add_categories( get_groups_conf_file_categories(data, execution_id)) users_conf_file.add_categories( get_users_conf_file_categories(data, execution_id)) wn_list_file.add_categories(get_wn_list_file_categories(data)) wn_info_file.generate_output_file() groups_conf_file.generate_output_file() users_conf_file.generate_output_file()