def run_func(func_names=[], *args, **kwargs):
fabrun_filter = []
is_filter = False
env.is_test = False
env.is_help = False
if len(args) > 0:
if args[0] == 'test':
env.is_test = True
args = args[1:]
elif args[0] == 'help':
env.is_help = True
args = args[1:]
if 'f' in kwargs:
fabrun_filter = kwargs['f'].split('+')
fabrun_filter = [re.compile(f) for f in fabrun_filter]
is_filter = True
func_patterns = [re.compile(name) for name in func_names]
host_filter = {}
for run in env.runs:
cluster_name = run['cluster']
cluster = env.cluster_map[run['cluster']]
env.cluster = cluster
log.init_logger(cluster_name)
for cluster_run in run['runs']:
run_results = []
script_name = cluster_run['fabscript']
if is_filter:
for f in fabrun_filter:
if f.search(script_name):
break
else:
continue
if env.is_check or env.is_manage:
# 二重実行を防ぐ
hosts = host_filter.get(script_name, [])
tmp_hosts = list(set(cluster_run['hosts']) - set(hosts))
cluster_run['hosts'] = tmp_hosts
hosts.extend(tmp_hosts)
host_filter[script_name] = hosts
if len(cluster_run['hosts']) == 0:
continue
hosts = []
for host in cluster_run['hosts']:
env.node_map[host] = env.node_map.get(host, {
'host': host,
'bootstrap_status': -1,
})
if env.node_map[host]['bootstrap_status'] != status.FAILED_CHECK:
hosts.append(host)
env.script_name = script_name
env.hosts = hosts
# override env
override_env = env.cluster.get('env', {})
override_env.update(cluster_run.get('env', {}))
default_env = {}
for key, value in override_env.items():
default_env[key] = getattr(env, key, None)
setattr(env, key, value)
env.cluster_status = env.cluster['__status']
env.node_status_map = env.cluster_status['node_map']
env.fabscript_status_map = env.cluster_status['fabscript_map']
env.fabscript = env.fabscript_status_map[script_name]
log.info('hosts: {0}'.format(env.hosts))
log.info('run: {0}: {1}'.format(script_name, env.fabscript))
log.debug('node_status_map: {0}'.format(env.node_status_map))
# check require
require = env.cluster['fabscript_map'][script_name]['require']
if env.is_setup:
is_require = True
for script, status_code in require.items():
required_status = env.fabscript_status_map[script]['status']
if required_status != status_code:
log.error('Require Error\n'
+ '{0} is require {1}:{2}.\nbut {1} status is {3}.'.format(
script_name, script, status_code, required_status))
is_require = False
break
if not is_require:
break
script = '.'.join([CONF.fabscript_module, script_name.replace('/', '.')])
# importlibは、2.7以上じゃないと使えない
# module = importlib.import_module(script)
module = __import__(script, globals(), locals(), ['*'], -1)
module_funcs = []
for member in inspect.getmembers(module):
if inspect.isfunction(member[1]):
module_funcs.append(member[0])
if env.is_help and len(func_patterns) == 0:
for candidate in module_funcs:
func = getattr(module, candidate)
print 'Task: {0}'.format(func.__name__)
print func.__doc__
continue
# func_patterns にマッチしたタスク関数をすべて実行する
is_expected = False
is_contain_unexpected = False
for func_pattern in func_patterns:
for candidate in module_funcs:
if not func_pattern.match(candidate):
continue
func = getattr(module, candidate)
# taskデコレータの付いているものだけ実行する
if not hasattr(func, 'is_task') or not func.is_task:
continue
if env.is_help:
print 'Task: {0}'.format(func.__name__)
print func.__doc__
continue
results = api.execute(func, *args, **kwargs)
# check results
data_map = {}
tmp_status = None
is_contain_failed = False
for host, result in results.items():
env.node_map[host].update(result['node'])
if not result or type(result) is not DictType:
result = {}
node_result = env.node_status_map[host]['fabscript_map'][script_name]
result_status = result.get('status')
task_status = result.get('task_status', status.SUCCESS)
msg = result.get('msg')
if msg is None:
if task_status is status.SUCCESS:
msg = status.FABSCRIPT_SUCCESS_MSG
else:
msg = status.FABSCRIPT_FAILED_MSG
if func.is_bootstrap:
if task_status == status.FAILED_CHECK or \
task_status == status.FAILED_CHECK_PING:
env.node_map[host]['bootstrap_status'] = status.FAILED_CHECK
else:
env.node_map[host]['bootstrap_status'] = status.SUCCESS
node_result['task_status'] = task_status
tmp_data_map = result.get('data_map')
if tmp_data_map is not None:
for map_name, tmp_map_data in tmp_data_map.items():
if tmp_map_data['type'] == 'table':
map_data = data_map.get(map_name, {
'name': map_name,
'type': 'table',
'data': [],
})
tmp_data = {'!!host': host}
tmp_data.update(tmp_map_data['data'])
map_data['data'].append(tmp_data)
data_map[map_name] = map_data
elif tmp_map_data['type'] == 'multi-table':
map_data = data_map.get(map_name, {
'name': map_name,
'type': 'multi-table',
'data': [],
})
tmp_data = {'!!host': host}
tmp_data.update(tmp_map_data['data'])
map_data['data'].append(tmp_data)
data_map[map_name] = map_data
elif tmp_map_data['type'] == 'line-chart':
map_data = data_map.get(map_name, {
'name': map_name,
'type': 'line-chart',
'data': [],
})
map_data['ex_data'] = tmp_map_data['ex_data']
map_data['layout'] = tmp_map_data['layout']
tmp_data = {'!!host': host}
tmp_data.update(tmp_map_data['data'])
map_data['data'].append(tmp_data)
data_map[map_name] = map_data
if env.is_setup:
node_result['msg'] = msg
if result_status is not None:
tmp_status = result_status
node_result['status'] = result_status
expected = cluster_run['expected_status']
if expected == result_status:
is_expected = True
log.info('{0}: {1} is expected status.'.format(
host, msg, result_status))
else:
is_contain_unexpected = True
log.error('expected status is {0}, bad status is {1}.'.format( # noqa
expected, result_status), host)
elif env.is_check:
node_result['check_msg'] = msg
if result_status is None:
result_status = status.SUCCESS
node_result.update({
'check_status': result_status,
})
if result_status != status.SUCCESS:
log.error('Failed check {0}.{1} [{2}]. {3}'.format( # noqa
script_name, candidate, result_status, msg), host)
if task_status == status.SUCCESS:
log.info('Success task {0}.{1} [{2}]. {3}'.format(
script_name, candidate, task_status, msg), host)
else:
log.error('Failed task {0}.{1} [{2}]. {3}'.format(
script_name, candidate, task_status, msg), host)
is_contain_failed = True
# end for host, result in results.items():
if len(data_map) > 0:
util.dump_datamap(data_map)
if is_contain_failed:
log.error('Failed task {0}.{1}. Exit setup.'.format(
script_name, candidate))
util.dump_status()
exit()
if tmp_status is not None:
env.fabscript['tmp_status'] = tmp_status
# for candidate in module_funcs:
# for func_pattern in func_patterns:
if env.is_setup:
if is_expected and not is_contain_unexpected or cluster_run['expected_status'] == 0:
env.cluster['__status']['fabscript_map'][script_name] = {
'status': cluster_run['expected_status'],
'task_status': status.SUCCESS,
}
util.dump_status()
else:
log.error('bad status.')
exit()
elif env.is_check:
env.cluster['__status']['fabscript_map'][script_name]['task_status'] = status.SUCCESS # noqa
util.dump_status()
elif env.is_manage:
env.cluster['__status']['fabscript_map'][script_name]['task_status'] = status.SUCCESS # noqa
util.dump_status()
# reset env
for key, value in default_env.items():
if value is not None:
setattr(env, key, value)
def node(*options):
len_options = len(options)
# init env
env.cmd_history = [] # for debug
env.last_runs = []
env.node = {}
env.node_map = {}
env.fabscript = {}
env.fabscript_map = {}
env.cluster = {}
env.cluster_map = {}
if len_options == 0 or options[0] in ['recent', 'r', 'error', 'e']:
with open(CONF._node_meta_pickle) as f:
node_meta = pickle.load(f)
recent_clusters = node_meta['recent_clusters']
if len(recent_clusters) == 0:
print 'There are no recent clusters'
return 0
if len_options > 1:
index = int(options[1])
else:
index = 0
cluster = recent_clusters[index]
cluster_dir = os.path.join(CONF._node_dir, cluster)
cluster_yaml = os.path.join(cluster_dir, '__cluster.yml')
cluster_pickle = os.path.join(cluster_dir, '__cluster.pickle')
node_cluster = None
if os.path.exists(cluster_pickle):
with open(cluster_pickle) as f:
node_cluster = pickle.load(f)
elif os.path.exists(cluster_yaml):
with open(cluster_yaml) as f:
node_cluster = yaml.load(f)
with open(cluster_pickle, 'w') as f:
pickle.dump(node_cluster, f)
if len_options > 0 and options[0] in ['error', 'e']:
is_only_error = True
else:
is_only_error = False
util.print_cluster(cluster, node_cluster, is_only_error)
return 0
else:
query = options[0]
find_depth = 1
is_yes = False
if len_options > 1:
option = options[1]
if option.isdigit():
find_depth = int(option)
elif option == 'yes':
is_yes = True
elif option == 'local':
is_yes = True
env.is_local = True
query += CONF.host
for task in env.tasks:
if task.find('setup') == 0:
env.is_setup = True
elif task.find('check') == 0:
env.is_check = True
elif task.find('manage') == 0:
env.is_manage = True
elif task.find('datamap') == 0:
env.is_datamap = True
util.load_runs(query, find_depth=find_depth)
util.print_runs()
if len(env.tasks) > 1:
if is_yes or util.confirm(
'Are you sure you want to run task on above nodes?',
'Canceled'):
if (env.is_setup or env.is_check or env.is_manage) \
and (not env.password or env.password == ''):
print 'Enter your password.\n'
if platform.system().find('CYGWIN') == 0:
env.password = getpass.getpass()
else:
sudo('hostname', shell=False)
util.decode_cluster_map()
util.dump_status()
else:
exit()
def node(*options):
len_options = len(options)
# init env
env.cmd_history = [] # for debug
env.last_runs = []
env.node = {}
env.node_map = {}
env.fabscript = {}
env.fabscript_map = {}
env.cluster = {}
env.cluster_map = {}
if len_options == 0 or options[0] in ['recent', 'r', 'error', 'e']:
with open(CONF._node_meta_pickle) as f:
node_meta = pickle.load(f)
recent_clusters = node_meta['recent_clusters']
if len(recent_clusters) == 0:
print 'There are no recent clusters'
return 0
if len_options > 1:
index = int(options[1])
else:
index = 0
cluster = recent_clusters[index]
cluster_dir = os.path.join(CONF._node_dir, cluster)
cluster_yaml = os.path.join(cluster_dir, '__cluster.yml')
cluster_pickle = os.path.join(cluster_dir, '__cluster.pickle')
node_cluster = None
if os.path.exists(cluster_pickle):
with open(cluster_pickle) as f:
node_cluster = pickle.load(f)
elif os.path.exists(cluster_yaml):
with open(cluster_yaml) as f:
node_cluster = yaml.load(f)
with open(cluster_pickle, 'w') as f:
pickle.dump(node_cluster, f)
if len_options > 0 and options[0] in ['error', 'e']:
is_only_error = True
else:
is_only_error = False
util.print_cluster(cluster, node_cluster, is_only_error)
return 0
else:
query = options[0]
find_depth = 1
is_yes = False
if len_options > 1:
option = options[1]
if option.isdigit():
find_depth = int(option)
elif option == 'yes':
is_yes = True
elif option == 'local':
is_yes = True
env.is_local = True
query += CONF.host
for task in env.tasks:
if task.find('setup') == 0:
env.is_setup = True
elif task.find('check') == 0:
env.is_check = True
elif task.find('manage') == 0:
env.is_manage = True
elif task.find('datamap') == 0:
env.is_datamap = True
util.load_runs(query, find_depth=find_depth)
util.print_runs()
if len(env.tasks) > 1:
if is_yes or util.confirm(
'Are you sure you want to run task on above nodes?', 'Canceled'):
if (env.is_setup or env.is_check or env.is_manage) \
and (not env.password or env.password == ''):
print 'Enter your password.\n'
if platform.system().find('CYGWIN') == 0:
env.password = getpass.getpass()
else:
sudo('hostname', shell=False)
util.decode_cluster_map()
util.dump_status()
else:
exit()
