setup.py

import os
import shutil

from skbuild import setup


def extra_cmake_args():
    # Warning: if you use paths on Windows, you should use "\\"
    # for the path delimiter to work on CI.
    env = os.getenv('EXTRA_CMAKE_ARGS')
    return env.split(';') if env else []


cmake_args = [] + extra_cmake_args()

if os.name == 'nt':
    # Need to export all headers on windows...
    cmake_args.append('-DCMAKE_WINDOWS_EXPORT_ALL_SYMBOLS=TRUE')


if os.getenv('USE_PYTHON_IN_PATH'):
    python_exe = shutil.which('python')
    if python_exe:
        # For this program, we use find_package(Python3 ...)
        cmake_args.append(f'-DPython3_EXECUTABLE={python_exe}')


with open('requirements.txt') as f:
    install_requires = f.read()

setup(
    name='stempy',
    use_scm_version=True,
    description='A package for the ingestion of 4D STEM data.',
    long_description='A package for the ingestion of 4D STEM data.',
    url='https://github.com/OpenChemistry/stempy',
    author='Kitware Inc',
    license='BSD 3-Clause',
    classifiers=[
        'Development Status :: 3 - Alpha',
        'License :: OSI Approved :: BSD License',
        'Programming Language :: Python :: 3',
    ],
    keywords='',
    packages=['stempy'],
    install_requires=install_requires,
    cmake_args=cmake_args,
)