from distutils.core import Extension, setup
import distutils.core
from distutils.command.build import build
import os

class my_build(build):
    def run(self):
        os.system ('mv -f ${CMAKE_CURRENT_SOURCE_DIR}/cigi.py ${CMAKE_CURRENT_SOURCE_DIR}/__init__.py')
        build.run(self)

module = Extension('cigi._cigi',
                   sources=['${CMAKE_CURRENT_SOURCE_DIR}/swig.i'],
                   libraries = ['cigicl'],
                   library_dirs = ['${CMAKE_CURRENT_BINARY_DIR}/../lib'],
                   include_dirs=['${CMAKE_CURRENT_SOURCE_DIR}/../include'],
                   swig_opts=['-c++', '-py3', '-I${CMAKE_CURRENT_SOURCE_DIR}/../include', '-builtin', '-DCIGI_SPEC'],
                   extra_compile_args=['-Wno-write-strings'])

setup(cmdclass=dict(build=my_build),
      name='cigi',
      version='${PACKAGE_VERSION}',
      ext_modules=[module],
      packages = ['cigi'],
      package_dir={ '': '${CMAKE_CURRENT_SOURCE_DIR}/python',
                    'cigi': '${CMAKE_CURRENT_SOURCE_DIR}' }
)