]w @ s d d l Z d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z m Z d d l
m Z m Z m
Z
d d l m Z d d l m Z d d l Z e j Z e j Z e j Z e j Z e j e j e j e j f Z e j d k o e j! d& k Z" e" rd Z# d
Z$ d Z% n d Z# d
Z$ d Z% d d d d d d i Z& Gd d d Z' Gd d d Z( d d Z) d d Z* d d Z+ d d Z, d d d! Z- e) Z. e. d" k re d# e. Gd$ d% d% e Z/ d S)' N)DistutilsExecErrorDistutilsPlatformError CompileErrorLibError LinkError) CCompilergen_preprocess_optionsgen_lib_options)log) get_platformwin32 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\.NETFrameworkx86z win-amd64amd64zwin-ia64Zia64c @ sp e Z d Z d d Z e e Z d d Z e e Z d d Z e e Z d d Z e e Z d S)
Regc C sM x: t D]2 } | j | | } | r | | k r | | Sq Wt | d S)N)HKEYS read_valuesKeyError)clspathkeybased r 2/opt/alt/python35/lib64/python3.5/msvc9compiler.py get_valueA s
z
Reg.get_valuec C s y t | | } Wn t k
r+ d SYn Xg } d } xD y t | | } Wn t k
rc PYn X| j | | d 7} q; W| S)Nr ) RegOpenKeyExRegError
RegEnumKeyappend)r r r handleLikr r r read_keysI s
z
Reg.read_keysc C s y t | | } Wn t k
r+ d SYn Xi } d } xh y t | | \ } } } Wn t k
rl PYn X| j } | j | | | j | <| d 7} q; W| S)Nr r )r r RegEnumValuelower convert_mbcs) r r r r" r r$ namevaluetyper r r r [ s
zReg.read_valuesc C sG t | d d } | d k rC y | d } Wn t k
rB Yn X| S)Ndecodembcs)getattr UnicodeError)sZdecr r r r) q s
zReg.convert_mbcsN) __name__
__module__ __qualname__r classmethodr& r r) staticmethodr r r r r = s r c @ s@ e Z d Z d d Z d d Z d d Z d d Z d S)
MacroExpanderc C s' i | _ t | | _ | j | d S)N)macrosVS_BASEvsbase load_macros)selfversionr r r __init__} s
zMacroExpander.__init__c C s! t j | | | j d | r? r; rH r r r r r7 { s r7 c C s d } t j j | } | d
k r( d S| t | } t j | d j d d \ } } t | d d d } | d k r | d 7} t | d d d } | d k r d } | d k r | | Sd S) NzMSC v.r r
g $@r )sysr= findlensplitint)prefixr$ r1 restZ majorVersionZ minorVersionr r r get_build_version s %
rV c C sF g } x9 | D]1 } t j j | } | | k r
| j | q
W| S)N)osr normpathr! )pathsZ
reduced_pathsrC Znpr r r normalize_and_reduce_paths s
rZ c C sX | j t j } g } x' | D] } | | k r | j | q Wt j j | } | S)N)rR rW pathsepr! join)variableZoldListZnewListr$ Z newVariabler r r removeDuplicates s
r^ c
C s^ t | } y t j d | d } Wn% t k
rK t j d d } Yn X| sf t j j | r d | } t j j
| d } | r t j j | r t j j | t j t j d } t j j
| } t j j | s t j d | d Sn t j d | | s"t j d d St j j | d } t j j | rM| St j 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)r9 r r r r
debugrW r isdirenvirongetr\ pardirabspathisfile)r= r: r@ ZtoolskeyZtoolsdir vcvarsallr r r find_vcvarsall s2
!
rg c C s t | } t d } i } | d k r6 t d t j d | | t j d | | f d t j d t j } z | j \ } } | j d
k r t | j
d | j
d } x | j d D] } t j
| } d
| k r q | j } | j d
d \ } }
| j } | | k r |
j t j rQ|
d d }
t |
| | VERSION_MSVCCompiler__versionZ_MSVCCompiler__root_MSVCCompiler__paths plat_name_MSVCCompiler__arch initialized)r< verbosedry_runforcer r r r> J s zMSVCCompiler.__init__Nc C s | d k r t } d$ } | | k r: t d | f d t j k r d t j k r | j d r d | _ d | _ d | _ d
| _ d | _ n
| t k s | d k r t
| } n t
t d t
| } t t | } | d
j
t j | _ | d t j d <| d t j d |)r z*\s*zI|)w)
openreadrw rer DOTALLrH searchwriteOSError)r< Z
manifest_fileZ
manifest_fZ manifest_bufpatternr r r r s.
z!MSVCCompiler._remove_visual_c_refc C s d | S)Nz /LIBPATH:r )r< dirr r r library_dir_option s z MSVCCompiler.library_dir_optionc C s t d d S)Nz r r r r r r r r r r r r r r r r r r , sP
X VR + r l )0rW rp rO r Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr winregZ OpenKeyExr ZEnumKeyr Z EnumValuer' errorr Z
HKEY_USERSHKEY_CURRENT_USERHKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr platformmaxsizeZ NATIVE_WIN64r9 rB rA r r r7 rV rZ r^ rg r r r r r r r sL ( >. #)