Created a client/common_lib/utils/import_site_class function to take
care of site specific importing of classes. Modified code to use it.
Signed-off-by: Mihai Rusu <dizzy@google.com>
git-svn-id: http://test.kernel.org/svn/autotest/trunk@2626 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/client/bin/sysinfo.py b/client/bin/sysinfo.py
index 5a92431..3d88e3b 100755
--- a/client/bin/sysinfo.py
+++ b/client/bin/sysinfo.py
@@ -1,14 +1,9 @@
+from autotest_lib.client.common_lib import utils
from autotest_lib.client.bin import base_sysinfo
-try:
- from autotest_lib.client.bin import site_sysinfo
-except ImportError:
- # no site_sysinfo, just make a class using the base version
- class sysinfo(base_sysinfo.base_sysinfo):
- pass
-else:
- # otherwise, use the site version (should inherit from the base)
- class sysinfo(site_sysinfo.site_sysinfo):
- pass
+
+sysinfo = utils.import_site_class(__file__,
+ "autotest_lib.client.bin.site_sysinfo",
+ "site_sysinfo", base_sysinfo.base_sysinfo)
# pull in some data stucture stubs from base_sysinfo, for convenience
logfile = base_sysinfo.logfile