function.py

import time

from PyQt5.QtCore import Qt, QThread
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QTableWidgetItem
from PyQt5.uic.properties import QtWidgets
from aliyunsdkcore.client import AcsClient
from aliyunsdkecs.request.v20140526.DescribeInstancesRequest import DescribeInstancesRequest
from aliyunsdkecs.request.v20140526.CreateCommandRequest import CreateCommandRequest
from aliyunsdkecs.request.v20140526.InvokeCommandRequest import InvokeCommandRequest
from aliyunsdkecs.request.v20140526.DescribeSecurityGroupAttributeRequest import DescribeSecurityGroupAttributeRequest
from aliyunsdkecs.request.v20140526.DescribeCloudAssistantStatusRequest import DescribeCloudAssistantStatusRequest
from aliyunsdkecs.request.v20140526.DescribeInvocationResultsRequest import DescribeInvocationResultsRequest
import json, base64, random, math
import OSSTools


def info_message(ui):
    print('msg')
    QtWidgets.QMessageBox.information(ui.pushButton, "提示", 'msg')

def err_message(ui):
    print('msg')
    QtWidgets.QMessageBox.critical(ui.pushButton, "警告", 'msg')

def DescribeCloudAssistantStatus(AccessKeyID, AccessKeySecret, ZoneId, InstanceId):
    #print(AccessKeyID, AccessKeySecret, ZoneId, InstanceId)
    client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)
    #client = AcsClient('LTAI4FzRuhWj54NnbKRrWPGW', '0FnQHlxmJ0tJQK3lTsadUU9FBODPWu', 'cn-hangzhou')

    request = DescribeCloudAssistantStatusRequest()
    request.set_accept_format('json')

    request.set_InstanceIds([InstanceId])

    response = client.do_action_with_exception(request)
    # python2:  print(response)
    #print(str(response, encoding='utf-8'))
    return json.loads(response)


def CreateCommand(ui, AccessKeyID, AccessKeySecret, com_type, command, ZoneId, InstanceId):

    item = ui.tableWidget.findItems(InstanceId, Qt.MatchExactly)
    row = item[0].row()
    Status = ui.tableWidget.item(row, 8).text()
    if Status == 'false':
        OSSTools.show_message(ui,'未安装云助手，不能执行命令。')
        return

    else:
        client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)

        request = CreateCommandRequest()
        request.set_accept_format('json')
        name = ''.join(random.sample(
            ['z', 'y', 'x', 'w', 'v', 'u', 't', 's', 'r', 'q', 'p', 'o', 'n', 'm', 'l', 'k', 'j', 'i', 'h', 'g', 'f', 'e',
             'd', 'c', 'b', 'a'], 5))
        # name = 'test11111'
        try:
            request.set_Name(name)
            request.set_Type(com_type)
            command = base64.b64encode(command.encode()).decode()

            request.set_CommandContent(command)

            response = client.do_action_with_exception(request)
            # python2:  print(response)
            # print(str(response, encoding='utf-8'))
            CloudAssistantStatus = DescribeCloudAssistantStatus(AccessKeyID, AccessKeySecret, ZoneId, InstanceId)
            Status = CloudAssistantStatus['InstanceCloudAssistantStatusSet']['InstanceCloudAssistantStatus'][0][
                'CloudAssistantStatus']
            item = ui.tableWidget.findItems(InstanceId, Qt.MatchExactly)
            row = item[0].row()
            yunzhushou = QTableWidgetItem(Status)
            ui.tableWidget.setItem(row, 8, yunzhushou)
            return json.loads(response, encoding='utf-8')['CommandId']
        except:
            CloudAssistantStatus = DescribeCloudAssistantStatus(AccessKeyID, AccessKeySecret, ZoneId, InstanceId)
            Status = CloudAssistantStatus['InstanceCloudAssistantStatusSet']['InstanceCloudAssistantStatus'][0]['CloudAssistantStatus']


            print('命令创建失败')
            OSSTools.show_message(ui,'命令创建失败')


