Index: tools/telemetry/telemetry/core/util.py |
diff --git a/tools/telemetry/telemetry/core/util.py b/tools/telemetry/telemetry/core/util.py |
index 5b11640fe38ad9fb1f443f914fe5075902ee7d2c..fddab1c3c5eb5e6f09c4085e736413cd38d0e4e1 100644 |
--- a/tools/telemetry/telemetry/core/util.py |
+++ b/tools/telemetry/telemetry/core/util.py |
@@ -4,11 +4,19 @@ |
import inspect |
import os |
import socket |
+import sys |
import time |
class TimeoutException(Exception): |
pass |
+def GetBaseDir(): |
+ main_module = sys.modules['__main__'] |
+ if hasattr(main_module, '__file__'): |
+ return os.path.dirname(os.path.abspath(main_module.__file__)) |
+ else: |
+ return os.getcwd() |
+ |
def GetTelemetryDir(): |
return os.path.normpath(os.path.join( |
__file__, os.pardir, os.pardir, os.pardir)) |