158 lines
5.6 KiB
Python
158 lines
5.6 KiB
Python
# coding: utf8
|
|
import os
|
|
from pathlib import Path
|
|
from PySide6 import QtWidgets, QtCore, QtGui
|
|
from util_ext import ProfilesData, DeleteThread, DeleteThreadManager
|
|
from global_vars import accept_warning
|
|
|
|
|
|
def sort_profiles_id_func(profile_id: str) -> int:
|
|
if profile_id == "Default":
|
|
return 0
|
|
else:
|
|
seq = profile_id.split(" ", 1)[-1]
|
|
try:
|
|
return int(seq)
|
|
except ValueError:
|
|
# if the id is weird
|
|
return 999
|
|
|
|
|
|
class ProfilesModel(QtCore.QAbstractTableModel):
|
|
|
|
def __init__(self,
|
|
profiles_data: ProfilesData,
|
|
ext_id: str,
|
|
profiles: list[str],
|
|
parent=None):
|
|
super().__init__(parent)
|
|
self.ext_id = ext_id
|
|
self.profiles = profiles
|
|
self.profiles.sort(key=sort_profiles_id_func)
|
|
self.profiles_data = profiles_data
|
|
|
|
def rowCount(self, parent: QtCore.QModelIndex = ...):
|
|
return len(self.profiles)
|
|
|
|
def columnCount(self, parent: QtCore.QModelIndex = ...):
|
|
return 2
|
|
|
|
def data(self, index: QtCore.QModelIndex, role: int = ...):
|
|
row = index.row()
|
|
col = index.column()
|
|
|
|
if col == 0:
|
|
if role == QtCore.Qt.ItemDataRole.DisplayRole:
|
|
return self.profiles[row]
|
|
elif col == 1:
|
|
if role == QtCore.Qt.ItemDataRole.DisplayRole:
|
|
return self.profiles_data[self.profiles[row]].name
|
|
|
|
def headerData(self, section: int, orientation: QtCore.Qt.Orientation, role: int = ...):
|
|
if role == QtCore.Qt.ItemDataRole.DisplayRole:
|
|
if section == 0:
|
|
return "ID"
|
|
if section == 1:
|
|
return "名称"
|
|
|
|
|
|
class DaShowProfiles(QtWidgets.QDialog):
|
|
|
|
def __init__(self,
|
|
browser: str,
|
|
is_compat: bool,
|
|
profiles_data: ProfilesData,
|
|
ext_id: str,
|
|
ext_name: str,
|
|
ext_icon: QtGui.QIcon,
|
|
profiles: list[str],
|
|
parent: QtWidgets.QWidget = None):
|
|
super().__init__(parent)
|
|
self.setWindowTitle(ext_name)
|
|
self.setWindowIcon(ext_icon)
|
|
self.browser = browser
|
|
self.is_compat = is_compat
|
|
|
|
self.process = QtCore.QProcess(self)
|
|
|
|
# ========== UI ==============
|
|
|
|
self.vly_m = QtWidgets.QVBoxLayout()
|
|
self.setLayout(self.vly_m)
|
|
|
|
self.lne_info = QtWidgets.QLineEdit(ext_id, self)
|
|
self.lne_info.setReadOnly(True)
|
|
self.vly_m.addWidget(self.lne_info)
|
|
|
|
self.trv_profiles = QtWidgets.QTreeView(self)
|
|
self.trv_profiles.setSelectionMode(QtWidgets.QAbstractItemView.SelectionMode.ExtendedSelection)
|
|
self.vly_m.addWidget(self.trv_profiles)
|
|
|
|
self.pgb_del = QtWidgets.QProgressBar(self)
|
|
self.pgb_del.setAlignment(QtCore.Qt.AlignmentFlag.AlignCenter)
|
|
self.vly_m.addWidget(self.pgb_del)
|
|
|
|
self.hly_bot = QtWidgets.QHBoxLayout()
|
|
self.pbn_delete_selected = QtWidgets.QPushButton("删除所选", self)
|
|
self.pbn_open = QtWidgets.QPushButton("打开", self)
|
|
self.pbn_cancel = QtWidgets.QPushButton("取消", self)
|
|
self.hly_bot.addWidget(self.pbn_delete_selected)
|
|
self.hly_bot.addStretch(1)
|
|
self.hly_bot.addWidget(self.pbn_open)
|
|
self.hly_bot.addWidget(self.pbn_cancel)
|
|
self.vly_m.addLayout(self.hly_bot)
|
|
|
|
self.profiles_model = ProfilesModel(profiles_data, ext_id, profiles, self)
|
|
self.trv_profiles.setModel(self.profiles_model)
|
|
|
|
self.pbn_delete_selected.clicked.connect(self.on_pbn_delete_selected_clicked)
|
|
self.pbn_open.clicked.connect(self.on_pbn_open_clicked)
|
|
self.pbn_cancel.clicked.connect(self.reject)
|
|
|
|
def sizeHint(self):
|
|
return QtCore.QSize(400, 360)
|
|
|
|
def on_pbn_open_clicked(self):
|
|
us = QtCore.QSettings()
|
|
exec_path = str(us.value(f"{self.browser}Exec", ""))
|
|
if len(exec_path) == 0 or not os.path.exists(exec_path):
|
|
QtWidgets.QMessageBox.critical(self, "错误", f"无法找到 {self.browser} 浏览器的执行路径")
|
|
return
|
|
|
|
indexes = self.trv_profiles.selectedIndexes()
|
|
cmd = rf'"{exec_path}" --profile-directory="{{0}}"'
|
|
for idx in indexes:
|
|
if idx.column() != 0:
|
|
continue
|
|
profile_id = self.profiles_model.data(idx, QtCore.Qt.ItemDataRole.DisplayRole)
|
|
# setProgram 不行，不知道为什么，莫名其妙
|
|
self.process.startCommand(cmd.format(profile_id))
|
|
self.process.waitForFinished(10000)
|
|
|
|
def on_pbn_delete_selected_clicked(self):
|
|
us = QtCore.QSettings()
|
|
user_data_path = str(us.value(f"{self.browser}Data", ""))
|
|
if self.is_compat:
|
|
pref_name = "Preferences"
|
|
else:
|
|
pref_name = "Secure Preferences"
|
|
|
|
indexes = self.trv_profiles.selectedIndexes()
|
|
total = len(indexes) // 2
|
|
if accept_warning(self, True, "警告", f"确定要删除这 {total} 项吗？"):
|
|
return
|
|
|
|
del_thd_mgr = DeleteThreadManager(total, self.pgb_del, self)
|
|
|
|
for idx in indexes:
|
|
if idx.column() != 0:
|
|
continue
|
|
profile_id = self.profiles_model.data(idx, QtCore.Qt.ItemDataRole.DisplayRole)
|
|
del_thd = DeleteThread(
|
|
str(Path(user_data_path, profile_id)),
|
|
pref_name,
|
|
[self.lne_info.text()],
|
|
self
|
|
)
|
|
del_thd_mgr.start(del_thd)
|