U
e5dRN @ s d Z ddlZddlZddlZddlZddlZddlmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZ dd Zd
d Zd d
dddZdd Zdd Zd ddZd ddddZG dd d e
Z dS ) a distutils._msvccompiler
Contains MSVCCompiler, an implementation of the abstract CCompiler class
for Microsoft Visual Studio 2015.
The module is compatible with VS 2015 and later. You can find legacy support
for older versions in distutils.msvc9compiler and distutils.msvccompiler.
N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_lib_options)log) get_platform)countc C s z t jt jdt jt jB d} W n tk
r> td Y dS X d}d }| t D ]}zt | |\}}}W n tk
r Y qY nX |rT|t j
krTt j
|rTztt|}W n ttfk
r Y qTY nX |dkrT||krT|| }}qTW 5 Q R X ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)accessz Visual C++ is not registeredNNr )winregZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEYOSErrorr debugr Z EnumValueZREG_SZospathisdirintfloat
ValueError TypeError)key best_versionbest_dirivZvc_dirZvtversion r //usr/lib64/python3.8/distutils/_msvccompiler.py _find_vc2015 s2
r" c
C s ddl } tjdp tjd}|s(dS z8tjtj|dddd d
d d d
dddg ddd }W n tj t
t fk
r~ Y dS X tj|ddd}tj |rd|fS dS )aJ Returns "15, path" based on the result of invoking vswhere.exe
If no install is found, returns "None, None"
The version is returned to avoid unnecessarily changing the function
result. It may be ignored when the path is not None.
If vswhere.exe is not available, by definition, VS 2017 is not
installed.
r NzProgramFiles(x86)Z ProgramFilesr
zMicrosoft Visual StudioZ Installerz vswhere.exez-latestz -prereleasez -requiresz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64z -propertyZinstallationPathz -products*mbcsstrict)encodingerrorsZVCZ AuxiliaryZBuild )
jsonr environget
subprocess check_outputr joinstripCalledProcessErrorr UnicodeDecodeErrorr )r) rootr r r r! _find_vc2017: s2
r3 x86Zx64ZarmZarm64)r4 x86_amd64x86_arm x86_arm64c C s\ t \}}|st \}}|s*td dS tj|d}tj|sTtd| dS |d fS )Nz$No suitable Visual C++ version foundr
z
vcvarsall.batz%s cannot be found)r3 r" r r r r r. isfile) plat_spec_r r vcvarsallr r r! _find_vcvarsallc s
r< c
C s t dr dd t j D S t| \}}|s6tdz&tjd|| tj dj
ddd }W n@ tj k
r } z t
|j td
|jW 5 d }~X Y nX d d d d
| D D }|S )NZDISTUTILS_USE_SDKc S s i | ]\}}| |qS r lower).0r valuer r r!
w s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)stderrzutf-16lereplace)r' zError executing {}c S s$ i | ] \}}}|r|r| |qS r r= )r? r r: r@ r r r! rA s
c s s | ]}| d V qdS )=N) partition)r? liner r r! s z _get_vc_env..)r getenvr* itemsr< r r, r- formatZSTDOUTdecoder0 r erroroutputcmd
splitlines)r9 r; r: outexcenvr r r! _get_vc_envu s0
rS c C sN |st dt j}|D ].}t jt j|| }t j|r| S q| S )at Return path to an MSVC executable program.
Tries to find the program in several places: first, one of the
MSVC program search paths from the registry; next, the directories
in the PATH environment variable. If any of those work, return an
absolute path that is known to exist. If none of them work, just
return the original program name, 'exe'.
r )r rH splitpathsepr r. abspathr8 )Zexepathspfnr r r! _find_exe s
rZ r5 r6 r7 )Zwin32z win-amd64z win-arm32z win-arm64c
s e Zd ZdZdZi ZdgZdddgZdgZdgZ ee e e Z
d Z d
Z d Z
d Zd
ZZdZd(ddZd)ddZd*ddZd+ddZd,ddZd-d d Z fd d Zd d! Zd"d# Zd$d% Zd.d&d'Z Z S )/ MSVCCompilerzwConcrete class that implements an interface to Microsoft Visual C++,
as defined by the CCompiler abstract class.Zmsvcz.cz.ccz.cppz.cxx.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t | ||| d | _d| _d S )NF)r __init__ plat_name initialized)selfverboseZdry_runZforcer r r! r] s zMSVCCompiler.__init__Nc C s
| j rtd|d kr t }|tkr6tdttt| }t|}|sRtd|dd| _ | j
t j }t
d|| _t
d|| _t
d|| _t
d || _t
d
|| _t
d || _|d d
t j D ]}|r| |t j q|d
d
t j D ]}|r| |t j qd | _ddddddg| _ddddddg| _dddg}ddddg}|d!| _|d"| _ |d#| _ |d$| _ || _ || _ t!j"d f| jt!j"d f| jt!j"d f| j t!j#d f| j t!j#d f| j t!j#d f| j t!j$d f| j t!j$d f| j t!j$d f| j i | _%d | _ d S )%Nzdon't init multiple timesz --plat-name must be one of {}z7Unable to find a compatible Visual Studio installation.r zcl.exezlink.exezlib.exezrc.exezmc.exezmt.exeZincludelibz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCGz /DEBUG:FULL/MANIFEST:EMBED,ID=1/DLL/MANIFEST:EMBED,ID=2/MANIFESTUAC:NOFT)rd )rd )re rf rg )re rf rg )&r_ AssertionErrorr
PLAT_TO_VCVARSr rJ tuplerS r+ _pathsrT r rU rZ cclinkerrc rcmcZmtZadd_include_dirrstripsepZadd_library_dirZpreprocess_optionscompile_optionscompile_options_debugZ ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY_ldflags)r` r^ r9 Zvc_envrW dirldflagsZ
ldflags_debugr r r!
initialize s
zMSVCCompiler.initializerb c sT fddj D fddjj D p4d fdd}tt||S )Nc s i | ] }| j qS r )
obj_extensionr? extr` r r! rA & s z1MSVCCompiler.object_filenames..c s i | ] }| j qS r )
res_extensionry r{ r r! rA ' s rb c s t j| \}}r"t j|}n2t j|\}}|t jjt jjfrT|dd }zt j| | W S t k
r t
d | Y nX d S )N z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithrq altsepr. LookupErrorr rJ )rX baserz r: )ext_map
output_dir strip_dirr r!
make_out_path, s z4MSVCCompiler.object_filenames..make_out_path)src_extensions_rc_extensions_mc_extensionslistmap)r` Zsource_filenamesr r r r )r r r` r r! object_filenames! s z MSVCCompiler.object_filenamesc C s | j s| | ||||||} | \}}
}} } |p6g }
|
d |rT|
| j n |
| j d}|
D ]}z| | \}}W n tk
r Y qhY nX |rtj
|}|| j krd| }nD|| j krd| }d}n*|| j
kr@|}d| }z | | jg| ||g W qh tk
r: } z t|W 5 d }~X Y qhX qhn|| jkrtj |}tj |}z\| | jd|d||g tj tj |\}}tj ||d }| | jd| |g W qh tk
r } z t|W 5 d }~X Y qhX qhntd
||| jg|
| }|r"|d || |d | || z| | W qh tk
r~ } z t|W 5 d }~X Y qhX qh|
S )
Nz/cFz/Tcz/TpTz/foz-hz-rr\ z"Don't know how to compile {} to {}z/EHscz/Fo)r_ rw Z_setup_compileappendextendrs rr KeyErrorr r rV
_c_extensions_cpp_extensionsr spawnrn r r r dirnamero r~ r r. rJ rl )r` Zsourcesr ZmacrosZ include_dirsr
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsobjsrcrz Z input_optZ
output_optmsgZh_dirZrc_dirr r: Zrc_fileargsr r r! compileB sx
zMSVCCompiler.compilec
C s | j s| | ||\}}| j||d}| ||r|d| g }|rJz,td| jd| | | jg| W q t
k
r } z t |W 5 d }~X Y qX n td| d S )N)r /OUT:Executing "%s" %s skipping %s (up-to-date)) r_ rw _fix_object_argslibrary_filename
_need_linkr r rc r. r r r ) r` r Zoutput_libnamer r target_langoutput_filenameZlib_argsr r r r! create_static_lib s z MSVCCompiler.create_static_libc
C s | j s| | ||\}}| |||}|\}}}|rL| dt| t| |||}|d k rptj ||}|
||r| j || f }dd |pg D }|| | | d| g }tj |d } |d k r tj
tj|\}}tj | | |}|d| |
r |
|d d< | r.|| tj tj|}| | z,td| jd | | | jg| W n, tk
r } z t|W 5 d }~X Y nX n td | d S )
Nz5I don't know what to do with 'runtime_library_dirs': c S s g | ] }d | qS )z/EXPORT:r )r? Zsymr r r!
s z%MSVCCompiler.link..r r z/IMPLIB:r r r )r_ rw r Z
_fix_lib_argswarnstrr r r r. r rt r r~ r r r r rV Zmkpathr r rm r r r )r` Z target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r r Z
build_tempr Z
fixed_argsZlib_optsrv Z export_optsZld_argsZdll_nameZdll_extZ implib_filer r r r! link s`
zMSVCCompiler.linkc s8 t d}z | jt jd< t |W S |t jd< X d S )Nr )r rH r* rk superr )r` rN Zold_path __class__r r! r s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r r` ru r r r! library_dir_option s z MSVCCompiler.library_dir_optionc C s t dd S )Nz:don't know how to set runtime library search path for MSVC)r r r r r! runtime_library_dir_option
s z'MSVCCompiler.runtime_library_dir_optionc C s
| |S )N)r )r` rc r r r! library_option s zMSVCCompiler.library_optionc C s\ |r|d |g}n|g}|D ]:}|D ]0}t j|| |}t j|r$| S q$q d S )NZ_d)r r r. r r8 )r` dirsrc r Z try_namesru nameZlibfiler r r! find_library_file s z MSVCCompiler.find_library_file)r r r )N)r rb )NNNr NNN)Nr N)
NNNNNr NNNN)r ) __name__
__module__ __qualname____doc__Z
compiler_typeZ executablesr r r r r r| rx Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr] rw r r r r r r r r r
__classcell__r r r r! r[ sf
P
"
]
E r[ )N) r r Zshutilstatr, r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
itertoolsr r" r3 ZPLAT_SPEC_TO_RUNTIMEr< rS rZ ri r[ r r r r! s4 #