julian/KpsUnifier
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.2.0

Browse Source
This commit is contained in:
Julian Freeman
2024-08-11 17:01:46 -04:00
parent 362db77ff7
commit 030e5eec55
10 changed files with 164 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/da_target_login.py
View File

@@ -1,4 +1,6 @@
# coding: utf8
from pathlib import Path
from PySide6 import QtWidgets, QtCore, QtGui
from pykeepass import PyKeePass
from pykeepass.exceptions import HeaderChecksumError, CredentialsError
@@ -45,10 +47,11 @@ class UiDaTargetLogin(object):
class DaTargetLogin(QtWidgets.QDialog):
def __init__(self, parent=None):
def __init__(self, config: dict, parent=None):
super().__init__(parent)
self.ui = UiDaTargetLogin(self)
self.tar_kp: PyKeePass | None = None
self.config = config
self.ui.pbn_browse.clicked.connect(self.on_pbn_browse_clicked)
self.ui.pbn_eye.clicked.connect(self.on_pbn_eye_clicked)
@@ -56,11 +59,12 @@ class DaTargetLogin(QtWidgets.QDialog):
self.ui.pbn_cancel.clicked.connect(self.on_pbn_cancel_clicked)
def on_pbn_browse_clicked(self):
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "选择", "../",
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "选择", self.config["last_open_path"],
filter="KeePass 2 数据库 (*.kdbx);;所有文件 (*)")
if len(filename) == 0:
return
self.ui.lne_path.setText(filename)
self.config["last_open_path"] = str(Path(filename).parent)
def on_pbn_eye_clicked(self):
if self.ui.lne_password.echoMode() == QtWidgets.QLineEdit.EchoMode.Password:

3
src/gbx_kps_login.py
View File

