setup.py

import typing as tp
from setuptools import setup, Extension
from setuptools.command.build_ext import build_ext
import site, os
from pathlib import Path

AK_VERSION = Path('VERSION').read_text(encoding='utf-8').strip()


def get_ext_dir(*components: tp.Iterable[str]) -> tp.Sequence[str]:
 dirs = []
 # Check user site-packages
 user_site = site.getusersitepackages()
 if user_site:
 fp = os.path.join(user_site, *components)
 if os.path.exists(fp):
 dirs.append(fp)
 # Check system site-packages
 for sp in site.getsitepackages():
 fp = os.path.join(sp, *components)
 if os.path.exists(fp):
 dirs.append(fp)
 return dirs


ext_modules = [
 Extension(
 name='arraykit._arraykit',
 sources=[
 'src/_arraykit.c',
 'src/array_go.c',
 'src/array_to_tuple.c',
 'src/block_index.c',
 'src/delimited_to_arrays.c',
 'src/methods.c',
 'src/tri_map.c',
 'src/auto_map.c',
 ],
 include_dirs=get_ext_dir('numpy', '_core', 'include') + ['src'],
 library_dirs=get_ext_dir('numpy', '_core', 'lib'),
 define_macros=[('AK_VERSION', AK_VERSION)],
 libraries=['npymath'],
 )
]

setup(ext_modules=ext_modules)