All/Python/DoorKeep/SearchDeviceDemo/SearchDeviceDemo.py

# coding=utf-8
import sys
import time
import gc

from PyQt5.QtCore import QThread
from PyQt5.QtWidgets import QApplication, QMainWindow, QMessageBox, QTableWidgetItem, QDialog
from SearchDeviceUI import Ui_MainWindow
from InitDevAccountUI import Ui_InitDevAccount
from queue import Queue

from NetSDK.NetSDK import NetClient
from NetSDK.SDK_Enum import *
from NetSDK.SDK_Struct import *
from NetSDK.SDK_Callback import fSearchDevicesCBEx, fSearchDevicesCB
from ctypes import *

import socket
import struct

global wnd
device_queue = Queue(maxsize=0)
nUpdateNum = 0


class Mythread(QThread):
    # 定义信号,定义参数为int, int类型

    def __init__(self, parent=None):
        super().__init__(parent)

    def run(self):
        global nUpdateNum
        while not device_queue.empty():
            wnd.update_UItable(device_queue.get())
            device_queue.task_done()
            nUpdateNum += 1
            if (nUpdateNum % 10 == 0):
                wnd.tableWidget.update()
                wnd.tableWidget.viewport().update()
                nUpdateNum = 0
                time.sleep(0.1)


@CB_FUNCTYPE(None, C_LLONG, POINTER(DEVICE_NET_INFO_EX2), c_void_p)
def search_device_callback(lSearchHandle, pDevNetInfo, pUserData):
    try:
        buf = cast(pDevNetInfo, POINTER(DEVICE_NET_INFO_EX2)).contents
        print("Enter in search_device_callback")
        if (buf.stuDevInfo.iIPVersion == 4):
            device_queue.put(list((buf.stuDevInfo.byInitStatus, buf.stuDevInfo.iIPVersion, buf.stuDevInfo.szIP,
                                   buf.stuDevInfo.nPort, buf.stuDevInfo.szSubmask, buf.stuDevInfo.szGateway,
                                   buf.stuDevInfo.szMac, buf.stuDevInfo.szDeviceType, buf.stuDevInfo.szDetailType,
                                   buf.stuDevInfo.nHttpPort, buf.stuDevInfo.byPwdResetWay, buf.szLocalIP)))
        wnd.thread.run()
    except Exception as e:
        print(e)


@CB_FUNCTYPE(None, POINTER(DEVICE_NET_INFO_EX), c_void_p)
def search_devie_byIp_callback(pDevNetInfo, pUserData):
    try:
        buf = cast(pDevNetInfo, POINTER(DEVICE_NET_INFO_EX)).contents
        print("Enter in search_devie_byIp_callback")
        if (buf.iIPVersion == 4):
            device_queue.put(list((buf.byInitStatus, buf.iIPVersion, buf.szIP, buf.nPort, buf.szSubmask, buf.szGateway,
                                   buf.szMac, buf.szDeviceType, buf.szDetailType, buf.nHttpPort, buf.byPwdResetWay,
                                   None)))
        wnd.thread.run()
    except Exception as e:
        print(e)


class MyMainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super(MyMainWindow, self).__init__(parent)
        self.setupUi(self)
        self.init_ui()

        self.sdk = NetClient()
        self.sdk.InitEx(None, 0)

        self.thread = Mythread()
        self.thread.start()

    # 初始化界面
    def init_ui(self):
        nUpdateNum = 0
        self.row = 0
        self.column = 0
        self.device_info_list = []
        self.device_mac_list = []
        self.lSearchHandle_list = []
        self.SearchDeviceButton.setEnabled(True)

    # 获取多网卡ip
    def getIPAddrs(self):
        ip_list = socket.gethostbyname_ex(socket.gethostname())
        for ips in ip_list:
            if type(ips) == list and len(ips) != 0:
                IPlist = ips[0:]
        del ip_list
        return IPlist

    # 组播和广播搜索
    def start_search_device(self):
        # 获取本地IP，考虑到多网卡的情况下搜索
        # 有几张网卡，就得调用几次搜索接口
        IPList = self.getIPAddrs()
        nSuccess = 0
        for i in range(IPList.__len__()):
            startsearch_in = NET_IN_STARTSERACH_DEVICE()
            startsearch_in.dwSize = sizeof(NET_IN_STARTSERACH_DEVICE)
            startsearch_in.emSendType = EM_SEND_SEARCH_TYPE.MULTICAST_AND_BROADCAST
            startsearch_in.cbSearchDevices = search_device_callback
            startsearch_in.szLocalIp = IPList[i].encode()
            startsearch_out = NET_OUT_STARTSERACH_DEVICE()
            startsearch_out.dwSize = sizeof(NET_OUT_STARTSERACH_DEVICE)
            lSearchHandle = self.sdk.StartSearchDevicesEx(startsearch_in, startsearch_out)
            if lSearchHandle != 0:
                nSuccess += 1
                self.lSearchHandle_list.append(lSearchHandle)
        if (IPList.__len__() > 0):
            del IPList
        if (nSuccess > 0):
            return True
        else:
            return False

    # 单播搜索
    def start_search_device_byIP(self, start_IP, end_IP):  # 这里要注意每个ip的有效性
        startsearchbyIp_in = DEVICE_IP_SEARCH_INFO()
        startsearchbyIp_in.dwSize = sizeof(DEVICE_IP_SEARCH_INFO)
        start = struct.unpack("!I", socket.inet_aton(start_IP))[0]  # 网络序转字节序
        end = struct.unpack("!I", socket.inet_aton(end_IP))[0]
        if (end - start > 255):
            QMessageBox.about(self, '提示(prompt)', "IP数量超过最大限制256(Number of IP addresses exceeds the upper limit 256.)")
            return False

        startsearchbyIp_in.nIpNum = end - start + 1

        for i in range(startsearchbyIp_in.nIpNum):
            ip = DEVICE_IP_SEARCH_INFO_IP()
            ip.IP = socket.inet_ntoa(struct.pack("!I", start + i)).encode()
            startsearchbyIp_in.szIP[i] = ip

        wait_time = int(wnd.Searchtime_lineEdit.text())
        # 获取本地IP，考虑到多网卡的情况下搜索
        # 有几张网卡，就得调用几次搜索接口
        IPList = self.getIPAddrs()
        nSuccessNum = 0
        for i in range(IPList.__len__()):
            result = self.sdk.SearchDevicesByIPs(startsearchbyIp_in, search_devie_byIp_callback, 0, IPList[i].encode(),
                                                 wait_time)
            if result:
                nSuccessNum = + 1
        if (IPList.__len__() > 0):
            del IPList
        if (nSuccessNum > 0):
            return True
        else:
            return False

    # 停止搜索,配合start_search_device使用
    def stop_search_device(self):
        for i in range(self.lSearchHandle_list.__len__()):
            result = self.sdk.StopSearchDevices(self.lSearchHandle_list[i])
        nUpdateNum = 0
        self.lSearchHandle_list.clear()
        self.device_info_list.clear()
        self.device_mac_list.clear()
        self.tableWidget.clear()
        self.row = 0
        self.column = 0
        device_queue.queue.clear()
        if (not device_queue.empty()):
            device_queue.task_done()
        self.tableWidget.setHorizontalHeaderLabels(
            ['序号(No.)', '状态(Status)', 'IP版本(IP Version)', 'IP地址(IP Address)', '端口(Port)', '子网掩码(Subnet Mask)',
             '网关(Gateway)', '物理地址(Mac Address)', '设备类型(Device Type)', '详细类型(Detail Type)', 'Http(Http)'])
        return

    # 检查是否输入正确的ip地址
    def check_ip(self, ipaddr):
        addr = ipaddr.split('.')  # 切割IP地址为一个列表
        if len(addr) != 4:  # 切割后列表必须有4个参数
            del addr
            return False

        for i in range(4):
            try:
                addr[i] = int(addr[i])  # 每个参数必须为数字，否则校验失败
            except:
                del addr
                return False

            if addr[i] <= 255 and addr[i] >= 0:  # 每个参数值必须在0-255之间
                pass
            else:
                del addr
                return False
        del addr
        gc.collect()
        return True

    # 初始化账号
    def init_device_accout(self, device_info: list):
        child = QDialog()
        child_ui = Ui_InitDevAccount()
        child_ui.setupUi(child)
        if (1 == (device_info[3] & 1)):
            # 手机
            child_ui.way_lineEdit.setText('手机(Phone)')
        elif (1 == (device_info[3] >> 1 & 1)):
            # 邮箱
            child_ui.way_lineEdit.setText('邮箱(Mail)')
        value = child.exec()
        if (value == 0):
            return False
        init_Account_In = NET_IN_INIT_DEVICE_ACCOUNT()
        init_Account_In.dwSize = sizeof(init_Account_In)
        init_Account_In.szMac = device_info[2]
        username = child_ui.username_lineEdit.text()
        password = child_ui.password_lineEdit.text()
        confirm_password = child_ui.confirm_password_lineEdit.text()
        if (password != confirm_password):
            QMessageBox.about(self, '提示(prompt)', "确认密码不一致，请重新输入(Confirm password is wrong，please input again)")
            return
        init_Account_In.szUserName = username.encode()
        init_Account_In.szPwd = password.encode()
        init_Account_In.szCellPhone = child_ui.reset_way_lineEdit.text().encode()
        if (1 == (device_info[3] & 1)):
            # 手机
            init_Account_In.szCellPhone = child_ui.reset_way_lineEdit.text().encode()
        elif (1 == (device_info[3] >> 1 & 1)):
            # 邮箱
            init_Account_In.szMail = child_ui.reset_way_lineEdit.text().encode()
        init_Account_In.byPwdResetWay = device_info[3]
        init_Account_Out = NET_OUT_INIT_DEVICE_ACCOUNT()
        init_Account_Out.dwSize = sizeof(init_Account_Out)

        result = self.sdk.InitDevAccount(init_Account_In, init_Account_Out, 5000, device_info[4])
        if result:
            return True
        else:
            QMessageBox.about(self, '提示(prompt)', 'error:' + str(self.sdk.GetLastError()))
            return False

    def search_Device_Btn(self):
        self.stop_search_device()
        result = self.start_search_device()

    def search_Device_ByIp_Btn(self):
        self.stop_search_device()
        self.tableWidget.setHorizontalHeaderLabels(
            ['序号(No.)', '状态(Status)', 'IP版本(IP Version)', 'IP地址(IP Address)', '端口(Port)', '子网掩码(Subnet Mask)',
             '网关(Gateway)', '物理地址(Mac Address)', '设备类型(Device Type)', '详细类型(Detail Type)', 'Http(Http)'])
        start_IP = self.StartIP_lineEdit.text()
        end_IP = self.EndIP_lineEdit.text()
        if (start_IP != '') and (end_IP != ''):
            if (self.check_ip(start_IP) == True) and (self.check_ip(end_IP) == True):
                result = self.start_search_device_byIP(start_IP, end_IP)
            else:
                QMessageBox.about(self, '提示(prompt)', "IP不正确(IP is wrong)")
                pass
        else:
            QMessageBox.about(self, '提示(prompt)', "IP不能为空(IP can not be empty)")
            pass

    def Init_Btn(self):
        # 获取选中行的ip和初始化信息
        currentRow = self.tableWidget.currentRow()
        if ((len(self.device_info_list) == 0) or ((self.device_info_list[currentRow][0] & 3) != 1)):
            QMessageBox.about(self, '提示(prompt)', "请选择未初始化设备(Please select not initialized device)")
        else:
            result = self.init_device_accout(self.device_info_list[currentRow])
            if result == True:
                QMessageBox.about(self, '提示(prompt)', "初始化成功(Initialize Success)")
                item = QTableWidgetItem("已初始化(Initialize)")
                self.device_info_list[currentRow][0] = 2
                self.tableWidget.setItem(currentRow, 1, item)
                self.tableWidget.update()
                self.tableWidget.viewport().update()

    def update_UItable(self, device_list):
        # 将重复的设备进行过滤
        if device_list[6] in self.device_mac_list:
            return
        if device_list[1] != 4:
            return
        self.device_mac_list.append(device_list[6])
        self.device_info_list.append(
            list((device_list[0], device_list[2], device_list[6], device_list[10], device_list[11])))
        self.tableWidget.setRowCount(self.row + 1)
        item = QTableWidgetItem(str(self.row + 1))
        self.tableWidget.setItem(self.row, self.column, item)
        if ((device_list[0] & 3) == 1):
            item1 = QTableWidgetItem("未初始化(Uninitialize)")
            self.tableWidget.setItem(self.row, self.column + 1, item1)
        else:
            item1 = QTableWidgetItem("已初始化(Initialize)")
            self.tableWidget.setItem(self.row, self.column + 1, item1)
        item2 = QTableWidgetItem(str(device_list[1]))
        self.tableWidget.setItem(self.row, self.column + 2, item2)
        item3 = QTableWidgetItem(str(device_list[2].decode()))
        self.tableWidget.setItem(self.row, self.column + 3, item3)
        item4 = QTableWidgetItem(str(device_list[3]))
        self.tableWidget.setItem(self.row, self.column + 4, item4)
        item5 = QTableWidgetItem(str(device_list[4].decode()))
        self.tableWidget.setItem(self.row, self.column + 5, item5)
        item6 = QTableWidgetItem(str(device_list[5].decode()))
        self.tableWidget.setItem(self.row, self.column + 6, item6)
        item7 = QTableWidgetItem(str(device_list[6].decode()))
        self.tableWidget.setItem(self.row, self.column + 7, item7)
        item8 = QTableWidgetItem(str(device_list[7].decode()))
        self.tableWidget.setItem(self.row, self.column + 8, item8)
        item9 = QTableWidgetItem(str(device_list[8].decode()))
        self.tableWidget.setItem(self.row, self.column + 9, item9)
        item10 = QTableWidgetItem(str(device_list[9]))
        self.tableWidget.setItem(self.row, self.column + 10, item10)

        self.row += 1
        # 刷新table
        device_list.clear()
        del device_list

    # 关闭主窗口时清理资源
    def closeEvent(self, event):
        event.accept()
        print('exit')
        self.stop_search_device()
        self.sdk.Cleanup()
        del self.lSearchHandle_list
        del self.device_info_list
        del self.device_mac_list


if __name__ == '__main__':
    app = QApplication(sys.argv)
    my_wnd = MyMainWindow()
    wnd = my_wnd
    my_wnd.show()
    app.processEvents()
    sys.exit(app.exec_())