blob: 25e0fda0b57c461d9c3c5dd70368887e6a03c414 [file] [log] [blame]
# 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(), 'dependency_manager'))
_AddDirToPythonPath(os.path.join(util.GetCatapultDir(), 'third_party', 'mock'))
# mox3 is needed for pyfakefs usage, but not for pylint.
_AddDirToPythonPath(os.path.join(util.GetCatapultDir(), 'third_party', 'mox3'))
_AddDirToPythonPath(
os.path.join(util.GetCatapultDir(), 'third_party', 'pyfakefs'))