def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
cc_set_select = cc_format_tree_mode_id(
data.get_one_of_inputs('cc_set_select'))
for set_id in cc_set_select:
cc_kwargs = {
"bk_biz_id": biz_cc_id,
"bk_supplier_account": supplier_account,
"bk_set_id": set_id,
"data": {
"bk_service_status":
data.get_one_of_inputs('cc_set_status')
}
}
cc_result = client.cc.update_set(cc_kwargs)
if not cc_result['result']:
message = cc_handle_api_error('cc.update_set', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
return True
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
# 查询主机id
ip_list = get_ip_by_regex(data.get_one_of_inputs('cc_host_ip'))
host_result = cc_get_host_id_by_innerip(executor, biz_cc_id, ip_list,
supplier_account)
if not host_result['result']:
data.set_outputs('ex_data', host_result['message'])
return False
cc_module_select = cc_format_tree_mode_id(
data.get_one_of_inputs('cc_module_select'))
cc_is_increment = data.get_one_of_inputs('cc_is_increment')
cc_kwargs = {
"bk_biz_id": biz_cc_id,
"bk_supplier_account": supplier_account,
"bk_host_id": [int(host_id) for host_id in host_result['data']],
"bk_module_id": cc_module_select,
"is_increment": True if cc_is_increment == 'true' else False
}
cc_result = client.cc.transfer_host_module(cc_kwargs)
if cc_result['result']:
return True
else:
message = cc_handle_api_error('cc.transfer_host_module', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
# 查询主机id
ip_list = get_ip_by_regex(data.get_one_of_inputs('cc_host_ip'))
host_result = cc_get_host_id_by_innerip(executor, biz_cc_id, ip_list,
supplier_account)
if not host_result['result']:
data.set_outputs('ex_data', host_result['message'])
return False
transfer_kwargs = {
'bk_supplier_account': supplier_account,
'bk_biz_id': biz_cc_id,
'bk_host_id': [int(host_id) for host_id in host_result['data']]
}
transfer_result = client.cc.transfer_host_to_idlemodule(
transfer_kwargs)
if transfer_result['result']:
return True
else:
message = cc_handle_api_error('cc.transfer_host_to_idlemodule',
transfer_kwargs, transfer_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
kwargs = {
"bk_biz_id": biz_cc_id,
"bk_supplier_account": supplier_account
}
tree_data = client.cc.search_biz_inst_topo(kwargs)
if not tree_data['result']:
message = cc_handle_api_error('cc.search_biz_inst_topo', kwargs,
tree_data)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
cc_module_select = cc_format_tree_mode_id(
data.get_one_of_inputs('cc_module_select'))
cc_module_property = data.get_one_of_inputs('cc_module_property')
if cc_module_property == "bk_module_type":
bk_module_type = cc_format_prop_data(
executor, 'module', 'bk_module_type',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_module_type['result']:
data.set_outputs('ex_data', bk_module_type['message'])
return False
cc_module_prop_value = bk_module_type['data'].get(
data.get_one_of_inputs('cc_module_prop_value'))
if not cc_module_prop_value:
data.set_outputs('ex_data', _(u"模块类型校验失败，请重试并填写正确的模块类型"))
return False
else:
cc_module_prop_value = data.get_one_of_inputs(
'cc_module_prop_value')
for module_id in cc_module_select:
cc_kwargs = {
"bk_biz_id": biz_cc_id,
"bk_supplier_account": supplier_account,
"bk_set_id": get_module_set_id(tree_data['data'], module_id),
"bk_module_id": module_id,
"data": {
cc_module_property: cc_module_prop_value
}
}
cc_result = client.cc.update_module(cc_kwargs)
if not cc_result['result']:
message = cc_handle_api_error('cc.update_module', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
return True
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
cc_set_select = cc_format_tree_mode_id(
data.get_one_of_inputs('cc_set_select'))
cc_set_property = data.get_one_of_inputs('cc_set_property')
if cc_set_property == "bk_service_status":
bk_service_status = cc_format_prop_data(
executor, 'set', 'bk_service_status',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_service_status['result']:
data.set_outputs('ex_data', bk_service_status['message'])
return False
cc_set_prop_value = bk_service_status['data'].get(
data.get_one_of_inputs('cc_set_prop_value'))
if not cc_set_prop_value:
data.set_outputs('ex_data', _(u"服务状态校验失败，请重试并修改为正确的服务状态"))
return False
elif cc_set_property == "bk_set_env":
bk_set_env = cc_format_prop_data(
executor, 'set', 'bk_set_env',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_set_env['result']:
data.set_outputs('ex_data', bk_set_env['message'])
return False
cc_set_prop_value = bk_set_env['data'].get(
data.get_one_of_inputs('cc_set_prop_value'))
if not cc_set_prop_value:
data.set_outputs('ex_data', _(u"环境类型校验失败，请重试并修改为正确的环境类型"))
return False
else:
cc_set_prop_value = data.get_one_of_inputs('cc_set_prop_value')
for set_id in cc_set_select:
cc_kwargs = {
"bk_biz_id": biz_cc_id,
"bk_supplier_account": supplier_account,
"bk_set_id": set_id,
"data": {
cc_set_property: cc_set_prop_value
}
}
cc_result = client.cc.update_set(cc_kwargs)
if not cc_result['result']:
message = cc_handle_api_error('cc.update_set', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
return True
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
cc_set_parent_select = cc_format_tree_mode_id(
data.get_one_of_inputs('cc_set_parent_select'))
cc_set_info = data.get_one_of_inputs('cc_set_info')
cc_kwargs = {
'bk_biz_id': biz_cc_id,
'bk_supplier_account': supplier_account,
'data': {}
}
for parent_id in cc_set_parent_select:
cc_kwargs['data']['bk_parent_id'] = parent_id
for set_params in cc_set_info:
for key, value in set_params.items():
if value:
if key == "bk_set_env":
bk_set_env = cc_format_prop_data(
executor, 'set', 'bk_set_env',
parent_data.get_one_of_inputs('language'),
supplier_account)
if not bk_set_env['result']:
data.set_outputs('ex_data',
bk_set_env['message'])
return False
value = bk_set_env['data'].get(value)
if not value:
data.set_outputs('ex_data',
_(u"环境类型校验失败，请重试并修改为正确的环境类型"))
return False
elif key == "bk_service_status":
bk_service_status = cc_format_prop_data(
executor, 'set', 'bk_service_status',
parent_data.get_one_of_inputs('language'),
supplier_account)
if not bk_service_status['result']:
data.set_outputs('ex_data',
bk_service_status['message'])
return False
value = bk_service_status['data'].get(value)
if not value:
data.set_outputs('ex_data',
_(u"服务状态校验失败，请重试并修改为正确的服务状态"))
return False
cc_kwargs['data'][key] = value
cc_result = client.cc.create_set(cc_kwargs)
if not cc_result['result']:
message = cc_handle_api_error('cc.create_set', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
return True
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
cc_hosts = data.get_one_of_inputs('cc_host_replace_detail')
# 查询主机可编辑属性
search_attr_kwargs = {
'bk_obj_id': 'host',
'bk_supplier_account': supplier_account
}
search_attr_result = client.cc.search_object_attribute(
search_attr_kwargs)
if not search_attr_result['result']:
message = cc_handle_api_error('cc.search_object_attribute',
search_attr_kwargs,
search_attr_result)
self.logger.error(message)
data.outputs.ex_data = message
return False
editable_attrs = []
for item in search_attr_result['data']:
if item['editable']:
editable_attrs.append(item['bk_property_id'])
# 拉取所有主机信息
search_kwargs = {
'bk_biz_id':
biz_cc_id,
'bk_supplier_account':
supplier_account,
'condition': [{
'bk_obj_id': 'module',
'fields': ['bk_module_id'],
'condition': []
}]
}
fault_replace_ip_map = {}
for item in cc_hosts:
fault_replace_ip_map[''.join(get_ip_by_regex(
item['cc_fault_ip']))] = ''.join(
get_ip_by_regex(item['cc_new_ip']))
all_hosts = []
all_hosts.extend(fault_replace_ip_map.keys())
all_hosts.extend(fault_replace_ip_map.values())
search_kwargs['ip'] = {
'data': all_hosts,
'exact': 1,
'flag': 'bk_host_innerip'
}
hosts_result = client.cc.search_host(search_kwargs)
if not hosts_result['result']:
message = cc_handle_api_error('cc.search_host', search_attr_kwargs,
hosts_result)
self.logger.error(message)
data.outputs.ex_data = message
return False
# 更新替换机信息
batch_update_kwargs = {
'bk_obj_id': 'host',
'bk_supplier_account': supplier_account,
'update': []
}
host_dict = {
host_info['host']['bk_host_innerip']: host_info['host']
for host_info in hosts_result['data']['info']
}
host_id_to_ip = {
host_info['host']['bk_host_id']:
host_info['host']['bk_host_innerip']
for host_info in hosts_result['data']['info']
}
fault_replace_id_map = {}
for fault_ip, new_ip in fault_replace_ip_map.items():
fault_host = host_dict.get(fault_ip)
new_host = host_dict.get(new_ip)
if not fault_host:
data.outputs.ex_data = _(
u"无法查询到 %s 机器信息，请确认该机器是否在当前业务下") % fault_ip
return False
if not new_host:
data.outputs.ex_data = _(
u"无法查询到 %s 机器信息，请确认该机器是否在当前业务下") % new_ip
return False
update_item = {'datas': {}, 'inst_id': new_host['bk_host_id']}
for attr in [
attr for attr in editable_attrs if attr in fault_host
]:
update_item['datas'][attr] = fault_host[attr]
batch_update_kwargs['update'].append(update_item)
fault_replace_id_map[
fault_host['bk_host_id']] = new_host['bk_host_id']
update_result = client.cc.batch_update_inst(batch_update_kwargs)
if not update_result['result']:
message = cc_handle_api_error('cc.batch_update_inst',
batch_update_kwargs, update_result)
self.logger.error(message)
data.outputs.ex_data = message
return False
# 将主机上交至故障机模块
fault_transfer_kwargs = {
'bk_supplier_account': supplier_account,
'bk_biz_id': biz_cc_id,
'bk_host_id': list(fault_replace_id_map.keys())
}
fault_transfer_result = client.cc.transfer_host_to_faultmodule(
fault_transfer_kwargs)
if not fault_transfer_result['result']:
message = cc_handle_api_error('cc.transfer_host_to_faultmodule',
fault_transfer_kwargs,
fault_transfer_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False
# 转移主机模块
transfer_kwargs_list = []
for host_info in hosts_result['data']['info']:
new_host_id = fault_replace_id_map.get(
host_info['host']['bk_host_id'])
if new_host_id:
transfer_kwargs_list.append({
'bk_biz_id':
biz_cc_id,
'bk_supplier_account':
supplier_account,
'bk_host_id': [new_host_id],
'bk_module_id': [
module_info['bk_module_id']
for module_info in host_info['module']
],
'is_increment':
True
})
success = []
for kwargs in transfer_kwargs_list:
transfer_result = client.cc.transfer_host_module(kwargs)
if not transfer_result['result']:
message = cc_handle_api_error('cc.transfer_host_module',
kwargs, transfer_result)
self.logger.error(message)
data.outputs.ex_data = u"{msg}\n{success}".format(
msg=message, success=_(u"成功替换的机器: %s") % ','.join(success))
return False
success.append(host_id_to_ip[kwargs['bk_host_id'][0]])
def execute(self, data, parent_data):
executor = parent_data.get_one_of_inputs('executor')
client = get_client_by_user(executor)
if parent_data.get_one_of_inputs('language'):
setattr(client, 'language',
parent_data.get_one_of_inputs('language'))
translation.activate(parent_data.get_one_of_inputs('language'))
biz_cc_id = data.get_one_of_inputs('biz_cc_id',
parent_data.inputs.biz_cc_id)
supplier_account = supplier_account_for_business(biz_cc_id)
# 查询主机id
ip_list = get_ip_by_regex(data.get_one_of_inputs('cc_host_ip'))
host_result = cc_get_host_id_by_innerip(executor, biz_cc_id, ip_list,
supplier_account)
if not host_result['result']:
data.set_outputs('ex_data', host_result['message'])
return False
# 更新主机属性
cc_host_property = data.get_one_of_inputs('cc_host_property')
if cc_host_property == "bk_isp_name":
bk_isp_name = cc_format_prop_data(
executor, 'host', 'bk_isp_name',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_isp_name['result']:
data.set_outputs('ex_data', bk_isp_name['message'])
return False
cc_host_prop_value = bk_isp_name['data'].get(
data.get_one_of_inputs('cc_host_prop_value'))
if not cc_host_prop_value:
data.set_outputs('ex_data', _(u"所属运营商校验失败，请重试并修改为正确的所属运营商"))
return False
elif cc_host_property == "bk_state_name":
bk_state_name = cc_format_prop_data(
executor, 'host', 'bk_state_name',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_state_name['result']:
data.set_outputs('ex_data', bk_state_name['message'])
return False
cc_host_prop_value = bk_state_name['data'].get(
data.get_one_of_inputs('cc_host_prop_value'))
if not cc_host_prop_value:
data.set_outputs('ex_data', _(u"所在国家校验失败，请重试并修改为正确的所在国家"))
return False
elif cc_host_property == "bk_province_name":
bk_province_name = cc_format_prop_data(
executor, 'host', 'bk_province_name',
parent_data.get_one_of_inputs('language'), supplier_account)
if not bk_province_name['result']:
data.set_outputs('ex_data', bk_province_name['message'])
return False
cc_host_prop_value = bk_province_name['data'].get(
data.get_one_of_inputs('cc_host_prop_value'))
if not cc_host_prop_value:
data.set_outputs('ex_data', _(u"所在省份校验失败，请重试并修改为正确的所在省份"))
return False
else:
cc_host_prop_value = data.get_one_of_inputs('cc_host_prop_value')
cc_kwargs = {
"bk_host_id": ",".join(host_result['data']),
"bk_supplier_account": supplier_account,
"data": {
cc_host_property: cc_host_prop_value
}
}
cc_result = client.cc.update_host(cc_kwargs)
if cc_result['result']:
return True
else:
message = cc_handle_api_error('cc.update_host', cc_kwargs,
cc_result)
self.logger.error(message)
data.set_outputs('ex_data', message)
return False