def InvokeCommand(ui, AccessKeyID, AccessKeySecret, ZoneId, InstanceId, CommandId):
    client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)

    try:
        request = InvokeCommandRequest()
        request.set_accept_format('json')

        request.set_CommandId(CommandId)
        request.set_InstanceIds([InstanceId])

        response = client.do_action_with_exception(request)
        # python2:  print(response)
        # print(str(response, encoding='utf-8'))
        if json.loads(response)['InvokeId'] == '':
            print('命令执行错误')
            OSSTools.show_message(ui, '命令执行错误')
        else:
            print('命令执行完成')
            #print(str(response, encoding='utf-8'))
            #OSSTools.show_message(ui, '命令执行完成')
            #print(json.loads(response)['InvokeId'])
            return json.loads(response)['InvokeId']
        # return str(response)
    except:
        print('命令执行失败')
        OSSTools.show_message(ui, '命令执行失败')

def DescribeInvocationResults(AccessKeyID, AccessKeySecret, ZoneId, InvokeID):
    client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)

    request = DescribeInvocationResultsRequest()
    request.set_accept_format('json')

    request.set_InvokeId(InvokeID)

    response = client.do_action_with_exception(request)
    # python2:  print(response)
    #print(str(response, encoding='utf-8'))
    return json.loads(response)

def DescribeSecurityGroupAttribute(ui, AccessKeyID, AccessKeySecret, ZoneId, SecurityGroupId):
    client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)

    try:
        request = DescribeSecurityGroupAttributeRequest()
        request.set_accept_format('json')

        request.set_SecurityGroupId(SecurityGroupId)

        response = client.do_action_with_exception(request)
        item = ui.listWidget.item(1)
        item.setText("1111111111111")
    except:
        print('请输入正确参数')
        OSSTools.show_message(ui, '请输入正确参数')
        return
    # python2:  print(response)
    print(str(response, encoding='utf-8'))


def DescribeInstances(ui, AccessKeyID, AccessKeySecret):
    ui.pushButton.setEnabled(False)
    ui.pushButton.setText('正在查询')
    RegionIds = {"cn-hangzhou": "华东1（杭州）", "cn-chengdu": "成都", "cn-qingdao": "华北1（青岛）", "cn-beijing": "华北2（北京）",
                 "cn-huhehaote": "华北5（呼和浩特）", "cn-shanghai": "华东2（上海）",
                 "cn-shenzhen": "华南1（深圳）", "cn-zhangjiakou": "华北3（张家口）",
                 "cn-hongkong": "香港", "ap-southeast-1": "新加坡", "ap-southeast-2": "澳大利亚（悉尼）",
                 "ap-southeast-3": "马来西亚（吉隆坡）",
                 "ap-southeast-5": "印度尼西亚（雅加达）", "ap-northeast-1": "日本（东京）", "us-west-1": "美国（硅谷）",
                 "us-east-1": "美国（弗吉尼亚）",
                 "eu-central-1": "德国（法兰克福）", "me-east-1": "阿联酋（迪拜）", "ap-south-1": "印度（孟买）", "eu-west-1": "英国（伦敦）",
                 "cn-shenzhen-finance-1": "深圳金融云", "cn-shanghai-finance-1": "上海金融云",
                 "cn-north-2-gov-1": "华北 2 阿里政务云1", "cn-heyuan": "华南2（河源）"}
    for RegionId in RegionIds:
        client = AcsClient(AccessKeyID, AccessKeySecret, RegionId)
        # print(RegionId)

        try:
            request = DescribeInstancesRequest()
            request.set_accept_format('json')
            request.set_PageNumber(1)
            request.set_PageSize(100)

            response = client.do_action_with_exception(request)
            #print(json.loads(response))
        except:
            #print('请检查输入Key与Secret值,或重新执行')
            ui.pushButton.setText('查询')
            ui.pushButton.setEnabled(True)
            return ui, '请检查输入Key与Secret值,或重新执行'
        if json.loads(response)['TotalCount'] == 0:
            #print(RegionId)
            if RegionId == 'cn-heyuan':
                ui.pushButton.setText('查询')
                ui.pushButton.setEnabled(True)
                return ui,'查询完成'
            else:

                continue
        # python2:  print(response)
        # print(str(response, encoding='utf-8'))
        # item1 = json.loads(response)['Instances']['Instance'][0]['OSName']
        PrivateIps = ''
        PublicIps = ''

        TotalCount = json.loads(response)['TotalCount']
        pages = math.ceil(float(TotalCount/100))
        for page in range(pages):
            print(page)
            if page == pages-1:
                num = TotalCount%100
            else:
                num = 100
            for i in range(num):

                row_cnt = ui.tableWidget.rowCount()  # 读取行
                ui.tableWidget.insertRow(row_cnt)  # 创建行
                #print(json.loads(response)['Instances']['Instance'][i]['VpcAttributes']['PrivateIpAddress']['IpAddress'])
                for PrivateIp in json.loads(response)['Instances']['Instance'][i]['VpcAttributes']['PrivateIpAddress']['IpAddress']:
                    PrivateIps = str(PrivateIp) + ';'
                    # print(PrivateIps)
                for PublicIp in json.loads(response)['Instances']['Instance'][i]['PublicIpAddress']['IpAddress']:
                    PublicIps = str(PublicIp) + ';'
                    # print(PrivateIps)
                for SecurityGroup in json.loads(response)['Instances']['Instance'][i]['SecurityGroupIds'][
                    'SecurityGroupId']:
                    SecurityGroups = str(SecurityGroup) + ';'
                    # print(PrivateIps)

                InstanceId = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['InstanceId'])
                RegionId = QTableWidgetItem(RegionIds[json.loads(response)['Instances']['Instance'][i]['RegionId']])
                HostName = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['HostName'])
                OSName = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['OSName'])
                InstanceNetworkType = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['Status'])
                PrivateIpAddress = QTableWidgetItem(PrivateIps)
                PublicIpAddress = QTableWidgetItem(PublicIps)
                SecurityGroupIds = QTableWidgetItem(SecurityGroup)
                yunzhushou = QTableWidgetItem('待检测')
                performance = QTableWidgetItem(
                    str(json.loads(response)['Instances']['Instance'][i]['Cpu']) + '核处理器\n' + str(
                        json.loads(response)['Instances']['Instance'][0]['Memory']) + 'MB内存')
                CreationTime = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['CreationTime'])
                ExpiredTime = QTableWidgetItem(json.loads(response)['Instances']['Instance'][i]['ExpiredTime'])

                ui.tableWidget.setItem(row_cnt, 0, InstanceId)
                ui.tableWidget.setItem(row_cnt, 1, RegionId)
                ui.tableWidget.setItem(row_cnt, 2, HostName)
                ui.tableWidget.setItem(row_cnt, 3, OSName)
                ui.tableWidget.setItem(row_cnt, 4, InstanceNetworkType)
                ui.tableWidget.setItem(row_cnt, 5, PrivateIpAddress)
                ui.tableWidget.setItem(row_cnt, 6, PublicIpAddress)
                ui.tableWidget.setItem(row_cnt, 7, SecurityGroupIds)
                ui.tableWidget.setItem(row_cnt, 8, yunzhushou)
                ui.tableWidget.setItem(row_cnt, 9, performance)
                ui.tableWidget.setItem(row_cnt, 10, CreationTime)
                ui.tableWidget.setItem(row_cnt, 11, ExpiredTime)
