setup.py

# (c) 2017 Marcel Bargull
# (c) 2016-2017 Continuum Analytics, Inc. / http://continuum.io
# All Rights Reserved
#
# constructor-mb is distributed under the terms of the BSD 3-clause license.
# Consult LICENSE.txt or http://opensource.org/licenses/BSD-3-Clause.

import re
import sys
from os.path import abspath, dirname, join
from contextlib import contextmanager

from setuptools import setup, Distribution
from distutils.util import get_platform
try:
    from wheel.bdist_wheel import bdist_wheel
except ImportError:
    bdist_wheel = None


NAME = "constructor-mb"
PACKAGE_NAME = "constructor_mb"
SETUP_PY_DIR = dirname(abspath(__file__))


def get_package_data(platform):
    platform_package_data = {
        "win32": ["nsis/*", "ttf/*"],
        "unix": ["header.sh"],
    }
    if platform is None:
        package_data = sum(platform_package_data.values(), [])
    else:
        platform = platform if platform in platform_package_data else "unix"
        package_data = platform_package_data[platform]
    return {PACKAGE_NAME: package_data}


class PlatformSpecificDistribution(Distribution):
    def run_commands(self):
        for cmd in self.commands:
            with self.patch_dist_data(cmd):
                self.run_command(cmd)

    @contextmanager
    def patch_dist_data(dist, cmd):
        old_have_run = dist.have_run
        old_command_obj = dist.command_obj
        old_package_data = dist.package_data
        old_has_ext_modules = dist.has_ext_modules
        try:
            # rerun everthing for top cmds since patched dist data may differ
            dist.have_run = {}
            dist.command_obj = {}
            if cmd in {"build", "install", "bdist", "bdist_egg", "bdist_wheel"}:
                # use platform dependent package data
                dist.package_data = get_package_data(sys.platform)
            if cmd in {"bdist_egg"}:
                # force platform specific build for bdist_egg
                dist.has_ext_modules = lambda: True
            yield
        finally:
            dist.have_run = old_have_run
            dist.command_obj = old_command_obj
            dist.package_data = old_package_data
            dist.has_ext_modules = old_has_ext_modules


cmdclass = {}

if bdist_wheel:
    class BDistWheel(bdist_wheel):
        def finalize_options(self):
            # make wheels platform but not Python specific
            self.plat_name = self.plat_name or get_platform()
            self.universal = True
            super(BDistWheel, self).finalize_options()


    cmdclass["bdist_wheel"] = BDistWheel


# read version from PACKAGE_NAME/__init__.py
data = open(join(SETUP_PY_DIR, PACKAGE_NAME, "__init__.py")).read()
version = re.search(r"^__version__\s*=\s*(['\"])(\S*)\1", data, re.M).group(2)

setup(
    name=NAME,
    # name="constructor",
    version=version,
    author="Marcel Bargull",
    author_email="marcel.bargull@udo.edu",
    url="https://github.com/mbargull/constructor",
    # author="Ilan Schnell",
    # author_email="ilan@continuum.io",
    # url="https://github.com/conda/constructor",
    license="BSD",
    description="create installer from conda packages (fork of constructor)",
    # description="create installer from conda packages",
    long_description=open(join(SETUP_PY_DIR, "README.md")).read(),
    packages=[PACKAGE_NAME, "%s.tests" % PACKAGE_NAME],
    entry_points={
        "console_scripts": ["%s = %s.main:main" % (NAME, PACKAGE_NAME)],
    },
    install_requires=[
        "libconda",
        "pyyaml",
        "pillow >=3.1 ; platform_system=='Windows'",
        # non-python dependency: "nsis >=3.01 ; platform_system=='Windows'",
    ],
    package_data=get_package_data(None),
    cmdclass=cmdclass,
    distclass=PlatformSpecificDistribution,
)