from subprocess import check_output from sys import argv, platform, exit from shlex import split as s_split from distutils import sysconfig from distutils.command import build from setuptools import setup, Extension from setuptools.command.test import test as TestCommand from setuptools import dist # Determine if a base directory has been provided with the --basedir option basedir = None in_tree = False # Add compiler flags if debug is set compile_args = [] link_args = [] for arg in argv: if arg.startswith('--debug'): # Note from GCC manual: # If you use multiple -O options, with or without level numbers, # the last such option is the one that is effective. compile_args.extend(['-Wall', '-O0', '-g']) elif arg.startswith('--basedir='): basedir = arg.split('=')[1] in_tree = True # If a base directory has been provided, we use it if in_tree: base_cmd = '{0}/net-snmp-config {{{0}}}'.format(basedir) libs_cmd = base_cmd.format('--build-lib-dirs {0}'.format(basedir)) incl_cmd = base_cmd.format('--build-includes {0}'.format(basedir)) netsnmp_libs = check_output(base_cmd.format('--libs'), shell=True).decode() libdirs = check_output(libs_cmd, shell=True).decode() incdirs = check_output(incl_cmd, shell=True).decode() libs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == '-l'] libdirs = [flag[2:] for flag in s_split(libdirs) if flag[:2] == '-L'] incdirs = [flag[2:] for flag in s_split(incdirs) if flag[:2] == '-I'] # Otherwise, we use the system-installed SNMP libraries else: netsnmp_libs = check_output('net-snmp-config --libs', shell=True).decode() pass_next = False has_arg = ('-framework',) for flag in s_split(netsnmp_libs): if pass_next: link_args.append(flag) pass_next = False elif flag in has_arg: # -framework CoreFoundation link_args.append(flag) pass_next = True elif flag[:2] == '-f': # -flat_namespace link_args.append(flag) pass_next = False # link_args += [flag for flag in s_split(netsnmp_libs) if flag[:2] == '-f'] libs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == '-l'] libdirs = [flag[2:] for flag in s_split(netsnmp_libs) if flag[:2] == '-L'] incdirs = [] if platform == 'darwin': # OS X brew = check_output('brew info net-snmp', shell=True).decode() if 'command not found' not in brew: # /usr/local/opt is the default brew `opt` prefix, however the user # may have installed it elsewhere. The `brew info ` includes # an apostrophe, which breaks shlex. We'll simply replace it buildvars = list( map(lambda e: e.split('"', 1)[1].strip('"'), filter(lambda var: '="' in var, brew.split()))) libdirs += [flag[2:] for flag in buildvars if flag[:2] == '-L'] incdirs += [flag[2:] for flag in buildvars if flag[:2] == '-I'] # The homebrew version also depends on the Openssl keg openssl_ver = list(filter(lambda o: 'openssl' in o, *map(str.split, filter(lambda l: 'openssl' in l, str(brew.replace('\'', '')).split('\n') ))))[0] brew = check_output( 'brew info {0}'.format(openssl_ver), shell=True ).decode() buildvars = list( map(lambda e: e.split('"', 1)[1].strip('"'), filter(lambda var: '="' in var, brew.split()))) libdirs += [flag[2:] for flag in buildvars if flag[:2] == '-L'] incdirs += [flag[2:] for flag in buildvars if flag[:2] == '-I'] # Setup the py.test class for use with the test command class PyTest(TestCommand): user_options = [('pytest-args=', 'a', 'Arguments to pass to py.test')] def initialize_options(self): TestCommand.initialize_options(self) self.pytest_args = [] def finalize_options(self): TestCommand.finalize_options(self) self.test_args = [] self.test_suite = True def run_tests(self): # Import here, cause outside the eggs aren't loaded import pytest errno = pytest.main(self.pytest_args) exit(errno) # Read the long description from README.rst with open('README.rst') as f: long_description = f.read() setup( name='easysnmp', version='0.2.6', description='A blazingly fast and Pythonic SNMP library based on the ' 'official Net-SNMP bindings', long_description=long_description, author='Kent Coble', author_email='coblekent@gmail.com', url='https://github.com/kamakazikamikaze/easysnmp', license='BSD', packages=['easysnmp'], tests_require=['pytest-cov', 'pytest-flake8', 'pytest-sugar', 'pytest'], cmdclass={'test': PyTest}, ext_modules=[ Extension( 'easysnmp.interface', ['easysnmp/interface.c'], library_dirs=libdirs, include_dirs=incdirs, libraries=libs, extra_compile_args=compile_args, extra_link_args=link_args ) ], classifiers=[ 'Development Status :: 4 - Beta', 'Intended Audience :: Developers', 'License :: OSI Approved :: BSD License', 'Operating System :: OS Independent', 'Programming Language :: Python', 'Programming Language :: Python :: 2.7', 'Programming Language :: Python :: 3.5', 'Programming Language :: Python :: 3.6', 'Programming Language :: Python :: 3.7', 'Programming Language :: Python :: 3.8', 'Programming Language :: Python :: 3.9', 'Programming Language :: Python :: 3.10', 'Topic :: System :: Networking', 'Topic :: System :: Networking :: Monitoring' ] ) if platform == 'darwin': # Newer Net-SNMP dylib may not be linked to properly b = build.build(dist.Distribution()) # Dynamically determine build path b.finalize_options() ext = sysconfig.get_config_var('EXT_SUFFIX') or '.so' # None for Python 2 linked = check_output(( "otool -L {0}/easysnmp/interface{1} | " r"egrep 'libnetsnmp\.' | " "tr -s '\t' ' ' | " "cut -d' ' -f2").format( b.build_platlib, ext ), shell=True).decode().strip() target_libs = check_output( "find {0} -name libnetsnmp.*.dylib".format(' '.join(libdirs)), shell=True).decode().strip().split() prefix = check_output( "net-snmp-config --prefix", shell=True).decode().strip() for lib in target_libs: if prefix in lib: target_lib = lib break _ = check_output( 'install_name_tool -change {0} {1} {2}/easysnmp/interface{3}'.format( linked, target_lib, b.build_platlib, ext), shell=True)