'''
    row_cnt = ui.tableWidget.rowCount()
    ui.tableWidget.insertRow(row_cnt)
    print(row_cnt)

    ui.pushButton.setText('查询')
    ui.pushButton.setEnabled(True)
    return ui,'查询完成'
'''
def test(ui):
    row_cnt = ui.tableWidget.rowCount()
    ui.tableWidget.insertRow(row_cnt)
    print(row_cnt)


def DescribeSecurityGroupAttribute(ui, AccessKeyID, AccessKeySecret, ZoneId, SecurityGroupId):
    # print(AccessKeyID, AccessKeySecret, ZoneId, SecurityGroupId)
    client = AcsClient(AccessKeyID, AccessKeySecret, ZoneId)

    request = DescribeSecurityGroupAttributeRequest()
    request.set_accept_format('json')

    request.set_SecurityGroupId(SecurityGroupId)

    response = client.do_action_with_exception(request)
    # python2:  print(response)
    print(str(response, encoding='utf-8'))


def server_check_input(ui):
    RegionIds = {'华东1（杭州）': 'cn-hangzhou', '成都': 'cn-chengdu', '华北1（青岛）': 'cn-qingdao', '华北2（北京）': 'cn-beijing',
                 '华北5（呼和浩特）': 'cn-huhehaote', '华东2（上海）': 'cn-shanghai', '华南1（深圳）': 'cn-shenzhen',
                 '华北3（张家口）': 'cn-zhangjiakou', '香港': 'cn-hongkong', '新加坡': 'ap-southeast-1',
                 '澳大利亚（悉尼）': 'ap-southeast-2', '马来西亚（吉隆坡）': 'ap-southeast-3', '印度尼西亚（雅加达）': 'ap-southeast-5',
                 '日本（东京）': 'ap-northeast-1', '美国（硅谷）': 'us-west-1', '美国（弗吉尼亚）': 'us-east-1', '德国（法兰克福）': 'eu-central-1',
                 '阿联酋（迪拜）': 'me-east-1', '印度（孟买）': 'ap-south-1', '英国（伦敦）': 'eu-west-1',
                 '深圳金融云': 'cn-shenzhen-finance-1', '上海金融云': 'cn-shanghai-finance-1', '华北 2 阿里政务云1': 'cn-north-2-gov-1',
                 '华南2（河源）': 'cn-heyuan'}
    AccessKeyID = ui.lineEdit.text()
    AccessKeySecret = ui.lineEdit_2.text()

    for rowNum in range(0, ui.tableWidget.rowCount())[::-1]:
        ui.tableWidget.removeRow(rowNum)

    if AccessKeyID == '':
        #print('请输入AccessKeyID')
        return ui,'请输入AccessKeyID'
    elif AccessKeySecret == '':
        #print('请输入AccessKeySecret')
        return ui,'请输入AccessKeySecret'
    else:
        # RegionIds = {"cn-hangzhou": "华东1（杭州）"}

        #return DescribeInstances(ui, AccessKeyID, AccessKeySecret)
        DescribeInstances(ui, AccessKeyID, AccessKeySecret)
        row_cnt = ui.tableWidget.rowCount()
        for i in range(row_cnt):
            InstanceId = ui.tableWidget.item(i, 0).text()
            ZoneId = ui.tableWidget.item(i, 1).text()
            Status = DescribeCloudAssistantStatus(AccessKeyID, AccessKeySecret, RegionIds[ZoneId], InstanceId)
            #print(Status)
            yunzhushou = QTableWidgetItem(Status['InstanceCloudAssistantStatusSet']['InstanceCloudAssistantStatus'][0]['CloudAssistantStatus'])
            ui.tableWidget.setItem(i, 8, yunzhushou)

        ui.pushButton.setText('查询')
        ui.pushButton.setEnabled(True)
        return ui,'查询完成'



