opm-common/python/setup.py

from setuptools import setup, find_packages

from setuptools import Extension
from setuptools.command.build_ext import build_ext
import sys
import setuptools

import glob
import os
import subprocess
import argparse


parser = argparse.ArgumentParser()
parser.add_argument("build")
parser.add_argument("build_ext")
parser.add_argument("--build-lib")
parser.add_argument("--prefix")
parser.add_argument("--library-dirs")
parser.add_argument("--include-dirs")
parser.add_argument("--dry-run", action='store_true')
args = parser.parse_args()

#Circumnventing setuptools' PYTHONPATH check.
#This is necessary during install
if args.prefix:
   python_version = 'python' + str(sys.version_info.major) + '.' + str(sys.version_info.minor)
   pkg_path_root = os.path.join(args.prefix, 'lib', python_version, 'site-packages')
   if 'PYTHONPATH' in os.environ:
       os.environ['PYTHONPATH'] += ':' + pkg_path_root
   else:
       os.environ['PYTHONPATH'] = pkg_path_root

setupdir = os.path.dirname(__file__)
if setupdir != '':
  os.chdir( setupdir )

try:
  subprocess.call(["c++", "--version"])
  subprocess.call(['ccache', '--version'])
  os.environ['CC'] = 'ccache c++'
except OSError as e:
  print('\nNOTE: please install ccache for faster compilation of python bindings.\n')

if 'build' in sys.argv:
  if not 'build_ext' in sys.argv:
    raise TypeError("Missing option 'build_ext'.")

ext_modules = [
    Extension(
        'libopmcommon_python',
        [
                'cxx/connection.cpp',
                'cxx/deck.cpp',
                'cxx/deck_keyword.cpp',
                'cxx/eclipse_3d_properties.cpp',
                'cxx/eclipse_config.cpp',
                'cxx/eclipse_grid.cpp',
                'cxx/eclipse_state.cpp',
                'cxx/group.cpp',
                'cxx/group_tree.cpp',
                'cxx/parsecontext.cpp',
                'cxx/parser.cpp',
                'cxx/schedule.cpp',
                'cxx/common_state.cpp',
                'cxx/table_manager.cpp',
                'cxx/well.cpp',
                'cxx/common.cpp'
        ],
        libraries=['opmcommon', 'boost_filesystem', 'boost_regex', 'ecl'],
        language='c++',
        undef_macros=["NDEBUG"],
        include_dirs=["pybind11/include"]
    ),
]

setup(
    name='Opm',
    package_dir = {'': 'python'},
    packages=[
                'opm',
                'opm.io',
                'opm.io.parser',
                'opm.io.schedule',
                'opm.io.config',
                'opm.tools'
            ],
    ext_modules=ext_modules,
    package_data={
        '': ['cxx/libopmcommon_python.so']
    },
    include_package_data=True,
    license='Open Source',
    zip_safe=False,
    tests_suite='tests',
    setup_requires=["pytest-runner", 'setuptools_scm'],
)