3
Xoh @ s d Z ddlZddlZddlZddlZddlZddlmZ ddl Z ddl
m Z ddl Z g Z
dd Zdd Zd d
Zd d Zd
d Zdd Zdd Zdd ZdS )z
Monkey patching of distutils.
N)
import_module)sixc C s" t j dkr| f| j S tj| S )am
Returns the bases classes for cls sorted by the MRO.
Works around an issue on Jython where inspect.getmro will not return all
base classes if multiple classes share the same name. Instead, this
function will return a tuple containing the class itself, and the contents
of cls.__bases__. See https://github.com/pypa/setuptools/issues/1024.
ZJython)platformZpython_implementation __bases__inspectZgetmro)cls r /usr/lib/python3.6/monkey.py_get_mro s r
c C s0 t | tjrtnt | tjr tndd }|| S )Nc S s d S )Nr )itemr r r <lambda>* s z get_unpatched.<locals>.<lambda>)
isinstancer Z class_typesget_unpatched_classtypes FunctionTypeget_unpatched_function)r lookupr r r
get_unpatched& s r c C s: dd t | D }t|}|jjds6d| }t||S )zProtect against re-patching the distutils if reloaded
Also ensures that no other distutils extension monkeypatched the distutils
first.
c s s | ]}|j jd s|V qdS )
setuptoolsN)
__module__
startswith).0r r r r <genexpr>6 s z&get_unpatched_class.<locals>.<genexpr> distutilsz(distutils has already been patched by %r)r
nextr r AssertionError)r Zexternal_basesbasemsgr r r r / s r c C s t jtj_tjd k} | r"t jtj_tjd k pxd
tj k o@dk n pxdtj k oZdk n pxdtj k otdkn }|rd }|tjj _
t x"tj tjtj
fD ]}t j j|_qW t jjtj_t jjtj_d
tjk rt jjtjd
_t d S )N
r z https://upload.pypi.org/legacy/zdistutils.command.build_ext)r r r )r r! r" )r r )r r r! )r r# )r r# r$ )r r )r r r )r ZCommandr Zcoresys version_infofindallZfilelistconfigZ
PyPIRCCommandZDEFAULT_REPOSITORY+_patch_distribution_metadata_write_pkg_filedistcmdZ Distribution extensionZ Extensionmodules#patch_for_msvc_specialized_compiler)Zhas_issue_12885Zneeds_warehouseZ warehousemoduler r r patch_allA s&
r0 c C s t jjtjj_dS )zDPatch write_pkg_file to also write Requires-Python/Requires-ExternalN)r r* Zwrite_pkg_filer ZDistributionMetadatar r r r r) j s r) c C s* t ||}t| jd| t||| dS )z
Patch func_name in target_mod with replacement
Important - original must be resolved by name to avoid
patching an already patched function.
unpatchedN)getattrvars
setdefaultsetattr)Z replacementZ
target_mod func_nameZoriginalr r r
patch_funcq s
r7 c C s
t | dS )Nr1 )r2 ) candidater r r r s r c s t d tj dkrdS fdd} tj| d}tj| d}y t|d t|d W n tk
rl Y nX yt|d
W n tk
r Y nX yt|d W n tk
r Y nX dS ) z\
Patch functions in distutils to use standalone Microsoft Visual C++
compilers.
zsetuptools.msvcZWindowsNc sL d| kr dnd}||j d }t |}t| }t||sBt||||fS )zT
Prepare the parameters for patch_func to patch indicated function.
msvc9Zmsvc9_Zmsvc14__)lstripr2 r hasattr ImportError)Zmod_namer6 Z repl_prefixZ repl_namereplmod)msvcr r patch_params s
z9patch_for_msvc_specialized_compiler.<locals>.patch_paramszdistutils.msvc9compilerzdistutils._msvccompilerZfind_vcvarsallZquery_vcvarsallZ _get_vc_envZgen_lib_options)r r system functoolspartialr7 r= )rA r9 Zmsvc14r )r@ r r. s&
r. )__doc__r% Zdistutils.filelistr r r rC importlibr r Zsetuptools.externr r __all__r
r r r0 r) r7 r r. r r r r <module> s$ )