Index: third_party/google-endpoints/setuptools/monkey.py |
diff --git a/third_party/google-endpoints/setuptools/monkey.py b/third_party/google-endpoints/setuptools/monkey.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..5843c46bbc044bdfb17fc9153fa2bcf3fcaeefe8 |
--- /dev/null |
+++ b/third_party/google-endpoints/setuptools/monkey.py |
@@ -0,0 +1,191 @@ |
+""" |
+Monkey patching of distutils. |
+""" |
+ |
+import sys |
+import distutils.filelist |
+import platform |
+import types |
+import functools |
+import inspect |
+ |
+from .py26compat import import_module |
+import six |
+ |
+import setuptools |
+ |
+__all__ = [] |
+""" |
+Everything is private. Contact the project team |
+if you think you need this functionality. |
+""" |
+ |
+ |
+def get_unpatched(item): |
+ lookup = ( |
+ get_unpatched_class if isinstance(item, six.class_types) else |
+ get_unpatched_function if isinstance(item, types.FunctionType) else |
+ lambda item: None |
+ ) |
+ return lookup(item) |
+ |
+ |
+def get_unpatched_class(cls): |
+ """Protect against re-patching the distutils if reloaded |
+ |
+ Also ensures that no other distutils extension monkeypatched the distutils |
+ first. |
+ """ |
+ external_bases = ( |
+ cls |
+ for cls in inspect.getmro(cls) |
+ if not cls.__module__.startswith('setuptools') |
+ ) |
+ base = next(external_bases) |
+ if not base.__module__.startswith('distutils'): |
+ msg = "distutils has already been patched by %r" % cls |
+ raise AssertionError(msg) |
+ return base |
+ |
+ |
+def patch_all(): |
+ # we can't patch distutils.cmd, alas |
+ distutils.core.Command = setuptools.Command |
+ |
+ has_issue_12885 = ( |
+ sys.version_info < (3, 4, 6) |
+ or |
+ (3, 5) < sys.version_info <= (3, 5, 3) |
+ or |
+ (3, 6) < sys.version_info |
+ ) |
+ |
+ if has_issue_12885: |
+ # fix findall bug in distutils (http://bugs.python.org/issue12885) |
+ distutils.filelist.findall = setuptools.findall |
+ |
+ needs_warehouse = ( |
+ sys.version_info < (2, 7, 13) |
+ or |
+ (3, 0) < sys.version_info < (3, 3, 7) |
+ or |
+ (3, 4) < sys.version_info < (3, 4, 6) |
+ or |
+ (3, 5) < sys.version_info <= (3, 5, 3) |
+ or |
+ (3, 6) < sys.version_info |
+ ) |
+ |
+ if needs_warehouse: |
+ warehouse = 'https://upload.pypi.org/legacy/' |
+ distutils.config.PyPIRCCommand.DEFAULT_REPOSITORY = warehouse |
+ |
+ _patch_distribution_metadata_write_pkg_file() |
+ _patch_distribution_metadata_write_pkg_info() |
+ |
+ # Install Distribution throughout the distutils |
+ for module in distutils.dist, distutils.core, distutils.cmd: |
+ module.Distribution = setuptools.dist.Distribution |
+ |
+ # Install the patched Extension |
+ distutils.core.Extension = setuptools.extension.Extension |
+ distutils.extension.Extension = setuptools.extension.Extension |
+ if 'distutils.command.build_ext' in sys.modules: |
+ sys.modules['distutils.command.build_ext'].Extension = ( |
+ setuptools.extension.Extension |
+ ) |
+ |
+ patch_for_msvc_specialized_compiler() |
+ |
+ |
+def _patch_distribution_metadata_write_pkg_file(): |
+ """Patch write_pkg_file to also write Requires-Python/Requires-External""" |
+ distutils.dist.DistributionMetadata.write_pkg_file = ( |
+ setuptools.dist.write_pkg_file |
+ ) |
+ |
+ |
+def _patch_distribution_metadata_write_pkg_info(): |
+ """ |
+ Workaround issue #197 - Python 3 prior to 3.2.2 uses an environment-local |
+ encoding to save the pkg_info. Monkey-patch its write_pkg_info method to |
+ correct this undesirable behavior. |
+ """ |
+ environment_local = (3,) <= sys.version_info[:3] < (3, 2, 2) |
+ if not environment_local: |
+ return |
+ |
+ distutils.dist.DistributionMetadata.write_pkg_info = ( |
+ setuptools.dist.write_pkg_info |
+ ) |
+ |
+ |
+def patch_func(replacement, target_mod, func_name): |
+ """ |
+ Patch func_name in target_mod with replacement |
+ |
+ Important - original must be resolved by name to avoid |
+ patching an already patched function. |
+ """ |
+ original = getattr(target_mod, func_name) |
+ |
+ # set the 'unpatched' attribute on the replacement to |
+ # point to the original. |
+ vars(replacement).setdefault('unpatched', original) |
+ |
+ # replace the function in the original module |
+ setattr(target_mod, func_name, replacement) |
+ |
+ |
+def get_unpatched_function(candidate): |
+ return getattr(candidate, 'unpatched') |
+ |
+ |
+def patch_for_msvc_specialized_compiler(): |
+ """ |
+ Patch functions in distutils to use standalone Microsoft Visual C++ |
+ compilers. |
+ """ |
+ # import late to avoid circular imports on Python < 3.5 |
+ msvc = import_module('setuptools.msvc') |
+ |
+ if platform.system() != 'Windows': |
+ # Compilers only availables on Microsoft Windows |
+ return |
+ |
+ def patch_params(mod_name, func_name): |
+ """ |
+ Prepare the parameters for patch_func to patch indicated function. |
+ """ |
+ repl_prefix = 'msvc9_' if 'msvc9' in mod_name else 'msvc14_' |
+ repl_name = repl_prefix + func_name.lstrip('_') |
+ repl = getattr(msvc, repl_name) |
+ mod = import_module(mod_name) |
+ if not hasattr(mod, func_name): |
+ raise ImportError(func_name) |
+ return repl, mod, func_name |
+ |
+ # Python 2.7 to 3.4 |
+ msvc9 = functools.partial(patch_params, 'distutils.msvc9compiler') |
+ |
+ # Python 3.5+ |
+ msvc14 = functools.partial(patch_params, 'distutils._msvccompiler') |
+ |
+ try: |
+ # Patch distutils.msvc9compiler |
+ patch_func(*msvc9('find_vcvarsall')) |
+ patch_func(*msvc9('query_vcvarsall')) |
+ except ImportError: |
+ pass |
+ |
+ try: |
+ # Patch distutils._msvccompiler._get_vc_env |
+ patch_func(*msvc14('_get_vc_env')) |
+ except ImportError: |
+ pass |
+ |
+ try: |
+ # Patch distutils._msvccompiler.gen_lib_options for Numpy |
+ patch_func(*msvc14('gen_lib_options')) |
+ except ImportError: |
+ pass |