add udp_pkt.py example

2 files changed, 10 insertions, 3 deletions

diff --git a/scripts/automation/trex_control_plane/client/outer_packages.py b/scripts/automation/trex_control_plane/client/outer_packages.py
index f7e2bac9..2565be08 100755
--- a/scripts/automation/trex_control_plane/client/outer_packages.py
+++ b/scripts/automation/trex_control_plane/client/outer_packages.py
@@ -3,6 +3,7 @@
 import sys
 import os
 
+
 CURRENT_PATH        = os.path.dirname(os.path.realpath(__file__))
 ROOT_PATH           = os.path.abspath(os.path.join(CURRENT_PATH, os.pardir))     # path to trex_control_plane directory
 PATH_TO_PYTHON_LIB  = os.path.abspath(os.path.join(ROOT_PATH, os.pardir, os.pardir, 'external_libs'))
diff --git a/scripts/automation/trex_control_plane/client/trex_stateless_client.py b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
index ebc9a6ad..95fd2a69 100755
--- a/scripts/automation/trex_control_plane/client/trex_stateless_client.py
+++ b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
@@ -28,6 +28,7 @@ from trex_async_client import CTRexAsyncClient
 from yaml import YAMLError
 
 
+
 ############################     logger     #############################
 ############################                #############################
 ############################                #############################
@@ -1309,17 +1310,22 @@ class STLClient(object):
             # convert to new style stream object
             streams = [HACKSTLStream(stream) for stream in stream_list.compiled]
         except YAMLError:
-            # try python
+            # try python loader
             try:
                 basedir = os.path.dirname(filename)
+
                 sys.path.append(basedir)
                 file    = os.path.basename(filename).split('.')[0]
                 module = __import__(file, globals(), locals(), [], -1)
+                reload(module) # reload the update 
 
                 streams = module.register().get_streams()
 
-            except (AttributeError, ImportError):
-                raise STLError("bad format input file '{0}'".format(filename))
+            except Exception as e :
+                print str(e);
+                traceback.print_exc(file=sys.stdout)
+                raise STLError("Unexpected error: '{0}'".format(filename))
+
 
         self.add_streams(streams, ports)