diff options
Diffstat (limited to 'host-control')
-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() |