From ed7e44f1396362d6d8dc2a11419fe3dc58bd5d5f Mon Sep 17 00:00:00 2001 From: Paul Kocialkowski Date: Sat, 28 Jan 2023 18:50:01 +0100 Subject: host-control: Support running in sudo Signed-off-by: Paul Kocialkowski --- host-control | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/host-control b/host-control index 71863a0..1b60b2f 100755 --- a/host-control +++ b/host-control @@ -7,7 +7,7 @@ import socket import subprocess import yaml -data_path = os.path.expanduser("~") + "/.config/host-control.yaml" +data_path_base = ".config/host-control.yaml" class host_control(): host = None @@ -395,6 +395,14 @@ class host_control(): # main def data_load(self): + if "SUDO_USER" in os.environ: + user = os.environ["SUDO_USER"] + else: + user = os.environ["USER"] + + data_path_user = os.path.expanduser("~" + user) + data_path = os.path.join(data_path_user, data_path_base) + s = open(data_path, "r") y = yaml.load(s, Loader = yaml.SafeLoader) s.close() -- cgit v1.2.3