def commad_check_input(ui,uc,echo_id):
    cmd = uc.lineEdit.text()
    uc.textBrowser.append("Command :<font color='red'>"+str(cmd)+ "<font>")
    RegionIds = {'华东1（杭州）': 'cn-hangzhou', '成都': 'cn-chengdu', '华北1（青岛）': 'cn-qingdao', '华北2（北京）': 'cn-beijing',
                 '华北5（呼和浩特）': 'cn-huhehaote', '华东2（上海）': 'cn-shanghai', '华南1（深圳）': 'cn-shenzhen',
                 '华北3（张家口）': 'cn-zhangjiakou', '香港': 'cn-hongkong', '新加坡': 'ap-southeast-1',
                 '澳大利亚（悉尼）': 'ap-southeast-2', '马来西亚（吉隆坡）': 'ap-southeast-3', '印度尼西亚（雅加达）': 'ap-southeast-5',
                 '日本（东京）': 'ap-northeast-1', '美国（硅谷）': 'us-west-1', '美国（弗吉尼亚）': 'us-east-1', '德国（法兰克福）': 'eu-central-1',
                 '阿联酋（迪拜）': 'me-east-1', '印度（孟买）': 'ap-south-1', '英国（伦敦）': 'eu-west-1',
                 '深圳金融云': 'cn-shenzhen-finance-1', '上海金融云': 'cn-shanghai-finance-1', '华北 2 阿里政务云1': 'cn-north-2-gov-1',
                 '华南2（河源）': 'cn-heyuan'}
    AccessKeyID = ui.lineEdit.text()
    AccessKeySecret = ui.lineEdit_2.text()
    try:
        InstanceId_row = ui.tableWidget.selectedItems()[0].row()
        InstanceId = ui.tableWidget.item(InstanceId_row, 0).text()
    except:
        InstanceId = ui.lineEdit_3.text()

    com_type = ui.comboBox.currentIndex()
    command = ui.lineEdit_4.text()

    if com_type == 0:
        com_type = 'RunShellScript'
    elif com_type == 1:
        com_type = 'RunBatScript'
    elif com_type == 2:
        com_type = 'RunPowerShellScript'
    elif com_type == 3:
        com_type = 'RunShellScript'

    if AccessKeyID == '':
        print('请输入AccessKeyID')
        OSSTools.show_message(ui,'请输入AccessKeyID')
        uc.pushButton.setEnabled(True)
        uc.pushButton.setText("执行")
        return
    elif AccessKeySecret == '':
        print('请输入AccessKeySecret')
        OSSTools.show_message(ui, '请输入AccessKeySecret')
        uc.pushButton.setEnabled(True)
        uc.pushButton.setText("执行")
        return
    elif InstanceId == '':
        print('请输入实例ID,或者选中实例ID')
        OSSTools.show_message(ui, '请输入实例ID,或者选中实例ID')
        uc.pushButton.setEnabled(True)
        uc.pushButton.setText("执行")
        return
    elif command == '' and echo_id == 0:
        print('请输入执行命令')
        OSSTools.show_message(ui, '请输入执行命令')
        uc.pushButton.setEnabled(True)
        uc.pushButton.setText("执行")
        return
    else:
        try:
            item = ui.tableWidget.findItems(InstanceId, Qt.MatchExactly)
            row = item[0].row()
            ZoneId = RegionIds[ui.tableWidget.item(row, 1).text()]
            Status = ui.tableWidget.item(row, 4).text()
        except:
            print('未获取到区域ID，请先获取实例信息，或者输入正确实例ID')
            OSSTools.show_message(ui, '未获取到区域ID，请先获取实例信息，或者输入正确实例ID')
            uc.pushButton.setEnabled(True)
            uc.pushButton.setText("执行")
            return

        if Status == 'Stopped':
            OSSTools.show_message(ui, '示例未运行,请选择运行状态示例执行命令')
            uc.pushButton.setEnabled(True)
            uc.pushButton.setText("执行")
            return
        if echo_id == 1:
            command = uc.lineEdit.text()
            command_ID = CreateCommand(ui, AccessKeyID, AccessKeySecret, com_type, command, ZoneId, InstanceId)
            InvokeID = InvokeCommand(ui, AccessKeyID, AccessKeySecret, ZoneId, InstanceId, command_ID)
            time.sleep(1)
            Result = DescribeInvocationResults(AccessKeyID, AccessKeySecret, ZoneId, InvokeID)
            output = Result['Invocation']['InvocationResults']['InvocationResult'][0]['Output']
            print(base64.b64decode(output).decode())
            uc.textBrowser.append(base64.b64decode(output).decode())
            uc.pushButton.setEnabled(True)
            uc.pushButton.setText("执行")
            return

        else:
            command_ID = CreateCommand(ui, AccessKeyID, AccessKeySecret, com_type, command, ZoneId, InstanceId)
            InvokeCommand(ui, AccessKeyID, AccessKeySecret, ZoneId, InstanceId, command_ID)
            uc.pushButton.setEnabled(True)
            uc.pushButton.setText("执行")
            OSSTools.show_message(ui, '命令执行完成')
            return


        # print(a)

        # DescribeInstances(ui,AccessKeyID,AccessKeySecret,'cn-hangzhou')


