From 59f00264f39dfb95a62401776b4ed8c433682fa7 Mon Sep 17 00:00:00 2001
From: Yaroslav Brustinov <ybrustin@cisco.com>
Date: Thu, 24 Mar 2016 20:00:27 +0200
Subject: gather stateful client to stf folder, create trex_client package

---
 .../trex_control_plane/stf/general_utils.py        | 95 ++++++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100755 scripts/automation/trex_control_plane/stf/general_utils.py

(limited to 'scripts/automation/trex_control_plane/stf/general_utils.py')

diff --git a/scripts/automation/trex_control_plane/stf/general_utils.py b/scripts/automation/trex_control_plane/stf/general_utils.py
new file mode 100755
index 00000000..d2521f02
--- /dev/null
+++ b/scripts/automation/trex_control_plane/stf/general_utils.py
@@ -0,0 +1,95 @@
+#!/router/bin/python
+
+import sys
+import site
+import string
+import random
+import os
+
+try:
+    import pwd
+except ImportError:
+    import getpass
+    pwd = None
+
+using_python_3 = True if sys.version_info.major == 3 else False
+
+
+def user_input():
+    if using_python_3:
+        return input()
+    else:
+        # using python version 2
+        return raw_input()
+
+def get_current_user():
+  if pwd:
+      return pwd.getpwuid(os.geteuid()).pw_name
+  else:
+      return getpass.getuser()
+
+def import_module_list_by_path (modules_list):
+    assert(isinstance(modules_list, list))
+    for full_path in modules_list:
+        site.addsitedir(full_path)
+
+def find_path_to_pardir (pardir, base_path = os.getcwd() ):
+    """
+    Finds the absolute path for some parent dir `pardir`, starting from base_path
+
+    The request is only valid if the stop initiator is the same client as the TRex run initiator.
+            
+    :parameters:        
+        pardir : str
+            name of an upper-level directory to which we want to find an absolute path for
+        base_path : str
+            a full (usually nested) path from which we want to find a parent folder.
+
+            default value : **current working dir**
+
+    :return: 
+        string representation of the full path to 
+
+    """
+    components = base_path.split(os.sep)
+    return str.join(os.sep, components[:components.index(pardir)+1])
+
+
+def random_id_gen(length=8):
+    """
+    A generator for creating a random chars id of specific length
+
+    :parameters:
+        length : int
+            the desired length of the generated id
+
+            default: 8
+
+    :return:
+        a random id with each next() request.
+    """
+    id_chars = string.ascii_lowercase + string.digits
+    while True:
+        return_id = ''
+        for i in range(length):
+            return_id += random.choice(id_chars)
+        yield return_id
+
+def id_count_gen():
+    """
+    A generator for creating an increasing id for objects, starting from 0
+
+    :parameters:
+        None
+
+    :return:
+        an id (unsigned int) with each next() request.
+    """
+    return_id = 0
+    while True:
+        yield return_id
+        return_id += 1
+
+
+if __name__ == "__main__":
+    pass
-- 
cgit 1.2.3-korg