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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/router/bin/python
import sys
import os
import warnings
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 = ['dpkt-1.8.6',
'yaml-3.11',
'texttable-0.8.4',
'scapy-2.3.1'
]
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)
sys.path.insert(1, full_path)
import_platform_dirs()
def import_platform_dirs ():
# handle platform dirs
# try fedora 18 first and then cel5.9
# we are using the ZMQ module to determine the right platform
full_path = os.path.join(PATH_TO_PYTHON_LIB, 'platform/fedora18')
fix_path = os.path.normcase(full_path)
sys.path.insert(0, full_path)
try:
# try to import and delete it from the namespace
import zmq
del zmq
return
except:
sys.path.pop(0)
pass
full_path = os.path.join(PATH_TO_PYTHON_LIB, 'platform/cel59')
fix_path = os.path.normcase(full_path)
sys.path.insert(0, full_path)
try:
# try to import and delete it from the namespace
import zmq
del zmq
return
except:
sys.path.pop(0)
sys.modules['zmq'] = None
warnings.warn("unable to determine platform type for ZMQ import")
import_client_utils_modules()
|
