250 lines
10 KiB
Python
Executable File
250 lines
10 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
|
|
|
|
def __init__(self, icon):
|
|
QSystemTrayIcon.__init__(self, icon)
|
|
|
|
global w, contatore, srpm, lista, NetsrpmsArg, nOre, desktop_session
|
|
nOre = 2
|
|
srpm = [
|
|
"skype", "win32codecs", "msttcorefonts", "b43-firmware",
|
|
"virtualbox-extension-pack", "chromium-widevine", "spotify"
|
|
]
|
|
NetsrpmsArg = "-c"
|
|
lista = []
|
|
contatore = 0
|
|
self.menu = QMenu()
|
|
|
|
try:
|
|
desktop_session = os.path.basename(os.getenv('DESKTOP_SESSION'))
|
|
except:
|
|
desktop_session = 'default'
|
|
|
|
if desktop_session == 'default':
|
|
desktop_session = 'plasma'
|
|
elif desktop_session[:3] == 'kde':
|
|
desktop_session = 'kde'
|
|
|
|
# refresh packages cache timer
|
|
global refreshPackagesTimer, refreshPackagesCacheProcess,\
|
|
checkSRPMSUpdatesTimer, SRPMUpdateProcess, networkOnlineProcess
|
|
refreshPackagesTimer = QTimer(self)
|
|
checkSRPMSUpdatesTimer = QTimer(self)
|
|
refreshPackagesCacheProcess = QProcess(self)
|
|
SRPMUpdateProcess = QProcess(self)
|
|
networkOnlineProcess = QProcess(self)
|
|
refreshPackagesTimer.timeout.connect(self.refreshPackagesCache)
|
|
refreshPackagesCacheProcess.finished.connect(self.restartRefreshPackagesTimer)
|
|
checkSRPMSUpdatesTimer.timeout.connect(self.SRPMSCheckUpgradeList)
|
|
|
|
# software menu
|
|
global softwareManagerProcess, softwareCheckUpgradesProcess,\
|
|
softwareBaseInstallsProcess
|
|
softwareManagerProcess = QProcess(self)
|
|
softwareCheckUpgradesProcess = QProcess(self)
|
|
softwareBaseInstallsProcess = QProcess(self)
|
|
|
|
softwareMenu = self.menu.addMenu(QIcon.fromTheme("applications-system"),_("Software"))
|
|
|
|
softwareManageAction = softwareMenu.addAction(QIcon.fromTheme("applications-other"),_("Add/Remove software packages..."))
|
|
softwareManageAction.triggered.connect(self.softwareManager)
|
|
|
|
softwareCheckUpgradesAction = softwareMenu.addAction(QIcon.fromTheme("system-software-update"),_("Check for updates now..."))
|
|
softwareCheckUpgradesAction.triggered.connect(self.softwareCheckUpgrades)
|
|
|
|
softwareBaseInstallsAction = softwareMenu.addAction(QIcon.fromTheme("mambabase"),_("{} components...").format(distro.name()))
|
|
softwareBaseInstallsAction.triggered.connect(self.softwareBaseInstalls)
|
|
|
|
# network menu
|
|
global networkFirewallDisableProcess, networkFirewallEnableProcess,\
|
|
networkFirewallConfigureProcess, networkProxyProcess
|
|
networkFirewallDisableProcess = QProcess(self)
|
|
networkFirewallEnableProcess = QProcess(self)
|
|
networkFirewallConfigureProcess = QProcess(self)
|
|
networkProxyProcess = QProcess(self)
|
|
|
|
networkMenu = self.menu.addMenu(QIcon.fromTheme("preferences-system-network"),
|
|
_("Network"))
|
|
|
|
networkFirewallMenu = networkMenu.addMenu(QIcon.fromTheme("security-medium"),_("Firewall"))
|
|
networkFirewallDisableAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-low"),_("Disable"))
|
|
networkFirewallDisableAction.triggered.connect(self.networkFirewallDisable)
|
|
networkFirewallDisableProcess.finished.connect(self.networkFirewallDisableMessage)
|
|
networkFirewallEnableAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-high"),_("Enable"))
|
|
networkFirewallEnableAction.triggered.connect(self.networkFirewallEnable)
|
|
networkFirewallEnableProcess.finished.connect(self.networkFirewallEnableMessage)
|
|
networkFirewallConfigureAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-medium"),_("Configure..."))
|
|
networkFirewallConfigureAction.triggered.connect(self.networkFirewallConfigure)
|
|
networkProxyAction = networkMenu.addAction(QIcon.fromTheme("network-server"),_("Proxy..."))
|
|
networkProxyAction.triggered.connect(self.networkProxy)
|
|
SRPMUpdateProcess.finished.connect(self.SRPMSCheckUpgrade)
|
|
|
|
# about box
|
|
aboutAction = self.menu.addSeparator()
|
|
aboutAction = self.menu.addAction(QIcon.fromTheme("system-logo"),_("About"))
|
|
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
|
|
refreshPackagesTimer.start( 5 * 60000 )
|
|
# the first time check for SRPMS updates after 4 minutes
|
|
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 desktop_session != 'plasmawayland':
|
|
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 networkOnlineProcess.execute("/usr/bin/nm-online",['-q','-x'])
|
|
|
|
def softwareManager(self):
|
|
softwareManagerProcess.execute("/usr/bin/plasma-discover")
|
|
|
|
def softwareCheckUpgrades(self):
|
|
softwareCheckUpgradesProcess.start("/usr/bin/pkcon", ['refresh', 'force'])
|
|
|
|
def refreshPackagesCache(self):
|
|
refreshPackagesTimer.stop()
|
|
refreshPackagesCacheProcess.start("/usr/bin/pkcon", ['refresh', 'force'])
|
|
|
|
def restartRefreshPackagesTimer(self,status):
|
|
# check every 4 hours and 31 minutes */
|
|
refreshPackagesTimer.start( (4 * 60 + 31) * 60000)
|
|
|
|
def softwareBaseInstalls(self):
|
|
softwareBaseInstallsProcess.start("/usr/bin/mambabase")
|
|
|
|
def networkFirewallDisable(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):
|
|
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 desktop_session == 'kde' or desktop_session.startswith('plasma'):
|
|
networkFirewallConfigureProcess.execute("/usr/bin/kcmshell5", list(set(['firewall'])))
|
|
else:
|
|
networkFirewallConfigureProcess.execute("/usr/bin/gufw")
|
|
|
|
def networkProxy(self):
|
|
if desktop_session.startswith('plasma'):
|
|
networkProxyProcess.execute("/usr/bin/kcmshell5", list(set(['proxy'])))
|
|
elif desktop_session == 'kde':
|
|
networkProxyProcess.execute("/opt/kde/bin/kcmshell4", list(set(['proxy'])))
|
|
elif desktop_session == 'gnome':
|
|
networkProxyProcess.execute("/usr/bin/gnome-control-center", "network")
|
|
|
|
def SRPMSCheckUpgradeList(self):
|
|
checkSRPMSUpdatesTimer.stop()
|
|
global contatore, lista, NetsrpmsArg
|
|
if (self.networkOnline() == 0):
|
|
try:
|
|
SRPMUpdateProcess.start("/usr/bin/openmamba-netsrpms",[NetsrpmsArg,srpm[contatore]])
|
|
except:
|
|
contatore = 0
|
|
if NetsrpmsArg == '-c':
|
|
self.SRPMSAskAndInstall()
|
|
else:
|
|
lista = []
|
|
NetsrpmsArg = '-c'
|
|
checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)
|
|
|
|
# run for each netsrpm at the end of SRPMUpdateProcess
|
|
def SRPMSCheckUpgrade(self, status):
|
|
global contatore, lista
|
|
if status == 1:
|
|
lista.append(srpm[contatore])
|
|
contatore += 1
|
|
self.SRPMSCheckUpgradeList()
|
|
|
|
# netsrpm updates ask and install
|
|
def SRPMSAskAndInstall(self):
|
|
global NetsrpmsArg, lista
|
|
if not lista == []:
|
|
Lista = self.Lista()
|
|
msg = QMessageBox.question(self.menu,
|
|
(_("Not Open Source {} components").format(distro.name())),
|
|
(_("The following components are available for update: ") +
|
|
"\n\n" + Lista + "\n\n" + _("Ok to proceed?")),
|
|
QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
|
|
if msg == QMessageBox.Yes:
|
|
NetsrpmsArg = '-u'
|
|
self.SRPMSCheckUpgradeList()
|
|
else:
|
|
lista = []
|
|
checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)
|
|
else:
|
|
checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)
|
|
|
|
# starting from the list "lista" returns the list in readable format
|
|
# FIXME: make this code more clear and clean
|
|
def Lista(self):
|
|
testo = ''
|
|
for a in lista:
|
|
testo = testo + a + ', '
|
|
return testo[0:len(testo)-2]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app = QApplication(sys.argv)
|
|
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())
|