setup.py

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 <pkg>` 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)