#!/usr/bin/python3 import sys import sqlite3 import signal import sisyphus from collections import OrderedDict from PyQt6 import QtCore, QtGui, QtWidgets, uic class Sisyphus(QtWidgets.QMainWindow): def __init__(self): super(Sisyphus, self).__init__() uic.loadUi('/usr/share/sisyphus/ui/sisyphus.ui', self) signal.signal(signal.SIGTERM, self.handleSigterm) self.centerOnScreen() self.show() self.filterApplications = OrderedDict([ ('Package Name', 'pn'), ('Package Category', 'cat'), ('Package Description', 'descr') ]) self.applicationFilter.addItems(self.filterApplications.keys()) self.applicationFilter.setCurrentText('Package Name') self.applicationFilter.currentIndexChanged.connect( self.setApplicationFilter) Sisyphus.appFilter = self.filterApplications['Package Name'] self.filterDatabases = OrderedDict([ ('All Packages', 'all'), ('Installed Packages', 'installed'), ('Alien Packages', 'alien'), ('Available Packages', 'available'), ('Upgradable Packages', 'upgradable') ]) self.databaseFilter.addItems(self.filterDatabases.keys()) self.databaseFilter.setCurrentText('All Packages') self.databaseFilter.currentIndexChanged.connect(self.setDatabaseFilter) Sisyphus.dbFilter = self.filterDatabases['All Packages'] Sisyphus.searchTerm = "'%%'" self.progressWindow = None self.databaseTable.clicked.connect(self.rowClicked) self.inputBox.textEdited.connect(self.searchDatabase) self.progressButton.clicked.connect(self.showProgressWindow) self.settingsButton.clicked.connect(self.showSettingsWindow) self.updateWorker = MainWorker() self.updateThread = QtCore.QThread() self.updateWorker.moveToThread(self.updateThread) self.updateWorker.started.connect(self.showProgress) self.updateThread.started.connect(self.updateWorker.startUpdate) self.updateThread.finished.connect(self.hideProgress) self.updateWorker.finished.connect(self.updateThread.quit) self.installButton.clicked.connect(self.packageInstall) self.installWorker = MainWorker() self.installThread = QtCore.QThread() self.installWorker.moveToThread(self.installThread) self.installWorker.started.connect(self.showProgress) self.installThread.started.connect(self.installWorker.startInstall) self.installThread.finished.connect(self.hideProgress) self.installWorker.finished.connect(self.installThread.quit) self.uninstallButton.clicked.connect(self.packageUninstall) self.uninstallWorker = MainWorker() self.uninstallThread = QtCore.QThread() self.uninstallWorker.moveToThread(self.uninstallThread) self.uninstallWorker.started.connect(self.showProgress) self.uninstallThread.started.connect( self.uninstallWorker.startUninstall) self.uninstallThread.finished.connect(self.hideProgress) self.uninstallWorker.finished.connect(self.uninstallThread.quit) self.upgradeButton.clicked.connect(self.systemUpgrade) self.upgradeWorker = MainWorker() self.upgradeThread = QtCore.QThread() self.upgradeWorker.moveToThread(self.upgradeThread) self.upgradeWorker.started.connect(self.showProgress) self.upgradeThread.started.connect(self.upgradeWorker.startUpgrade) self.upgradeThread.finished.connect(self.hideProgress) self.upgradeWorker.finished.connect(self.upgradeThread.quit) self.autoremoveButton.clicked.connect(self.autoRemove) self.autoremoveWorker = MainWorker() self.autoremoveThread = QtCore.QThread() self.autoremoveWorker.moveToThread(self.autoremoveThread) self.autoremoveWorker.started.connect(self.showProgress) self.autoremoveThread.started.connect( self.autoremoveWorker.startAutoremove) self.autoremoveThread.finished.connect(self.hideProgress) self.autoremoveWorker.finished.connect(self.autoremoveThread.quit) self.updateSystem() self.exitButton.clicked.connect(self.closeMainWindow) def centerOnScreen(self): screenGeometry = QtGui.QGuiApplication.primaryScreen().geometry() windowGeometry = self.geometry() horizontalPosition = int( (screenGeometry.width() - windowGeometry.width()) / 2) verticalPosition = int( (screenGeometry.height() - windowGeometry.height()) / 2) self.move(horizontalPosition, verticalPosition) def rowClicked(self): Sisyphus.pkgSelect = len( self.databaseTable.selectionModel().selectedRows()) self.showPackageCount() def showPackageCount(self): self.statusBar().showMessage("Found: %d, Selected: %d packages" % (Sisyphus.pkgCount, Sisyphus.pkgSelect)) def setApplicationFilter(self): Sisyphus.appFilter = self.filterApplications[self.applicationFilter.currentText( )] self.loadDatabase() def setDatabaseFilter(self): Sisyphus.dbFilter = self.filterDatabases[self.databaseFilter.currentText( )] Sisyphus.SELECT = self.databaseFilter.currentText() self.loadDatabase() def loadDatabase(self): filter = Sisyphus.dbFilter cat = '%' + self.inputBox.text() + \ '%' if self.applicationFilter.currentText() == 'Package Category' else '' pn = '%' + self.inputBox.text() + \ '%' if self.applicationFilter.currentText() == 'Package Name' else '' desc = '%' + self.inputBox.text() + \ '%' if self.applicationFilter.currentText() == 'Package Description' else '' query = sisyphus.querydb.start(filter, cat, pn, desc) with sqlite3.connect(sisyphus.getfs.lcl_db) as db: cursor = db.cursor() cursor.execute(query) rows = cursor.fetchall() Sisyphus.pkgCount = len(rows) Sisyphus.pkgSelect = 0 model = QtGui.QStandardItemModel(len(rows), 5) model.setHorizontalHeaderLabels( ['Package Category', 'Package Name', 'Installed Version', 'Available Version', 'Package Description']) for row in rows: indx = rows.index(row) for column in range(0, 5): item = QtGui.QStandardItem("%s" % (row[column])) model.setItem(indx, column, item) self.databaseTable.setModel(model) self.showPackageCount() def searchDatabase(self): Sisyphus.searchTerm = "'%" + self.inputBox.text() + "%'" self.loadDatabase() def updateSystem(self): self.loadDatabase() self.statusBar().showMessage("I am syncing myself, hope to finish soon ...") self.updateThread.start() def getSelectedPackages(self): def byRow(e): return e['row'] pkg_categs = [{'row': pkg.row(), 'cat': pkg.data()} for pkg in self.databaseTable.selectionModel().selectedRows(0)] pkg_names = [{'row': pkg.row(), 'name': pkg.data()} for pkg in self.databaseTable.selectionModel().selectedRows(1)] pkg_categs = sorted(pkg_categs, key=byRow) pkg_names = sorted(pkg_names, key=byRow) selected_pkgs = [pkg_categs[i]['cat'] + '/' + pkg_names[i]['name'] for i in range(len(pkg_categs))] return (selected_pkgs) def packageInstall(self): if not self.databaseTable.selectionModel().hasSelection(): self.statusBar().showMessage("No package selected, please pick at least one!") else: Sisyphus.pkgname = self.getSelectedPackages() self.statusBar().showMessage("I am installing requested package(s), please wait ...") self.installThread.start() def packageUninstall(self): if not self.databaseTable.selectionModel().hasSelection(): self.statusBar().showMessage("No package selected, please pick at least one!") else: Sisyphus.pkgname = self.getSelectedPackages() self.statusBar().showMessage("I am removing requested package(s), please wait ...") self.uninstallThread.start() def systemUpgrade(self): self.statusBar().showMessage("I am upgrading the system, please be patient ...") self.upgradeThread.start() def autoRemove(self): self.statusBar().showMessage("I am busy with some cleaning, please don't rush me ...") self.autoremoveThread.start() def hideProgressButton(self): self.progressButton.hide() def showProgressButton(self): self.progressButton.show() def hideSettingsButton(self): self.settingsButton.hide() def showSettingsButton(self): self.settingsButton.show() def hideUiButtons(self): self.installButton.hide() self.uninstallButton.hide() self.autoremoveButton.hide() self.upgradeButton.hide() self.exitButton.hide() def showUiButtons(self): self.installButton.show() self.uninstallButton.show() self.autoremoveButton.show() self.upgradeButton.show() self.exitButton.show() def hideProgressBar(self): self.progressBar.setRange(0, 1) self.progressBar.setValue(1) self.progressBar.hide() def showProgressBar(self): self.progressBar.setRange(0, 0) self.progressBar.show() def setInputFocus(self): self.inputBox.setFocus() def showProgress(self): self.hideUiButtons() self.hideSettingsButton() self.showProgressButton() self.showProgressBar() self.setInputFocus() def hideProgress(self): self.showUiButtons() self.showSettingsButton() self.hideProgressButton() self.hideProgressBar() self.setInputFocus() self.loadDatabase() def showProgressWindow(self): if self.progressWindow is None: self.progressWindow = ProgressWindow(self) self.installWorker.workerOutput.connect( self.progressWindow.updateProgressWindow) self.uninstallWorker.workerOutput.connect( self.progressWindow.updateProgressWindow) self.upgradeWorker.workerOutput.connect( self.progressWindow.updateProgressWindow) self.autoremoveWorker.workerOutput.connect( self.progressWindow.updateProgressWindow) self.progressWindow.show() def showSettingsWindow(self): self.settingsWindow = SettingsWindow(self) self.settingsWindow.show() def closeMainWindow(self): self.close() def handleSigterm(self, signum, frame): self.close() def __del__(self): sys.stdout = sys.__stdout__ class ProgressWindow(QtWidgets.QMainWindow): progress_messages = [] def __init__(self, parent=None): super(ProgressWindow, self).__init__(parent) uic.loadUi('/usr/share/sisyphus/ui/progress.ui', self) self.setWindowFlag(QtCore.Qt.WindowType.FramelessWindowHint) self.centerOnScreen() self.refreshProgressWindow() self.clearButton.clicked.connect(self.clearProgressWindow) self.hideButton.clicked.connect(self.hideProgressWindow) sys.stdout = MainWorker(workerOutput=self.updateProgressWindow) def centerOnScreen(self): screenGeometry = QtGui.QGuiApplication.primaryScreen().geometry() windowGeometry = self.geometry() horizontalPosition = int( (screenGeometry.width() - windowGeometry.width()) / 2) verticalPosition = int( (screenGeometry.height() - windowGeometry.height()) / 2) self.move(horizontalPosition, verticalPosition) def updateProgressWindow(self, workerMessage): ProgressWindow.progress_messages.append(workerMessage) self.progressBox.insertPlainText(workerMessage) self.progressBox.ensureCursorVisible() def refreshProgressWindow(self): self.progressBox.clear() self.progressBox.setPlainText( ''.join(ProgressWindow.progress_messages)) self.progressBox.ensureCursorVisible() def clearProgressWindow(self): self.progressBox.clear() ProgressWindow.progress_messages.clear() def hideProgressWindow(self): self.hide() class SettingsWindow(QtWidgets.QMainWindow): def __init__(self, parent=None): super(SettingsWindow, self).__init__(parent) uic.loadUi('/usr/share/sisyphus/ui/settings.ui', self) self.centerOnScreen() self.MIRRORLIST = sisyphus.setmirror.getList() self.updateMirrorList() self.applyButton.pressed.connect(self.writeMirrorList) self.applyButton.released.connect(self.closeSettingsWindow) self.mirrorCombo.activated.connect(self.setMirrorList) def centerOnScreen(self): screenGeometry = QtGui.QGuiApplication.primaryScreen().geometry() windowGeometry = self.geometry() horizontalPosition = int( (screenGeometry.width() - windowGeometry.width()) / 2) verticalPosition = int( (screenGeometry.height() - windowGeometry.height()) / 2) self.move(horizontalPosition, verticalPosition) def updateMirrorList(self): model = QtGui.QStandardItemModel() for row in self.MIRRORLIST: indx = self.MIRRORLIST.index(row) item = QtGui.QStandardItem() item.setText(row['Url']) model.setItem(indx, item) if row['isActive']: self.ACTIVEMIRRORINDEX = indx self.mirrorCombo.setModel(model) self.mirrorCombo.setCurrentIndex(self.ACTIVEMIRRORINDEX) def setMirrorList(self): self.MIRRORLIST[self.ACTIVEMIRRORINDEX]['isActive'] = False self.ACTIVEMIRRORINDEX = self.mirrorCombo.currentIndex() self.MIRRORLIST[self.ACTIVEMIRRORINDEX]['isActive'] = True def writeMirrorList(self): sisyphus.setmirror.writeList(self.MIRRORLIST) def closeSettingsWindow(self): self.close() class MainWorker(QtCore.QObject): started = QtCore.pyqtSignal() finished = QtCore.pyqtSignal() workerOutput = QtCore.pyqtSignal(str) def write(self, text): self.workerOutput.emit(str(text)) def flush(self): pass def fileno(self): return 0 @QtCore.pyqtSlot() def startUpdate(self): self.started.emit() sisyphus.setjobs.start() sisyphus.syncall.start.__wrapped__(gfx_ui=True) # undecorate self.finished.emit() @QtCore.pyqtSlot() def startInstall(self): self.started.emit() pkgname = Sisyphus.pkgname sisyphus.pkgadd.start( pkgname, ebuild=False, gfx_ui=True, oneshot=False, nodeps=False) self.finished.emit() @QtCore.pyqtSlot() def startUninstall(self): self.started.emit() pkgname = Sisyphus.pkgname sisyphus.pkgremove.start( pkgname, depclean=True, gfx_ui=True, unmerge=False) self.finished.emit() @QtCore.pyqtSlot() def startUpgrade(self): self.started.emit() sisyphus.sysupgrade.start(ebuild=False, gfx_ui=True) self.finished.emit() @QtCore.pyqtSlot() def startAutoremove(self): self.started.emit() sisyphus.sysclean.start(gfx_ui=True) self.finished.emit() if __name__ == '__main__': app = QtWidgets.QApplication(sys.argv) app.setStyle('Breeze') window = Sisyphus() window.inputBox.setFocus() sys.exit(app.exec())