summaryrefslogtreecommitdiff
path: root/src/leap/bitmask/vpn/privilege.py
diff options
context:
space:
mode:
authorKali Kaneko (leap communications) <kali@leap.se>2017-05-31 18:00:48 +0200
committerKali Kaneko (leap communications) <kali@leap.se>2017-05-31 18:53:42 +0200
commit72d93a61cf9aff5996952d55e56981eb5a22d989 (patch)
tree6b36868fdb46132c413c6fbcfe72f20d3df3fc79 /src/leap/bitmask/vpn/privilege.py
parent59807917612e301ac37d59f0d4d832979316896c (diff)
[refactor] factor out installer function
Diffstat (limited to 'src/leap/bitmask/vpn/privilege.py')
-rw-r--r--src/leap/bitmask/vpn/privilege.py24
1 files changed, 10 insertions, 14 deletions
diff --git a/src/leap/bitmask/vpn/privilege.py b/src/leap/bitmask/vpn/privilege.py
index d918a30..c7cf119 100644
--- a/src/leap/bitmask/vpn/privilege.py
+++ b/src/leap/bitmask/vpn/privilege.py
@@ -41,31 +41,27 @@ log = Logger()
# TODO wrap the install/uninstall helper functions around the policychecker
# classes below.
+
def install_helpers():
- if IS_LINUX:
- cmd = 'bitmask_helpers install'
- if STANDALONE:
- binary_path = os.path.join(here(), "bitmask")
- cmd = "%s %s" % (binary_path, cmd)
- if os.getuid() != 0:
- cmd = 'pkexec ' + cmd
- retcode, _ = commands.getstatusoutput(cmd)
- if retcode != 0:
- raise Exception('Could not install helpers')
- else:
- raise Exception('No install mechanism for this platform')
+ _helper_installer('install')
def uninstall_helpers():
+ _helper_installer('uninstall')
+
+
+def _helper_installer(action):
+ if action not in ('install', 'uninstall'):
+ raise Exception('Wrong action: %s' % action)
+
if IS_LINUX:
- cmd = 'bitmask_helpers uninstall'
+ cmd = 'bitmask_helpers ' + action
if STANDALONE:
binary_path = os.path.join(here(), "bitmask")
cmd = "%s %s" % (binary_path, cmd)
if os.getuid() != 0:
cmd = 'pkexec ' + cmd
retcode, _ = commands.getstatusoutput(cmd)
- commands.getoutput('pkexec ' + cmd)
if retcode != 0:
raise Exception('Could not uninstall helpers')
else: