mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-20 20:39:55 +00:00
zhangtao
4254a7951f
新增 Windows 批处理脚本 run_win.bat 和 Linux shell 脚本 run_linux.sh，提供跨平台支持 移除旧版 run.sh 脚本，统一使用平台专用脚本
268 lines
6.2 KiB
Batchfile
Executable File
268 lines
6.2 KiB
Batchfile
Executable File
@echo off
|
|
setlocal EnableDelayedExpansion
|
|
|
|
set "SCRIPT_DIR=%~dp0"
|
|
set "REPO_ROOT=%SCRIPT_DIR%.."
|
|
|
|
set "tty_red="
|
|
set "tty_green="
|
|
set "tty_yellow="
|
|
set "tty_blue="
|
|
set "tty_cyan="
|
|
set "tty_purple="
|
|
set "tty_bold="
|
|
set "tty_reset="
|
|
|
|
set "RED=[0;31m"
|
|
set "GREEN=[0;32m"
|
|
set "YELLOW=[33m"
|
|
set "BLUE=[0;34m"
|
|
set "CYAN=[0;36m"
|
|
set "LIGHT_GRAY=[0;37m"
|
|
set "PURPLE=[0;35m"
|
|
set "BOLD=[1m"
|
|
set "RESET=[0m"
|
|
|
|
set "ERROR_COUNT=0"
|
|
|
|
goto :main_menu
|
|
|
|
:info
|
|
echo !tty_green!OK: %~1!tty_reset!
|
|
goto :eof
|
|
|
|
:warn
|
|
echo !tty_yellow!WARNING: %~1!tty_reset!
|
|
goto :eof
|
|
|
|
:error
|
|
echo !tty_red!ERROR: %~1!tty_reset!
|
|
goto :eof
|
|
|
|
:pause
|
|
pause
|
|
goto :eof
|
|
|
|
:print_separator
|
|
echo !LIGHT_GRAY!---------------------------------------------------------------!RESET!
|
|
goto :eof
|
|
|
|
:show_banner
|
|
echo.
|
|
echo Welcome to FastAPI Init Script
|
|
echo Version: 1.0.0
|
|
echo Author: coderxslee
|
|
echo.
|
|
goto :eof
|
|
|
|
:load_db_config
|
|
set "env_file=%SCRIPT_DIR%env\.env.dev"
|
|
|
|
if not exist "%env_file%" (
|
|
call :error "Not found: %env_file%"
|
|
exit /b 1
|
|
)
|
|
|
|
for /f "usebackq tokens=1,* delims==" %%a in ("%env_file%") do (
|
|
set "line=%%a"
|
|
set "value=%%b"
|
|
set "line=!line:~0,13!"
|
|
if "!line!"=="DATABASE_HOST" set "DATABASE_HOST=%%b"
|
|
if "!line!"=="DATABASE_PORT" set "DATABASE_PORT=%%b"
|
|
if "!line!"=="DATABASE_USER" set "DATABASE_USER=%%b"
|
|
if "!line!"=="DATABASE_PASSWORD" set "DATABASE_PASSWORD=%%b"
|
|
if "!line!"=="DATABASE_NAME" set "DATABASE_NAME=%%b"
|
|
if "!line!"=="DATABASE_TYPE" set "DATABASE_TYPE=%%b"
|
|
)
|
|
|
|
if "!DATABASE_HOST!"=="" (
|
|
call :error "Database configuration incomplete"
|
|
exit /b 1
|
|
)
|
|
exit /b 0
|
|
|
|
:start_dev_server
|
|
call :print_separator
|
|
echo Starting dev server...
|
|
call :load_db_config
|
|
if errorlevel 1 exit /b 1
|
|
|
|
echo Checking database...
|
|
echo Database host: !DATABASE_HOST!
|
|
echo Database port: !DATABASE_PORT!
|
|
echo Database user: !DATABASE_USER!
|
|
echo Database name: !DATABASE_NAME!
|
|
|
|
call :print_separator
|
|
cd /d "%SCRIPT_DIR%"
|
|
uv run main.py run --env=dev
|
|
exit /b 0
|
|
|
|
:create_migration
|
|
call :print_separator
|
|
echo Generating migration files...
|
|
cd /d "%SCRIPT_DIR%"
|
|
uv run main.py revision --env=dev
|
|
exit /b 0
|
|
|
|
:apply_migration
|
|
call :print_separator
|
|
echo Applying migrations...
|
|
cd /d "%SCRIPT_DIR%"
|
|
uv run main.py upgrade --env=dev
|
|
exit /b 0
|
|
|
|
:reset_migration_records
|
|
call :print_separator
|
|
echo Resetting migration records...
|
|
echo WARNING: This will reset migration records in the database!
|
|
set /p confirm="Continue? (y/N): "
|
|
if /i not "!confirm!"=="y" (
|
|
echo Operation cancelled
|
|
exit /b 0
|
|
)
|
|
|
|
call :load_db_config
|
|
if errorlevel 1 exit /b 1
|
|
|
|
cd /d "%SCRIPT_DIR%"
|
|
echo Resetting migration records...
|
|
echo DELETE FROM alembic_version; | PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d %DATABASE_NAME%
|
|
exit /b 0
|
|
|
|
:clean_database
|
|
call :print_separator
|
|
echo Cleaning database...
|
|
echo WARNING: This will delete all tables in the database!
|
|
set /p confirm="Continue? (y/N): "
|
|
if /i not "!confirm!"=="y" (
|
|
echo Operation cancelled
|
|
exit /b 0
|
|
)
|
|
|
|
call :load_db_config
|
|
if errorlevel 1 exit /b 1
|
|
|
|
cd /d "%SCRIPT_DIR%"
|
|
echo Cleaning database, dropping all tables...
|
|
echo DROP SCHEMA public CASCADE; CREATE SCHEMA public; | PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d %DATABASE_NAME%
|
|
exit /b 0
|
|
|
|
:drop_database
|
|
call :print_separator
|
|
echo Deleting database...
|
|
set /p db_name="Enter database name: "
|
|
|
|
if "!db_name!"=="" (
|
|
call :error "Database name cannot be empty"
|
|
exit /b 1
|
|
)
|
|
|
|
echo WARNING: This will permanently delete database '!db_name!' and all its data!
|
|
echo This action cannot be undone!
|
|
|
|
set /p confirm1="Confirm deletion of '!db_name!'? (y/N): "
|
|
if /i not "!confirm1!"=="y" (
|
|
echo Operation cancelled
|
|
exit /b 0
|
|
)
|
|
|
|
set /p confirm2="Enter 'DELETE' to confirm: "
|
|
if not "!confirm2!"=="DELETE" (
|
|
echo Operation cancelled
|
|
exit /b 0
|
|
)
|
|
|
|
call :load_db_config
|
|
if errorlevel 1 exit /b 1
|
|
|
|
cd /d "%SCRIPT_DIR%"
|
|
echo Dropping database '!db_name!'...
|
|
echo SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = '!db_name!' AND pid <> pg_backend_pid(); | PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d postgres
|
|
echo DROP DATABASE IF EXISTS "!db_name!"; | PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d postgres
|
|
exit /b 0
|
|
|
|
:init_sql_data
|
|
call :print_separator
|
|
echo Initializing SQL data...
|
|
|
|
set "sql_dir=%REPO_ROOT%\backend\sql\postgres\init_data"
|
|
|
|
if not exist "%sql_dir%" (
|
|
call :error "SQL directory not found: %sql_dir%"
|
|
exit /b 1
|
|
)
|
|
|
|
echo Available SQL files:
|
|
echo 0. Execute all SQL files
|
|
|
|
set "i=0"
|
|
set "file_count=0"
|
|
for %%f in ("%sql_dir%\*.sql") do (
|
|
set /a file_count+=1
|
|
echo !file_count!. %%~nxf
|
|
)
|
|
|
|
set /p choice="Select SQL to initialize (enter number): "
|
|
|
|
call :load_db_config
|
|
if errorlevel 1 exit /b 1
|
|
|
|
if "!choice!"=="0" (
|
|
echo Executing all SQL files...
|
|
for %%f in ("%sql_dir%\*.sql") do (
|
|
echo Executing: %%~nxf
|
|
PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d %DATABASE_NAME% -f "%%f"
|
|
)
|
|
) else (
|
|
set /a index=choice-1
|
|
set "selected_file="
|
|
set "j=0"
|
|
for %%f in ("%sql_dir%\*.sql") do (
|
|
if "!j!"=="!index!" (
|
|
set "selected_file=%%f"
|
|
)
|
|
set /a j+=1
|
|
)
|
|
if "!selected_file!"=="" (
|
|
call :error "Invalid selection"
|
|
exit /b 1
|
|
)
|
|
echo Executing: !selected_file!
|
|
PGPASSWORD=%DATABASE_PASSWORD% psql -h %DATABASE_HOST% -p %DATABASE_PORT% -U %DATABASE_USER% -d %DATABASE_NAME% -f "!selected_file!"
|
|
)
|
|
exit /b 0
|
|
|
|
:main_menu
|
|
cls
|
|
call :show_banner
|
|
echo Please select an operation:
|
|
echo.
|
|
echo 1. Start dev server (uv run dev)
|
|
echo 2. Generate migration files
|
|
echo 3. Apply migrations
|
|
echo 4. Reset migration records
|
|
echo 5. Clean database (drop all tables)
|
|
echo 6. Delete database
|
|
echo 7. Initialize SQL data
|
|
echo 0. Exit
|
|
echo.
|
|
|
|
set /p option="Select operation: "
|
|
|
|
if "!option!"=="1" call :start_dev_server
|
|
if "!option!"=="2" call :create_migration
|
|
if "!option!"=="3" call :apply_migration
|
|
if "!option!"=="4" call :reset_migration_records
|
|
if "!option!"=="5" call :clean_database
|
|
if "!option!"=="6" call :drop_database
|
|
if "!option!"=="7" call :init_sql_data
|
|
if "!option!"=="0" exit /b 0
|
|
|
|
call :pause
|
|
goto :main_menu
|
|
|
|
:end
|
|
endlocal
|
|
exit /b 0
|