You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

133 lines
5.3 KiB

#!/usr/bin/env python3
# just import all python3 modules used by osmo-gsm-tester to make sure they are
# installed.
import os
import sys
import argparse
import pprint
import subprocess
feature_module_map = {
'2g': ['bsc_osmo', 'bts_nanobts', 'bts_oc2g', 'bts_octphy', 'bts_osmo', 'bts_osmotrx', 'bts_osmovirtual', 'bts_sysmo', 'bts', 'bts', 'esme', 'ggsn_osmo', 'hlr_osmo', 'mgcpgw_osmo', 'mgw_osmo', 'ms_ofono', 'ms_driver', 'msc_osmo', 'nitb_osmo', 'osmo_ctrl', 'osmocon', 'pcap_recorder', 'pcu_oc2g', 'pcu_osmo', 'pcu_sysmo', 'pcu', 'sgsn_osmo', 'sms', 'smsc', 'stp_osmo'],
'4g': [],
'srs': ['enb_srs', 'epc_srs', 'ms_srs'],
'powersupply': ['powersupply', 'powersupply_intellinet', 'powersupply_sispm'],
'rfemu': ['rfemu', 'rfemu_amarisoftctrl', 'rfemu_minicircuits'],
def skip_features_to_skip_modules(skip_features):
skip_obj_modules = []
for skip_feature in skip_features:
if skip_feature in feature_module_map:
for skip_module in feature_module_map[skip_feature]:
return skip_obj_modules
def import_runtime_dependencies():
# we don't have any right now, but in the future if we import a module during runtime (eg inside a function), then we need to place it here:
# import foobar
def import_all_py_in_dir(rel_path, skip_modules=[]):
selfdir = os.path.dirname(os.path.abspath(__file__))
dir = os.path.join(selfdir, rel_path)
print('importing files in directory %s' % dir)
for entry in os.listdir(dir):
full_entry = os.path.join(selfdir, rel_path, entry)
if not os.path.isfile(full_entry):
if args.verbose:
print('skipping entry %s' % full_entry)
if not full_entry.endswith('.py'):
if args.verbose:
print('skipping file %s' % full_entry)
modulename = entry[:-3]
if modulename in skip_modules:
if args.verbose:
print('skipping module %s' % modulename)
modulepath = rel_path.replace('/', '.') + '.' + modulename
print('importing %s' % modulepath)
__import__(modulepath, globals(), locals())
def get_module_names():
all_modules_filtered = {}
for mname, m in all_modules:
if not hasattr(m, '__file__'):
continue # skip built-in modules
if mname.startswith('_'):
continue # skip internal modules
if mname.startswith('src.osmo_') or 'osmo_gsm_tester' in mname or 'osmo_ms_driver' in mname:
continue # skip our own local modules
mname = mname.split('.')[0] # store only main module
if m not in all_modules_filtered.values():
all_modules_filtered[mname] = m
return all_modules_filtered
def print_deb_packages(modules):
packages_deb = []
modules_err = []
for mname, m in modules.items():
proc = subprocess.Popen(["dpkg", "-S", m.__file__], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
outs, errs = proc.communicate()
if args.verbose:
print('out: %s, err: %s' %(outs, errs))
if len(errs): # error -> package not found (installed through pip?)
modules_err.append((mname, errs.decode('utf-8')))
elif len(outs):
outs = outs.decode('utf-8')
outs = outs.split()[0].rstrip(':') # first part is debian package name
if not outs in packages_deb:
print('WARNING: dpkg returns empty!')
print('Debian packages:')
for pkgname in packages_deb:
print("\t" + pkgname)
print('Modules without debian package (pip or setuptools?):')
for mname, err in modules_err:
print("\t" + mname.ljust(20) + " [" + err.rstrip() +"]")
parser = argparse.ArgumentParser(epilog=__doc__, formatter_class=argparse.RawTextHelpFormatter)
parser.add_argument('-s', '--skip-feature', dest='skip_features', action='append',
help='''All osmo-gsm-tester features not used by the user running the script''')
parser.add_argument('-p', '--distro-packages', dest='distro_packages', action='store_true',
help='Print distro packages installing modules')
parser.add_argument('-v', '--verbose', dest='verbose', action='store_true',
help='Print a lot more information')
args = parser.parse_args()
skip_obj_modules = skip_features_to_skip_modules(list(args.skip_features or []))
print('Skip checking modules: %r' % skip_obj_modules)
# We need to add it for cross-references between osmo_ms_driver and osmo_gsm_tester to work:
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), 'src/'))
import_all_py_in_dir('src/osmo_gsm_tester/obj', skip_obj_modules)
print('Importing dependencies ok, all installed')
print('Retreiving list of imported modules...')
modules = get_module_names()
if args.verbose:
for mname, m in modules.items():
print('%s --> %s' %(mname, m.__file__))
if args.distro_packages:
print('Generating distro package list from imported module list...')