def _process_non_multi_part(self, user_data):
if user_data.startswith(b'#cloud-config'):
user_data_plugins = factory.load_plugins()
cloud_config_plugin = user_data_plugins.get('text/cloud-config')
ret_val = cloud_config_plugin.process(user_data)
else:
ret_val = userdatautils.execute_user_data_script(user_data)
return self._get_plugin_return_value(ret_val)
def _process_non_multi_part(self, user_data):
if user_data.startswith('#cloud-config'):
user_data_plugins = factory.load_plugins()
cloud_config_plugin = user_data_plugins.get('text/cloud-config')
ret_val = cloud_config_plugin.process(user_data)
else:
ret_val = userdatautils.execute_user_data_script(user_data)
return self._get_plugin_return_value(ret_val)
def _process_user_data(self, user_data):
plugin_status = base.PLUGIN_EXECUTION_DONE
reboot = False
if user_data.startswith(b'Content-Type: multipart'):
user_data_plugins = factory.load_plugins()
user_handlers = {}
for part in self._parse_mime(user_data):
(plugin_status,
reboot) = self._process_part(part, user_data_plugins,
user_handlers)
if reboot:
break
if not reboot:
for handler_func in list(set(user_handlers.values())):
self._end_part_process_event(handler_func)
return (plugin_status, reboot)
else:
return self._process_non_multi_part(user_data)
def _process_user_data(self, user_data):
plugin_status = base.PLUGIN_EXECUTION_DONE
reboot = False
LOG.debug('User data content:\n%s' % user_data)
if user_data.startswith('Content-Type: multipart'):
user_data_plugins = factory.load_plugins()
user_handlers = {}
for part in self._parse_mime(user_data):
(plugin_status, reboot) = self._process_part(part,
user_data_plugins,
user_handlers)
if reboot:
break
if not reboot:
for handler_func in list(set(user_handlers.values())):
self._end_part_process_event(handler_func)
return (plugin_status, reboot)
else:
return self._process_non_multi_part(user_data)
def test_process(self, mock_load_class):
response = factory.load_plugins()
self.assertTrue(response is not None)