@@ -97,6 +97,9 @@ class GbxKpsLogin(QtWidgets.QGroupBox):
QtWidgets.QMessageBox.critical(self, "密码错误",
f"{self.lne_path.text()}\n密码错误")
return
except ValueError as e:
QtWidgets.QMessageBox.critical(self, "错误", str(e))
return
self.file_kp[self.lne_path.text()] = kp
self.sec_sqh.insert_into("secrets", insert_sec_columns, [

8
src/mw_kps_unifier.py
View File

@@ -1,4 +1,6 @@
# coding: utf8
from pathlib import Path
from PySide6 import QtWidgets, QtCore, QtGui
from pykeepass import PyKeePass
@@ -131,18 +133,20 @@ class KpsUnifier(QtWidgets.QMainWindow):
self.ui.lne_db_path.setText(filename)
def on_act_new_triggered(self):
filename, _ = QtWidgets.QFileDialog.getSaveFileName(self, "新建", "../",
filename, _ = QtWidgets.QFileDialog.getSaveFileName(self, "新建", self.config["last_open_path"],
filter="数据库 (*.db);;所有文件 (*)")
if len(filename) == 0:
return
self.update_db(filename)
self.config["last_open_path"] = str(Path(filename).parent)
def on_act_open_triggered(self):
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开", "../",
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开", self.config["last_open_path"],
filter="数据库 (*.db);;所有文件 (*)")
if len(filename) == 0:
return
self.update_db(filename)
self.config["last_open_path"] = str(Path(filename).parent)
def on_act_load_triggered(self):
self.ui.sw_m.setCurrentIndex(0)

16
src/page_load.py
View File

@@ -109,15 +109,17 @@ class PageLoad(QtWidgets.QWidget):
self.pbn_clear_loaded_mem.clicked.connect(self.on_pbn_clear_loaded_mem_clicked)
def update_sqh(self, sqh: Sqlite3Worker):
self.sqh = sqh
self.wg_sa.update_sqh(sqh)
def on_pbn_add_kps_clicked(self):
filenames, _ = QtWidgets.QFileDialog.getOpenFileNames(self, "选择", "../",
filenames, _ = QtWidgets.QFileDialog.getOpenFileNames(self, "选择", self.config["last_open_path"],
filter="KeePass 2 数据库 (*.kdbx);;所有文件 (*)")
if len(filenames) == 0:
return
for filename in filenames:
self.wg_sa.add_kps(filename)
self.config["last_open_path"] = str(Path(filenames[0]).parent)
def on_pbn_clear_db_clicked(self):
if accept_warning(self, True, "警告", "你确定要清空当前数据库吗？"):
@@ -141,12 +143,16 @@ class PageLoad(QtWidgets.QWidget):
self.wg_sa.update_load_status(wg)
def on_pbn_clear_loaded_mem_clicked(self):
if accept_warning(self, True, "警告", "你确定要清空所有加载记忆吗？"):
if accept_warning(self, True, "警告", "你确定要清空当前加载记忆吗？"):
return
loaded_mem: dict = self.config["loaded_memory"]
loaded_mem.clear()
QtWidgets.QMessageBox.information(self, "提示", "已清空加载记忆")
filename = str(Path(self.sqh.db_name).name)
loaded_mem: list = self.config["loaded_memory"].get(filename, None)
if loaded_mem is None:
QtWidgets.QMessageBox.warning(self, "警告", f"没有找到 {filename} 的加载记忆")
else:
loaded_mem.clear()
QtWidgets.QMessageBox.information(self, "提示", "已清空加载记忆")
# 更新kps加载状态
for wg in self.wg_sa.kps_wgs:

52
src/page_query.py
View File

@@ -1,5 +1,6 @@
# coding: utf8
import json
from pathlib import Path
from uuid import UUID
from PySide6 import QtWidgets, QtCore, QtGui
from pykeepass import PyKeePass
@@ -117,7 +118,13 @@ class UiPageQuery(object):
self.vly_sa_wg.addStretch(1)
self.pbn_read_filters = QtWidgets.QPushButton("更多过滤", window)
self.lne_entries_count = QtWidgets.QLineEdit(self.sa_wg)
self.lne_entries_count.setDisabled(True)
self.lne_entries_count.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
self.lne_entries_count.setMinimumWidth(config["button_min_width"])
self.vly_sa_wg.addWidget(self.lne_entries_count)
self.pbn_read_filters = QtWidgets.QPushButton("更多过滤", self.sa_wg)
self.pbn_read_filters.setMinimumWidth(config["button_min_width"])
self.vly_sa_wg.addWidget(self.pbn_read_filters)
@@ -181,11 +188,11 @@ class PageQuery(QtWidgets.QWidget):
def set_filter_button(self, fil: dict):
pbn_fil = PushButtonWithData(fil, self.ui.sa_wg, fil["name"])
pbn_fil.setMinimumWidth(self.config["button_min_width"])
self.ui.vly_sa_wg.insertWidget(self.ui.vly_sa_wg.count() - 2, pbn_fil)
self.ui.vly_sa_wg.insertWidget(self.ui.vly_sa_wg.count() - 3, pbn_fil)
pbn_fil.clicked_with_data.connect(self.on_custom_filters_clicked_with_data)
def on_pbn_read_filters_clicked(self):
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开", "../",
filename, _ = QtWidgets.QFileDialog.getOpenFileName(self, "打开", self.config["last_open_path"],
filter="JSON 文件 (*.json);;所有文件 (*)")
if len(filename) == 0:
return
@@ -194,12 +201,14 @@ class PageQuery(QtWidgets.QWidget):
for fil in filter_ls:
self.set_filter_button(fil)
self.config["last_open_path"] = str(Path(filename).parent)
def on_custom_filters_clicked_with_data(self, data: dict):
_, results = self.sqh.select("entries", query_columns,
where=Expression(data["where"]).and_(Operand(deleted_col).equal_to(0)))
model = QueryTableModel(results, self)
self.ui.trv_m.setModel(model)
self.ui.lne_entries_count.setText(str(model.rowCount()))
def update_sqh(self, sqh: Sqlite3Worker):
self.sqh = sqh
@@ -209,6 +218,7 @@ class PageQuery(QtWidgets.QWidget):
where=Operand(deleted_col).equal_to(0))
model = QueryTableModel(results, self)
self.ui.trv_m.setModel(model)
self.ui.lne_entries_count.setText(str(model.rowCount()))
def on_pbn_deleted_clicked(self):
_, results = self.sqh.select("entries", query_columns,
@@ -247,13 +257,10 @@ class PageQuery(QtWidgets.QWidget):
kp = self.file_kp[filepath]
return kp
def delete_the_delete_and_transfer(self, transfer: bool = False):
cond = Operand(status_col).equal_to("delete")
if transfer is True:
cond = cond.or_(Operand(status_col).equal_to("transfer"), high_priority=True)
cond = cond.and_(Operand(deleted_col).equal_to(0))
_, results = self.sqh.select("entries", sim_columns, where=cond)
def delete_the_delete(self):
_, results = self.sqh.select("entries", sim_columns,
where=Operand(status_col).equal_to("delete")
.and_(Operand(deleted_col).equal_to(0)))
file_uuids = get_filepath_uuids_map(results)
total, success, invalid = 0, 0, 0
@@ -272,10 +279,11 @@ class PageQuery(QtWidgets.QWidget):
continue
kp.delete_entry(entry)
success += 1
self.sqh.update("entries", [(deleted_col, 1)],
where=Operand(uuid_col).equal_to(u).and_(
Operand(filepath_col).equal_to(blob_fy(file))))
success += 1
kp.save()
@@ -309,17 +317,15 @@ class PageQuery(QtWidgets.QWidget):
invalid += 1
continue
self.tar_kp.add_entry(
dest_group,
entry.title or "",
entry.username or "",
entry.password or "",
entry.url,
entry.notes,
otp=entry.otp,
force_creation=True
)
kp.move_entry(entry, dest_group)
success += 1
self.sqh.update("entries", [(deleted_col, 1)],
where=Operand(uuid_col).equal_to(u).and_(
Operand(filepath_col).equal_to(blob_fy(file))))
kp.save()
self.tar_kp.save()
QtWidgets.QMessageBox.information(self, "提示",
f"共 {total} 条转移条目，成功 {success} 条，失败 {invalid} 条。")
@@ -336,10 +342,10 @@ class PageQuery(QtWidgets.QWidget):
if transfer:
self.transfer_the_transfer()
self.delete_the_delete_and_transfer(transfer)
self.delete_the_delete()
def on_pbn_set_target_clicked(self):
da_target_login = DaTargetLogin(self)
da_target_login = DaTargetLogin(self.config, self)
da_target_login.exec()
self.tar_kp = da_target_login.tar_kp