Adding New/Updated Automation Files

1 files changed, 33 insertions, 0 deletions

diff --git a/SharedModules/Utils/serverutils.py b/SharedModules/Utils/serverutils.py
new file mode 100644
index 0000000..618ee23
--- /dev/null
+++ b/SharedModules/Utils/serverutils.py
@@ -0,0 +1,33 @@
+"""serverutils module
+"""
+import re
+import hostutils
+from atfglobals import GlobalObj
+
+def execute_on_brick(brickkey, command, commandInput=None):
+    """
+    """
+    env = GlobalObj.getTestenvObj()
+
+    raw_brick_obj = env.getRawBrick(brickkey)
+    if not raw_brick_obj:
+        print "InValid Brick. %s not defined in TestEnvironment" % brickkey
+        return 1
+    serverkey = re.split("\.", raw_brick_obj.hostname, maxsplit=1)[0]
+    
+    brick_obj = env.getBrick(brickkey)
+    if not brick_obj:
+        print "InValid Brick. %s not defined in TestEnvironment" % brickkey
+        return 1
+    exportdirpath = brick_obj.path
+
+    command = "cd " + exportdirpath + ";" + command
+    return_status = hostutils.execute_command(serverkey, command, commandInput)
+    return return_status
+
+__all__ = ['execute_on_brick']
+
+
+    
+    
+