added vpn module

* shows current VPN connection
* allows to establish a VPN connection
This commit is contained in:
Bernhard B 2019-06-10 20:05:32 +02:00
parent 315328b41f
commit a7a1da6b5d
2 changed files with 93 additions and 0 deletions

90
bumblebee/modules/vpn.py Normal file
View file

@ -0,0 +1,90 @@
# pylint: disable=C0111,R0903
""" Displays the VPN profile that is currently in use.
Left click opens a popup menu that lists all available VPN profiles and allows to establish
a VPN connection using that profile.
Prerequisites:
* nmcli needs to be installed and configured properly.
To quickly test, whether nmcli is working correctly, type "nmcli -g NAME,TYPE,DEVICE con" which
lists all the connection profiles that are configured. Make sure that your VPN profile is in that list!
e.g: to import a openvpn profile via nmcli:
sudo nmcli connection import type openvpn file </path/to/your/openvpn/profile.ovpn>
"""
import logging
import bumblebee.input
import bumblebee.output
import bumblebee.engine
import functools
class Module(bumblebee.engine.Module):
def __init__(self, engine, config):
super(Module, self).__init__(engine, config,
bumblebee.output.Widget(full_text=self.vpn_status)
)
self._connected_vpn_profile = None
self._selected_vpn_profile = None
res = bumblebee.util.execute("nmcli -g NAME,TYPE c")
lines = res.splitlines()
self._vpn_profiles = []
for line in lines:
info = line.split(':')
if info[1] == "vpn":
self._vpn_profiles.append(info[0])
engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
cmd=self.popup)
def update(self, widgets):
try:
res = bumblebee.util.execute("nmcli -g NAME,TYPE,DEVICE con")
lines = res.splitlines()
self._connected_vpn_profile = None
for line in lines:
info = line.split(':')
if info[1] == "vpn" and info[2] != "":
self._connected_vpn_profile = info[0]
except Exception as e:
logging.exception("Couldn't get VPN status")
self._connected_vpn_profile = None
def vpn_status(self, widget):
if self._connected_vpn_profile is None:
return "off"
return self._connected_vpn_profile
def _on_vpn_disconnect(self):
try:
bumblebee.util.execute("nmcli c down " + self._connected_vpn_profile)
self._connected_vpn_profile = None
except Exception as e:
logging.exception("Couldn't disconnect VPN connection")
def _on_vpn_connect(self, name):
self._selected_vpn_profile = name
try:
bumblebee.util.execute("nmcli c up " + self._selected_vpn_profile)
self._connected_vpn_profile = name
except Exception as e:
logging.exception("Couldn't establish VPN connection")
self._connected_vpn_profile = None
def popup(self, widget):
menu = bumblebee.popup_v2.PopupMenu()
if self._connected_vpn_profile is not None:
menu.add_menuitem("Off", callback=self._on_vpn_disconnect)
for vpn_profile in self._vpn_profiles:
menu.add_menuitem(vpn_profile, callback=functools.partial(self._on_vpn_connect, vpn_profile))
menu.show(widget)
def state(self, widget):
return []

View file

@ -196,5 +196,8 @@
"pihole": { "pihole": {
"enabled": { "prefix": "" }, "enabled": { "prefix": "" },
"disabled": { "prefix": "" } "disabled": { "prefix": "" }
},
"vpn": {
"prefix": ""
} }
} }