-
Notifications
You must be signed in to change notification settings - Fork 29.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
build,windows: implement PEP514 python detection
- Loading branch information
Showing
2 changed files
with
69 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
@IF NOT DEFINED DEBUG_HELPER @ECHO OFF | ||
SETLOCAL | ||
:: If python.exe is in %Path%, just validate | ||
FOR /F "delims=" %%a IN ('where python 2^> NUL') DO ( | ||
SET need_path=0 | ||
SET p=%%~dpa | ||
IF NOT ERRORLEVEL 1 GOTO :validate | ||
) | ||
|
||
:: Query the 3 locations mentioned in PEP 514 for a python2 InstallPath | ||
FOR %%K IN ( "HKCU\Software", "HKLM\SOFTWARE", "HKLM\Software\Wow6432Node") DO ( | ||
SET need_path=1 | ||
CALL :find-main-branch %%K | ||
:: If validate returns 0 just jump to the end | ||
IF NOT ERRORLEVEL 1 GOTO :validate | ||
) | ||
EXIT /B 1 | ||
|
||
:: Helper subroutine to handle quotes in %1 | ||
:find-main-branch | ||
SET main_key="%~1\Python\PythonCore" | ||
REG QUERY %main_key% /s | findstr "2." | findstr InstallPath > NUL 2> NUL | ||
IF NOT ERRORLEVEL 1 CALL :find-key %main_key% | ||
EXIT /B | ||
|
||
:: Query registry sub-tree for InstallPath | ||
:find-key | ||
FOR /F "delims=" %%a IN ('REG QUERY %1 /s ^| findstr "2." ^| findstr InstallPath') DO IF NOT ERRORLEVEL 1 CALL :find-path %%a | ||
EXIT /B | ||
|
||
:: Parse the value of %1 as the path for python.exe | ||
:find-path | ||
FOR /F "tokens=3*" %%a IN ('REG QUERY %1 /ve') DO ( | ||
SET pt=%%a | ||
IF NOT ERRORLEVEL 1 SET p=%pt% | ||
EXIT /B 0 | ||
) | ||
EXIT /B 1 | ||
|
||
:: Check if %p% holds a path to a real python2 executable | ||
:validate | ||
IF NOT EXIST "%p%python.exe" EXIT /B 1 | ||
:: Check if %p% is python2 | ||
%p%python.exe -V 2>&1 | findstr /R "^Python.2.*" > NUL | ||
IF ERRORLEVEL 1 EXIT /B %ERRORLEVEL% | ||
:: We can wrap it up | ||
ENDLOCAL & SET pt=%p%& SET need_path_ext=%need_path% | ||
SET VCBUILD_PYTHON_LOCATION=%pt%python.exe | ||
IF %need_path_ext%==1 SET Path=%Path%;%pt% | ||
SET need_path_ext= | ||
EXIT /B %ERRORLEVEL% |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters