3
\KT @ s d Z ddlZddlZddlZddlZddlZddlmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ ddlmZ dd Zd
d Zd d
Zdd ZdddZdddZedgZ G dd de
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 y t jt jdt jt jB d} W n tk
r< tjd dS X d}d }| xt D ]}yt j | |\}}}W n tk
r P Y nX |rT|t j
krTt j j
|rTytt|}W n ttfk
r wTY nX |dkrT||krT|| }}qTW W d Q R X ||fS )Nz'Software\Microsoft\VisualStudio\SxS\VC7)accessz Visual C++ is not registeredr )NN)winregZ OpenKeyExHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_32KEYOSErrorr debugr Z EnumValueZREG_SZospathisdirintfloat
ValueError TypeError)key best_versionbest_dirivvc_dirZvtversion r //usr/lib64/python3.6/distutils/_msvccompiler.py _find_vc2015 s0
r" c
s dd l dd l} d }d }g fdd}| j|d}|j |j xD ]~\}}}}d|krLtjj|ddd} tjj| sqLy td d
|j d D }
W n t
t fk
r wLY nX |
|krL|
| }}qLW y |d }W n t k
r d }Y nX ||fS )
Nr c s j j d S )N)extendfindallr )_distutils_findvs all_packagesr r! _getallD s z _find_vc2017.._getall)targetz1Microsoft.VisualStudio.Component.VC.Tools.x86.x64ZVCZ AuxiliaryZBuildc s s | ]}t |V qd S )N)r ).0r r r r! P s z _find_vc2017...)r )
r% threadingZThreadstartjoinr r r tuplesplitr r
IndexError) r, r r r' tnameZ version_strr Zpackagesr r r )r% r& r! _find_vc2017: s2
r4 c C s
t \}}d }d| krdnd}|rztjj|ddddddd }y d
d l}|j|d d d }W n tttfk
rx d }Y nX |st \}}|rtjj|d|dd }|st j
d dS tjj|d}tjj |st j
d| dS | ptjj | rt j
d| d }||fS )Namd64Zx64x86z..ZredistZMSVCz**zMicrosoft.VC141.CRTzvcruntime140.dllr T) recursive zMicrosoft.VC140.CRTz$No suitable Visual C++ version foundz
vcvarsall.batz%s cannot be found)NN)NN) r4 r r r. glob ImportErrorr LookupErrorr" r r isfile) plat_specr r vcruntimeZvcruntime_platZvcredistr: vcvarsallr r r! _find_vcvarsall[ s6
rA c C s t jdr dd t jj D S t| \}}|s6tdy&tjdj|| tj dj
ddd }W n@ tj k
r } z"t j
|j td
j|jW Y d d }~X nX d d d d
|j D D }|r||d< |S )NZDISTUTILS_USE_SDKc S s i | ]\}}||j qS r )lower)r) r valuer r r!
s z _get_vc_env..z Unable to find vcvarsall.batzcmd /u /c "{}" {} && set)stderrzutf-16lereplace)errorszError executing {}c S s$ i | ] \}}}|r|r||j qS r )rB )r) r _rC r r r! rD s c s s | ]}|j d V qdS )=N) partition)r) liner r r! r* s z _get_vc_env..py_vcruntime_redist)r getenvenvironitemsrA r
subprocessZ check_outputformatZSTDOUTdecodeZCalledProcessErrorr erroroutputcmd
splitlines)r> r@ r? outexcenvr r r! _get_vc_env} s(
rZ c C sN |st jdjt j}x2|D ]*}t jjt jj|| }t jj|r |S q W | 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 rM r0 pathsepr r. abspathr= )Zexepathspfnr r r! _find_exe s
r` r6 Z x86_amd64)Zwin32z win-amd64zvcruntime140.dllc
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 Zd d Z fd d!Zd"d# Zd$d% Zd&d' Z d0d(d)Z Z S )1 MSVCCompilerzwConcrete class that implements an interface to Microsoft Visual C++,
as defined by the CCompiler abstract class.Zmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s t j| ||| d | _d| _d S )NF)r __init__ plat_name initialized)selfverboseZdry_runZforcer r r! rb s zMSVCCompiler.__init__Nc C sN | j std|d kr t }|tkr8tdjttt| }t|}|sTtd|jdd| _ | j j
t j }t
d|| _t
d|| _t
d|| _t
d || _t
d
|| _t
d || _|jd d| _x2|jd
dj
t j D ]}|r| j|jt j qW x6|jddj
t j D ] }|r| j|jt j qW d | _dddddg| _| jj| jrZdnd ddddddg| _ dddg}| js|j d% dddd g}|d&| _ |d'| _ |d(| _ |d)| _!|| _"|| _#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!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.exerL includelibz/nologoz/Oxz/W3z/GLz/DNDEBUGz/MDz/MTz/Odz/MDdz/Ziz/D_DEBUGz/INCREMENTAL:NOz/LTCG/nodefaultlib:libucrt.libucrt.libz /DEBUG:FULL/MANIFEST:EMBED,ID=1/DLL/MANIFEST:EMBED,ID=2/MANIFESTUAC:NOFT)rj rk )rl )rl )rm rn ro )rm rn ro ))rd AssertionErrorr
PLAT_TO_VCVARSr rQ r/ rZ get_pathsr0 r r[ r` cclinkerri rcmcZmt_vcruntime_redistZadd_include_dirrstripsepZadd_library_dirZpreprocess_optionscompile_optionsappendcompile_options_debugr# Z ldflags_exeZldflags_exe_debugZldflags_sharedZldflags_shared_debugZldflags_staticZldflags_static_debugr Z
EXECUTABLEZ
SHARED_OBJECTZSHARED_LIBRARY_ldflags)re rc r> Zvc_envr] dirldflagsZ
ldflags_debugr r r!
initialize sb
zMSVCCompiler.initializerg c sT fddj D fddjj D p4d fdd}tt||S )Nc s i | ] } j |qS r )
obj_extension)r) ext)re r r! rD 8 s z1MSVCCompiler.object_filenames..c s i | ] } j |qS r )
res_extension)r) r )re r r! rD 9 s rg c s t jj| \}}r"t jj|}n2t jj|\}}|jt jjt jjfrT|dd }yt jj| | S t k
r t
dj | Y nX d S )Nr8 z Don't know how to compile {}) r r splitextbasename
splitdrive
startswithrz altsepr. r< r rQ )r^ baser rH )ext_map
output_dir strip_dirr r!
make_out_path> s z4MSVCCompiler.object_filenames..make_out_path)src_extensions_rc_extensions_mc_extensionslistmap)re Zsource_filenamesr r r r )r r re r r! object_filenames3 s
z MSVCCompiler.object_filenamesc C s | j s| j | j||||||} | \}}
}} } |p6g }
|
jd |rT|
j| j n |
j| j d}x |
D ]}y| | \}}W n tk
r wlY nX |rtj j
|}|| j krd| }nD|| j krd| }d}n*|| j
krB|}d| }y | j| jg| ||g W ql tk
r< } zt|W Y d d }~X qlX qln|| jkrtj j|}tj j|}y\| j| jd|d||g tj jtj j|\}}tj j||d }| j| jd| |g W ql tk
r } zt|W Y d d }~X qlX qlntd
j||| jg|
| }|r$|jd |j| |jd | |j| y| j| W ql tk
r } zt|W Y d d }~X qlX qlW |
S )
Nz/cFz/Tcz/TpTz/foz-hz-rz.rcz"Don't know how to compile {} to {}z/EHscz/Fo)rd r Z_setup_compiler| r# r} r{ KeyErrorr r r\
_c_extensions_cpp_extensionsr spawnrv r r r dirnamerw r r r. rQ rt )re Zsourcesr ZmacrosZ include_dirsr
extra_preargsextra_postargsZdependsZ compile_infoobjectsZpp_optsZbuildZ compile_optsZ add_cpp_optsobjsrcr Z input_optZ
output_optmsgZh_dirZrc_dirr rH Zrc_fileargsr r r! compileT sn
zMSVCCompiler.compilec C s | j s| j | j||\}}| j||d}| j||r|d| g }|rJy,tjd| jdj| | j | jg| W q t
k
r } zt |W Y d d }~X qX n tjd| d S )N)r z/OUT:zExecuting "%s" %s zskipping %s (up-to-date)) rd r _fix_object_argslibrary_filename
_need_linkr r ri r. r r r ) re r Zoutput_libnamer r target_langoutput_filenameZlib_argsr r r r! create_static_lib s z MSVCCompiler.create_static_libc C s | j s| j | j||\}}| j|||}|\}}}|rL| jdt| t| |||}|d k rptjj ||}| j
||r| j || f }dd |pg D }|| | | d| g }tjj |d } |d k r tjj
tjj|\}}tjj | | j|}|jd| |
r |
|d d< | r.|j| tjj tjj|}| j| y6tjd| jdj | | j| jg| | j| W n, tk
r } zt|W Y d d }~X nX n tjd | 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..z/OUT:r z/IMPLIB:zExecuting "%s" %sr zskipping %s (up-to-date))rd r r Z
_fix_lib_argswarnstrr r r r. r r~ r r r r r| r# r\ Zmkpathr r ru r _copy_vcruntimer r )re Z target_descr r r Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr r r Z
build_tempr Z
fixed_argsZlib_optsr Z export_optsZld_argsZdll_nameZdll_extZ implib_filer r r r! link sN
zMSVCCompiler.linkc C s` | j }| stjj| r d S tjj|j tkr6d S tjd| t j
||}tj |t j
d S )Nz Copying "%s")rx r r r= r rB
_BUNDLED_DLLSr r shutilcopychmodstatS_IWRITE)re r r? r r r! r s z MSVCCompiler._copy_vcruntimec s4 t jd}z| jt jd< t j|S |t jd< X d S )Nr )r rM rs rN superr )re rU Zold_path) __class__r r! r s
zMSVCCompiler.spawnc C s d| S )Nz /LIBPATH:r )re r 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 )re r r r r! runtime_library_dir_option) s z'MSVCCompiler.runtime_library_dir_optionc C s
| j |S )N)r )re ri r r r! library_option- s zMSVCCompiler.library_optionc C s` |r|d |g}n|g}xB|D ]6}x0|D ](}t jj|| j|}t jj|r(|S q(W q W d S d S )NZ_d)r r r. r r= )re dirsri r Z try_namesr r3 Zlibfiler r r! find_library_file0 s
z MSVCCompiler.find_library_file)r r r )N)r rg )NNNr NNN)Nr N)
NNNNNr NNNN)r ) __name__
__module__ __qualname____doc__Z
compiler_typeZ executablesr r r r r r r Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionrb r r r r r r r r r r r
__classcell__r r )r r! ra sN
T
[
9 ra )N) r r r r rP r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r Z distutilsr Zdistutils.utilr
itertoolsr r" r4 rA rZ r` rq frozensetr ra r r r r! s& !"