bearhub/fpakman/view/qt/apps_table.py

from typing import List

from PyQt5.QtCore import Qt, QUrl
from PyQt5.QtGui import QPixmap, QIcon, QColor, QCursor
from PyQt5.QtNetwork import QNetworkRequest, QNetworkAccessManager
from PyQt5.QtWidgets import QTableWidget, QTableView, QMenu, QAction, QTableWidgetItem, QToolButton, QWidget, \
    QMessageBox, QHeaderView

from fpakman.core import resource
from fpakman.core.controller import FlatpakController
from fpakman.view.qt import dialog


class UpdateToggleButton(QToolButton):

    def __init__(self, model: dict, root: QWidget, checked: bool = True):
        super(UpdateToggleButton, self).__init__()
        self.model = model
        self.root = root
        self.setCheckable(True)
        self.clicked.connect(self.change_state)
        self.icon_on = QIcon(resource.get_path('img/toggle_on.svg'))
        self.icon_off = QIcon(resource.get_path('img/toggle_off.svg'))
        self.setIcon(self.icon_on)
        self.setStyleSheet('border: 0px;')

        if not checked:
            self.click()

    def change_state(self, not_checked: bool):
        self.model['update_checked'] = not not_checked
        self.setIcon(self.icon_on if not not_checked else self.icon_off)
        self.root.change_update_state()


class AppsTable(QTableWidget):

    def __init__(self, parent: QWidget, controller: FlatpakController, columns: List[str]):
        super(AppsTable, self).__init__()
        self.parent = parent
        self.controller = controller
        self.columns = columns
        self.setColumnCount(len(columns))
        self.setFocusPolicy(Qt.NoFocus)
        self.setShowGrid(False)
        self.verticalHeader().setVisible(False)
        self.setSelectionBehavior(QTableView.SelectRows)
        self.setHorizontalHeaderLabels(columns)
        self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.icon_flathub = QIcon(resource.get_path('img/flathub_45.svg'))
        self.icon_uninstall = QIcon(resource.get_path('img/uninstall.svg'))
        self.icon_downgrade = QIcon(resource.get_path('img/downgrade.svg'))

        self.network_man = QNetworkAccessManager()
        self.network_man.finished.connect(self._load_icon)

        self.icon_cache = {}

    def contextMenuEvent(self, QContextMenuEvent):  # selected row right click event
        menu_row = QMenu()

        action_uninstall = QAction(self.parent.locale_keys["manage_window.apps_table.row.actions.uninstall"])
        action_uninstall.setIcon(self.icon_uninstall)
        action_uninstall.triggered.connect(self._uninstall_app)
        menu_row.addAction(action_uninstall)

        app = self._get_selected_app()

        if not app['model']['runtime']:  # downgrade only allowed for apps
            action_downgrade = QAction(self.parent.locale_keys["manage_window.apps_table.row.actions.downgrade"])
            action_downgrade.triggered.connect(self._downgrade_app)
            action_downgrade.setIcon(self.icon_downgrade)
            menu_row.addAction(action_downgrade)

        menu_row.adjustSize()
        menu_row.popup(QCursor.pos())
        menu_row.exec_()

    def _get_selected_app(self):
        return self.parent.apps[self.currentRow()]

    def _uninstall_app(self):
        selected_app = self._get_selected_app()

        if dialog.ask_confirmation(title=self.parent.locale_keys['manage_window.apps_table.row.actions.uninstall.popup.title'],
                                   body=self.parent.locale_keys['manage_window.apps_table.row.actions.uninstall.popup.body'].format(selected_app['model']['name']),
                                   locale_keys=self.parent.locale_keys):
            self.parent.uninstall_app(selected_app['model']['ref'])

    def _downgrade_app(self):
        selected_app = self._get_selected_app()

        if dialog.ask_confirmation(title=self.parent.locale_keys['manage_window.apps_table.row.actions.downgrade'],
                                   body=self.parent.locale_keys['manage_window.apps_table.row.actions.downgrade.popup.body'].format(selected_app['model']['name']),
                                   locale_keys=self.parent.locale_keys):
            self.parent.downgrade_app(selected_app['model']['ref'])

    def _load_icon(self, http_response):
        icon_url = http_response.url().toString()
        pixmap = QPixmap()
        pixmap.loadFromData(http_response.readAll())
        icon = QIcon(pixmap)
        self.icon_cache[icon_url] = icon

        for idx, app in enumerate(self.parent.apps):
            if app['model']['icon'] == icon_url:
                self.item(idx, 0).setIcon(icon)
                self.parent.resize_and_center()
                break

    def update_apps(self, apps: List[dict]):
        self.setEnabled(True)
        self.setRowCount(len(apps) if apps else 0)

        if apps:
            for idx, app in enumerate(apps):

                col_name = QTableWidgetItem()
                col_name.setText(app['model']['name'])
                col_name.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)

                if not app['model']['icon']:
                    col_name.setIcon(self.icon_flathub)
                else:
                    cached_icon = self.icon_cache.get(app['model']['icon'])

                    if cached_icon:
                        col_name.setIcon(cached_icon)
                    else:
                        col_name.setIcon(self.icon_flathub)
                        self.network_man.get(QNetworkRequest(QUrl(app['model']['icon'])))

                self.setItem(idx, 0, col_name)

                col_version = QTableWidgetItem()
                col_version.setText(app['model']['version'])
                col_version.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 1, col_version)

                col_release = QTableWidgetItem()
                col_release.setText(app['model']['latest_version'])
                col_release.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 2, col_release)

                if app['model']['version'] and app['model']['latest_version'] and app['model']['version'] < app['model']['latest_version']:
                    col_release.setForeground(QColor('orange'))

                col_branch = QTableWidgetItem()
                col_branch.setText(app['model']['branch'])
                col_branch.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 3, col_branch)

                col_arch = QTableWidgetItem()
                col_arch.setText(app['model']['arch'])
                col_arch.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 4, col_arch)

                col_package = QTableWidgetItem()
                col_package.setText(app['model']['ref'])
                col_package.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 5, col_package)

                col_origin = QTableWidgetItem()
                col_origin.setText(app['model']['origin'])
                col_origin.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
                self.setItem(idx, 6, col_origin)

                col_update = UpdateToggleButton(app['model'], self.parent, app['model']['update']) if app['model']['update'] else None
                self.setCellWidget(idx, 7, col_update)

    def change_headers_policy(self, policy: QHeaderView = QHeaderView.ResizeToContents):
        header_horizontal = self.horizontalHeader()
        for i in range(0, len(self.columns)):
            header_horizontal.setSectionResizeMode(i, policy)