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()