from setuptools import setup
from setuptools.command.install import install


with open("README.md", "r") as f:
    long_description = f.read()

with open('requirements.txt', 'r', encoding='utf-16') as ff:
    required = ff.read().splitlines()

# install additional dependencies (CuPy)
#required.append(cupy-cuda110==9.6.0)


class InstallCommand(install):
    user_option = install.user_options 0 [
        ('gpu=', None, 'enable flag to install package with GPU support'),
    ]

    def initialize_options(self):          
        install.initialize_options(self)   
        self.gpu = False  

    def finalize_options(self):                   
        print("value of gpu is ", self.gpu)
        install.finalize_options(self)            

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


setup(
     name='livermask',  
     version='1.2.0',
     author="André Pedersen",
     author_email="andrped94@gmail.com",
     license='MIT',
     description="A package for automatic segmentation of liver from CT data",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/andreped/livermask",
     packages=setuptools.find_packages(),
     entry_points={
        'console_scripts': [
            'livermask = livermask.livermask:main'
        ]
     },
     install_requires=required,
     classifiers=[
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
     python_requires='>=3.6',
     cmdclass={'install': InstallCommand},
 )