YuKiKo/get-cookies.bat at main · dwgx/YuKiKo · GitHub

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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
@echo off
setlocal EnableExtensions

set "ROOT_DIR=%~dp0"
cd /d "%ROOT_DIR%"

set "PYTHON_KIND="
if exist "%ROOT_DIR%\.venv\Scripts\python.exe" (
    set "PYTHON_KIND=venv"
    goto python_ready
)

where py >nul 2>nul
if not errorlevel 1 (
    set "PYTHON_KIND=py"
    goto python_ready
)

where python >nul 2>nul
if not errorlevel 1 (
    set "PYTHON_KIND=python"
    goto python_ready
)

echo [ERROR] Python was not found. Install Python or use the repo .venv.
pause
exit /b 1

:python_ready
if not "%~1"=="" goto run_args

echo.
echo ============================================
echo YuKiKo Cookie Launcher
echo ============================================
echo 1. bilibili
echo 2. douyin
echo 3. qzone
echo 4. kuaishou
echo 5. zhihu
echo 6. weibo
echo 7. all
echo.
echo Shortcuts: b=bilibili  d=douyin  q=qzone  k=kuaishou  a=all
set /p SITE_CHOICE=Select site [1-7 or shortcut, default 1]:
if "%SITE_CHOICE%"=="" set "SITE_CHOICE=1"

set "SITE_ARG=--site bilibili"
if "%SITE_CHOICE%"=="2" set "SITE_ARG=--site douyin"
if /I "%SITE_CHOICE%"=="b" set "SITE_ARG=--site bilibili"
if /I "%SITE_CHOICE%"=="d" set "SITE_ARG=--site douyin"
if "%SITE_CHOICE%"=="3" set "SITE_ARG=--site q"
if /I "%SITE_CHOICE%"=="q" set "SITE_ARG=--site q"
if "%SITE_CHOICE%"=="4" set "SITE_ARG=--site kuaishou"
if /I "%SITE_CHOICE%"=="k" set "SITE_ARG=--site kuaishou"
if "%SITE_CHOICE%"=="5" set "SITE_ARG=--site zhihu"
if "%SITE_CHOICE%"=="6" set "SITE_ARG=--site weibo"
if "%SITE_CHOICE%"=="7" set "SITE_ARG=--all"
if /I "%SITE_CHOICE%"=="a" set "SITE_ARG=--all"

echo.
echo Browser:
echo 1. auto
echo 2. edge
echo 3. chrome
echo 4. firefox
set /p BROWSER_CHOICE=Select browser [1-4, default 1]:

set "BROWSER_ARG="
if "%BROWSER_CHOICE%"=="2" set "BROWSER_ARG=--browser edge"
if "%BROWSER_CHOICE%"=="3" set "BROWSER_ARG=--browser chrome"
if "%BROWSER_CHOICE%"=="4" set "BROWSER_ARG=--browser firefox"

set "QR_ARG="
if "%SITE_CHOICE%"=="1" (
    echo.
    set /p USE_QR=Use bilibili QR login? [y/N]:
    if /I "%USE_QR%"=="Y" set "QR_ARG=--qr"
)
if /I "%SITE_CHOICE%"=="b" (
    echo.
    set /p USE_QR=Use bilibili QR login? [y/N]:
    if /I "%USE_QR%"=="Y" set "QR_ARG=--qr"
)

echo.
echo Starting cookie extraction...
echo Output files:
echo   data\cookies\*_cookies.json
echo   data\cookies\*_cookie.txt
echo   data\cookies\*_for_yukiko.yml
echo   data\cookies\yukiko_cookie_import.yml
echo.
call :run_python scripts\get_cookies_windows.py %SITE_ARG% %BROWSER_ARG% %QR_ARG%
set "EXIT_CODE=%ERRORLEVEL%"
echo.
if not "%EXIT_CODE%"=="0" (
    echo [ERROR] Cookie extraction failed with exit code: %EXIT_CODE%
) else (
    echo [OK] Cookie extraction finished.
)
pause
exit /b %EXIT_CODE%

:run_args
call :run_python scripts\get_cookies_windows.py %*
exit /b %ERRORLEVEL%

:run_python
if "%PYTHON_KIND%"=="venv" (
    "%ROOT_DIR%\.venv\Scripts\python.exe" %*
    exit /b %ERRORLEVEL%
)
if "%PYTHON_KIND%"=="py" (
    py -3 %*
    exit /b %ERRORLEVEL%
)
python %*
exit /b %ERRORLEVEL%