From 68ed79b989008006897108c7978d5c65fef47711 Mon Sep 17 00:00:00 2001
From: Yaroslav Brustinov <ybrustin@cisco.com>
Date: Sun, 27 Mar 2016 19:06:48 +0300
Subject: fix find_python usage

---
 scripts/stl-sim | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

(limited to 'scripts/stl-sim')

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
-- 
cgit 1.2.3-korg