setup.py

from __future__ import with_statement

import os
import sys

try:
    import setuptools
    from setuptools import setup, Extension
except ImportError:
    from distutils.core import setup, Extension

__version__ = "0.7.878"

# Set up the long_description

loc = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(loc, 'py_src', 'README.rst'), 'r') as fd:
    long_description = fd.read()

# Determine whether to build C binaries
# The exception is made for bdist_wheel because it genuinely uses the --universal flag

__USE_C__ = '--universal' not in sys.argv and os.path.isfile(
    os.path.join(loc, 'cp_src', 'base.cpp'))
if '--universal' in sys.argv and 'bdist_wheel' not in sys.argv:
    sys.argv.remove('--universal')

__DEBUG__ = [("CP2P_DEBUG_FLAG", "a")] if ('--debug' in sys.argv and
                                           __USE_C__) else []

# This sets up the program's classifiers

classifiers = [
    'Development Status :: 4 - Beta',
    'Intended Audience :: Developers',
    'License :: OSI Approved :: GNU Lesser General Public License v3 (LGPLv3)',
    'Operating System :: OS Independent',
    'Topic :: Communications',
    'Topic :: Internet',
    'Programming Language :: C',
    'Programming Language :: C++',
    'Programming Language :: JavaScript',
    'Programming Language :: Other',
    'Programming Language :: Python'
]

classifiers.extend((('Programming Language :: Python :: %s' % x)
                    for x in '2 3 2.7 3.3 3.4 3.5 3.6'.split()))


def has_environment_marker_support():
    #type: () -> bool
    """
    Tests that setuptools has support for PEP-426 environment marker support.
    The first known release to support it is 0.7 (and the earliest on PyPI seems to be 0.7.2
    so we're using that), see: http://pythonhosted.org/setuptools/history.html#id142
    References:
    * https://wheel.readthedocs.io/en/latest/index.html#defining-conditional-dependencies
    * https://www.python.org/dev/peps/pep-0426/#environment-markers
    Method extended from pytest. Credit goes to developers there.
    """
    try:
        from pkg_resources import parse_version
        return parse_version(setuptools.__version__) >= parse_version('0.7.2')
    except Exception as exc:
        sys.stderr.write("Could not test setuptool's version: %s\n" % exc)
        return False


def main():
    #type: () -> None
    ext_modules = []
    install_requires = [
        'custom_inherit',
        'async_promises',
        'u-msgpack-python',
        'pyee',
        'typing',
        'base58'
    ]
    extras_require = {'SSL': ['cryptography'], 'snappy': ['python-snappy']}
    if has_environment_marker_support():
        pass
    else:
        pass

    if __USE_C__:
        ext_modules.append(
            Extension(
                'py2p.cbase',
                sources=[
                    os.path.join(loc, 'c_src', 'base_wrapper.c'), os.path.join(
                        loc, 'c_src', 'sha', 'sha2.c')
                ] + [
                    os.path.join(loc, 'c_src', 'msgpack-c', 'src', file_)
                    for file_ in os.listdir(
                        os.path.join(loc, 'c_src', 'msgpack-c', 'src'))
                    if file_.endswith('.c')
                ],
                include_dirs=[
                    os.path.join(loc, 'c_src', 'msgpack-c', 'include')
                ],
                define_macros=__DEBUG__))

    try:
        setup(
            name='py2p',
            description='A python library for peer-to-peer networking',
            long_description=long_description,
            version=__version__,
            author='Gabe Appleton',
            author_email='gappleto97+development@gmail.com',
            url='https://github.com/gappleto97/p2p-project',
            license='LGPLv3',
            packages=['py2p', 'py2p.test'],
            package_dir={'py2p': os.path.join(loc, 'py_src')},
            package_data={'py2p': [os.path.join(loc,
                                                'py_src',
                                                'seeders.msgpack')]},
            ext_modules=ext_modules,
            classifiers=classifiers,
            install_requires=install_requires,
            extras_require=extras_require,
            entry_points='''
            [console_scripts]
            py2p=py2p.__main__:cli
        ''')
    except:
        print("Not building C code due to errors")
        setup(
            name='py2p',
            description='A python library for peer-to-peer networking',
            long_description=long_description,
            version=__version__,
            author='Gabe Appleton',
            author_email='gappleto97+development@gmail.com',
            url='https://github.com/gappleto97/p2p-project',
            license='LGPLv3',
            packages=['py2p', 'py2p.test'],
            package_dir={'py2p': 'py_src'},
            package_data={'py2p': [os.path.join(loc,
                                                'py_src',
                                                'seeders.msgpack')]},
            classifiers=classifiers,
            install_requires=install_requires,
            extras_require=extras_require,
            entry_points='''
            [console_scripts]
            py2p=py2p.__main__:cli
        ''')


if __name__ == "__main__":
    main()