Add sys.path to include chromite
BUG=chromium-os:37911
TEST=Run generate_test_report.py, it prints help information
Change-Id: Ia02a792d6fb49527aeb6cc62577b939a64008ff4
Reviewed-on: https://gerrit.chromium.org/gerrit/42070
Reviewed-by: Chris Sosa <sosa@chromium.org>
Commit-Queue: Yunlian Jiang <yunlian@chromium.org>
Tested-by: Yunlian Jiang <yunlian@chromium.org>
diff --git a/utils_py/generate_test_report.py b/utils_py/generate_test_report.py
index 27f85cc..c02368b 100755
--- a/utils_py/generate_test_report.py
+++ b/utils_py/generate_test_report.py
@@ -18,7 +18,13 @@
import re
import sys
-from chromite.lib import cros_build_lib
+try:
+ from chromite.lib import cros_build_lib
+except ImportError:
+ script_path = os.path.dirname(os.path.abspath(__file__))
+ cros_path = os.path.join(script_path, "../../../..")
+ sys.path.append(cros_path)
+ from chromite.lib import cros_build_lib
from chromite.lib import terminal
_STDOUT_IS_TTY = hasattr(sys.stdout, 'isatty') and sys.stdout.isatty()
