| import re |
| import functools |
| import distutils.core |
| import distutils.errors |
| import distutils.extension |
| |
| from setuptools.extern.six.moves import map |
| |
| from .monkey import get_unpatched |
| |
| |
| def _have_cython(): |
| """ |
| Return True if Cython can be imported. |
| """ |
| cython_impl = 'Cython.Distutils.build_ext' |
| try: |
| # from (cython_impl) import build_ext |
| __import__(cython_impl, fromlist=['build_ext']).build_ext |
| return True |
| except Exception: |
| pass |
| return False |
| |
| |
| # for compatibility |
| have_pyrex = _have_cython |
| |
| _Extension = get_unpatched(distutils.core.Extension) |
| |
| |
| class Extension(_Extension): |
| """Extension that uses '.c' files in place of '.pyx' files""" |
| |
| def __init__(self, name, sources, *args, **kw): |
| # The *args is needed for compatibility as calls may use positional |
| # arguments. py_limited_api may be set only via keyword. |
| self.py_limited_api = kw.pop("py_limited_api", False) |
| _Extension.__init__(self, name, sources, *args, **kw) |
| |
| def _convert_pyx_sources_to_lang(self): |
| """ |
| Replace sources with .pyx extensions to sources with the target |
| language extension. This mechanism allows language authors to supply |
| pre-converted sources but to prefer the .pyx sources. |
| """ |
| if _have_cython(): |
| # the build has Cython, so allow it to compile the .pyx files |
| return |
| lang = self.language or '' |
| target_ext = '.cpp' if lang.lower() == 'c++' else '.c' |
| sub = functools.partial(re.sub, '.pyx$', target_ext) |
| self.sources = list(map(sub, self.sources)) |
| |
| |
| class Library(Extension): |
| """Just like a regular Extension, but built as a library instead""" |
