B
²µ7›Aâ»ã @ sN d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ d dlZejZejZejZejZejej ej ej fZ ej dko®ej!dkZ"e"rÂdZ#d Z$d
Z%n d Z#d Z$d
Z%dddœZ&G dd„ dƒZ'G dd„ dƒZ(dd„ Z)dd„ Z*dd„ Z+dd „ Z,d#d d „Z-e)ƒ Z.e.d k r:ed e. ƒ‚G d!d"„ d"e ƒZ/dS )$é N)ÚDistutilsExecErrorÚDistutilsPlatformErrorÚ CompileErrorÚLibErrorÚ LinkError)Ú CCompilerÚgen_preprocess_optionsÚgen_lib_options)Úlog)Ú get_platformÚwin32l z1Software\Wow6432Node\Microsoft\VisualStudio\%0.1fz5Software\Wow6432Node\Microsoft\Microsoft SDKs\Windowsz,Software\Wow6432Node\Microsoft\.NETFrameworkz%Software\Microsoft\VisualStudio\%0.1fz)Software\Microsoft\Microsoft SDKs\Windowsz Software\Microsoft\.NETFrameworkÚx86Zamd64)r z win-amd64c @ sL e Zd Zdd„ ZeeƒZdd„ ZeeƒZdd„ ZeeƒZdd„ ZeeƒZd S )
ÚRegc C s: x,t D ]$}| ||¡}|r||kr|| S qW t|ƒ‚d S )N)ÚHKEYSÚ read_valuesÚKeyError)ÚclsÚpathÚkeyÚbaseÚd© r ú t d¡ d }Y nX |rPtj |¡sÆd| }tj
|d ¡}|r¸tj |¡r¸tj |tj tj d¡}tj
|¡}tj |¡sÆt d| ¡ d S nt d| ¡ |sØt d¡ d S tj |d ¡}tj |¡rö|S t d
¡ d S ) Nz %s\Setup\VCr= z%Unable to find productdir in registryzVS%0.f0COMNTOOLSZVCz%s is not a valid directoryz Env var %s is not set or invalidzNo productdir foundz
vcvarsall.batz Unable to find vcvarsall.bat)r6 r r r r
ÚdebugrU r ÚisdirÚenvironÚgetrZ ÚpardirÚabspathÚisfile)r: r7 r= ZtoolskeyZtoolsdirÚ vcvarsallr r r Úfind_vcvarsallÚ s2
re c C s< t | ƒ}ddddh}i }|d kr(tdƒ‚t d|| ¡ tjd||f tjtjd}z¨| ¡ \}}| ¡ d krzt| d
¡ƒ‚| d
¡}xr|
d ¡D ]d}t |¡}d |kr¨q|
¡ }|
d d
¡\} }
| ¡ } | |kr|
tj¡rè|
d d… }
t|
ƒ|| < qW W d |j ¡ |j ¡ X t|ƒt|ƒkr8ttt| ¡ ƒƒƒ‚|S )NÚincludeÚlibZlibpathr z Unable to find vcvarsall.batz'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)ÚstdoutÚstderrr r+ Ú
ú=r rF )re r r
r] Ú
subprocessÚPopenÚPIPEZ communicateÚwaitr* rP r r& Ústripr% ÚendswithrU rY r\ rh Úcloseri rO Ú
ValueErrorÚstrÚlistÚkeys) r: Zarchrd Z interestingÚresultÚpopenrh ri Úliner r( r r r Úquery_vcvarsallý s<
rz g @z(VC %0.1f is not supported by this modulec
@ sÚ e Zd ZdZi ZdgZdddgZdgZdgZee e e Z dZ
d Z d
Z d Z
d ZZd
Zd-dd„Zd.dd„Zd/dd„Zd0dd„Zd1dd„Zd2dd „Zd d „ Zd d „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Z d3d)d*„Z d+d,„ Z dS )4Ú MSVCCompilerZmsvcz.cz.ccz.cppz.cxxz.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s8 t | |||¡ t| _d| _g | _d | _d | _d| _d S )Nz Software\Microsoft\VisualStudioF) r r; ÚVERSIONÚ_MSVCCompiler__versionZ_MSVCCompiler__rootÚ_MSVCCompiler__pathsÚ plat_nameÚ_MSVCCompiler__archÚ initialized)r9 ÚverboseZdry_runÚforcer r r r; I s zMSVCCompiler.__init__Nc C s |d krt ƒ }d}||kr(td|f ƒ‚dtjkrfdtjkrf| d¡rfd| _d| _d| _d| _d | _ nÂ|t ƒ ksx|d
kr‚t
| }nt
t ƒ d t
| }t t |ƒ}|d
tj¡| _|d
tjd
< |d tjd< t| jƒdkrìtd| j ƒ‚| d¡| _| d¡| _| d¡| _| d¡| _| d ¡| _ y,x&tjd
d¡D ]}| j |¡ q|| j
kr>|}d| }y"| | jg| |g |g ¡ W qh tk
r8 } z t|ƒ‚W d d }~X Y qhX qhnÒ|| jkr tj |¡}tj |¡}yl| | jgd|d|g |g ¡ tj tj |¡¡\}}tj ||d ¡}| | jgd| g |g ¡ W qh tk
rú } z t|ƒ‚W d d }~X Y qhX qhntd||f ƒ‚d | }y&| | jg|
| ||g | ¡ W qh tk
rj } z t|ƒ‚W d d }~X Y qhX qhW |
S )
Nz/cz/Tcz/Tpz/foz-hz-rz.rcz"Don't know how to compile %s to %sz/Fo)r r‘ Z_setup_compiler ÚextendrŽ r r rU r rb Ú
_c_extensionsÚ_cpp_extensionsr˜ ÚspawnrŠ r r rš Údirnamer‹ r“ r— rZ rˆ )r9 Zsourcesrœ r5 Z include_dirsr] Ú
extra_preargsÚextra_postargsZdependsZ compile_infoÚobjectsZpp_optsZbuildZ compile_optsÚobjÚsrcr Z input_optZ
output_optÚmsgZh_dirZrc_dirr r„ Zrc_filer r r ÚcompileÈ sj
zMSVCCompiler.compilec
C sž | j s| ¡ | ||¡\}}| j||d}| ||¡rŽ|d| g }|rJy| | jg| ¡ W qš tk
rŠ } z t|ƒ‚W d d }~X Y qšX n t
d|¡ d S )N)rœ z/OUT:zskipping %s (up-to-date)) r r‘ Ú_fix_object_argsÚlibrary_filenameÚ
_need_linkr¢ rg r r r
r] ) r9 r¦ Zoutput_libnamerœ r] Ú target_langÚoutput_filenameZlib_argsr© r r r Úcreate_static_lib s z MSVCCompiler.create_static_libc
C sX | j s| ¡ | ||¡\}}| |||¡}|\}}}|rL| dt|ƒ ¡ t| |||ƒ}|d k rptj ||¡}|
||¡rH|t j kr¬| rœ| j
dd … }q¾| jdd … }n| r¸| j
}n| j}g }x |pÊg D ]}| d| ¡ qÌW || | | d| g }tj |d ¡} |d k rPtj tj |¡¡\}}tj | | |¡¡}| d| ¡ | || |¡ |
rp|
|d d…< | r€| | ¡ | tj |¡¡ y| | jg| ¡ W n, tk
rÔ } z t|ƒ‚W d d }~X Y nX | ||¡}|d k rT|\}}d||f }y| dd d
||g¡ W n, tk
rD } z t|ƒ‚W d d }~X Y nX n t d |¡ d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:z/OUT:r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestzskipping %s (up-to-date)) r r‘ r« Z
_fix_lib_argsÚwarnrt r rU r rZ r r Ú
EXECUTABLEr r r r£ r“ r— r¬ Úmanifest_setup_ldargsrŸ Zmkpathr¢ r‰ r r Úmanifest_get_embed_infor
r] ) r9 Ú target_descr¦ r¯ rœ Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsr] r¤ r¥ Ú
build_tempr® Z
fixed_argsZlib_optsZldflagsZ export_optsZsymÚld_argsZdll_nameZdll_extZ implib_filer© ZmfinfoZ
mffilenameÚmfidZout_argr r r Úlink6 sl
zMSVCCompiler.linkc C s, t j |t j |¡d ¡}| d| ¡ d S )Nz .manifestz/MANIFESTFILE:)rU r rZ r— r )r9 r¯ r¶ r· Ú
temp_manifestr r r r³ • s z"MSVCCompiler.manifest_setup_ldargsc C s` x,|D ] }| d¡r| dd¡d }P qW d S |tjkr>d}nd}| |¡}|d krXd S ||fS )Nz/MANIFESTFILE:ú:r rK )Ú
startswithrP r r² Ú_remove_visual_c_ref)r9 rµ r· Úargrº r¸ r r r r´ ¡ s
z$MSVCCompiler.manifest_get_embed_infoc C s² y˜t |ƒ}z | ¡ }W d | ¡ X t dtj¡}t |d|¡}d}t |d|¡}t dtj¡}t ||¡d krrd S t |dƒ}z| |¡ |S | ¡ X W n t k
r¬ Y nX d S )NzU|)r’ z*\s*zI|)Úw)
ÚopenÚreadrr Úrerª ÚDOTALLrE ÚsearchÚwriteÚOSError)r9 Z
manifest_fileZ
manifest_fZ manifest_bufÚpatternr r r r½ ¹ s.
z!MSVCCompiler._remove_visual_c_refc C s d| S )Nz /LIBPATH:r )r9 Údirr r r Úlibrary_dir_optionä s z MSVCCompiler.library_dir_optionc C s t dƒ‚d S )Nztjd d¡D ]*}tj tj |¡|¡}tj |¡rH|S qHW |S )NZPathr… )r~ rU r rZ rb rc r_ rP )r9 Zexer@ Úfnr r r r‡ s zMSVCCompiler.find_exe)r r r )N)r r’ )NNNr NNN)Nr N)
NNNNNr NNNN)r ) r/ r0 r1 Z
compiler_typeZ executablesr r¡ r˜ rš r– r™ r› Zstatic_lib_extensionZshared_lib_extensionZstatic_lib_formatZshared_lib_formatZ
exe_extensionr; r‘ rž rª r° r¹ r³ r´ r½ rÉ rÊ rË rÎ r‡ r r r r r{ + sP
W
V
R +
r{ )r
)0rU rl rM r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr ÚwinregZ OpenKeyExr ZEnumKeyr Z EnumValuer$ Úerrorr Z
HKEY_USERSÚHKEY_CURRENT_USERÚHKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr ÚplatformÚmaxsizeZ NATIVE_WIN64r6 r? r> rŒ r r4 rT rX r\ re rz r| r{ r r r r Ú sJ >. #
)