summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/leap/bitmask/gui/app.py17
1 files changed, 16 insertions, 1 deletions
diff --git a/src/leap/bitmask/gui/app.py b/src/leap/bitmask/gui/app.py
index 38a37c4..8ec15b2 100644
--- a/src/leap/bitmask/gui/app.py
+++ b/src/leap/bitmask/gui/app.py
@@ -47,6 +47,11 @@ else:
from PyQt5.QtCore import QSize
from PyQt5.QtCore import QObject, pyqtSlot
from PyQt5.QtWidgets import QApplication
+ from PyQt5.QtGui import QIcon
+ from PyQt5.QtWidgets import QMenu
+ from PyQt5.QtWidgets import QSystemTrayIcon
+ from PyQt5.QtWidgets import QDialog
+
try:
from PyQt5.QtWebKitWidgets import QWebView
from PyQt5.QtWebKit import QWebSettings
@@ -66,7 +71,14 @@ bitmaskd = None
browser = None
-class BrowserWindow(QWebView):
+class WithTrayIcon(QDialog):
+
+ def createTrayIcon(self):
+ self.trayIcon = QSystemTrayIcon(self)
+ self.trayIcon.setContextMenu(self.trayIconMenu)
+
+
+class BrowserWindow(QWebView, WithTrayIcon):
"""
This qt-webkit window exposes a couple of callable objects through the
python-js bridge:
@@ -216,6 +228,9 @@ def launch_gui():
print('ERROR: ' + e.message)
sys.exit(1)
+ browser.createTrayIcon()
+ browser.trayIcon.show()
+
qApp.setQuitOnLastWindowClosed(True)
qApp.lastWindowClosed.connect(browser.shutdown)