Index: recipe_engine/third_party/setuptools/extension.py |
diff --git a/recipe_engine/third_party/setuptools/extension.py b/recipe_engine/third_party/setuptools/extension.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8178ed33d7f095f59c624e728fd850f938bd37e8 |
--- /dev/null |
+++ b/recipe_engine/third_party/setuptools/extension.py |
@@ -0,0 +1,57 @@ |
+import sys |
+import re |
+import functools |
+import distutils.core |
+import distutils.errors |
+import distutils.extension |
+ |
+from .dist import _get_unpatched |
+from . import msvc9_support |
+ |
+_Extension = _get_unpatched(distutils.core.Extension) |
+ |
+msvc9_support.patch_for_specialized_compiler() |
+ |
+def have_pyrex(): |
+ """ |
+ Return True if Cython or Pyrex can be imported. |
+ """ |
+ pyrex_impls = 'Cython.Distutils.build_ext', 'Pyrex.Distutils.build_ext' |
+ for pyrex_impl in pyrex_impls: |
+ try: |
+ # from (pyrex_impl) import build_ext |
+ __import__(pyrex_impl, fromlist=['build_ext']).build_ext |
+ return True |
+ except Exception: |
+ pass |
+ return False |
+ |
+ |
+class Extension(_Extension): |
+ """Extension that uses '.c' files in place of '.pyx' files""" |
+ |
+ def __init__(self, *args, **kw): |
+ _Extension.__init__(self, *args, **kw) |
+ self._convert_pyx_sources_to_lang() |
+ |
+ 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_pyrex(): |
+ # 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""" |
+ |
+distutils.core.Extension = Extension |
+distutils.extension.Extension = Extension |
+if 'distutils.command.build_ext' in sys.modules: |
+ sys.modules['distutils.command.build_ext'].Extension = Extension |