setup_eb.py

import os
import re
import toml.decoder

from setuptools import setup, find_packages


ROOT_DIR = os.path.abspath(os.path.dirname(__file__))

PYPROJECT_TOML = toml.decoder.load(os.path.join(ROOT_DIR, 'pyproject.toml'))
POETRY_DATA = PYPROJECT_TOML['tool']['poetry']

_CARET_MATCH = re.compile(r"[\^]([0-9]+)([.].*)?$")
_TILDE_MATCH = re.compile(r"[~]([0-9]+[.])([0-9]+)([.].*)?$")


def fix_requirement(requirement):
    if isinstance(requirement, dict) and "version" in requirement:
        requirement = requirement["version"]
    m = _CARET_MATCH.match(requirement)
    if m:
        return ">=%s%s,<%s" % (m.group(1), m.group(2), int(m.group(1)) + 1)
    m = _TILDE_MATCH.match(requirement)
    if m:
        return ">=%s%s%s,<%s%s" % (m.group(1), m.group(2), m.group(3), m.group(1), int(m.group(2)) + 1)
    if requirement[0].isdigit():
        return "==" + requirement
    else:
        return requirement


_EMAIL_MATCH = re.compile(r"^([^<]*)[<]([^>]*)[>]$")


def author_and_email(authorship_spec):
    m = _EMAIL_MATCH.match(authorship_spec)
    if m:
        return m.group(1), m.group(2)
    else:
        raise ValueError("Expect authorship in format 'human_name <email_name@email_host>': %s" % authorship_spec)


def get_requirements(kind='dependencies'):
    return [
        pkg + fix_requirement(requirement)
        for pkg, requirement in POETRY_DATA[kind].items()
        if pkg != "python"
    ]


def flatten_config_data(key, dictionary):
    return "%s\n%s\n\n" % (key, "\n".join([
        key + " = " + val
        for key, val in dictionary.items()
    ]))


def entry_points():
    result = flatten_config_data("[console_scripts]", POETRY_DATA['scripts'])
    paste_dict = PYPROJECT_TOML['paste']
    for subkey in paste_dict:
        result += flatten_config_data('[paste.%s]' % subkey, paste_dict[subkey])
    return result


ENTRY_POINTS = entry_points()

PACKAGE_NAME = POETRY_DATA['name'].replace('dcic','')
SUBDIR = POETRY_DATA['packages'][0]['from']
README = open(os.path.join(ROOT_DIR, 'README.rst')).read()
DESCRIPTION = POETRY_DATA['description']
LONG_DESCRIPTION = README
AUTHOR, AUTHOR_EMAIL = author_and_email(POETRY_DATA['authors'][0])
URL = 'http://data.4dnucleome.org'
LICENSE = 'MIT'
INSTALL_REQUIRES = get_requirements()
TESTS_REQUIRE = get_requirements('dev-dependencies')
VERSION = POETRY_DATA['version']

if __name__ == '__main__':

    setup(
        name=PACKAGE_NAME,
        version=VERSION,
        description=DESCRIPTION,
        long_description=LONG_DESCRIPTION,
        packages=find_packages(SUBDIR),
        package_dir={'': SUBDIR},
        include_package_data=True,
        zip_safe=False,
        author=AUTHOR,
        author_email=AUTHOR_EMAIL,
        url=URL,
        license=LICENSE,
        install_requires=INSTALL_REQUIRES,
        tests_require=TESTS_REQUIRE,
        extras_require={'test': TESTS_REQUIRE},
        entry_points=ENTRY_POINTS,
    )