mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 00:04:15 +02:00
390 lines
16 KiB
Python
390 lines
16 KiB
Python
import os
|
|
from threading import Lock
|
|
from typing import List
|
|
|
|
from PyQt5.QtCore import Qt, QUrl
|
|
from PyQt5.QtGui import QPixmap, QIcon, QCursor
|
|
from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkRequest
|
|
from PyQt5.QtWidgets import QTableWidget, QTableView, QMenu, QAction, QTableWidgetItem, QWidget, \
|
|
QHeaderView, QLabel, QHBoxLayout, QPushButton, QToolButton, QToolBar
|
|
|
|
from fpakman.core import resource
|
|
from fpakman.core.model import ApplicationStatus
|
|
from fpakman.util import util
|
|
from fpakman.util.cache import Cache
|
|
from fpakman.view.qt import dialog
|
|
from fpakman.view.qt.view_model import ApplicationView, ApplicationViewStatus
|
|
|
|
INSTALL_BT_STYLE = 'background: {back}; color: white; font-size: 10px; font-weight: bold'
|
|
|
|
|
|
class IconButton(QWidget):
|
|
|
|
def __init__(self, icon_path: str, action, background: str = None, align: int = Qt.AlignCenter, tooltip: str = None):
|
|
super(IconButton, self).__init__()
|
|
self.bt = QToolButton()
|
|
self.bt.setIcon(QIcon(icon_path))
|
|
self.bt.clicked.connect(action)
|
|
|
|
if background:
|
|
self.bt.setStyleSheet('QToolButton { color: white; background: ' + background + '}')
|
|
|
|
if tooltip:
|
|
self.bt.setToolTip(tooltip)
|
|
|
|
layout = QHBoxLayout()
|
|
layout.setContentsMargins(0, 0, 0, 0)
|
|
layout.setAlignment(align)
|
|
layout.addWidget(self.bt)
|
|
self.setLayout(layout)
|
|
|
|
|
|
class UpdateToggleButton(QWidget):
|
|
|
|
def __init__(self, app_view: ApplicationView, root: QWidget, locale_keys: dict, checked: bool = True):
|
|
super(UpdateToggleButton, self).__init__()
|
|
|
|
self.app_view = app_view
|
|
self.root = root
|
|
|
|
layout = QHBoxLayout()
|
|
layout.setContentsMargins(2, 2, 2, 0)
|
|
layout.setAlignment(Qt.AlignCenter)
|
|
self.setLayout(layout)
|
|
|
|
self.bt = QToolButton()
|
|
self.bt.setCheckable(True)
|
|
self.bt.clicked.connect(self.change_state)
|
|
|
|
self.bt.setIcon(QIcon(resource.get_path('img/app_update.svg')))
|
|
self.bt.setStyleSheet('QToolButton { background: #4EC306 } ' +
|
|
'QToolButton:checked { background: gray }')
|
|
layout.addWidget(self.bt)
|
|
|
|
self.setToolTip(locale_keys['manage_window.apps_table.upgrade_toggle.tooltip'])
|
|
|
|
if not checked:
|
|
self.bt.click()
|
|
|
|
def change_state(self, not_checked: bool):
|
|
self.app_view.update_checked = not not_checked
|
|
self.root.change_update_state(change_filters=False)
|
|
|
|
|
|
class AppsTable(QTableWidget):
|
|
|
|
def __init__(self, parent: QWidget, icon_cache: Cache, disk_cache: bool, download_icons: bool):
|
|
super(AppsTable, self).__init__()
|
|
self.setParent(parent)
|
|
self.window = parent
|
|
self.disk_cache = disk_cache
|
|
self.download_icons = download_icons
|
|
self.column_names = ['' for _ in range(7)]
|
|
self.setColumnCount(len(self.column_names))
|
|
self.setFocusPolicy(Qt.NoFocus)
|
|
self.setShowGrid(False)
|
|
self.verticalHeader().setVisible(False)
|
|
self.horizontalHeader().setVisible(False)
|
|
self.setSelectionBehavior(QTableView.SelectRows)
|
|
self.setHorizontalHeaderLabels(self.column_names)
|
|
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
|
|
self.icon_flathub = QIcon(resource.get_path('img/flathub.svg'))
|
|
self.icon_logo = QIcon(resource.get_path('img/logo.svg'))
|
|
|
|
self.network_man = QNetworkAccessManager()
|
|
self.network_man.finished.connect(self._load_icon_and_cache)
|
|
|
|
self.icon_cache = icon_cache
|
|
self.lock_async_data = Lock()
|
|
|
|
def has_any_settings(self, app_v: ApplicationView):
|
|
return app_v.model.can_be_refreshed() or \
|
|
app_v.model.has_history() or \
|
|
app_v.model.can_be_downgraded()
|
|
|
|
def show_app_settings(self, app: ApplicationView):
|
|
menu_row = QMenu()
|
|
|
|
if app.model.installed:
|
|
if app.model.can_be_refreshed():
|
|
action_history = QAction(self.window.locale_keys["manage_window.apps_table.row.actions.refresh"])
|
|
action_history.setIcon(QIcon(resource.get_path('img/refresh.svg')))
|
|
|
|
def refresh():
|
|
self.window.refresh(app)
|
|
|
|
action_history.triggered.connect(refresh)
|
|
menu_row.addAction(action_history)
|
|
|
|
if app.model.has_history():
|
|
action_history = QAction(self.window.locale_keys["manage_window.apps_table.row.actions.history"])
|
|
action_history.setIcon(QIcon(resource.get_path('img/history.svg')))
|
|
|
|
def show_history():
|
|
self.window.get_app_history(app)
|
|
|
|
action_history.triggered.connect(show_history)
|
|
menu_row.addAction(action_history)
|
|
|
|
if app.model.can_be_downgraded():
|
|
action_downgrade = QAction(self.window.locale_keys["manage_window.apps_table.row.actions.downgrade"])
|
|
|
|
def downgrade():
|
|
if dialog.ask_confirmation(
|
|
title=self.window.locale_keys['manage_window.apps_table.row.actions.downgrade'],
|
|
body=self.window.locale_keys['manage_window.apps_table.row.actions.downgrade.popup.body'].format(app.model.base_data.name),
|
|
locale_keys=self.window.locale_keys):
|
|
self.window.downgrade_app(app)
|
|
|
|
action_downgrade.triggered.connect(downgrade)
|
|
action_downgrade.setIcon(QIcon(resource.get_path('img/downgrade.svg')))
|
|
menu_row.addAction(action_downgrade)
|
|
|
|
menu_row.adjustSize()
|
|
menu_row.popup(QCursor.pos())
|
|
menu_row.exec_()
|
|
|
|
def fill_async_data(self):
|
|
if self.window.apps:
|
|
|
|
for idx, app_v in enumerate(self.window.apps):
|
|
|
|
if app_v.visible and app_v.status == ApplicationViewStatus.LOADING and app_v.model.status == ApplicationStatus.READY:
|
|
|
|
if self.download_icons:
|
|
self.network_man.get(QNetworkRequest(QUrl(app_v.model.base_data.icon_url)))
|
|
|
|
app_name = self.item(idx, 0).text()
|
|
|
|
if not app_name or app_name == '...':
|
|
self.item(idx, 0).setText(app_v.model.base_data.name)
|
|
|
|
self._set_col_version(idx, app_v)
|
|
self._set_col_description(idx, app_v)
|
|
app_v.status = ApplicationViewStatus.READY
|
|
|
|
self.window.resize_and_center()
|
|
|
|
def get_selected_app(self) -> ApplicationView:
|
|
return self.window.apps[self.currentRow()]
|
|
|
|
def get_selected_app_icon(self) -> QIcon:
|
|
return self.item(self.currentRow(), 0).icon()
|
|
|
|
def _uninstall_app(self, app_v: ApplicationView):
|
|
if dialog.ask_confirmation(title=self.window.locale_keys['manage_window.apps_table.row.actions.uninstall.popup.title'],
|
|
body=self.window.locale_keys['manage_window.apps_table.row.actions.uninstall.popup.body'].format(app_v),
|
|
locale_keys=self.window.locale_keys):
|
|
self.window.uninstall_app(app_v)
|
|
|
|
def _downgrade_app(self):
|
|
selected_app = self.get_selected_app()
|
|
|
|
if dialog.ask_confirmation(title=self.window.locale_keys['manage_window.apps_table.row.actions.downgrade'],
|
|
body=self.window.locale_keys['manage_window.apps_table.row.actions.downgrade.popup.body'].format(selected_app.model.base_data.name),
|
|
locale_keys=self.window.locale_keys):
|
|
self.window.downgrade_app(selected_app)
|
|
|
|
def _refresh_app(self):
|
|
self.window.refresh(self.get_selected_app())
|
|
|
|
def _get_app_info(self):
|
|
self.window.get_app_info(self.get_selected_app())
|
|
|
|
def _get_app_history(self):
|
|
self.window.get_app_history(self.get_selected_app())
|
|
|
|
def _install_app(self, app_v: ApplicationView):
|
|
|
|
if dialog.ask_confirmation(
|
|
title=self.window.locale_keys['manage_window.apps_table.row.actions.install.popup.title'],
|
|
body=self.window.locale_keys['manage_window.apps_table.row.actions.install.popup.body'].format(app_v),
|
|
locale_keys=self.window.locale_keys):
|
|
|
|
self.window.install_app(app_v)
|
|
|
|
def _load_icon_and_cache(self, http_response):
|
|
icon_url = http_response.url().toString()
|
|
|
|
icon_data = self.icon_cache.get(icon_url)
|
|
icon_was_cached = True
|
|
|
|
if not icon_data:
|
|
icon_bytes = http_response.readAll()
|
|
|
|
if not icon_bytes:
|
|
return
|
|
|
|
icon_was_cached = False
|
|
pixmap = QPixmap()
|
|
|
|
pixmap.loadFromData(icon_bytes)
|
|
icon = QIcon(pixmap)
|
|
icon_data = {'icon': icon, 'bytes': icon_bytes}
|
|
self.icon_cache.add(icon_url, icon_data)
|
|
|
|
for idx, app in enumerate(self.window.apps):
|
|
if app.model.base_data.icon_url == icon_url:
|
|
col_name = self.item(idx, 0)
|
|
col_name.setIcon(icon_data['icon'])
|
|
|
|
if self.disk_cache and app.model.supports_disk_cache():
|
|
if not icon_was_cached or not os.path.exists(app.model.get_disk_icon_path()):
|
|
self.window.manager.cache_to_disk(app=app.model, icon_bytes=icon_data['bytes'], only_icon=True)
|
|
|
|
def update_apps(self, app_views: List[ApplicationView], update_check_enabled: bool = True):
|
|
self.setRowCount(len(app_views) if app_views else 0)
|
|
self.setEnabled(True)
|
|
|
|
if app_views:
|
|
for idx, app_v in enumerate(app_views):
|
|
self._set_col_name(idx, app_v)
|
|
self._set_col_version(idx, app_v)
|
|
self._set_col_description(idx, app_v)
|
|
self._set_col_type(idx, app_v)
|
|
self._set_col_installed(idx, app_v)
|
|
|
|
self._set_col_settings(idx, app_v)
|
|
|
|
col_update = None
|
|
|
|
if update_check_enabled and app_v.model.update:
|
|
col_update = UpdateToggleButton(app_v, self.window, self.window.locale_keys, app_v.model.update)
|
|
|
|
self.setCellWidget(idx, 6, col_update)
|
|
|
|
def _gen_row_button(self, text: str, style: str, callback) -> QWidget:
|
|
col = QWidget()
|
|
col_bt = QPushButton()
|
|
col_bt.setText(text)
|
|
col_bt.setStyleSheet('QPushButton { ' + style + '}')
|
|
col_bt.clicked.connect(callback)
|
|
|
|
layout = QHBoxLayout()
|
|
layout.setContentsMargins(2, 2, 2, 0)
|
|
layout.setAlignment(Qt.AlignCenter)
|
|
layout.addWidget(col_bt)
|
|
col.setLayout(layout)
|
|
|
|
return col
|
|
|
|
def _set_col_installed(self, idx: int, app_v: ApplicationView):
|
|
|
|
if app_v.model.installed:
|
|
if app_v.model.can_be_uninstalled():
|
|
def uninstall():
|
|
self._uninstall_app(app_v)
|
|
|
|
col = self._gen_row_button(self.window.locale_keys['uninstall'].capitalize(), INSTALL_BT_STYLE.format(back='#cc0000'), uninstall)
|
|
else:
|
|
col = QLabel()
|
|
col.setPixmap((QPixmap(resource.get_path('img/checked.svg'))))
|
|
col.setAlignment(Qt.AlignCenter)
|
|
col.setToolTip(self.window.locale_keys['installed'])
|
|
elif app_v.model.can_be_installed():
|
|
def install():
|
|
self._install_app(app_v)
|
|
col = self._gen_row_button(self.window.locale_keys['install'].capitalize(), INSTALL_BT_STYLE.format(back='#088A08'), install)
|
|
else:
|
|
col = None
|
|
|
|
self.setCellWidget(idx, 4, col)
|
|
|
|
def _set_col_type(self, idx: int, app_v: ApplicationView):
|
|
col_type = QLabel()
|
|
pixmap = QPixmap(app_v.model.get_default_icon_path())
|
|
col_type.setPixmap(pixmap.scaled(16, 16, Qt.KeepAspectRatio, Qt.SmoothTransformation))
|
|
col_type.setAlignment(Qt.AlignCenter)
|
|
col_type.setToolTip('{}: {}'.format(self.window.locale_keys['type'], app_v.model.get_type()))
|
|
self.setCellWidget(idx, 3, col_type)
|
|
|
|
def _set_col_version(self, idx: int, app_v: ApplicationView):
|
|
label_version = QLabel(app_v.model.base_data.version if app_v.model.base_data.version else '?')
|
|
label_version.setAlignment(Qt.AlignCenter)
|
|
|
|
col_version = QWidget()
|
|
col_version.setLayout(QHBoxLayout())
|
|
col_version.layout().addWidget(label_version)
|
|
|
|
if app_v.model.base_data.version:
|
|
tooltip = self.window.locale_keys['version.installed'] if app_v.model.installed else self.window.locale_keys['version']
|
|
else:
|
|
tooltip = self.window.locale_keys['version.unknown']
|
|
|
|
if app_v.model.update:
|
|
label_version.setStyleSheet("color: #4EC306; font-weight: bold")
|
|
tooltip = self.window.locale_keys['version.installed_outdated']
|
|
|
|
if app_v.model.installed and app_v.model.base_data.version and app_v.model.base_data.latest_version and app_v.model.base_data.version < app_v.model.base_data.latest_version:
|
|
tooltip = '{}. {}: {}'.format(tooltip, self.window.locale_keys['version.latest'], app_v.model.base_data.latest_version)
|
|
label_version.setText(label_version.text() + ' > {}'.format(app_v.model.base_data.latest_version))
|
|
|
|
col_version.setToolTip(tooltip)
|
|
self.setCellWidget(idx, 1, col_version)
|
|
|
|
def _set_col_name(self, idx: int, app_v: ApplicationView):
|
|
col = QTableWidgetItem()
|
|
col.setText(app_v.model.base_data.name if app_v.model.base_data.name else '...')
|
|
col.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
|
|
col.setToolTip(self.window.locale_keys['app.name'].lower())
|
|
|
|
if self.disk_cache and app_v.model.supports_disk_cache() and os.path.exists(app_v.model.get_disk_icon_path()):
|
|
with open(app_v.model.get_disk_icon_path(), 'rb') as f:
|
|
icon_bytes = f.read()
|
|
pixmap = QPixmap()
|
|
pixmap.loadFromData(icon_bytes)
|
|
icon = QIcon(pixmap)
|
|
self.icon_cache.add_non_existing(app_v.model.base_data.icon_url, {'icon': icon, 'bytes': icon_bytes})
|
|
|
|
elif not app_v.model.base_data.icon_url:
|
|
icon = QIcon(app_v.model.get_default_icon_path())
|
|
else:
|
|
icon_data = self.icon_cache.get(app_v.model.base_data.icon_url)
|
|
icon = icon_data['icon'] if icon_data else QIcon(app_v.model.get_default_icon_path())
|
|
|
|
col.setIcon(icon)
|
|
self.setItem(idx, 0, col)
|
|
|
|
def _set_col_description(self, idx: int, app_v: ApplicationView):
|
|
col = QTableWidgetItem()
|
|
col.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
|
|
|
|
if app_v.model.base_data.description is not None or app_v.model.is_library() or app_v.model.status == ApplicationStatus.READY:
|
|
desc = app_v.model.base_data.description
|
|
else:
|
|
desc = '...'
|
|
|
|
if desc and desc != '...':
|
|
desc = util.strip_html(desc[0:25]) + '...'
|
|
|
|
col.setText(desc)
|
|
|
|
if app_v.model.base_data.description:
|
|
col.setToolTip(app_v.model.base_data.description)
|
|
|
|
self.setItem(idx, 2, col)
|
|
|
|
def _set_col_settings(self, idx: int, app_v: ApplicationView):
|
|
tb = QToolBar()
|
|
|
|
if app_v.model.has_info():
|
|
|
|
def get_info():
|
|
self.window.get_app_info(app_v)
|
|
|
|
tb.addWidget(IconButton(icon_path=resource.get_path('img/app_info.svg'), action=get_info, background='#2E68D3'))
|
|
|
|
def handle_click():
|
|
self.show_app_settings(app_v)
|
|
|
|
if self.has_any_settings(app_v):
|
|
bt = IconButton(icon_path=resource.get_path('img/app_settings.svg'), action=handle_click, background='#12ABAB')
|
|
tb.addWidget(bt)
|
|
|
|
self.setCellWidget(idx, 5, tb)
|
|
|
|
def change_headers_policy(self, policy: QHeaderView = QHeaderView.ResizeToContents):
|
|
header_horizontal = self.horizontalHeader()
|
|
for i in range(self.columnCount()):
|
|
header_horizontal.setSectionResizeMode(i, policy)
|