setup.py

import pathlib

import setuptools
from pkg_resources import parse_requirements

import shutil

MODULE_NAME = "nanovare_yolov5"
VERSION = "0.3.0"


def copy_to_module():
    current_path = pathlib.Path(".").resolve()
    module_root_path = current_path / MODULE_NAME
    module_root_path = module_root_path.relative_to(current_path)

    # Delete folder where we we will create build
    if module_root_path.is_dir():
        shutil.rmtree(module_root_path)

    modules_moved = []
    python_files = current_path.rglob("*.py")
    python_files = list(filter(lambda x: (".venv" not in x.parts) and x.stem != "__init__", python_files))

    for python_file_path in python_files:
        python_file_path = python_file_path.relative_to(current_path)
        if str(python_file_path) == "setup.py":
            continue
        new_path = module_root_path / python_file_path
        modules_moved.append(".".join((list(map(str, python_file_path.parents))[::-1][1:] + [python_file_path.stem])))
        if not new_path.parent.is_dir():
            new_path.parent.mkdir(parents=True, exist_ok=True)

            init_file = new_path.parent / "__init__.py"
            # Make __init__ files and put the __version__ in the root __init__
            if init_file.parent.stem == MODULE_NAME:
                with init_file.open("w")as f:
                    f.write(f"__version__ = \"{VERSION}\"")
            else:
                init_file.touch()

        shutil.copyfile(python_file_path, new_path)

    # Rename modules
    strings_to_replace = {}
    for module_name in modules_moved:
        strings_to_replace[f"from {module_name} "] = f"from {MODULE_NAME}.{module_name} "
        # TODO handle cases for "import module_name as"
        strings_to_replace[f"import {module_name}"] = f"import {MODULE_NAME}.{module_name} as {module_name}"
    strings_to_search_for = tuple(strings_to_replace)
    new_python_files = list(module_root_path.glob("**/*.py"))
    lines_changed = []

    for python_file_path in new_python_files:
        data = []
        with python_file_path.open("r", encoding="utf8") as f:
            for line in f:
                if line.startswith(strings_to_search_for):
                    lines_changed.append(line)
                    for key in strings_to_search_for:
                        if line.startswith(key):
                            print(line.rstrip())
                            line = line.replace(key, strings_to_replace[key], 1)
                            print(line.rstrip())
                            print()
                data.append(line)
        with python_file_path.open("w", encoding="utf8") as f:
            for line in data:
                f.write(line)

    print(f"Changed {len(lines_changed)} lines to accommodate for imports.")


with open("README.md", "r", encoding="utf8") as fh:
    long_description = fh.read()

with pathlib.Path('requirements.txt').open(encoding="utf8") as requirements_txt:
    install_requires = [
        str(requirement)
        for requirement
        in parse_requirements(requirements_txt)
    ]

copy_to_module()
modules = setuptools.find_packages(include=[MODULE_NAME, f"{MODULE_NAME}.*"])

setuptools.setup(
    name=MODULE_NAME,
    version=VERSION,
    author="Nanovare SAS",
    author_email="vincent@mojofertility.co, robin@mojofertility.co",
    description="yolov5 modifications for Nanovare SAS",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/robin-maillot/yolov5",
    packages=modules,
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    python_requires='>=3.7',
    dependency_links=[
            "https://download.pytorch.org/whl/torch_stable.html",
        ],
    install_requires=install_requires
)