setup.py

from distutils.core import setup
import os
from typing import Any

from setuptools import Distribution, setup
from setuptools.command.develop import develop
from setuptools.command.install import install


class PreDevelopCommand(develop):
    """Pre-installation for development mode."""

    def __init__(self, dist: Distribution, **kw: Any):
        make_inits()
        super().__init__(dist, **kw)

    def run(self):
        make_inits()
        develop.run(self)
        make_inits()


class PreInstallCommand(install):
    """Pre-installation for installation mode."""

    def __init__(self, dist: Distribution) -> None:
        make_inits()
        super().__init__(dist)

    def run(self):
        make_inits()
        install.run(self)
        make_inits()


try:
    from wheel.bdist_wheel import bdist_wheel as _bdist_wheel

    class bdist_wheel(_bdist_wheel):
        def __init__(self, dist: Distribution) -> None:
            make_inits()
            super().__init__(dist)

        def finalize_options(self):
            _bdist_wheel.finalize_options(self)
            self.root_is_pure = False


except ImportError:
    bdist_wheel = None


def make_inits():
    currentDir = os.path.dirname(__file__)
    touch(os.path.join(currentDir, "./src/flowtron/__init__.py"))
    touch(os.path.join(currentDir, "./src/flowtron/tacotron2/__init__.py"))
    touch(os.path.join(currentDir, "./src/flowtron/tacotron2/waveglow/__init__.py"))


def touch(path: str):
    basedir = os.path.dirname(path)
    if not os.path.exists(basedir):
        os.makedirs(basedir)

    with open(path, "a"):
        os.utime(path, None)


setup(
    name="flowtron-module",
    version="0.1.5",
    description="Pipelining to allow using flowtron as a pip module",
    author="NVIDIA",
    author_email="rafaelvalle@nvidia.com",
    package_dir={
        "flowtron": "src/flowtron",
        "text": "src/flowtron/text",
        "tacotron2": "src/flowtron/tacotron2",
        "waveglow": "src/flowtron/tacotron2/waveglow",
        },
    packages=[
        "flowtron",
        "tacotron2",
        "text",
        "waveglow",
    ],
    cmdclass={
        "develop": PreDevelopCommand,
        "install": PreInstallCommand,
        "bdist_wheel": bdist_wheel,
    },
    setup_requires=["wheel"],
    install_requires=[
        "torch==1.7.1",
        "matplotlib==3.3.2",
        "numpy==1.19.2",
        "inflect==4.1.0",
        "librosa==0.6.3",
        "scipy==1.5.2",
        "Unidecode==1.0.22",
        "pillow",
        "tensorboardX",
    ],
    license="Apache-2.0 License",
)