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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
|
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)
|