-
Notifications
You must be signed in to change notification settings - Fork 8
/
QQKey_bug_fixer.py
94 lines (86 loc) · 4.56 KB
/
QQKey_bug_fixer.py
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
# -*- coding: utf-8 -*-
from PyQt5 import QtCore, QtGui, QtWidgets
import sys
import subprocess
import os
def get_file_path(name):
if getattr(sys, 'frozen', None):
basedir = sys._MEIPASS
else:
basedir = os.path.dirname(__file__)
file_path = os.path.join(basedir, name)
return file_path
class Ui_Dialog(object):
def setupUi(self, Dialog):
Dialog.setObjectName("Dialog")
Dialog.resize(252, 72)
font = QtGui.QFont()
font.setFamily("微软雅黑")
Dialog.setFont(font)
Dialog.setWindowFlags(QtCore.Qt.WindowMinimizeButtonHint | QtCore.Qt.WindowCloseButtonHint)
Dialog.setFixedSize(Dialog.width(), Dialog.height())
Dialog.setWindowIcon(QtGui.QIcon(get_file_path("./icon.ico")))
self.centralwidget = QtWidgets.QWidget(Dialog)
self.centralwidget.setObjectName("centralwidget")
self.pushButton = QtWidgets.QPushButton(Dialog)
self.pushButton.setGeometry(QtCore.QRect(20, 30, 91, 31))
self.pushButton.setObjectName("pushButton")
self.pushButton.clicked.connect(lambda:self.write_host(0))
self.pushButton_2 = QtWidgets.QPushButton(Dialog)
self.pushButton_2.setGeometry(QtCore.QRect(140, 30, 91, 31))
self.pushButton_2.setObjectName("pushButton_2")
self.pushButton_2.clicked.connect(lambda:self.write_host(1))
self.label = QtWidgets.QLabel(Dialog)
self.label.setGeometry(QtCore.QRect(21, 7, 131, 16))
self.label.setObjectName("label")
self.retranslateUi(Dialog)
QtCore.QMetaObject.connectSlotsByName(Dialog)
def retranslateUi(self, Dialog):
_translate = QtCore.QCoreApplication.translate
Dialog.setWindowTitle(_translate("Dialog", "QQKey漏洞修复器"))
self.pushButton.setText(_translate("Dialog", "修复"))
self.pushButton_2.setText(_translate("Dialog", "恢复"))
with open('C:\Windows\System32\drivers\etc\hosts') as f:
content = f.read()
self.label.setText(_translate("Dialog", f"当前状态:{'已修复' if '0.0.0.0 localhost.ptlogin2.qq.com' in content else '未修复'}"))
def write_host(self,choose):
try:
_translate = QtCore.QCoreApplication.translate
if choose == 0:
with open('C:\Windows\System32\drivers\etc\hosts',encoding='utf-8') as f:
content = f.read()
with open('C:\Windows\System32\drivers\etc\hosts','w+',encoding='utf-8') as f:
if '0.0.0.0 localhost.ptlogin2.qq.com' not in content:
f.write(content+'\n0.0.0.0 localhost.ptlogin2.qq.com')
else:
f.write(content)
else:
with open('C:\Windows\System32\drivers\etc\hosts',encoding='utf-8') as f:
content = f.readlines()
with open('C:\Windows\System32\drivers\etc\hosts','w+',encoding='utf-8') as f:
for i in range(0,len(content)):
if '0.0.0.0 localhost.ptlogin2.qq.com' in content[i]:
del content[i]
f.write(''.join(content))
subprocess.call("ipconfig /flushdns",creationflags=subprocess.CREATE_NO_WINDOW)
with open('C:\Windows\System32\drivers\etc\hosts') as f:
content = f.read()
self.label.setText(_translate("Dialog",
f"当前状态:{'已修复' if '0.0.0.0 localhost.ptlogin2.qq.com' in content else '未修复'}"))
QtWidgets.QMessageBox.information(self.centralwidget,"提示","操作完成!")
except PermissionError:
QtWidgets.QMessageBox.critical(self.centralwidget,"错误","检测到权限不足,请尝试以管理员身份运行程序并检查是否有程序占用hosts文件!")
except Exception as e:
QtWidgets.QMessageBox.critical(self.centralwidget,"错误","出现未知错误,请尝试以管理员身份运行程序!")
if __name__ == '__main__':
QtGui.QGuiApplication.setHighDpiScaleFactorRoundingPolicy(QtCore.Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
QtGui.QGuiApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling)
# 获取UIC窗口操作权限
app = QtWidgets.QApplication(sys.argv)
Dialog = QtWidgets.QDialog()
# 调自定义的界面（即刚转换的.py对象）
Ui = Ui_Dialog()
Ui.setupUi(Dialog)
# 显示窗口并释放资源
Dialog.show()
sys.exit(app.exec_())