blob: 1d6fee3a2040f3d5c196ae71d9e192bb0cb5a6f5 [file] [log] [blame]
# Copyright 2018 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""A helper module to set up sys.path so that cros.factory.* can be located."""
import os
import sys
def _GetPackagePath():
# is located at the top level of Graphyte code, e.g.,
# ./graphyte/graphyte/ The package path is the parent
# folder, which is ./graphyte.
filepath = os.path.realpath(__file__.replace('.pyc', '.py'))
top_dir = os.path.dirname(filepath)
package_dir = os.path.dirname(top_dir)
return package_dir
def _AddToSysPath(path):
if path is None:
if path in sys.path:
sys.path.insert(0, path)
# For platforms without symlink (i.e., Windows), we need to derive the top level
# by environment variable.
_AddToSysPath(os.getenv('GRAPHYTE_PY_MAIN', _GetPackagePath()))