summaryrefslogtreecommitdiffstats
path: root/scripts/automation/trex_control_plane/client_utils/outer_packages.py
blob: a6c9a2ebe8e425b1b7e90e5fe4b95482a08f3be2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/router/bin/python

import sys
import site
import os

CURRENT_PATH        = os.path.dirname(os.path.realpath(__file__))
ROOT_PATH           = os.path.abspath(os.path.join(CURRENT_PATH, os.pardir))     # path to trex_control_plane directory
PATH_TO_PYTHON_LIB  = os.path.abspath(os.path.join(ROOT_PATH, os.pardir, os.pardir, 'external_libs'))

CLIENT_UTILS_MODULES = ['zmq',
                        'dpkt-1.8.6'
                        ]


def import_client_utils_modules():
    # must be in a higher priority
    sys.path.insert(0, PATH_TO_PYTHON_LIB)
    sys.path.append(ROOT_PATH)
    import_module_list(CLIENT_UTILS_MODULES)


def import_module_list(modules_list):
    assert(isinstance(modules_list, list))
    for p in modules_list:
        full_path = os.path.join(PATH_TO_PYTHON_LIB, p)
        fix_path = os.path.normcase(full_path)
        site.addsitedir(full_path)

import_client_utils_modules()