setup.py

#!/usr/bin/env python

import sys

from setuptools import setup
from setuptools_rust import Binding, RustExtension

from setup_util.setup_commons import (
    path, parse_requirements, get_version,
    get_long_description, find_required_packages, PyInstaller,
    move_wheel, print_errors, DatabaseMigration)

from golem.docker.manager import DockerManager
from golem.tools.ci import in_appveyor, in_travis

building_wheel = 'bdist_wheel' in sys.argv
building_binary = 'pyinstaller' in sys.argv
building_migration = 'migration' in sys.argv

directory = path.abspath(path.dirname(__file__))
requirements, dependencies = parse_requirements(directory)

setup(
    name='golem',
    version=get_version(),
    platforms=sys.platform,
    description='Global, open sourced, decentralized supercomputer',
    long_description=get_long_description(directory),
    url='https://golem.network',
    author='Golem Team',
    author_email='contact@golem.network',
    license="GPL-3.0",
    classifiers=[
        'Development Status :: 3 - Alpha',
        'Intended Audience :: Developers',
        'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',
        'Natural Language :: English',
        'Programming Language :: Python :: 3.6',
    ],
    zip_safe=False,
    keywords='golem',
    packages=find_required_packages(),
    install_requires=requirements,
    dependency_links=dependencies,
    include_package_data=True,
    cmdclass={
        'pyinstaller': PyInstaller,
        'migration': DatabaseMigration
    },
    entry_points={
        'gui_scripts': [
            'golemapp = golemapp:start',
        ],
    },
    rust_extensions=[
        RustExtension(
            'rust.golem',
            'rust/golem/Cargo.toml',
            binding=Binding.RustCPython,
            debug=False,
        ),
    ],
    data_files=[
        (path.normpath('../../'), [
            'golemapp.py', 'loggingconfig.py'
        ]),
        (path.normpath('../../golem/'), [
            path.normpath('golem/CONCENT_TERMS.html'),
        ]),
        (path.normpath('../../golem/'), [
            path.normpath('golem/TERMS.html'),
        ]),
        (path.normpath('../../golem/apps'), [
            path.normpath('apps/registered.ini'),
            path.normpath('apps/registered_test.ini'),
            path.normpath('apps/images.ini')
        ]),
        (path.normpath(
            '../../golem/golem/envs/docker/benchmark/cpu/minilight'), [
            path.normpath(
                'golem/envs/docker/benchmark/cpu/minilight/cornellbox.ml.txt'),
        ]),
        (path.normpath(
            '../../golem/apps/blender/resources/images/entrypoints/scripts/'
            'render_tools/templates'), [
                path.normpath(
                    'apps/blender/resources/images/entrypoints/'
                    'scripts/render_tools/templates/blendercrop.py.template')]
        ),
        (path.normpath('../../golem/apps/dummy/resources/code_dir'), [
            path.normpath('apps/dummy/resources/code_dir/computing.py')
        ]),
        (path.normpath('../../golem/apps/dummy/test_data'), [
            path.normpath('apps/dummy/test_data/in.data')
        ]),
        (path.normpath('../../golem/network/concent/resources/ssl/certs'), [
            path.normpath('golem/network/concent/resources/ssl/certs/test.crt'),
        ]),
    ]
)

if not (in_appveyor() or in_travis() or
        building_wheel or building_binary):

    docker_manager = DockerManager()

    try:
        DockerManager().pull_images()
    except Exception as exc:  # pylint: disable=broad-except
        print('Exception occurred:', exc)
        DockerManager().build_images()

if building_wheel:
    move_wheel()

if not building_migration:
    from golem.database.migration.create import latest_migration_exists

    if not latest_migration_exists():
        raise RuntimeError("Database schema error: latest migration script "
                           "does not exist")


# test a potential docker package conflict

def _docker_conflict(e: Exception):
    raise RuntimeError(
        "Suspected conflict in python `docker` library.\n"
        "Please run `pip uninstall -y docker docker-py` "
        "and re-install golem's requirements. "
    ) from e


try:
    from docker import DockerClient as Client
except ImportError as import_error:
    _docker_conflict(import_error)

try:
    Client().api
except TypeError as type_error:
    _docker_conflict(type_error)