263 lines
11 KiB
Python
Executable File
263 lines
11 KiB
Python
Executable File
#!/usr/bin/python3
|
|
# mambatray 2
|
|
# Copyright (C) 2011-2024 by Silvan Calarco
|
|
# Copyright (C) 2011 by michiamophil
|
|
# Distributed under the terms of the GPL version 3 FLOSS License
|
|
#
|
|
import sys
|
|
import os
|
|
import gettext
|
|
import distro
|
|
from PyQt5.QtCore import Qt
|
|
from PyQt5.QtGui import QIcon, QCursor
|
|
from PyQt5.QtWidgets import QSystemTrayIcon, QApplication, QMenu, QMessageBox
|
|
from PyQt5.QtCore import QTimer, QProcess
|
|
|
|
DATADIR = os.path.dirname(os.path.realpath((__file__))) + "/"
|
|
|
|
|
|
class SystemTrayIcon(QSystemTrayIcon):
|
|
|
|
menu = None
|
|
contatore = 0
|
|
NetsrpmsArg = "-c"
|
|
lista = []
|
|
nOre = 2
|
|
srpm = [
|
|
"skype", "win32codecs", "msttcorefonts", "b43-firmware",
|
|
"virtualbox-extension-pack", "chromium-widevine", "spotify"
|
|
]
|
|
desktop_session = 'default'
|
|
xdg_session_type = 'x11'
|
|
|
|
refreshPackagesTimer = None
|
|
checkSRPMSUpdatesTimer = None
|
|
refreshPackagesCacheProcess = None
|
|
SRPMUpdateProcess = None
|
|
|
|
softwareManagerProcess = None
|
|
softwareCheckUpgradesProcess = None
|
|
softwareBaseInstallsProcess = None
|
|
|
|
networkOnlineProcess = None
|
|
networkFirewallDisableProcess = None
|
|
networkFirewallEnableProcess = None
|
|
networkFirewallConfigureProcess = None
|
|
networkProxyProcess = None
|
|
|
|
def __init__(self, icon):
|
|
QSystemTrayIcon.__init__(self, icon)
|
|
|
|
self.menu = QMenu()
|
|
|
|
try:
|
|
self.desktop_session = os.path.basename(os.getenv('DESKTOP_SESSION'))
|
|
except:
|
|
None
|
|
|
|
try:
|
|
self.xdg_session_type = os.path.basename(os.getenv('XDG_SESSION_TYPE'))
|
|
except:
|
|
None
|
|
|
|
if self.desktop_session == 'default':
|
|
self.desktop_session = 'plasma'
|
|
elif self.desktop_session[:3] == 'kde':
|
|
self.desktop_session = 'kde'
|
|
|
|
# refresh packages cache timer
|
|
self.refreshPackagesTimer = QTimer(self)
|
|
self.checkSRPMSUpdatesTimer = QTimer(self)
|
|
self.refreshPackagesCacheProcess = QProcess(self)
|
|
self.SRPMUpdateProcess = QProcess(self)
|
|
self.networkOnlineProcess = QProcess(self)
|
|
self.refreshPackagesTimer.timeout.connect(self.refreshPackagesCache)
|
|
self.refreshPackagesCacheProcess.finished.connect(self.restartRefreshPackagesTimer)
|
|
self.checkSRPMSUpdatesTimer.timeout.connect(self.SRPMSCheckUpgradeList)
|
|
|
|
# software menu
|
|
self.softwareManagerProcess = QProcess(self)
|
|
self.softwareCheckUpgradesProcess = QProcess(self)
|
|
self.softwareBaseInstallsProcess = QProcess(self)
|
|
|
|
self.softwareMenu = self.menu.addMenu(QIcon.fromTheme("applications-system"),_("Software"))
|
|
|
|
self.softwareManageAction = self.softwareMenu.addAction(QIcon.fromTheme("applications-other"),_("Add/Remove software packages..."))
|
|
self.softwareManageAction.triggered.connect(self.softwareManager)
|
|
|
|
self.softwareCheckUpgradesAction = self.softwareMenu.addAction(QIcon.fromTheme("system-software-update"),_("Check for updates now..."))
|
|
self.softwareCheckUpgradesAction.triggered.connect(self.softwareCheckUpgrades)
|
|
|
|
self.softwareBaseInstallsAction = self.softwareMenu.addAction(QIcon.fromTheme("mambabase"),_("{} components...").format(distro.name()))
|
|
self.softwareBaseInstallsAction.triggered.connect(self.softwareBaseInstalls)
|
|
|
|
# network menu
|
|
self.networkFirewallDisableProcess = QProcess(self)
|
|
self.networkFirewallEnableProcess = QProcess(self)
|
|
self.networkFirewallConfigureProcess = QProcess(self)
|
|
self.networkProxyProcess = QProcess(self)
|
|
|
|
self.networkMenu = self.menu.addMenu(QIcon.fromTheme("preferences-system-network"), _("Network"))
|
|
|
|
self.networkFirewallMenu = self.networkMenu.addMenu(QIcon.fromTheme("security-medium"),_("Firewall"))
|
|
self.networkFirewallDisableAction = self.networkFirewallMenu.addAction(QIcon.fromTheme("security-low"),_("Disable"))
|
|
self.networkFirewallDisableAction.triggered.connect(self.networkFirewallDisable)
|
|
self.networkFirewallDisableProcess.finished.connect(self.networkFirewallDisableMessage)
|
|
self.networkFirewallEnableAction = self.networkFirewallMenu.addAction(QIcon.fromTheme("security-high"),_("Enable"))
|
|
self.networkFirewallEnableAction.triggered.connect(self.networkFirewallEnable)
|
|
self.networkFirewallEnableProcess.finished.connect(self.networkFirewallEnableMessage)
|
|
self.networkFirewallConfigureAction = self.networkFirewallMenu.addAction(QIcon.fromTheme("security-medium"),_("Configure..."))
|
|
self.networkFirewallConfigureAction.triggered.connect(self.networkFirewallConfigure)
|
|
self.networkProxyAction = self.networkMenu.addAction(QIcon.fromTheme("network-server"),_("Proxy..."))
|
|
self.networkProxyAction.triggered.connect(self.networkProxy)
|
|
self.SRPMUpdateProcess.finished.connect(self.SRPMSCheckUpgrade)
|
|
|
|
# about box
|
|
self.aboutAction = self.menu.addSeparator()
|
|
self.aboutAction = self.menu.addAction(QIcon.fromTheme("system-logo"),_("About"))
|
|
self.aboutAction.triggered.connect(self.about)
|
|
|
|
self.setToolTip(_("{} control center").format(distro.name()))
|
|
self.activated.connect(self.iconActivated)
|
|
self.setContextMenu(self.menu)
|
|
|
|
# the first time refresh packages cache after 5 minutes
|
|
self.refreshPackagesTimer.start( 5 * 60000 )
|
|
# the first time check for SRPMS updates after 4 minutes
|
|
self.checkSRPMSUpdatesTimer.start( 4 * 60000 )
|
|
|
|
def iconActivated(self, reason):
|
|
if reason in (QSystemTrayIcon.Trigger, QSystemTrayIcon.DoubleClick, QSystemTrayIcon.Context):
|
|
# FIXME: on plasma wayland and left click menu will popup in the center of the screen
|
|
# disabling and only supporting right click.
|
|
if self.xdg_session_type != 'wayland':
|
|
self.contextMenu().popup(QCursor.pos())
|
|
|
|
def about(self):
|
|
aboutBox = QMessageBox(self.menu)
|
|
aboutBox.setWindowIcon(QIcon.fromTheme("mambatray"))
|
|
aboutBox.setWindowTitle(_("About"))
|
|
aboutBox.setText(
|
|
"<center><b>"+_("{} control center").format(distro.name()+"</b>"
|
|
"<br/><br/>"+distro.name(pretty=True)+
|
|
"<br/><br/><a href=\""+distro.os_release_info()['home_url']+
|
|
"\">" + distro.os_release_info()['home_url'] + "</a>"
|
|
"</center>"))
|
|
aboutBox.setTextFormat(Qt.RichText)
|
|
aboutBox.setIconPixmap(QIcon.fromTheme("system-logo").pixmap(32));
|
|
aboutBox.setWindowModality(Qt.NonModal);
|
|
aboutBox.exec()
|
|
|
|
def networkOnline(self):
|
|
return self.networkOnlineProcess.execute("/usr/bin/nm-online",['-q','-x'])
|
|
|
|
def softwareManager(self):
|
|
self.softwareManagerProcess.execute("/usr/bin/plasma-discover")
|
|
|
|
def softwareCheckUpgrades(self):
|
|
self.softwareCheckUpgradesProcess.start("/usr/bin/pkcon", ['refresh', 'force'])
|
|
|
|
def refreshPackagesCache(self):
|
|
self.refreshPackagesTimer.stop()
|
|
self.refreshPackagesCacheProcess.start("/usr/bin/pkcon", ['refresh', 'force'])
|
|
|
|
def restartRefreshPackagesTimer(self,status):
|
|
# check every 4 hours and 31 minutes */
|
|
self.refreshPackagesTimer.start( (4 * 60 + 31) * 60000)
|
|
|
|
def softwareBaseInstalls(self):
|
|
self.softwareBaseInstallsProcess.start("/usr/bin/mambabase")
|
|
|
|
def networkFirewallDisable(self):
|
|
self.networkFirewallDisableProcess.start("sudo",['/usr/sbin/ufw','disable'])
|
|
|
|
def networkFirewallDisableMessage(self,status):
|
|
if (status == QProcess.NormalExit):
|
|
self.showMessage("Firewall", _("Firewall disabled in current session"))
|
|
else:
|
|
self.showMessage("Firewall", _("Could not disable firewall"))
|
|
|
|
def networkFirewallEnable(self):
|
|
self.networkFirewallEnableProcess.start("sudo",['/usr/sbin/ufw','enable'])
|
|
|
|
def networkFirewallEnableMessage(self,status):
|
|
if (status == QProcess.NormalExit):
|
|
self.showMessage("Firewall", _("Firewall enabled in current session"))
|
|
else:
|
|
self.showMessage("Firewall", _("Could not enable firewall"))
|
|
|
|
def networkFirewallConfigure(self):
|
|
if self.desktop_session == 'kde' or self.desktop_session.startswith('plasma'):
|
|
self.networkFirewallConfigureProcess.execute("/usr/bin/kcmshell5", list(set(['firewall'])))
|
|
else:
|
|
self.networkFirewallConfigureProcess.execute("/usr/bin/gufw")
|
|
|
|
def networkProxy(self):
|
|
if self.desktop_session.startswith('plasma'):
|
|
self.networkProxyProcess.execute("/usr/bin/kcmshell5", list(set(['proxy'])))
|
|
elif self.desktop_session == 'kde':
|
|
self.networkProxyProcess.execute("/opt/kde/bin/kcmshell4", list(set(['proxy'])))
|
|
elif self.desktop_session == 'gnome':
|
|
self.networkProxyProcess.execute("/usr/bin/gnome-control-center", "network")
|
|
|
|
def SRPMSCheckUpgradeList(self):
|
|
self.checkSRPMSUpdatesTimer.stop()
|
|
if (self.networkOnline() == 0):
|
|
try:
|
|
self.SRPMUpdateProcess.start("/usr/bin/openmamba-netsrpms",[self.NetsrpmsArg,self.srpm[self.contatore]])
|
|
except:
|
|
self.contatore = 0
|
|
if self.NetsrpmsArg == '-c':
|
|
self.SRPMSAskAndInstall()
|
|
else:
|
|
self.lista = []
|
|
self.NetsrpmsArg = '-c'
|
|
self.checkSRPMSUpdatesTimer.start((self.nOre * 60) * 60000)
|
|
|
|
# run for each netsrpm at the end of SRPMUpdateProcess
|
|
def SRPMSCheckUpgrade(self, status):
|
|
if status == 1:
|
|
self.lista.append(self.srpm[self.contatore])
|
|
self.contatore += 1
|
|
self.SRPMSCheckUpgradeList()
|
|
|
|
# netsrpm updates ask and install
|
|
def SRPMSAskAndInstall(self):
|
|
if not self.lista == []:
|
|
self.Lista = self.PrintLista()
|
|
msg = QMessageBox.question(self.menu,
|
|
(_("Not Open Source {} components").format(distro.name())),
|
|
(_("The following components are available for update: ") +
|
|
"\n\n" + self.Lista + "\n\n" + _("Ok to proceed?")),
|
|
QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
|
|
if msg == QMessageBox.Yes:
|
|
self.NetsrpmsArg = '-u'
|
|
self.SRPMSCheckUpgradeList()
|
|
else:
|
|
self.lista = []
|
|
self.checkSRPMSUpdatesTimer.start((self.nOre * 60) * 60000)
|
|
else:
|
|
self.checkSRPMSUpdatesTimer.start((self.nOre * 60) * 60000)
|
|
|
|
# starting from the list "lista" returns the list in readable format
|
|
# FIXME: make this code more clear and clean
|
|
def PrintLista(self):
|
|
testo = ''
|
|
for a in self.lista:
|
|
testo = testo + a + ', '
|
|
return testo[0:len(testo)-2]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app = QApplication(sys.argv)
|
|
app.setDesktopFileName("mambatray")
|
|
|
|
if app.isSessionRestored():
|
|
app.exit(1)
|
|
else:
|
|
app.setQuitOnLastWindowClosed(0)
|
|
gettext.install('mambatray', '/usr/share/locale')
|
|
trayIcon = SystemTrayIcon(QIcon.fromTheme("mambatray"))
|
|
trayIcon.show()
|
|
sys.exit(app.exec())
|