o
    Ëý°ja  ã                   @   s’   d Z ddlZddlZddlZddlZddlZddlmZ ddl	Z	ddl
Z
g Z	 dd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )z
Monkey patching of distutils.
é    N)Úimport_modulec                 C   s"   t  ¡ dkr| f| j S t | ¡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.
    ÚJython)ÚplatformÚpython_implementationÚ	__bases__ÚinspectÚgetmro)Úcls© r
   ú„/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/setuptools/monkey.pyÚ_get_mro   s   	
r   c                 C   s.   t | tƒrtnt | tjƒrtndd„ }|| ƒS )Nc                 S   s   d S )Nr
   )Úitemr
   r
   r   Ú<lambda>(   s    zget_unpatched.<locals>.<lambda>)Ú
isinstanceÚtypeÚget_unpatched_classÚtypesÚFunctionTypeÚget_unpatched_function)r   Úlookupr
   r
   r   Úget_unpatched$   s
   ýr   c                 C   s:   dd„ t | ƒD ƒ}t|ƒ}|j d¡sd|  }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  d ¡s|V  qdS )Ú
setuptoolsN)Ú
__module__Ú
startswith)Ú.0r	   r
   r
   r   Ú	<genexpr>3   s   € 
ý
ÿz&get_unpatched_class.<locals>.<genexpr>Ú	distutilsz(distutils has already been patched by %r)r   Únextr   r   ÚAssertionError)r	   Úexternal_basesÚbaseÚmsgr
   r
   r   r   -   s   þr   c                  C   sÔ   t jtj_tjdk} | rt jtj_tjdk p/dtj  k o dk n  p/dtj  k o-dkn  }|r9d}|tjj	_
tƒ  tjtjtjfD ]}t jj|_qDt jjtj_t jjtj_dtjv ret jjtjd _tƒ  d S )N)é   é   r"   )é   é   é   )r"   é   )r"   r'   é   )r"   r#   zhttps://upload.pypi.org/legacy/zdistutils.command.build_ext)r   ÚCommandr   ÚcoreÚsysÚversion_infoÚfindallÚfilelistÚconfigÚPyPIRCCommandÚDEFAULT_REPOSITORYÚ_patch_distribution_metadataÚdistÚcmdÚDistributionÚ	extensionÚ	ExtensionÚmodulesÚ#patch_for_msvc_specialized_compiler)Úhas_issue_12885Úneeds_warehouseÚ	warehouseÚmoduler
   r
   r   Ú	patch_all?   s*   



þû


ÿ
r>   c                  C   s*   dD ]} t tj| ƒ}ttjj| |ƒ qdS )zDPatch write_pkg_file and read_pkg_file for higher metadata standards)Úwrite_pkg_fileÚread_pkg_fileÚget_metadata_versionN)Úgetattrr   r3   Úsetattrr   ÚDistributionMetadata)ÚattrÚnew_valr
   r
   r   r2   f   s   þr2   c                 C   s*   t ||ƒ}t| ƒ d|¡ t||| ƒ dS )zž
    Patch func_name in target_mod with replacement

    Important - original must be resolved by name to avoid
    patching an already patched function.
    Ú	unpatchedN)rB   ÚvarsÚ
setdefaultrC   )ÚreplacementÚ
target_modÚ	func_nameÚoriginalr
   r
   r   Ú
patch_funcm   s   
rN   c                 C   s
   t | dƒS )NrG   )rB   )Ú	candidater
   r
   r   r   ~   s   
r   c                     s¸   t dƒ‰ t ¡ dkrdS ‡ fdd„} t | d¡}t | d¡}zt|dƒŽ  t|d	ƒŽ  W n	 ty5   Y nw zt|d
ƒŽ  W n	 tyG   Y nw z	t|dƒŽ  W dS  ty[   Y dS w )z\
    Patch functions in distutils to use standalone Microsoft Visual C++
    compilers.
    zsetuptools.msvcÚWindowsNc                    sL   d| v rdnd}||  d¡ }tˆ |ƒ}t| ƒ}t||ƒs!t|ƒ‚|||fS )zT
        Prepare the parameters for patch_func to patch indicated function.
        Úmsvc9Úmsvc9_Úmsvc14_Ú_)ÚlstriprB   r   ÚhasattrÚImportError)Úmod_namerL   Úrepl_prefixÚ	repl_nameÚreplÚmod©Úmsvcr
   r   Úpatch_paramsŽ   s   


z9patch_for_msvc_specialized_compiler.<locals>.patch_paramszdistutils.msvc9compilerzdistutils._msvccompilerÚfind_vcvarsallÚquery_vcvarsallÚ_get_vc_envÚgen_lib_options)r   r   ÚsystemÚ	functoolsÚpartialrN   rW   )r_   rQ   Úmsvc14r
   r]   r   r9   ‚   s,   ÿÿÿr9   )Ú__doc__r+   Údistutils.filelistr   r   r   re   Ú	importlibr   r   r   Ú__all__r   r   r   r>   r2   rN   r   r9   r
   r
   r
   r   Ú<module>   s&    	'