1# Copyright (c) 2012 The Chromium OS Authors. All rights reserved. 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5# Sets up the cros.factory module path. This is necessary since there 6# is already a cros directory, and we need to rejigger things so that 7# cros.factory points to the correct path. 8 9import imp, logging, os, sys 10 11# If SYSROOT is present, also look in 12# $SYSROOT/usr/local/factory/py_pkg (necessary during the build step). 13sysroot = os.environ.get('SYSROOT') 14extra_path = ([os.path.join(sysroot, 'usr/local/factory/py_pkg')] 15 if sysroot else []) 16 17# Try to import cros, or just create a dummy module if it doesn't 18# exist. 19try: 20 import cros 21except: 22 cros = imp.load_module('cros', None, '', ('', '', imp.PKG_DIRECTORY)) 23 24# Load cros.factory, inserting it into the cros module. 25cros.factory = imp.load_module( 26 'cros.factory', 27 *imp.find_module('cros/factory', sys.path + extra_path)) 28