T1431: make it possible to obtain session environment and run vyos.config functions under it.

This is required for programs running outside a CLI session, like the future API daemon.

1 files changed, 9 insertions, 2 deletions

diff --git a/python/vyos/config.py b/python/vyos/config.py
index 9a5125eb9..96e9631e0 100644
--- a/python/vyos/config.py
+++ b/python/vyos/config.py
@@ -87,9 +87,13 @@ class Config(object):
     the only state it keeps is relative *config path* for convenient access to config
     subtrees.
     """
-    def __init__(self):
+    def __init__(self, session_env=None):
         self._cli_shell_api = "/bin/cli-shell-api"
         self._level = ""
+        if session_env:
+            self.__session_env = session_env
+        else:
+            self.__session_env = None
 
     def _make_command(self, op, path):
         args = path.split()
@@ -97,7 +101,10 @@ class Config(object):
         return cmd
 
     def _run(self, cmd):
-        p = subprocess.Popen(cmd, stdout=subprocess.PIPE)
+        if self.__session_env:
+            p = subprocess.Popen(cmd, stdout=subprocess.PIPE, env=self.__session_env)
+        else:
+            p = subprocess.Popen(cmd, stdout=subprocess.PIPE)
         out = p.stdout.read()
         p.wait()
         if p.returncode != 0: