244 lines
8.8 KiB
Python
244 lines
8.8 KiB
Python
# coding: utf8
|
|
from PySide6 import QtWidgets, QtCore, QtGui
|
|
from util_ext import scan_extensions, ExtensionsData
|
|
from global_vars import (
|
|
ExtensionStatusRole,
|
|
ExtensionIdRole,
|
|
)
|
|
|
|
|
|
class UiWgExtensions(object):
|
|
|
|
def __init__(self, window: QtWidgets.QWidget):
|
|
self.vly_m = QtWidgets.QVBoxLayout()
|
|
window.setLayout(self.vly_m)
|
|
|
|
self.hly_top = QtWidgets.QHBoxLayout()
|
|
self.vly_m.addLayout(self.hly_top)
|
|
self.cmbx_browsers = QtWidgets.QComboBox(window)
|
|
self.cbx_safe = QtWidgets.QCheckBox("安全", window)
|
|
self.cbx_unsafe = QtWidgets.QCheckBox("不安全", window)
|
|
self.cbx_unknown = QtWidgets.QCheckBox("未知", window)
|
|
self.cbx_safe.setChecked(True)
|
|
self.cbx_unsafe.setChecked(True)
|
|
self.cbx_unknown.setChecked(True)
|
|
self.cbx_chrome_compat = QtWidgets.QCheckBox("谷歌兼容模式", window)
|
|
self.pbn_update = QtWidgets.QPushButton("更新", window)
|
|
self.hly_top.addWidget(self.cmbx_browsers)
|
|
self.hly_top.addWidget(self.cbx_safe)
|
|
self.hly_top.addWidget(self.cbx_unsafe)
|
|
self.hly_top.addWidget(self.cbx_unknown)
|
|
self.hly_top.addStretch(1)
|
|
self.hly_top.addWidget(self.cbx_chrome_compat)
|
|
self.hly_top.addWidget(self.pbn_update)
|
|
|
|
self.lv_extensions = QtWidgets.QListView(window)
|
|
self.vly_m.addWidget(self.lv_extensions)
|
|
|
|
|
|
class BaseExtensionsListModel(QtCore.QAbstractListModel):
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.all_extensions = {} # type: ExtensionsData
|
|
self.names = [] # type: list[tuple[str, str]]
|
|
self.icons = {} # type: dict[str, QtGui.QIcon]
|
|
self.safe_info = {} # type: dict[str, dict]
|
|
self.blank_icon = QtGui.QIcon(":/images/blank_128.png")
|
|
|
|
def update(self, **kwargs):
|
|
raise NotImplementedError
|
|
|
|
def update_ext(self, browser: str, is_chrome_compat=False):
|
|
"""内部用"""
|
|
self.all_extensions.clear()
|
|
self.names.clear()
|
|
self.icons.clear()
|
|
|
|
self.all_extensions = scan_extensions(browser, is_chrome_compat)
|
|
for ext_id in self.all_extensions:
|
|
name = self.all_extensions[ext_id].name
|
|
icon = self.all_extensions[ext_id].icon
|
|
self.names.append((ext_id, name))
|
|
if len(icon) == 0:
|
|
self.icons[ext_id] = self.blank_icon
|
|
else:
|
|
self.icons[ext_id] = QtGui.QIcon(icon)
|
|
self.names.sort(key=lambda x: x[1].lower())
|
|
|
|
# print("updated " + browser)
|
|
|
|
def update_safe_info(self, safe_info: dict):
|
|
self.safe_info.clear()
|
|
self.safe_info = safe_info
|
|
|
|
def rowCount(self, parent: QtCore.QModelIndex = ...):
|
|
return len(self.names)
|
|
|
|
def data(self, index: QtCore.QModelIndex, role: int = ...):
|
|
row = index.row()
|
|
ext_id, name = self.names[row]
|
|
|
|
if role == QtCore.Qt.ItemDataRole.DisplayRole:
|
|
return name
|
|
if role == QtCore.Qt.ItemDataRole.DecorationRole:
|
|
return self.icons[ext_id]
|
|
if role == QtCore.Qt.ItemDataRole.BackgroundRole:
|
|
is_safe = self.data(index, ExtensionStatusRole)
|
|
if is_safe is True:
|
|
return QtGui.QBrush(QtGui.QColor("lightgreen"))
|
|
elif is_safe is False:
|
|
return QtGui.QBrush(QtGui.QColor("lightpink"))
|
|
else:
|
|
return QtGui.QBrush(QtCore.Qt.BrushStyle.NoBrush)
|
|
if role == ExtensionStatusRole:
|
|
if ext_id not in self.safe_info:
|
|
return None
|
|
else:
|
|
return self.safe_info[ext_id]["safe"]
|
|
if role == ExtensionIdRole:
|
|
return ext_id
|
|
|
|
|
|
class ChromeExtensionsListModel(BaseExtensionsListModel):
|
|
|
|
def update(self, is_chrome_compat=False):
|
|
super().update_ext("Chrome", is_chrome_compat)
|
|
|
|
|
|
class EdgeExtensionsListModel(BaseExtensionsListModel):
|
|
|
|
def update(self, **kwargs):
|
|
super().update_ext("Edge")
|
|
|
|
|
|
class BraveExtensionsListModel(BaseExtensionsListModel):
|
|
|
|
def update(self, **kwargs):
|
|
super().update_ext("Brave")
|
|
|
|
|
|
class BrowsersListModel(QtCore.QAbstractListModel):
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.browsers = ["Chrome", "Edge", "Brave"]
|
|
self.icons = [
|
|
QtGui.QIcon(":/images/browsers/chrome_32.png"),
|
|
QtGui.QIcon(":/images/browsers/edge_32.png"),
|
|
QtGui.QIcon(":/images/browsers/brave_32.png"),
|
|
]
|
|
|
|
def rowCount(self, parent: QtCore.QModelIndex = ...):
|
|
return len(self.browsers)
|
|
|
|
def data(self, index: QtCore.QModelIndex, role: int = ...):
|
|
row = index.row()
|
|
|
|
if role == QtCore.Qt.ItemDataRole.DisplayRole:
|
|
return self.browsers[row]
|
|
if role == QtCore.Qt.ItemDataRole.DecorationRole:
|
|
return self.icons[row]
|
|
|
|
|
|
class WgExtensions(QtWidgets.QWidget):
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.ui = UiWgExtensions(self)
|
|
|
|
self.browsers_list_model = BrowsersListModel(self)
|
|
self.ui.cmbx_browsers.setModel(self.browsers_list_model)
|
|
|
|
self.ext_list_models: dict[str, BaseExtensionsListModel] = {
|
|
"Chrome": ChromeExtensionsListModel(self),
|
|
"Edge": EdgeExtensionsListModel(self),
|
|
"Brave": BraveExtensionsListModel(self),
|
|
}
|
|
self.model_is_initial = {
|
|
"Chrome": True,
|
|
"Edge": True,
|
|
"Brave": True,
|
|
}
|
|
self.switch_model(self.get_current_browser())
|
|
|
|
self.ui.cbx_chrome_compat.clicked.connect(self.on_cbx_chrome_compat_clicked)
|
|
self.ui.cmbx_browsers.currentTextChanged.connect(self.on_cmbx_browsers_current_text_changed)
|
|
self.ui.cbx_safe.clicked.connect(self.on_cbx_safe_clicked)
|
|
self.ui.cbx_unsafe.clicked.connect(self.on_cbx_unsafe_clicked)
|
|
self.ui.cbx_unknown.clicked.connect(self.on_cbx_unknown_clicked)
|
|
self.ui.pbn_update.clicked.connect(self.on_pbn_update_clicked)
|
|
|
|
def get_current_browser(self) -> str:
|
|
return self.ui.cmbx_browsers.currentData(QtCore.Qt.ItemDataRole.DisplayRole)
|
|
|
|
def show_all_rows(self):
|
|
# 在 update 之前调用
|
|
self.filters_clicked(True, True)
|
|
self.filters_clicked(False, True)
|
|
self.filters_clicked(None, True)
|
|
|
|
def apply_rows_hidden(self):
|
|
# 在 update 之后调用
|
|
self.filters_clicked(True, self.ui.cbx_safe.isChecked())
|
|
self.filters_clicked(False, self.ui.cbx_unsafe.isChecked())
|
|
self.filters_clicked(None, self.ui.cbx_unknown.isChecked())
|
|
|
|
def update_model(self, browser: str):
|
|
model = self.ext_list_models[browser]
|
|
self.show_all_rows()
|
|
model.update(is_chrome_compat=self.ui.cbx_chrome_compat.isChecked())
|
|
self.apply_rows_hidden()
|
|
|
|
def switch_model(self, browser: str):
|
|
if self.model_is_initial[browser] is True:
|
|
self.update_model(browser)
|
|
self.model_is_initial[browser] = False
|
|
|
|
self.ui.lv_extensions.setModel(self.ext_list_models[browser])
|
|
|
|
def on_cbx_chrome_compat_clicked(self):
|
|
if self.get_current_browser() == "Chrome":
|
|
self.update_model("Chrome")
|
|
|
|
def on_cmbx_browsers_current_text_changed(self, text: str):
|
|
self.switch_model(text)
|
|
self.ui.cbx_chrome_compat.setVisible(self.ui.cmbx_browsers.currentText() == "Chrome")
|
|
|
|
def filters_clicked(self, safe_mark: bool | None, checked: bool):
|
|
model = self.ext_list_models[self.get_current_browser()]
|
|
for row in range(model.rowCount()):
|
|
idx = model.index(row)
|
|
is_safe = model.data(idx, ExtensionStatusRole)
|
|
if is_safe is safe_mark:
|
|
self.ui.lv_extensions.setRowHidden(row, not checked)
|
|
|
|
def on_cbx_safe_clicked(self, checked: bool):
|
|
self.filters_clicked(True, checked)
|
|
|
|
def on_cbx_unsafe_clicked(self, checked: bool):
|
|
self.filters_clicked(False, checked)
|
|
|
|
def on_cbx_unknown_clicked(self, checked: bool):
|
|
self.filters_clicked(None, checked)
|
|
|
|
def on_pbn_update_clicked(self):
|
|
self.update_model(self.get_current_browser())
|
|
|
|
def update_safe(self, safe_info: dict):
|
|
for browser in self.ext_list_models:
|
|
self.ext_list_models[browser].update_safe_info(safe_info)
|
|
|
|
def export_unknown(self) -> dict[str, dict]:
|
|
unknown_ext = {}
|
|
for browser in self.ext_list_models:
|
|
model = self.ext_list_models[browser]
|
|
for row in range(model.rowCount()):
|
|
idx = model.index(row)
|
|
is_safe = model.data(idx, ExtensionStatusRole)
|
|
if is_safe is None:
|
|
ext_id = model.data(idx, ExtensionIdRole)
|
|
name = model.data(idx, QtCore.Qt.ItemDataRole.DisplayRole)
|
|
unknown_ext[ext_id] = {"name": name}
|
|
return unknown_ext
|