host-control: Support running in sudo

Signed-off-by: Paul Kocialkowski <contact@paulk.fr>

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()