VatHistory: Add ability to trace file executions

[csit.git] / resources / libraries / python / VatExecutor.py

diff --git a/resources/libraries/python/VatExecutor.py b/resources/libraries/python/VatExecutor.py
index febfe9f..de6baab 100644 (file)
--- a/resources/libraries/python/VatExecutor.py
+++ b/resources/libraries/python/VatExecutor.py
@@ -152,6 +152,9 @@ class VatExecutor(object):
             vat=Constants.VAT_BIN_NAME,
             input=vat_name)
 
             vat=Constants.VAT_BIN_NAME,
             input=vat_name)
 

+        with open(vat_name, 'r') as tmp_f:
+            VatHistory.add_to_vat_history(node, tmp_f.read())
+

         try:
             (ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout)
         except SSHTimeout:
         try:
             (ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout)
         except SSHTimeout:


@@ -197,6 +200,10 @@ class VatExecutor(object):
                                     vat=Constants.VAT_BIN_NAME)
         cmd_input = "exec exec {0}".format(fname)
 
                                     vat=Constants.VAT_BIN_NAME)
         cmd_input = "exec exec {0}".format(fname)
 

+        VatHistory.add_to_vat_history(node, cmd_input)
+        with open(fname, 'r') as tmp_f:
+            VatHistory.add_to_vat_history(node, tmp_f.read())
+

         try:
             (ret_code, stdout, stderr) = ssh.exec_command_sudo(cmd, cmd_input,
                                                                timeout)
         try:
             (ret_code, stdout, stderr) = ssh.exec_command_sudo(cmd, cmd_input,
                                                                timeout)