def render_configuration_file(self, conf_template_string, params):
"""
Hacky override to make pillowtop config. Multiple configs within the conf file
"""
pillow_env_configs = params['pillow_env_configs']
configs = []
all_pillows = get_pillows_for_env(pillow_env_configs,
settings.PILLOWTOPS)
for pillow_config in all_pillows:
pillow_name = pillow_config.name
num_processes = pillow_config.params.get('num_processes', 1)
pillow_params = {
'pillow_name': pillow_name,
'pillow_option': ' --pillow-name %s' % pillow_name,
'num_processes': num_processes
}
pillow_params.update(params)
for process_number in range(num_processes):
pillow_params.update({'process_num': process_number})
pillow_rendering = super(Command,
self).render_configuration_file(
conf_template_string,
pillow_params)
configs.append(pillow_rendering)
return '\n\n'.join(configs)
def test_india_server_exclusions(self):
self.pillowtops['fluff'] = [
'custom.bihar.models.CareBiharFluffPillow',
'custom.opm.models.OpmUserFluffPillow',
]
configs = [
{'exclude_groups': ['mvp_indicators', 'fluff']},
{'include_pillows': ['custom.bihar.models.CareBiharFluffPillow']},
]
pillows = [c.name for c in get_pillows_for_env(configs, self.pillowtops)]
self.assertNotIn('OpmUserFluffPillow', pillows)
self.assertIn('CareBiharFluffPillow', pillows)
def render_configuration_file(self, conf_template_string, params):
"""
Hacky override to make pillowtop config. Multiple configs within the conf file
"""
pillow_env_configs = params['pillow_env_configs']
configs = []
all_pillows = get_pillows_for_env(pillow_env_configs, settings.PILLOWTOPS)
for pillow_config in all_pillows:
pillow_name = pillow_config.name
pillow_params = {
'pillow_name': pillow_name,
'pillow_option': ' --pillow-name %s' % pillow_name
}
pillow_params.update(params)
pillow_rendering = super(Command, self).render_configuration_file(conf_template_string, pillow_params)
configs.append(pillow_rendering)
return '\n\n'.join(configs)
def test_india_server_exclusions(self):
self.pillowtops['fluff'] = [
'custom.bihar.models.CareBiharFluffPillow',
'custom.opm.models.OpmUserFluffPillow',
]
configs = [
{
'exclude_groups': ['mvp_indicators', 'fluff']
},
{
'include_pillows':
['custom.bihar.models.CareBiharFluffPillow']
},
]
pillows = [
c.name for c in get_pillows_for_env(configs, self.pillowtops)
]
self.assertNotIn('OpmUserFluffPillow', pillows)
self.assertIn('CareBiharFluffPillow', pillows)
def render_configuration_file(self, conf_template_string, params):
"""
Hacky override to make pillowtop config. Multiple configs within the conf file
"""
pillow_env_configs = params['pillow_env_configs']
configs = []
all_pillows = get_pillows_for_env(pillow_env_configs, settings.PILLOWTOPS)
for pillow_config in all_pillows:
pillow_name = pillow_config.name
num_processes = pillow_config.params.get('num_processes', 1)
total_processes = pillow_config.params.get('total_processes', num_processes)
start_process = pillow_config.params.get('start_process', 0)
pillow_params = {
'pillow_name': pillow_name,
'pillow_option': ' --pillow-name %s' % pillow_name,
'total_processes': total_processes
}
pillow_params.update(params)
for process_number in range(num_processes):
pillow_params.update({'process_num': process_number + start_process})
pillow_rendering = super(Command, self).render_configuration_file(conf_template_string, pillow_params)
configs.append(pillow_rendering)
return '\n\n'.join(configs)
