def format_history(histories):
history_list = []
trans = MigrateDal.get_isp_trans()
for history in histories:
history_list.append({
'id':
history.id,
'migrate_rooms':
sorted(json.loads(history.migrate_rooms)),
'dst_rooms':
sorted(json.loads(history.dst_rooms)),
'migrate_isps':
sorted([trans[isp] for isp in json.loads(history.migrate_isps)]),
'cur':
history.cur,
'all':
history.all,
'state':
history.state,
'rtx_id':
history.rtx_id,
'update_at':
history.updated_time.strftime('%Y-%m-%d %H:%M:%S')
})
return history_list
def get_previewinfo():
trans = MigrateDal.get_isp_trans()
domain_count = ViewRecordDal.zone_domain_count()
migrate_list = []
histories = MigrateDal.get_migrated_history()
for history in histories:
migrate_list.append({
'migrate_rooms': sorted(json.loads(history.migrate_rooms)),
'dst_rooms': sorted(json.loads(history.dst_rooms)),
'migrate_isps': sorted([trans[isp] for isp in json.loads(history.migrate_isps)])
})
migrate_acl_subnet = ViewIspAclDal.get_migrate_subnet()
return {'domain_count': domain_count,
'migrate': migrate_list,
'acl_migrate': migrate_acl_subnet}
def format_history(histories):
history_list = []
trans = MigrateDal.get_isp_trans()
for history in histories:
history_list.append({
'id': history.id,
'migrate_rooms': sorted(json.loads(history.migrate_rooms)),
'dst_rooms': sorted(json.loads(history.dst_rooms)),
'migrate_isps': sorted([trans[isp] for isp in json.loads(history.migrate_isps)]),
'cur': history.cur,
'all': history.all,
'state': history.state,
'rtx_id': history.rtx_id,
'update_at': history.updated_time.strftime('%Y-%m-%d %H:%M:%S')
})
return history_list
def run(self):
with self.app.app_context():
for i in range(0, len(self.migrate_domain_list), self.step):
try:
MigrateDal.migrate_domains(self.migrate_domain_list[i: i + self.step], self.migrate_history_id)
except Exception as e:
log.error('migrate rooms failed:%s' % e)
MigrateDal.update_history_by_id(self.migrate_history_id, state='error')
return
# 更新serial
MigrateDal.increase_serial_num(NORMAL_TO_CNAME.values())
def run(self):
with self.app.app_context():
for i in range(0, len(self.migrate_domain_list), self.step):
try:
MigrateDal.migrate_domains(
self.migrate_domain_list[i:i + self.step],
self.migrate_history_id)
except Exception as e:
log.error('migrate rooms failed:%s' % e)
MigrateDal.update_history_by_id(self.migrate_history_id,
state='error')
return
# 更新serial
MigrateDal.increase_serial_num(NORMAL_TO_CNAME.values())
def list_migrate_domain():
json_data = request.get_json(force=True)
src_rooms = json_data.get('src_rooms', [])
dst_rooms = json_data.get("dst_rooms", [])
isps = json_data.get("isps", [])
return MigrateDal.list_migrate_domain(src_rooms, dst_rooms, isps)
def list_migrate_history():
return format_history(MigrateDal.get_last_few_history(limit=15))
def migrate_rooms(src_rooms, dst_rooms, to_migrate_isps, username):
history_id = MigrateDal.create_migrage_history(username)
migrated_isp_rooms = MigrateDal.get_all_abnormal_isps(key='isp', value='room')
to_migrate_dict = {isp: set(src_rooms) | migrated_isp_rooms.get(isp, set())
for isp in to_migrate_isps}
migrate_isp_domains = MigrateDal.list_migrate_domain_by_isp(to_migrate_dict, dst_rooms)
has_migrate_domains = False
for isp, migrate_domains_list in migrate_isp_domains.items():
migrate_domains_list = [domain for domain in migrate_domains_list if domain['after_enabled_rooms']]
if len(migrate_domains_list) == 0:
continue
has_migrate_domains = True
MigrateDal.update_history_total(history_id, len(migrate_domains_list))
m_thread = MigrateThread(username, history_id, migrate_domains_list)
m_thread.start()
if has_migrate_domains:
MigrateDal.add_batch_abnormal_isp(username, to_migrate_dict)
# send_alert_email("[FROM DNSDB]: 机房{}上运营商{}迁移到{}啦.".format(src_rooms, to_migrate_isps, dst_rooms))
MigrateDal.update_history_by_id(history_id,
migrate_rooms=json.dumps(src_rooms),
migrate_isps=json.dumps(to_migrate_isps),
dst_rooms=json.dumps(dst_rooms),
migrate_info=json.dumps({}))
else:
MigrateDal.delete_history_by_id(history_id)
raise BadParam("no domain can migrate, isp_rooms: %s"
% to_migrate_dict, msg_ch=u'没有可迁移的机房')
history_info = get_migrate_info(history_id)
return history_info
def get_migrate_info(history_id):
try:
history_id = int(history_id)
except Exception:
raise BadParam('id should be int')
return format_history([MigrateDal.get_history_info(history_id)])
def get_view_migrate_detail(migrate_id, domain, page, page_size):
return MigrateDal.get_view_migrate_detail(migrate_id, domain, page, page_size)
def get_view_migrate_detail(migrate_id, domain, page, page_size):
return MigrateDal.get_view_migrate_detail(migrate_id, domain, page,
page_size)
def onekey_recover_rooms():
return MigrateDal.onekey_recover_rooms()
def list_migrate_domain():
json_data = request.get_json(force=True)
src_rooms = json_data.get('src_rooms', [])
dst_rooms = json_data.get("dst_rooms", [])
isps = json_data.get("isps", [])
return MigrateDal.list_migrate_domain(src_rooms, dst_rooms, isps)
def onekey_recover_rooms():
return MigrateDal.onekey_recover_rooms()
def list_migrate_history():
return format_history(MigrateDal.get_last_few_history(limit=15))
def migrate_rooms(src_rooms, dst_rooms, to_migrate_isps, username):
history_id = MigrateDal.create_migrage_history(username)
migrated_isp_rooms = MigrateDal.get_all_abnormal_isps(key='isp',
value='room')
to_migrate_dict = {
isp: set(src_rooms) | migrated_isp_rooms.get(isp, set())
for isp in to_migrate_isps
}
migrate_isp_domains = MigrateDal.list_migrate_domain_by_isp(
to_migrate_dict, dst_rooms)
has_migrate_domains = False
for isp, migrate_domains_list in migrate_isp_domains.iteritems():
migrate_domains_list = [
domain for domain in migrate_domains_list
if domain['after_enabled_rooms']
]
if len(migrate_domains_list) == 0:
continue
has_migrate_domains = True
MigrateDal.update_history_total(history_id, len(migrate_domains_list))
m_thread = MigrateThread(username, history_id, migrate_domains_list)
m_thread.start()
if has_migrate_domains:
MigrateDal.add_batch_abnormal_isp(username, to_migrate_dict)
# send_alert_email("[FROM DNSDB]: 机房{}上运营商{}迁移到{}啦.".format(src_rooms, to_migrate_isps, dst_rooms))
MigrateDal.update_history_by_id(
history_id,
migrate_rooms=json.dumps(src_rooms),
migrate_isps=json.dumps(to_migrate_isps),
dst_rooms=json.dumps(dst_rooms),
migrate_info=json.dumps({}))
else:
MigrateDal.delete_history_by_id(history_id)
raise BadParam("no domain can migrate, isp_rooms: %s" %
to_migrate_dict,
msg_ch=u'没有可迁移的机房')
history_info = get_migrate_info(history_id)
return history_info
def get_migrate_info(history_id):
try:
history_id = int(history_id)
except Exception:
raise BadParam('id should be int')
return format_history([MigrateDal.get_history_info(history_id)])