1 files changed, 17 insertions, 1 deletions
diff --git a/scripts/automation/trex_control_plane/client_utils/general_utils.py b/scripts/automation/trex_control_plane/client_utils/general_utils.py
index 5488b9dd..69ad14b2 100755
--- a/scripts/automation/trex_control_plane/client_utils/general_utils.py
+++ b/scripts/automation/trex_control_plane/client_utils/general_utils.py
@@ -24,7 +24,7 @@ def user_input():
 
 def get_current_user():
   if pwd:
-      return pwd.getpwuid( os.geteuid() ).pw_name
+      return pwd.getpwuid(os.geteuid()).pw_name
   else:
       return getpass.getuser()
 
@@ -75,6 +75,22 @@ def random_id_gen(length=8):
             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