def SecurityGroup_check(ui):
    AccessKeyID = ui.lineEdit.text()
    AccessKeySecret = ui.lineEdit_2.text()

    try:
        SecurityGroup_row = ui.tableWidget.selectedItems()[0].row()
        SecurityGroupId = ui.tableWidget.item(SecurityGroup_row, 7).text()
    except:
        SecurityGroupId = ui.lineEdit_3.text()
    if AccessKeyID == '':
        print('请输入AccessKeyID')
        return
    elif AccessKeySecret == '':
        print('请输入AccessKeySecret')
        return
    elif SecurityGroupId == '':
        print('请输入安全组ID,或者选中安全组ID')
    else:
        item = ui.tableWidget.findItems(SecurityGroupId, Qt.MatchExactly)
        row = item[0].row()
        ZoneId = ui.tableWidget.item(row, 1).text()
        # SecurityGroup = ui.tableWidget.item(SecurityGroup_row, 7).text()

        DescribeSecurityGroupAttribute(ui, AccessKeyID, AccessKeySecret, ZoneId, SecurityGroupId)


def get_InstanceId(ui):
    try:
        row = ui.tableWidget.selectedItems()[0].row()
        InstanceId = ui.tableWidget.item(row, 0).text()
        ui.lineEdit_3.setText(InstanceId)
    except:
        return