from detection_metrics import __version__
import subprocess

from setuptools import setup
from setuptools.command.develop import develop
from setuptools.command.egg_info import egg_info
from setuptools.command.install import install


def custom_command():
    subprocess.call(["pip", "install", "numpy", "cython"])
    subprocess.call(["pip", "install", "-r", "requirements.txt", "--user"])

class CustomInstallCommand(install):
    def run(self):
        install.run(self)
        custom_command()

class CustomDevelopCommand(develop):
    def run(self):
        develop.run(self)
        custom_command()


class CustomEggInfoCommand(egg_info):
    def run(self):
        egg_info.run(self)
        custom_command()

setup(
    name="detection_metrics",
    description="COCO Metrics for Object Detection and Instance Segmentation",
    version=__version__,
    zip_safe=True,
    packages=["detection_metrics", "detection_metrics.pycocotools"],
    include_package_data=True,
    cmdclass={
        "install": CustomInstallCommand,
        "develop": CustomDevelopCommand,
        "egg_info": CustomEggInfoCommand,
    },
)