easyPing--python3.6/easyPing.py at master · LuckyDL/easyPing--python3.6 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# -*- coding: utf-8 -*-
# easyPing is a widget to check which IP is in use in your LAN

import sys
import re
import threading
import subprocess
from PyQt5 import QtCore, QtWidgets
from MyPing import Ui_MyPing


class EasyPing(QtWidgets.QWidget):
    '''
    检测某网段的IP使用情况
    '''
    _ping_signal = QtCore.pyqtSignal(bool, str)

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

        self.ui = Ui_MyPing()
        self.ui.setupUi(self)
        self.ui.startIP.editingFinished.connect(self.set_end_ip)
        self.ui.pingButton.clicked.connect(self.start_ping)
        self._ping_signal.connect(self.set_ui)

    def set_end_ip(self):
        '''
        填写起始地址后，默认填写结束地址为xxx.xxx.xxx.255
        '''
        startip = self.ui.startIP.text()
        pattern = r"((?:(?:25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d)))\.){3}(?:25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d)))$)"
        m = re.match(pattern, startip)      # 检查IP地址是否合法
        if m:
            startip = startip.split('.')
            startip[3] = '255'
            endip = '.'.join(startip)
            self.ui.endIP.setText(endip)
        else:
            QtWidgets.QMessageBox.warning(self, "easyPing", "IP地址错误")
            self.ui.startIP.setFocus()
            self.ui.startIP.selectAll()

    def start_ping(self):
        '''
        启动多线程
        '''
        self.reset_ui()
        startip = self.ui.startIP.text().split('.')
        endip = self.ui.endIP.text().split('.')
        tmp_ip = startip

        pthread_list = []
        for i in range(int(startip[3]), int(endip[3]) + 1):
            tmp_ip[3] = str(i)
            ip = '.'.join(tmp_ip)
            pthread_list.append(threading.Thread(target=self.get_ping_result, args=(ip,)))
        for item in pthread_list:
            item.setDaemon(True)
            item.start()

    def get_ping_result(self, ip):
        '''
        检查对应的IP是否被占用
        '''
        cmd_str = "ping {0} -n 1 -w 600".format(ip)
        DETACHED_PROCESS = 0x00000008   # 不创建cmd窗口
        try:
            subprocess.run(cmd_str, creationflags=DETACHED_PROCESS, check=True)  # 仅用于windows系统
        except subprocess.CalledProcessError as err:
            self._ping_signal.emit(False, ip)
        else:
            self._ping_signal.emit(True, ip)

    def reset_ui(self):
        ''' 初始化窗口IP窗格为灰色背景 '''
        for item in self.ui.label_list:
            item.setStyleSheet("background-color: rgb(203, 203, 203);")

    def set_ui(self, result, ip):
        '''
        设置窗口颜色
            result：线程ping的结果
            ip：为对于的IP地址
        '''
        index = int(ip.split('.')[3])
        if result:
            self.ui.label_list[index].setStyleSheet("background-color: rgb(85, 170, 127);") # 设置背景为绿色
        else:
            self.ui.label_list[index].setStyleSheet("background-color: rgb(255, 142, 119);")    # 设置背景为红色


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    easy = EasyPing()
    easy.show()
    sys.exit(app.exec_())