summaryrefslogtreecommitdiffstats
path: root/scripts/stl-sim
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/stl-sim')
-rwxr-xr-xscripts/stl-sim18
1 files changed, 16 insertions, 2 deletions
diff --git a/scripts/stl-sim b/scripts/stl-sim
index 57fe4fa8..198d1275 100755
--- a/scripts/stl-sim
+++ b/scripts/stl-sim
@@ -1,6 +1,20 @@
#!/bin/bash
-source find_python.sh
+INPUT_ARGS=$@
+
+if [[ $@ =~ '--python2' ]]; then
+ INPUT_ARGS=${@//--python2/}
+ source find_python.sh --python2
+fi
+
+if [[ $@ =~ '--python3' ]]; then
+ INPUT_ARGS=${@//--python3/}
+ source find_python.sh --python3
+fi
+
+if [ -z "$PYTHON" ]; then
+ source find_python.sh
+fi
export PYTHONPATH=automation/trex_control_plane/stl
-$PYTHON -m trex_stl_lib.trex_stl_sim -p $PWD $@
+$PYTHON -m trex_stl_lib.trex_stl_sim -p $PWD $INPUT_ARGS