# coding=utf8
import requests
import os
from requests_toolbelt.multipart.encoder import MultipartEncoder
import sys
import time
import json
import urllib

username='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 = 7500


headers = {'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 conn

def 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_name

def 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_ids

def 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_list

def 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_infos


def 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 e


def 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_list
if uninstall_list:
    uninstall(uninstall_list)
hosts=get_hosts(host_ips)
install(install_list,hosts)