统一管理导出文件路径

This commit is contained in:
SiYuan
2024-02-04 23:11:43 +08:00
parent 1fa8241aba
commit 4f868525dd
12 changed files with 111 additions and 108 deletions

View File

@@ -11,6 +11,7 @@ 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
from app.decrypt import get_wx_info, decrypt
from app.log import logger
from app.util import path
@@ -185,7 +186,7 @@ class DecryptControl(QWidget, decryptUi.Ui_Dialog, QCursorGif):
try:
os.makedirs('./app/data', exist_ok=True)
with open('./app/data/info.json', "w", encoding="utf-8") as f:
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:
@@ -215,7 +216,7 @@ class DecryptThread(QThread):
msg_db.close()
# micro_msg_db.close()
# hard_link_db.close()
output_dir = 'app/DataBase/Msg'
output_dir = db_dir
os.makedirs(output_dir, exist_ok=True)
tasks = []
if os.path.exists(self.db_path):
@@ -235,14 +236,14 @@ class DecryptThread(QThread):
self.signal.emit(str(i))
# print(self.db_path)
# 目标数据库文件
target_database = "app/DataBase/Msg/MSG.db"
target_database = os.path.join(db_dir, 'MSG.db')
# 源数据库文件列表
source_databases = [f"app/DataBase/Msg/MSG{i}.db" for i in range(1, 50)]
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)
try:
shutil.copy2("app/DataBase/Msg/MSG0.db", target_database) # 使用一个数据库文件作为模板
shutil.copy2(os.path.join(db_dir, 'MSG0.db'), target_database) # 使用一个数据库文件作为模板
except FileNotFoundError:
logger.error(traceback.format_exc())
self.errorSignal.emit(True)
@@ -254,13 +255,13 @@ class DecryptThread(QThread):
QMessageBox.critical("错误", "数据库不存在\n请检查微信版本是否为最新")
# 音频数据库文件
target_database = "app/DataBase/Msg/MediaMSG.db"
target_database = os.path.join(db_dir,'MediaMSG.db')
# 源数据库文件列表
if os.path.exists(target_database):
os.remove(target_database)
source_databases = [f"app/DataBase/Msg/MediaMSG{i}.db" for i in range(1, 50)]
source_databases = [os.path.join(db_dir,f"MediaMSG{i}.db") for i in range(1, 50)]
try:
shutil.copy2("app/DataBase/Msg/MediaMSG0.db", target_database) # 使用一个数据库文件作为模板
shutil.copy2(os.path.join(db_dir,'MediaMSG0.db'), target_database) # 使用一个数据库文件作为模板
except FileNotFoundError:
logger.error(traceback.format_exc())
self.errorSignal.emit(True)
@@ -277,7 +278,7 @@ class DecryptThread(QThread):
class MyThread(QThread):
signal = pyqtSignal(list)
def __init__(self,version_list = None):
def __init__(self, version_list=None):
super(MyThread, self).__init__()
self.version_list = version_list
@@ -328,7 +329,7 @@ class MyThread(QThread):
result = get_wx_info.get_info(version_bias)
else:
logger.info(f"从云端获取内存基址失败:{version}")
result = [-2,version]
result = [-2, version]
except:
logger.error(traceback.format_exc())
result = [-10086]