3
9��f� � @ 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 | tj�rtnt | tj�r tndd� }|| �S )Nc S s d S )Nr )�itemr r r �<lambda>* s z get_unpatched.<locals>.<lambda>)�
isinstancer Z class_types�get_unpatched_class�types� FunctionType�get_unpatched_function)r �lookupr r r �
get_unpatched& s r c C s: dd� t | �D �}t|�}|jjd�s6d| }t|��|S )z�Protect 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_bases�base�msgr 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 }|r�d }|tjj _
t � x"tj tjtj
fD ]}t j j|_q�W t jjtj_t jjtj_d
tjk� r�t 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 Zcore�sys� version_info�findallZfilelist�configZ
PyPIRCCommandZDEFAULT_REPOSITORY�+_patch_distribution_metadata_write_pkg_file�dist�cmdZ Distribution� extensionZ Extension�modules�#patch_for_msvc_specialized_compiler)Zhas_issue_12885Zneeds_warehouseZ warehouse�moduler 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)�getattr�vars�
setdefault�setattr)Z replacementZ
target_mod� func_name�originalr r r �
patch_funcq s
r8 c C s
t | d�S )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_name�repl�mod)�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� functools�partialr8 r> )rB r: Zmsvc14r )rA r r. � s&
r. )�__doc__r% Zdistutils.filelistr r r rD � importlibr r Zsetuptools.externr r �__all__r
r r r0 r) r8 r r. r r r r �<module> s$ )