init
This commit is contained in:
3
app/ui/tool/pc_decrypt/__init__.py
Normal file
3
app/ui/tool/pc_decrypt/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from .pc_decrypt import DecryptControl
|
||||
|
||||
__all__ = ['DecryptControl']
|
||||
206
app/ui/tool/pc_decrypt/decryptUi.py
Normal file
206
app/ui/tool/pc_decrypt/decryptUi.py
Normal file
@@ -0,0 +1,206 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Form implementation generated from reading ui file 'decryptUi.ui'
|
||||
#
|
||||
# Created by: PyQt5 UI code generator 5.15.10
|
||||
#
|
||||
# WARNING: Any manual changes made to this file will be lost when pyuic5 is
|
||||
# run again. Do not edit this file unless you know what you are doing.
|
||||
|
||||
|
||||
from PyQt5 import QtCore, QtGui, QtWidgets
|
||||
|
||||
|
||||
class Ui_Dialog(object):
|
||||
def setupUi(self, Dialog):
|
||||
Dialog.setObjectName("Dialog")
|
||||
Dialog.resize(611, 519)
|
||||
font = QtGui.QFont()
|
||||
font.setFamily("微软雅黑")
|
||||
font.setPointSize(12)
|
||||
Dialog.setFont(font)
|
||||
Dialog.setLayoutDirection(QtCore.Qt.LeftToRight)
|
||||
Dialog.setStyleSheet("QPushButton {\n"
|
||||
" background: rgb(238,244,249);\n"
|
||||
"}\n"
|
||||
"QPushButton:hover{\n"
|
||||
" background: rgb(230, 235, 240);\n"
|
||||
"}")
|
||||
self.verticalLayout_2 = QtWidgets.QVBoxLayout(Dialog)
|
||||
self.verticalLayout_2.setObjectName("verticalLayout_2")
|
||||
spacerItem = QtWidgets.QSpacerItem(20, 79, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
|
||||
self.verticalLayout_2.addItem(spacerItem)
|
||||
self.horizontalLayout_3 = QtWidgets.QHBoxLayout()
|
||||
self.horizontalLayout_3.setObjectName("horizontalLayout_3")
|
||||
spacerItem1 = QtWidgets.QSpacerItem(78, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_3.addItem(spacerItem1)
|
||||
self.verticalLayout = QtWidgets.QVBoxLayout()
|
||||
self.verticalLayout.setObjectName("verticalLayout")
|
||||
self.horizontalLayout_4 = QtWidgets.QHBoxLayout()
|
||||
self.horizontalLayout_4.setObjectName("horizontalLayout_4")
|
||||
spacerItem2 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_4.addItem(spacerItem2)
|
||||
self.btn_help = QtWidgets.QPushButton(Dialog)
|
||||
self.btn_help.setMinimumSize(QtCore.QSize(0, 40))
|
||||
self.btn_help.setMaximumSize(QtCore.QSize(200, 16777215))
|
||||
self.btn_help.setStyleSheet("")
|
||||
self.btn_help.setObjectName("btn_help")
|
||||
self.horizontalLayout_4.addWidget(self.btn_help)
|
||||
spacerItem3 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_4.addItem(spacerItem3)
|
||||
self.verticalLayout.addLayout(self.horizontalLayout_4)
|
||||
self.label_3 = QtWidgets.QLabel(Dialog)
|
||||
font = QtGui.QFont()
|
||||
font.setFamily("一纸情书")
|
||||
font.setPointSize(20)
|
||||
self.label_3.setFont(font)
|
||||
self.label_3.setAlignment(QtCore.Qt.AlignCenter)
|
||||
self.label_3.setObjectName("label_3")
|
||||
self.verticalLayout.addWidget(self.label_3)
|
||||
self.label_9 = QtWidgets.QLabel(Dialog)
|
||||
self.label_9.setAlignment(QtCore.Qt.AlignCenter)
|
||||
self.label_9.setObjectName("label_9")
|
||||
self.verticalLayout.addWidget(self.label_9)
|
||||
self.gridLayout = QtWidgets.QGridLayout()
|
||||
self.gridLayout.setObjectName("gridLayout")
|
||||
self.label = QtWidgets.QLabel(Dialog)
|
||||
self.label.setObjectName("label")
|
||||
self.gridLayout.addWidget(self.label, 0, 0, 1, 1)
|
||||
self.label_pid = QtWidgets.QLabel(Dialog)
|
||||
self.label_pid.setText("")
|
||||
self.label_pid.setObjectName("label_pid")
|
||||
self.gridLayout.addWidget(self.label_pid, 0, 1, 1, 1)
|
||||
self.label_7 = QtWidgets.QLabel(Dialog)
|
||||
self.label_7.setObjectName("label_7")
|
||||
self.gridLayout.addWidget(self.label_7, 1, 0, 1, 1)
|
||||
self.label_version = QtWidgets.QLabel(Dialog)
|
||||
self.label_version.setText("")
|
||||
self.label_version.setObjectName("label_version")
|
||||
self.gridLayout.addWidget(self.label_version, 1, 1, 1, 1)
|
||||
self.btn_getinfo = QtWidgets.QPushButton(Dialog)
|
||||
self.btn_getinfo.setMinimumSize(QtCore.QSize(0, 40))
|
||||
self.btn_getinfo.setObjectName("btn_getinfo")
|
||||
self.gridLayout.addWidget(self.btn_getinfo, 1, 2, 2, 1)
|
||||
self.checkBox = QtWidgets.QCheckBox(Dialog)
|
||||
self.checkBox.setText("")
|
||||
self.checkBox.setCheckable(False)
|
||||
self.checkBox.setObjectName("checkBox")
|
||||
self.gridLayout.addWidget(self.checkBox, 1, 3, 2, 1)
|
||||
self.label_2 = QtWidgets.QLabel(Dialog)
|
||||
self.label_2.setObjectName("label_2")
|
||||
self.gridLayout.addWidget(self.label_2, 2, 0, 1, 1)
|
||||
self.lineEdit_phone = QtWidgets.QLineEdit(Dialog)
|
||||
self.lineEdit_phone.setStyleSheet("background:transparent;\n"
|
||||
"\n"
|
||||
" border-radius:5px;\n"
|
||||
" border-top: 0px solid #b2e281;\n"
|
||||
" border-bottom: 2px solid black;\n"
|
||||
" border-right: 0px solid #b2e281;\n"
|
||||
" border-left: 0px solid #b2e281;\n"
|
||||
"\n"
|
||||
"\n"
|
||||
" border-style:outset\n"
|
||||
" ")
|
||||
self.lineEdit_phone.setFrame(False)
|
||||
self.lineEdit_phone.setObjectName("lineEdit_phone")
|
||||
self.gridLayout.addWidget(self.lineEdit_phone, 2, 1, 1, 1)
|
||||
self.label_5 = QtWidgets.QLabel(Dialog)
|
||||
self.label_5.setObjectName("label_5")
|
||||
self.gridLayout.addWidget(self.label_5, 3, 0, 1, 1)
|
||||
self.lineEdit_name = QtWidgets.QLineEdit(Dialog)
|
||||
self.lineEdit_name.setStyleSheet("background:transparent;\n"
|
||||
"\n"
|
||||
" border-radius:5px;\n"
|
||||
" border-top: 0px solid #b2e281;\n"
|
||||
" border-bottom: 2px solid black;\n"
|
||||
" border-right: 0px solid #b2e281;\n"
|
||||
" border-left: 0px solid #b2e281;\n"
|
||||
"\n"
|
||||
"\n"
|
||||
" border-style:outset\n"
|
||||
" ")
|
||||
self.lineEdit_name.setFrame(False)
|
||||
self.lineEdit_name.setObjectName("lineEdit_name")
|
||||
self.gridLayout.addWidget(self.lineEdit_name, 3, 1, 1, 1)
|
||||
self.label_4 = QtWidgets.QLabel(Dialog)
|
||||
self.label_4.setObjectName("label_4")
|
||||
self.gridLayout.addWidget(self.label_4, 4, 0, 1, 1)
|
||||
self.label_wxid = QtWidgets.QLabel(Dialog)
|
||||
self.label_wxid.setText("")
|
||||
self.label_wxid.setObjectName("label_wxid")
|
||||
self.gridLayout.addWidget(self.label_wxid, 4, 1, 1, 1)
|
||||
self.btn_db_dir = QtWidgets.QPushButton(Dialog)
|
||||
self.btn_db_dir.setMinimumSize(QtCore.QSize(0, 40))
|
||||
self.btn_db_dir.setObjectName("btn_db_dir")
|
||||
self.gridLayout.addWidget(self.btn_db_dir, 4, 2, 3, 1)
|
||||
self.label_6 = QtWidgets.QLabel(Dialog)
|
||||
self.label_6.setObjectName("label_6")
|
||||
self.gridLayout.addWidget(self.label_6, 5, 0, 1, 1)
|
||||
self.label_key = QtWidgets.QLabel(Dialog)
|
||||
self.label_key.setMaximumSize(QtCore.QSize(400, 16777215))
|
||||
self.label_key.setText("")
|
||||
self.label_key.setObjectName("label_key")
|
||||
self.gridLayout.addWidget(self.label_key, 5, 1, 1, 1)
|
||||
self.checkBox_2 = QtWidgets.QCheckBox(Dialog)
|
||||
self.checkBox_2.setText("")
|
||||
self.checkBox_2.setCheckable(False)
|
||||
self.checkBox_2.setObjectName("checkBox_2")
|
||||
self.gridLayout.addWidget(self.checkBox_2, 5, 3, 1, 1)
|
||||
self.label_8 = QtWidgets.QLabel(Dialog)
|
||||
self.label_8.setObjectName("label_8")
|
||||
self.gridLayout.addWidget(self.label_8, 6, 0, 1, 1)
|
||||
self.label_db_dir = QtWidgets.QLabel(Dialog)
|
||||
self.label_db_dir.setMaximumSize(QtCore.QSize(400, 300))
|
||||
self.label_db_dir.setText("")
|
||||
self.label_db_dir.setObjectName("label_db_dir")
|
||||
self.gridLayout.addWidget(self.label_db_dir, 6, 1, 1, 1)
|
||||
self.verticalLayout.addLayout(self.gridLayout)
|
||||
self.horizontalLayout_2 = QtWidgets.QHBoxLayout()
|
||||
self.horizontalLayout_2.setObjectName("horizontalLayout_2")
|
||||
spacerItem4 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_2.addItem(spacerItem4)
|
||||
self.btn_start = QtWidgets.QPushButton(Dialog)
|
||||
self.btn_start.setMinimumSize(QtCore.QSize(80, 40))
|
||||
self.btn_start.setMaximumSize(QtCore.QSize(200, 16777215))
|
||||
self.btn_start.setObjectName("btn_start")
|
||||
self.horizontalLayout_2.addWidget(self.btn_start)
|
||||
spacerItem5 = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_2.addItem(spacerItem5)
|
||||
self.verticalLayout.addLayout(self.horizontalLayout_2)
|
||||
self.horizontalLayout = QtWidgets.QHBoxLayout()
|
||||
self.horizontalLayout.setObjectName("horizontalLayout")
|
||||
self.label_ready = QtWidgets.QLabel(Dialog)
|
||||
self.label_ready.setObjectName("label_ready")
|
||||
self.horizontalLayout.addWidget(self.label_ready)
|
||||
self.progressBar = QtWidgets.QProgressBar(Dialog)
|
||||
self.progressBar.setProperty("value", 0)
|
||||
self.progressBar.setObjectName("progressBar")
|
||||
self.horizontalLayout.addWidget(self.progressBar)
|
||||
self.verticalLayout.addLayout(self.horizontalLayout)
|
||||
self.horizontalLayout_3.addLayout(self.verticalLayout)
|
||||
spacerItem6 = QtWidgets.QSpacerItem(128, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
|
||||
self.horizontalLayout_3.addItem(spacerItem6)
|
||||
self.verticalLayout_2.addLayout(self.horizontalLayout_3)
|
||||
spacerItem7 = QtWidgets.QSpacerItem(20, 79, QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Expanding)
|
||||
self.verticalLayout_2.addItem(spacerItem7)
|
||||
|
||||
self.retranslateUi(Dialog)
|
||||
QtCore.QMetaObject.connectSlotsByName(Dialog)
|
||||
|
||||
def retranslateUi(self, Dialog):
|
||||
_translate = QtCore.QCoreApplication.translate
|
||||
Dialog.setWindowTitle(_translate("Dialog", "Dialog"))
|
||||
self.btn_help.setText(_translate("Dialog", "使用说明"))
|
||||
self.label_3.setText(_translate("Dialog", "解密数据库"))
|
||||
self.label_9.setText(_translate("Dialog", "以下内容为自动获取,如获取失败请手动填写"))
|
||||
self.label.setText(_translate("Dialog", "PID"))
|
||||
self.label_7.setText(_translate("Dialog", "版本"))
|
||||
self.btn_getinfo.setText(_translate("Dialog", "获取信息"))
|
||||
self.label_2.setText(_translate("Dialog", "手机号"))
|
||||
self.label_5.setText(_translate("Dialog", "微信昵称"))
|
||||
self.label_4.setText(_translate("Dialog", "wxid"))
|
||||
self.btn_db_dir.setText(_translate("Dialog", "设置微信路径"))
|
||||
self.label_6.setText(_translate("Dialog", "密钥"))
|
||||
self.label_8.setText(_translate("Dialog", "微信路径"))
|
||||
self.btn_start.setText(_translate("Dialog", "开始启动"))
|
||||
self.label_ready.setText(_translate("Dialog", "未就绪"))
|
||||
340
app/ui/tool/pc_decrypt/pc_decrypt.py
Normal file
340
app/ui/tool/pc_decrypt/pc_decrypt.py
Normal file
@@ -0,0 +1,340 @@
|
||||
import json
|
||||
import os.path
|
||||
import sys
|
||||
import traceback
|
||||
from urllib.parse import urljoin
|
||||
|
||||
import requests
|
||||
from PyQt5.QtCore import pyqtSignal, QThread, QUrl
|
||||
from PyQt5.QtGui import QDesktopServices
|
||||
from PyQt5.QtWidgets import QWidget, QMessageBox, QFileDialog
|
||||
|
||||
from app.DataBase import msg_db, misc_db, close_db
|
||||
from app.DataBase.merge import merge_databases, merge_MediaMSG_databases
|
||||
from app.components.QCursorGif import QCursorGif
|
||||
from app.config import INFO_FILE_PATH, DB_DIR, SERVER_API_URL
|
||||
from app.decrypt import get_wx_info, decrypt
|
||||
from app.log import logger
|
||||
from app.util import path
|
||||
from . import decryptUi
|
||||
from ...Icon import Icon
|
||||
from ...menu.about_dialog import Decrypt
|
||||
|
||||
|
||||
class DecryptControl(QWidget, decryptUi.Ui_Dialog, QCursorGif):
|
||||
DecryptSignal = pyqtSignal(bool)
|
||||
get_wxidSignal = pyqtSignal(str)
|
||||
versionErrorSignal = pyqtSignal(str)
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super(DecryptControl, self).__init__(parent)
|
||||
self.max_val = 0
|
||||
self.setupUi(self)
|
||||
# 设置忙碌光标图片数组
|
||||
self.initCursor([':/icons/icons/Cursors/%d.png' %
|
||||
i for i in range(8)], self)
|
||||
self.setCursorTimeout(100)
|
||||
self.version_list = None
|
||||
self.btn_start.clicked.connect(self.decrypt)
|
||||
self.btn_getinfo.clicked.connect(self.get_info)
|
||||
self.btn_db_dir.clicked.connect(self.select_db_dir)
|
||||
# self.lineEdit.returnPressed.connect(self.set_wxid)
|
||||
# self.lineEdit.textChanged.connect(self.set_wxid_)
|
||||
self.lineEdit_name.returnPressed.connect(self.set_wxid)
|
||||
self.lineEdit_name.textChanged.connect(self.set_wxid_)
|
||||
self.lineEdit_phone.returnPressed.connect(self.set_wxid)
|
||||
self.lineEdit_phone.textChanged.connect(self.set_wxid_)
|
||||
self.btn_help.clicked.connect(self.show_help)
|
||||
self.btn_getinfo.setIcon(Icon.Get_info_Icon)
|
||||
self.btn_db_dir.setIcon(Icon.Folder_Icon)
|
||||
self.btn_start.setIcon(Icon.Start_Icon)
|
||||
self.btn_help.setIcon(Icon.Help_Icon)
|
||||
self.info = {}
|
||||
self.lineEdit_name.setFocus()
|
||||
self.ready = False
|
||||
self.wx_dir = None
|
||||
|
||||
def show_help(self):
|
||||
# 定义网页链接
|
||||
url = QUrl("https://blog.lc044.love/post/4")
|
||||
# 使用QDesktopServices打开网页
|
||||
QDesktopServices.openUrl(url)
|
||||
|
||||
# @log
|
||||
def get_info(self):
|
||||
self.startBusy()
|
||||
self.get_info_thread = MyThread(self.version_list)
|
||||
self.get_info_thread.signal.connect(self.set_info)
|
||||
self.get_info_thread.start()
|
||||
|
||||
def set_info(self, result):
|
||||
# print(result)
|
||||
if result[0] == -1:
|
||||
QMessageBox.critical(self, "错误", "请登录微信")
|
||||
elif result[0] == -2:
|
||||
self.versionErrorSignal.emit(result[1])
|
||||
QMessageBox.critical(self, "错误",
|
||||
"微信版本不匹配\n请手动填写信息")
|
||||
|
||||
elif result[0] == -3:
|
||||
QMessageBox.critical(self, "错误", "WeChat WeChatWin.dll Not Found")
|
||||
elif result[0] == -10086:
|
||||
QMessageBox.critical(self, "错误", "未知错误,请收集错误信息")
|
||||
else:
|
||||
self.ready = True
|
||||
self.info = result[0]
|
||||
self.label_key.setText(self.info['key'])
|
||||
self.label_wxid.setText(self.info['wxid'])
|
||||
self.lineEdit_name.setText(self.info['name'])
|
||||
self.lineEdit_phone.setText(self.info['mobile'])
|
||||
self.label_pid.setText(str(self.info['pid']))
|
||||
self.label_version.setText(self.info['version'])
|
||||
self.lineEdit_name.setFocus()
|
||||
self.checkBox.setCheckable(True)
|
||||
self.checkBox.setChecked(True)
|
||||
self.get_wxidSignal.emit(self.info['wxid'])
|
||||
directory = os.path.join(path.wx_path(), self.info['wxid'])
|
||||
if os.path.exists(directory):
|
||||
self.label_db_dir.setText(directory)
|
||||
self.wx_dir = directory
|
||||
self.checkBox_2.setCheckable(True)
|
||||
self.checkBox_2.setChecked(True)
|
||||
self.ready = True
|
||||
if self.ready:
|
||||
self.label_ready.setText('已就绪')
|
||||
if self.wx_dir and os.path.exists(os.path.join(self.wx_dir)):
|
||||
self.label_ready.setText('已就绪')
|
||||
self.stopBusy()
|
||||
|
||||
def set_wxid_(self):
|
||||
if self.sender() == self.lineEdit_name:
|
||||
self.info['name'] = self.lineEdit_name.text()
|
||||
elif self.sender() == self.lineEdit_phone:
|
||||
self.info['mobel'] = self.lineEdit_phone.text()
|
||||
|
||||
def set_wxid(self):
|
||||
if self.sender() == self.lineEdit_name:
|
||||
self.info['name'] = self.lineEdit_name.text()
|
||||
QMessageBox.information(self, "ok", f"昵称修改成功{self.info['name']}")
|
||||
elif self.sender() == self.lineEdit_phone:
|
||||
self.info['mobile'] = self.lineEdit_phone.text()
|
||||
QMessageBox.information(self, "ok", f"手机号修改成功{self.info['mobile']}")
|
||||
|
||||
def select_db_dir(self):
|
||||
directory = QFileDialog.getExistingDirectory(
|
||||
self, "选取微信文件保存目录——能看到Msg文件夹",
|
||||
path.wx_path()
|
||||
) # 起始路径
|
||||
db_dir = os.path.join(directory, 'Msg')
|
||||
if not os.path.exists(db_dir):
|
||||
QMessageBox.critical(self, "错误", "文件夹选择错误\n一般以wxid_xxx结尾")
|
||||
return
|
||||
|
||||
self.label_db_dir.setText(directory)
|
||||
self.wx_dir = directory
|
||||
self.checkBox_2.setCheckable(True)
|
||||
self.checkBox_2.setChecked(True)
|
||||
if self.ready:
|
||||
self.label_ready.setText('已就绪')
|
||||
|
||||
def decrypt(self):
|
||||
if not self.ready:
|
||||
QMessageBox.critical(self, "错误", "请先获取信息")
|
||||
return
|
||||
if not self.wx_dir:
|
||||
QMessageBox.critical(self, "错误", "请先选择微信安装路径")
|
||||
return
|
||||
if self.label_wxid.text() == 'None':
|
||||
QMessageBox.critical(self, "错误", "请填入wxid")
|
||||
return
|
||||
db_dir = os.path.join(self.wx_dir, 'Msg')
|
||||
if self.ready:
|
||||
if not os.path.exists(db_dir):
|
||||
QMessageBox.critical(self, "错误", "文件夹选择错误\n一般以wxid_xxx结尾")
|
||||
return
|
||||
if self.info.get('key') == 'None':
|
||||
QMessageBox.critical(self, "错误",
|
||||
"密钥错误\n请查看教程解决相关问题")
|
||||
close_db()
|
||||
self.thread2 = DecryptThread(db_dir, self.info['key'])
|
||||
self.thread2.maxNumSignal.connect(self.setProgressBarMaxNum)
|
||||
self.thread2.signal.connect(self.progressBar_view)
|
||||
self.thread2.okSignal.connect(self.btnExitClicked)
|
||||
self.thread2.errorSignal.connect(
|
||||
lambda x: QMessageBox.critical(self, "错误",
|
||||
"错误\n请检查微信版本是否为最新和微信路径是否正确\n或者关闭微信多开")
|
||||
)
|
||||
self.thread2.start()
|
||||
|
||||
def btnEnterClicked(self):
|
||||
# print("enter clicked")
|
||||
# 中间可以添加处理逻辑
|
||||
# QMessageBox.about(self, "解密成功", "数据库文件存储在app/DataBase/Msg文件夹下")
|
||||
self.progressBar_view(self.max_val)
|
||||
self.DecryptSignal.emit(True)
|
||||
# self.close()
|
||||
|
||||
def setProgressBarMaxNum(self, max_val):
|
||||
self.max_val = max_val
|
||||
self.progressBar.setRange(0, max_val)
|
||||
|
||||
def progressBar_view(self, value):
|
||||
"""
|
||||
进度条显示
|
||||
:param value: 进度0-100
|
||||
:return: None
|
||||
"""
|
||||
self.progressBar.setProperty('value', value)
|
||||
# self.btnExitClicked()
|
||||
# data.init_database()
|
||||
|
||||
def btnExitClicked(self):
|
||||
# print("Exit clicked")
|
||||
dic = {
|
||||
'wxid': self.info['wxid'],
|
||||
'wx_dir': self.wx_dir,
|
||||
'name': self.info['name'],
|
||||
'mobile': self.info['mobile'],
|
||||
'token': Decrypt.decrypt(self.info['wxid'])
|
||||
}
|
||||
try:
|
||||
with open(INFO_FILE_PATH, "w", encoding="utf-8") as f:
|
||||
json.dump(dic, f, ensure_ascii=False, indent=4)
|
||||
except:
|
||||
with open('./info.json', 'w', encoding='utf-8') as f:
|
||||
f.write(json.dumps(dic))
|
||||
self.progressBar_view(self.max_val)
|
||||
self.DecryptSignal.emit(True)
|
||||
self.close()
|
||||
|
||||
|
||||
class DecryptThread(QThread):
|
||||
signal = pyqtSignal(str)
|
||||
maxNumSignal = pyqtSignal(int)
|
||||
okSignal = pyqtSignal(str)
|
||||
errorSignal = pyqtSignal(bool)
|
||||
|
||||
def __init__(self, db_path, key):
|
||||
super(DecryptThread, self).__init__()
|
||||
self.db_path = db_path
|
||||
self.key = key
|
||||
self.textBrowser = None
|
||||
|
||||
def __del__(self):
|
||||
pass
|
||||
|
||||
def run(self):
|
||||
close_db()
|
||||
output_dir = DB_DIR
|
||||
os.makedirs(output_dir, exist_ok=True)
|
||||
tasks = []
|
||||
if os.path.exists(self.db_path):
|
||||
for root, dirs, files in os.walk(self.db_path):
|
||||
for file in files:
|
||||
if '.db' == file[-3:]:
|
||||
if 'xInfo.db' == file:
|
||||
continue
|
||||
inpath = os.path.join(root, file)
|
||||
# print(inpath)
|
||||
output_path = os.path.join(output_dir, file)
|
||||
tasks.append([self.key, inpath, output_path])
|
||||
else:
|
||||
try:
|
||||
name, suffix = file.split('.')
|
||||
if suffix.startswith('db_SQLITE'):
|
||||
inpath = os.path.join(root, file)
|
||||
# print(inpath)
|
||||
output_path = os.path.join(output_dir, name + '.db')
|
||||
tasks.append([self.key, inpath, output_path])
|
||||
except:
|
||||
continue
|
||||
self.maxNumSignal.emit(len(tasks))
|
||||
for i, task in enumerate(tasks):
|
||||
if decrypt.decrypt(*task) == -1:
|
||||
self.errorSignal.emit(True)
|
||||
self.signal.emit(str(i))
|
||||
# print(self.db_path)
|
||||
# 目标数据库文件
|
||||
target_database = os.path.join(DB_DIR, 'MSG.db')
|
||||
# 源数据库文件列表
|
||||
source_databases = [os.path.join(DB_DIR, f"MSG{i}.db") for i in range(1, 50)]
|
||||
import shutil
|
||||
if os.path.exists(target_database):
|
||||
os.remove(target_database)
|
||||
shutil.copy2(os.path.join(DB_DIR, 'MSG0.db'), target_database) # 使用一个数据库文件作为模板
|
||||
# 合并数据库
|
||||
merge_databases(source_databases, target_database)
|
||||
|
||||
# 音频数据库文件
|
||||
target_database = os.path.join(DB_DIR, 'MediaMSG.db')
|
||||
# 源数据库文件列表
|
||||
if os.path.exists(target_database):
|
||||
os.remove(target_database)
|
||||
source_databases = [os.path.join(DB_DIR, f"MediaMSG{i}.db") for i in range(1, 50)]
|
||||
shutil.copy2(os.path.join(DB_DIR, 'MediaMSG0.db'), target_database) # 使用一个数据库文件作为模板
|
||||
|
||||
# 合并数据库
|
||||
merge_MediaMSG_databases(source_databases, target_database)
|
||||
self.okSignal.emit('ok')
|
||||
# self.signal.emit('100')
|
||||
|
||||
|
||||
class MyThread(QThread):
|
||||
signal = pyqtSignal(list)
|
||||
|
||||
def __init__(self, version_list=None):
|
||||
super(MyThread, self).__init__()
|
||||
self.version_list = version_list
|
||||
|
||||
def __del__(self):
|
||||
pass
|
||||
|
||||
def get_bias_add(self, version):
|
||||
url = urljoin(SERVER_API_URL, 'wxBiasAddr')
|
||||
data = {
|
||||
'version': version
|
||||
}
|
||||
try:
|
||||
response = requests.get(url, json=data)
|
||||
print(response)
|
||||
print(response.text)
|
||||
if response.status_code == 200:
|
||||
update_info = response.json()
|
||||
return update_info
|
||||
else:
|
||||
return {}
|
||||
except:
|
||||
return {}
|
||||
|
||||
def run(self):
|
||||
if self.version_list:
|
||||
VERSION_LIST = self.version_list
|
||||
else:
|
||||
file_path = './app/resources/data/version_list.json'
|
||||
if not os.path.exists(file_path):
|
||||
resource_dir = getattr(sys, '_MEIPASS', os.path.abspath(os.path.dirname(__file__)))
|
||||
file_path = os.path.join(resource_dir, 'app', 'resources', 'data', 'version_list.json')
|
||||
with open(file_path, "r", encoding="utf-8") as f:
|
||||
VERSION_LIST = json.loads(f.read())
|
||||
try:
|
||||
result = get_wx_info.get_info(VERSION_LIST)
|
||||
if result == -1:
|
||||
result = [result]
|
||||
elif result == -2:
|
||||
result = [result]
|
||||
elif result == -3:
|
||||
result = [result]
|
||||
elif isinstance(result, str):
|
||||
version = result
|
||||
# version = '3.9.9.43'
|
||||
version_bias = self.get_bias_add(version)
|
||||
if version_bias.get(version):
|
||||
logger.info(f"从云端获取内存基址:{version_bias}")
|
||||
result = get_wx_info.get_info(version_bias)
|
||||
else:
|
||||
logger.info(f"从云端获取内存基址失败:{version}")
|
||||
result = [-2, version]
|
||||
except:
|
||||
logger.error(traceback.format_exc())
|
||||
result = [-10086]
|
||||
self.signal.emit(result)
|
||||
Reference in New Issue
Block a user