diff options
author | Paul Kocialkowski | 2023-01-28 18:50:01 +0100 |
---|---|---|
committer | Paul Kocialkowski | 2023-01-28 21:35:17 +0100 |
commit | ed7e44f1396362d6d8dc2a11419fe3dc58bd5d5f (patch) | |
tree | e9989a80b2c4811be85e713d462f50e6f1323621 | |
parent | 8df80d7254d370ddb3932de9f98072e736136af3 (diff) |
host-control: Support running in sudo
Signed-off-by: Paul Kocialkowski <contact@paulk.fr>
-rwxr-xr-x | host-control | 10 |
1 files changed, 9 insertions, 1 deletions
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() |