# coding=utf8import requestsimport osfrom requests_toolbelt.multipart.encoder import MultipartEncoderimport sysimport timeimport jsonimport urllibusername=‘admin‘password=‘Omp123!‘file_path=r‘C:\Users\Administrator\Desktop\UYUN-Pl
# coding=utf8import requestsimport osfrom requests_toolbelt.multipart.encoder import MultipartEncoderimport sysimport timeimport jsonimport urllibusername=‘admin‘password=‘Omp123!‘file_path=r‘C:\Users\Administrator\Desktop\UYUN-Platform-smc-V2.0.R16.0.tar.gz‘server_ip=‘10.1.61.123‘host_ips=‘10.1.61.123‘host_ips=host_ips.split(‘,‘)port = 7500headers = {‘Content-Type‘: ‘application/json‘}def login(username,password): global server_ip, port,headers login_data = { ‘username‘: username, ‘passwd‘: password } conn = requests.Session() login_url = ‘http://{}:{}/omp/api/v1/user/login‘.format(server_ip, port) response = conn.post( url=login_url, json=login_data, headers=headers ) if response.status_code != 200: print( ‘Login failed reason:{}‘.format(response.text.decode())) sys.exit(1) else: print ‘login success‘ return conndef upload_file(file_path): global server_ip, port, conn file_name = file_path.split(os.sep)[-1] url = ‘http://{}:{}/omp/api/v1/pkgs/upload‘.format(server_ip, port) encoded_name = urllib.quote(file_name) with open(file_path, ‘rb‘) as f: m = MultipartEncoder( fields={‘file‘: (encoded_name, f, ‘application/gzip‘)} ) decoded_m = m.to_string() response = conn.post(url, data=decoded_m, headers={‘Content-Type‘: m.content_type, ‘charset‘: ‘UTF-8‘}, verify=False) if response.status_code==200: print ‘upload success‘ else: print ‘upload fail‘, response.text sys.exit(1) return file_namedef analysis(file_name): global server_ip, port, conn url = ‘http://{}:{}/omp/api/v1/pkgs/analysis?fileName={}‘.format(server_ip, port, file_name) headers = {‘Content-Type‘: ‘application/json‘} response = conn.get(url, headers=headers, verify=False) if response.status_code==200: print ‘analysis success‘ else: print ‘analysis fail:‘, response.text sys.exit(1)def check_install_list(upload_time): global server_ip, port, conn products = {} pkg_ids = {} j = 1 while 1: main_url = ‘http://{}:{}/omp/api/v1/pkgs/query‘.format(server_ip, port) main_params = {‘pageSize‘: 40, ‘pageIndex‘: j} j += 1 response = conn.get(main_url, params=main_params) if response.status_code != 200: print(‘query error‘) sys.exit(1) else: res = json.loads(response.text) pkg_names = res.get(‘data‘) lens = 0 for i in pkg_names: pkg_name = i[‘pkgName‘] # print pkg_name,i[‘pkgs‘] for module in i[‘pkgs‘]: # print module lens += 1 if int(module[‘addTime‘]) >= int(upload_time) * 1000: pkg_id = module[‘pkgId‘] products.update({pkg_name: module[‘version‘]}) pkg_ids.update({pkg_name: pkg_id}) if lens < 40: break # print ‘install products‘, products, pkg_ids if not products: print ‘没找到要安装的模块包‘ sys.exit(1) else: print ‘check install list success‘ return products, pkg_idsdef check_uninstall_list(host_ips,products): global server_ip, port, conn uninstall_list = {} install_products = products.keys() j = 1 while 1: main_url = ‘http://{}:{}/omp/api/v1/installed_pkgs/query‘.format(server_ip, port) main_params = {‘pageSize‘: 40, ‘pageIndex‘: j} j += 1 response = conn.get(main_url, params=main_params) if response.status_code != 200: print(‘query error‘) sys.exit(1) else: res = json.loads(response.text) pkg_names = res.get(‘data‘) lens = 0 for i in pkg_names: pkg_name = i[‘pkgName‘] for pkg_module in i[‘installedDiffInfoVOs‘]: lens += 1 if pkg_module[‘ip‘] not in host_ips: continue # print pkg_name,pkg_module pkg_id = pkg_module[‘pkgId‘] host_ip = pkg_module[‘hostId‘] if products.get(pkg_name) and ( pkg_module[‘version‘] <= products.get(pkg_name) or pkg_module[‘status‘] != 1): # print pkg_name,pkg_id if not uninstall_list.get(host_ip): uninstall_list.update({host_ip: []}) uninstall_list[host_ip].append({pkg_name: pkg_id}) elif products.get(pkg_name) and pkg_module[‘version‘] > products.get(pkg_name) and pkg_module[ ‘status‘] == 1: install_products.remove(pkg_name) if lens < 40: break # print uninstall_list install_list = [] # print install_products # print pkg_ids for k, v in pkg_ids.items(): if k in install_products: install_list.append(v) print ‘check install list success‘ return install_list,uninstall_listdef uninstall(uninstall_list): global server_ip, port, conn main_url = ‘http://{}:{}/omp/api/v1/pkgs/uninstall‘.format(server_ip, port) for host_id, v in uninstall_list.items(): for pkg in v: pkg_name = pkg.keys()[0] pkg_id = pkg.values()[0] main_params = {‘pkgId‘: pkg_id, ‘hostIds‘: [host_id], ‘forceDelete‘: True} print main_params response = conn.post(main_url, json=main_params, headers=headers) if response.status_code != 200: print(‘{} uninstall {} error:‘.format(host_id, pkg_name)) print response.text sys.exit(1) else: print (‘{} uninstall {} success‘.format(host_id, pkg_name))def get_hosts(host_ips): global server_ip, port, conn host_ip=host_ips[0] url = ‘http://{}:{}/omp/api/v1/hosts/query‘.format(server_ip, port) response = conn.get(url) host_id_infos = [] content = response.content if response.status_code != 200: print(‘Get host info failed:{}‘.format(content.decode())) sys.exit(1) else: print content host_ids = json.loads(content) for i in host_ids: if host_ip and i[‘ip‘] == host_ip: host_id_infos = [i[‘hostId‘]] break host_id_infos.append(i[‘hostId‘]) print(‘host_id:‘, host_id_infos) return host_id_infosdef check_all(modules,pkg_install_info,host_id_infos,order=0): try: global server_ip, port # print pkg_install_info,order # print modules node_url = ‘http://{}:{}/omp/api/v1/pkgs/installation_info/nodes_check‘.format(server_ip, port) data = {"pkgId": list(pkg_install_info[order].values())[0], "pkgName": list(pkg_install_info[order].keys())[0], "hostIds": host_id_infos, "products": [{"modules": modules, "order": order}]} # print ‘check data‘,data response = conn.post(node_url, data=json.dumps(data), headers=headers) if response.status_code != 200: print(‘check failed:{}‘.format(response.text.decode())) sys.exit(1) else: print(response.text) return data except Exception as e: print edef install(install_list_temp,host_id_infos): pkg_install_info = {} install_list={} for i in install_list_temp: install_list.update({i: 0}) for pkg_id, status in install_list.items(): install_ids = [] install_names = [] print ‘开始安装‘, pkg_id check_url = ‘http://{}:{}/omp/api/v1/pkgs/verify_install‘.format(server_ip, port) # print check_url,pkg_id response = conn.get(check_url, params={‘pkgId‘: pkg_id}) if response.status_code != 200: print(‘verify_install error:‘) print response.text sys.exit(1) else: print ‘verify_install‘, response.text conf_url = ‘http://{}:{}/omp/api/v1/pkgs/dependent_pkgs/query‘.format(server_ip, port) response = conn.get(conf_url, params={‘pkgId‘: pkg_id}) if response.status_code != 200: print(‘query dependent pkgs error‘) print response.text sys.exit(1) else: print(‘query dependent pkgs success‘) pkgs = json.loads(response.text) # print ‘pkgs‘,pkgs for i in pkgs: # print i for pkg in i[‘modules‘]: if pkg[‘pkgId‘] == pkg_id: pkg_tag_name = pkg[‘name‘] pkg_install_info.update({pkg[‘order‘]: {pkg[‘name‘]: pkg[‘pkgId‘]}}) print ‘pkg_install_info‘, pkg_install_info length = len(pkg_install_info.keys()) # print(length) modules = [] for i in range(0, length): data = check_all(modules,pkg_install_info,host_id_infos, order=i) # print data modules.append({"order": i, "pkgId": data["pkgId"], "pkgName": data["pkgName"], "hostIds": host_id_infos}) print ‘模块检查‘, modules # 环境检测 check_url = ‘http://{}:{}/omp/api/v1/pkgs/installation_env/check‘.format(server_ip, port) install_datas = {} data = [] for k, v in pkg_install_info.items(): pkg_item_id = list(v.values())[0] if pkg_item_id in install_list.keys() and install_list[pkg_item_id] == 0: display = True install_ids.append(pkg_item_id) else: display = False data.append({"pkgId": pkg_item_id, "hostIds": host_id_infos, "display": display}) # print data response = conn.post(check_url, data=json.dumps(data), headers=headers) if response.status_code != 200: print(response.text) print(‘安装包检查报错，请检查‘) sys.exit(1) else: print(‘安装包检查成功‘) print(response.text) check_url = ‘http://{}:{}/omp/api/v1/pkgs/installation_info/confirm‘.format(server_ip, port) response = conn.post(check_url, data=json.dumps(data), headers=headers) if response.status_code != 200: print(response.text) print(‘安装时环境确认报错，请检查‘) sys.exit(1) else: print(response.text) print(‘环境确认成功‘) insert_url = ‘http://{}:{}/omp/api/v1/tasks/insert‘.format(server_ip, port) data = [] for pkg in pkgs[0][‘modules‘]: # print(pkg) pkg_install_info.update({pkg[‘order‘]: {pkg[‘name‘]: pkg[‘pkgId‘]}}) for k, v in pkg_install_info.items(): item_id = list(v.values())[0] if item_id in install_list.keys() and install_list[item_id] == 0: display = True else: display = False data.append({"pkgId": item_id, "hostIds": host_id_infos, "display": display, "order": k}) # print data response = conn.post(insert_url, data=json.dumps(data), headers=headers) if response.status_code != 200: print(response.text) print(‘插入任务错误，请检查‘) sys.exit(1) else: task_id = json.loads(response.text)[‘taskId‘] print(‘插入任务成功‘) t = time.time() add_time = int(round(t * 1000)) data = [] for k, v in pkg_install_info.items(): id = list(v.values())[0] if id in install_list.keys() and install_list[item_id] == 0: display = True install_names.append(list(v.keys())[0]) else: display = False data.append({"pkgId": id, "hostIds": host_id_infos, "display": display, "order": k, "opts": [], "taskId": task_id}) # print data install_url = ‘http://{}:{}/omp/api/v1/pkgs/install‘.format(server_ip, port) response = conn.post(install_url, data=json.dumps(data), headers=headers) if response.status_code != 200: print(response.text) print(‘执行安装任务错误，请检查‘) sys.exit(1) else: print(response.text) print(‘安装任务开始执行‘) print ‘pkg_install_info‘, pkg_install_info length = 0 nums = 1 while length < nums: time.sleep(60) query_url = ‘http://{}:{}/omp/api/v1/logs/exerecord/abnormal_info/count‘.format(server_ip, port) data = {‘taskId‘: task_id, ‘completed‘: 0} response = conn.get( url=query_url, params=data ) if response.status_code != 200: print ( ‘query warn failed reason:{}‘.format(response.content.decode())) sys.exit(1) else: res = json.loads(response.text) # print res if res.get(‘warningCnt‘) != 0: print (‘出现告警提示‘) elif res.get(‘errorCnt‘) != 0: print (‘出现报错‘) sys.exit(1) # 检查模块 print ‘检查模块‘,pkg_tag_name query_url = ‘http://{}:{}/omp/api/v1/installed_pkgs/query‘.format(server_ip, port) data = {‘keyWord‘: pkg_tag_name, ‘status‘: 1} response = conn.get( url=query_url, params=data ) if response.status_code != 200: print ( ‘query product pkg failed reason:{}‘.format(response.content.decode())) sys.exit(1) else: res = json.loads(response.text) print ‘res‘, res for i in res[‘data‘]: for j in i[‘installedDiffInfoVOs‘]: if j[‘pkgId‘] == pkg_id and j[‘status‘] != 1: print (‘部分模块安装出错‘) sys.exit(1) elif j[‘pkgId‘] == pkg_id: length += 1 # print ‘length‘, length if length == nums: print ‘{}安装成功‘.format(pkg_tag_name) break install_list.update({pkg_id: 1})conn=login(username,password)upload_time = int(time.time())file_name=upload_file(file_path)analysis(file_name)products, pkg_ids=check_install_list(upload_time)install_list,uninstall_list=check_uninstall_list(host_ips,products)print install_list,uninstall_listif uninstall_list: uninstall(uninstall_list)hosts=get_hosts(host_ips)install(install_list,hosts)