forked from ydb-platform/ydb
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ya.bat
executable file
·90 lines (84 loc) · 2.24 KB
/
ya.bat
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
@echo off
rem Ya Simple Windows launcher
setlocal
call :dbg Ya Simple Windows Launcher (Debug)
call :find_ya
if ERRORLEVEL 1 exit /b 1
call :dbg Ya: %YA_BAT_REAL%
call :fix_env
call :find_python
if ERRORLEVEL 1 exit /b 1
call :dbg Python: "%YA_BAT_PYTHON%"
call "%YA_BAT_PYTHON%" %YA_PY_KNOB% "%YA_BAT_REAL%" %*
exit /b %ERRORLEVEL%
:find_ya
call :dbg Searching for ya near ya.bat...
set YA_BAT_REAL=%~dp0ya
if exist "%YA_BAT_REAL%" exit /b 0
call :err Ya not found
exit /b 1
:fix_env
call :dbg Fixing environment...
for /f "delims=: tokens=2" %%F in ('chcp') do (
if "%%F" == " 65001" (
call :dbg -- Forcing I/O encoding for python: utf-8
set PYTHONIOENCODING=utf-8
set PYTHONUTF8=1
)
)
exit /b 0
:find_python
call :dbg Searching for python in PATH...
for /f "delims=" %%F in ('where python 2^>nul') do (
call :test_python %%~sF
if not ERRORLEVEL 1 (
set YA_BAT_PYTHON=%%F
if /I "%%F" == "C:\Windows\py.exe" (
set YA_PY_KNOB=-3
)
exit /b 0
)
)
call :dbg Searching for python in ftypes...
for /f delims^=^=^"^ tokens^=2 %%F in ('ftype Python.File 2^>nul') do (
call :test_python %%F
if not ERRORLEVEL 1 (
set YA_BAT_PYTHON=%%F
if /I "%%F" == "C:\Windows\py.exe" (
set YA_PY_KNOB=-3
)
exit /b 0
)
)
call :dbg Searching for python with Get-Command...
for /f "delims=" %%i in ('powershell -Command "Get-Command python | Select-Object -ExpandProperty Source"') do (
call :test_python %%i
if not ERRORLEVEL 1 (
set YA_BAT_PYTHON=%%i
exit /b 0
)
)
call :err Python not found
exit /b 1
:test_python
call :dbg -- Checking python: %1
if not exist %1 (
call :dbg ---- Not found
exit /b 1
)
for /f %%P in ('%1 -c "import os, sys; sys.stdout.write(os.name + '\n')" 2^>nul') do set YA_BAT_PYTHON_PLATFORM=%%P
if not defined YA_BAT_PYTHON_PLATFORM (
call :dbg ---- Not runnable
exit /b 2
)
if not "%YA_BAT_PYTHON_PLATFORM%"=="nt" (
call :dbg ---- Non-windows: %YA_BAT_PYTHON_PLATFORM%
exit /b 3
)
exit /b 0
:dbg
if defined YA_BAT_DEBUG echo [ya.bat] %* 1>&2
exit /b 0
:err
echo [ya.bat] Error: %* 1>&2
exit /b 0