Index: tools/telemetry/catapult_base/__init__.py |
diff --git a/tools/telemetry/catapult_base/__init__.py b/tools/telemetry/catapult_base/__init__.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..653dd32922af820f43a9c31378fe3dfa3800cb6c |
--- /dev/null |
+++ b/tools/telemetry/catapult_base/__init__.py |
@@ -0,0 +1,23 @@ |
+# Copyright 2015 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+# All files in this directory should be moved to catapult/base/ after moving |
+# to the new repo. |
+ |
+import os |
+import sys |
+ |
+from catapult_base import util |
+ |
+def _AddDirToPythonPath(*path_parts): |
+ path = os.path.abspath(os.path.join(*path_parts)) |
+ if os.path.isdir(path) and path not in sys.path: |
+ # Some callsite that use telemetry assumes that sys.path[0] is the directory |
+ # containing the script, so we add these extra paths to right after it. |
+ sys.path.insert(1, path) |
+ |
+_AddDirToPythonPath(os.path.join(util.GetCatapultDir(), 'third_party', 'mock')) |
+_AddDirToPythonPath(os.path.join(util.GetCatapultDir(), 'third_party', 'mox3')) |
+_AddDirToPythonPath( |
+ os.path.join(util.GetCatapultDir(), 'third_party', 'pyfakefs')) |