setup.py

#
# This file is part of snmpsim software.
#
# Copyright (c) 2010-2019, Ilya Etingof <etingof@gmail.com>
# License: http://snmplabs.com/snmpsim/license.html
#
"""SNMP Agents simulator

   SNMP Simulator is a tool that acts as multitude of SNMP Agents built
   into real physical devices, from SNMP Manager's point of view.
   Simulator builds and uses a database of physical devices' SNMP footprints 
   to respond like their original counterparts do.
"""
import glob
import os
import sys
import setuptools

classifiers = """\
Development Status :: 5 - Production/Stable
Environment :: Console
Intended Audience :: Developers
Intended Audience :: Education
Intended Audience :: Information Technology
Intended Audience :: System Administrators
Intended Audience :: Telecommunications Industry
License :: OSI Approved :: BSD License
Natural Language :: English
Operating System :: OS Independent
Programming Language :: Python :: 2
Programming Language :: Python :: 2.7
Programming Language :: Python :: 3
Programming Language :: Python :: 3.2
Programming Language :: Python :: 3.3
Programming Language :: Python :: 3.4
Programming Language :: Python :: 3.5
Programming Language :: Python :: 3.6
Programming Language :: Python :: 3.7
Topic :: Communications
Topic :: System :: Monitoring
Topic :: System :: Networking :: Monitoring
"""


def howto_install_setuptools():
    print("""
   Error: You need setuptools Python package!

   It's very easy to install it, just type:

   wget https://bootstrap.pypa.io/ez_setup.py
   python ez_setup.py

   Then you could make eggs from this package.
""")


if sys.version_info[:2] < (2, 7):
    print("ERROR: this package requires Python 2.7 or later!")
    sys.exit(1)

params = {
    'install_requires': ['pysnmp>=4.4.3,<5.0.0'],
    'zip_safe': False  # this is due to data and variation dirs
}

doclines = [x.strip() for x in (__doc__ or '').split('\n') if x]

params.update(
    {'name': 'thola-snmpsim',
     'version': open(os.path.join('snmpsim', '__init__.py')).read().split('\'')[1],
     'description': doclines[0],
     'long_description': ' '.join(doclines[1:]),
     'maintainer': 'Team Thola <snmplabs@thola.io>',
     'author': 'Ilya Etingof',
     'author_email': 'snmplabs@thola.io',
     'url': 'http://snmplabs.thola.io/snmpsim',
     'license': 'BSD',
     'platforms': ['any'],
     'classifiers': [x for x in classifiers.split('\n') if x],
     'packages': setuptools.find_packages(),
     'include_package_data': True,
     'entry_points': {
        'console_scripts': [
            'snmpsim-manage-records = snmpsim.commands.rec2rec:main',
            'snmpsim-record-mibs = snmpsim.commands.mib2rec:main',
            'snmpsim-record-traffic = snmpsim.commands.pcap2rec:main',
            'snmpsim-record-commands = snmpsim.commands.cmd2rec:main',
            'snmpsim-command-responder = snmpsim.commands.responder:main',
            'snmpsim-command-responder-lite = snmpsim.commands.responder_lite:main',
        ]
     }}
)

# install stock variation modules as data_files
params['data_files'] = [
    (os.path.join('share', 'snmpsim', 'variation'),
     glob.glob(os.path.join('variation', '*.py')))
]

# install sample .snmprec files as data_files
for x in os.walk('data'):
    files = []
    files.extend(glob.glob(os.path.join(x[0], '*.snmprec')))
    files.extend(glob.glob(os.path.join(x[0], '*.snmpwalk')))
    files.extend(glob.glob(os.path.join(x[0], '*.sapwalk')))

    params['data_files'].append(
        (os.path.join('share', 'snmpsim', *os.path.split(x[0])), files))

if 'py2exe' in sys.argv:

    # fix executables
    params['console'] = params['scripts']

    del params['scripts']
    # pysnmp used by snmpsim dynamically loads some of its *.py files
    params['options'] = {
        'py2exe': {
            'includes': [
                'pysnmp.smi.mibs.*',
                'pysnmp.smi.mibs.instances.*',
                'pysnmp.entity.rfc3413.oneliner.*'
            ],
            'bundle_files': 1,
            'compressed': True
        }
    }

    params['zipfile'] = None

    del params['data_files']  # no need to store these in .exe

    # additional modules used by snmpsimd but not seen by py2exe
    for m in ('dbm', 'gdbm', 'dbhash', 'dumbdb',
              'shelve', 'random', 'math', 'bisect',
              'sqlite3', 'subprocess', 'redis'):
        try:
            __import__(m)

        except ImportError:
            continue

        else:
            params['options']['py2exe']['includes'].append(m)

    print("!!! Make sure your pysnmp/pyasn1 packages are NOT .egg'ed!!!")

setuptools.setup(**params)