setup.py

import os

from setuptools import setup

from TestRunner import OperationRuns


def all_folders():
    folder_names = [
        "transformations/" + f
        for f in list(OperationRuns.get_all_folder_names())
    ]
    folder_names.extend(
        [
            "filters/" + f
            for f in list(OperationRuns.get_all_folder_names("filters"))
        ]
    )
    return folder_names


def read(fname):
    with open(os.path.join(os.path.dirname(__file__), fname)) as f:
        data = f.read()
    return data


def recursive_requirements():
    # (1) read main requirements.txt
    requirements = read("requirements.txt")
    # (1) read all requirements.txt in the folder.
    for folder in all_folders():
        r_file = os.path.join(
            os.path.dirname(__file__), folder + "/requirements.txt"
        )
        if os.path.isfile(r_file):
            with open(r_file) as f:
                requirements += f.read() + "\n"
    return requirements


setup(
    name="NL-Augmenter",
    version="0.0.1",
    description="The official repository of transformations.",
    long_description=read("README.md"),
    install_requires=recursive_requirements(),  # read("requirements.txt"),
    package_data={
        "": ["*.json", "*.txt", "*.tsv", "*.csv", "*.npz", "*.ckpt"]
    },
    include_package_data=True,
)