setup.py

#!/usr/bin/env python

import glob
import sys
import numpy as np

from setuptools import setup, find_packages
from Cython.Build import cythonize


# Determine shared library suffix
if sys.platform == "darwin":
    suffix = "dylib"
else:
    suffix = "so"

# Get version information from __init__.py. This is ugly, but more reliable than
# using an import.
with open("openmc/__init__.py", "r") as f:
    version = f.readlines()[-1].split()[-1].strip("'")

kwargs = {
    "name": "openmc",
    "version": version,
    "packages": find_packages(exclude=["tests*"]),
    "scripts": glob.glob("scripts/openmc*"),
    # Data files and libraries
    "package_data": {
        "openmc.lib": ["libopenmc.{}".format(suffix), "openmc"],
        "openmc.data": ["mass_1.mas20.txt", "BREMX.DAT", "half_life.json", "*.h5"],
        "openmc.data.effective_dose": ["*.txt"],
    },
    # Metadata
    "author": "The OpenMC Development Team",
    "author_email": "openmc@anl.gov",
    "description": "OpenMC",
    "url": "https://openmc.org",
    "download_url": "https://github.com/openmc-dev/openmc/releases",
    "project_urls": {
        "Issue Tracker": "https://github.com/openmc-dev/openmc/issues",
        "Documentation": "https://docs.openmc.org",
        "Source Code": "https://github.com/openmc-dev/openmc",
    },
    "classifiers": [
        "Development Status :: 4 - Beta",
        "Intended Audience :: Developers",
        "Intended Audience :: End Users/Desktop",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT License",
        "Natural Language :: English",
        "Topic :: Scientific/Engineering" "Programming Language :: C++",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
    ],
    # Dependencies
    "python_requires": ">=3.7",
    "install_requires": [
        "numpy>=1.9",
        "h5py",
        "scipy",
        "ipython",
        "matplotlib",
        "pandas",
        "lxml",
        "uncertainties",
    ],
    "extras_require": {
        "depletion-mpi": ["mpi4py"],
        "docs": [
            "sphinx",
            "sphinxcontrib-katex",
            "sphinx-numfig",
            "jupyter",
            "sphinxcontrib-svg2pdfconverter",
            "sphinx-rtd-theme",
        ],
        "test": ["pytest", "pytest-cov", "colorama"],
        "vtk": ["vtk"],
    },
    # Cython is used to add resonance reconstruction and fast float_endf
    "ext_modules": cythonize("openmc/data/*.pyx"),
    "include_dirs": [np.get_include()],
}

setup(**kwargs)