xref: /aosp_15_r20/external/autotest/cli/atest_venv_entry.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
1*9c5db199SXin Li#!/usr/bin/python3 -u
2*9c5db199SXin Liimport os
3*9c5db199SXin Liimport sys
4*9c5db199SXin Li
5*9c5db199SXin Liimport common
6*9c5db199SXin Li
7*9c5db199SXin Li_AUTOTEST_ROOT = os.path.realpath(os.path.join(__file__, '..', '..'))
8*9c5db199SXin Li_CHROMIUMOS_ROOT = os.path.abspath(
9*9c5db199SXin Li    os.path.join(_AUTOTEST_ROOT, '..', '..', '..', '..'))
10*9c5db199SXin Li_SKYLAB_INVENTORY_DIR = os.path.join(_CHROMIUMOS_ROOT, 'infra',
11*9c5db199SXin Li                                     'skylab_inventory', 'venv')
12*9c5db199SXin Li# In any valid chromiumos checkout
13*9c5db199SXin Lisys.path.append(_SKYLAB_INVENTORY_DIR)
14*9c5db199SXin Li# TODO: Where is this checked out on infra servers?
15*9c5db199SXin Li
16*9c5db199SXin Litry:
17*9c5db199SXin Li    import skylab_inventory  # pylint: disable=unused-import
18*9c5db199SXin Liexcept ImportError as e:
19*9c5db199SXin Li    raise Exception(
20*9c5db199SXin Li            'Error when importing skylab_inventory (venv dir: %s): %s' %
21*9c5db199SXin Li            (_SKYLAB_INVENTORY_DIR, e))
22*9c5db199SXin Li
23*9c5db199SXin Li# Import atest after 'import skylab_inventory' as it uses skylab_inventory
24*9c5db199SXin Lifrom autotest_lib.cli import atest
25*9c5db199SXin Li
26*9c5db199SXin Lisys.exit(atest.main())
27