blob: 4f8c79f7c76005f2ee77a853af0f93573054fa28 [file] [log] [blame]
# Copyright (c) 2012 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.
# Sets up the cros.factory module path. This is necessary since there
# is already a cros directory, and we need to rejigger things so that
# cros.factory points to the correct path.
import imp, logging, os, sys
# If SYSROOT is present, also look in
# $SYSROOT/usr/local/factory/py_pkg (necessary during the build step).
sysroot = os.environ.get('SYSROOT')
extra_path = ([os.path.join(sysroot, 'usr/local/factory/py_pkg')]
if sysroot else [])
# Try to import cros, or just create a dummy module if it doesn't
# exist.
import cros
cros = imp.load_module('cros', None, '', ('', '', imp.PKG_DIRECTORY))
# Load cros.factory, inserting it into the cros module.
cros.factory = imp.load_module(
*imp.find_module('cros/factory', sys.path + extra_path))