setup.py

#!/usr/bin/env python2.7

import os
import platform
import shutil
import sys
from importlib import import_module

from setuptools import Extension, setup
from setuptools.command.install import install

from src.version import softwareVersion

EXTRAS_REQUIRE = {
    'docs': ['sphinx'],
    'gir': ['pygobject'],
    'json': ['jsonrpclib'],
    'notify2': ['notify2'],
    'opencl': ['pyopencl', 'numpy'],
    'prctl': ['python_prctl'],  # Named threads
    'qrcode': ['qrcode'],
    'sound;platform_system=="Windows"': ['winsound'],
    'tor': ['stem'],
    'xdg': ['pyxdg'],
    'xml': ['defusedxml']
}


class InstallCmd(install):
    """Custom setuptools install command preparing icons"""

    def run(self):
        # prepare icons directories
        try:
            os.makedirs('desktop/icons/scalable')
        except os.error:
            pass
        shutil.copyfile(
            'desktop/can-icon.svg', 'desktop/icons/scalable/pybitmessage.svg')
        try:
            os.makedirs('desktop/icons/24x24')
        except os.error:
            pass
        shutil.copyfile(
            'desktop/icon24.png', 'desktop/icons/24x24/pybitmessage.png')

        return install.run(self)


if __name__ == "__main__":
    here = os.path.abspath(os.path.dirname(__file__))
    with open(os.path.join(here, 'README.md')) as f:
        README = f.read()

    with open(os.path.join(here, 'requirements.txt'), 'r') as f:
        requirements = list(f.readlines())

    bitmsghash = Extension(
        'pybitmessage.bitmsghash.bitmsghash',
        sources=['src/bitmsghash/bitmsghash.cpp'],
        libraries=['pthread', 'crypto'],
    )

    installRequires = ['six']
    packages = [
        'pybitmessage',
        'pybitmessage.bitmessageqt',
        'pybitmessage.bitmessagecurses',
        'pybitmessage.fallback',
        'pybitmessage.messagetypes',
        'pybitmessage.network',
        'pybitmessage.plugins',
        'pybitmessage.pyelliptic',
        'pybitmessage.storage'
    ]
    package_data = {'': [
        'bitmessageqt/*.ui', 'bitmsghash/*.cl', 'sslkeys/*.pem',
        'translations/*.ts', 'translations/*.qm', 'default.ini',
        'sql/*.sql', 'images/*.png', 'images/*.ico', 'images/*.icns',
        'bitmessagekivy/main.kv', 'bitmessagekivy/screens_data.json',
        'bitmessagekivy/kv/*.kv', 'images/kivy/payment/*.png',
        'images/kivy/*.gif', 'images/kivy/text_images*.png'
    ]}

    if sys.version_info[0] == 3:
        packages.extend([
            'pybitmessage.bitmessagekivy',
            'pybitmessage.bitmessagekivy.baseclass'
        ])

    if os.environ.get('INSTALL_TESTS', False):
        packages.extend([
            'pybitmessage.mockbm', 'pybitmessage.backend',
            'pybitmessage.bitmessagekivy.tests'
        ])
        package_data[''].extend(['bitmessagekivy/tests/sampleData/*.dat'])

    # this will silently accept alternative providers of msgpack
    # if they are already installed

    try:
        import msgpack
        installRequires.append(
            "msgpack-python" if msgpack.version[:2] < (0, 6) else "msgpack")
    except ImportError:
        try:
            import_module('umsgpack')
            installRequires.append("umsgpack")
        except ImportError:
            packages += ['pybitmessage.fallback.umsgpack']

    data_files = [
        ('share/applications/',
            ['desktop/pybitmessage.desktop']),
        ('share/icons/hicolor/scalable/apps/',
            ['desktop/icons/scalable/pybitmessage.svg']),
        ('share/icons/hicolor/24x24/apps/',
            ['desktop/icons/24x24/pybitmessage.png'])
    ]

    try:
        if platform.dist()[0] in ('Debian', 'Ubuntu'):
            data_files += [
                ("etc/apparmor.d/",
                    ['packages/apparmor/pybitmessage'])
            ]
    except AttributeError:
        pass  # FIXME: use distro for more recent python

    dist = setup(
        name='pybitmessage',
        version=softwareVersion,
        description="Reference client for Bitmessage: "
        "a P2P communications protocol",
        long_description=README,
        license='MIT',
        # TODO: add author info
        url='https://bitmessage.org',
        # TODO: add keywords
        install_requires=installRequires,
        tests_require=requirements,
        test_suite='tests.unittest_discover',
        extras_require=EXTRAS_REQUIRE,
        classifiers=[
            "License :: OSI Approved :: MIT License"
            "Operating System :: OS Independent",
            "Programming Language :: Python :: 2.7 :: Only",
            "Topic :: Internet",
            "Topic :: Security :: Cryptography",
            "Topic :: Software Development :: Libraries :: Python Modules",
        ],
        package_dir={'pybitmessage': 'src'},
        packages=packages,
        package_data=package_data,
        data_files=data_files,
        ext_modules=[bitmsghash],
        zip_safe=False,
        entry_points={
            'bitmessage.gui.menu': [
                'address.qrcode = pybitmessage.plugins.menu_qrcode [qrcode]'
            ],
            'bitmessage.notification.message': [
                'notify2 = pybitmessage.plugins.notification_notify2'
                '[gir, notify2]'
            ],
            'bitmessage.notification.sound': [
                'theme.canberra = pybitmessage.plugins.sound_canberra',
                'file.gstreamer = pybitmessage.plugins.sound_gstreamer'
                '[gir]',
                'file.fallback = pybitmessage.plugins.sound_playfile'
                '[sound]'
            ],
            'bitmessage.indicator': [
                'libmessaging ='
                'pybitmessage.plugins.indicator_libmessaging [gir]'
            ],
            'bitmessage.desktop': [
                'freedesktop = pybitmessage.plugins.desktop_xdg [xdg]'
            ],
            'bitmessage.proxyconfig': [
                'stem = pybitmessage.plugins.proxyconfig_stem [tor]'
            ],
            'console_scripts': [
                'pybitmessage = pybitmessage.bitmessagemain:main'
            ] if sys.platform[:3] == 'win' else []
        },
        scripts=['src/pybitmessage'],
        cmdclass={'install': InstallCmd},
        command_options={
            'build_sphinx': {
                'source_dir': ('setup.py', 'docs')}
        }
    )