---
 fence/agents/amt/fence_amt.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/fence/agents/amt/fence_amt.py b/fence/agents/amt/fence_amt.py
index dc2fd12..edab61a 100644
--- a/fence/agents/amt/fence_amt.py
+++ b/fence/agents/amt/fence_amt.py
@@ -1,6 +1,6 @@
 #!/usr/bin/python -tt
 
-import sys, re
+import sys, re, os
 import atexit
 from pipes import quote
 sys.path.append("@FENCEAGENTSLIBDIR@")
@@ -14,7 +14,7 @@ BUILD_DATE=""
 #END_VERSION_GENERATION
 
 def get_power_status(_, options):
-       output = run_command(options, create_command(options, "status"))
+       output = amt_run_command(options, create_command(options, "status"))
        match = re.search('Powerstate:[\\s]*(..)', str(output))
        status = match.group(1) if match else None
 
@@ -26,18 +26,20 @@ def get_power_status(_, options):
                return "off"
 
 def set_power_status(_, options):
-       run_command(options, create_command(options, options["--action"]))
+       amt_run_command(options, create_command(options, options["--action"]))
        return
 
 def reboot_cycle(_, options):
        (status, _, _) = run_command(options, create_command(options, "cycle"))
        return not bool(status)
 
-def create_command(options, action):
-       # --password / -p
-       cmd = "AMT_PASSWORD=" + quote(options["--password"])
+def amt_run_command(options, command, timeout = None):
+       env = os.environ.copy()
+       env["AMT_PASSWORD"] = quote(options["--password"])
+       return run_command(options, command, timeout, env)
 
-       cmd += " " + options["--amttool-path"]
+def create_command(options, action):
+       cmd = options["--amttool-path"]
 
        # --ip / -a
        cmd += " " + options["--ip"]
-- 
1.8.3.1

Reply via email to