129 lines
4.0 KiB
Python
129 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
MySQL 初始化模块
|
||
"""
|
||
|
||
import os
|
||
import subprocess
|
||
import shutil
|
||
from pathlib import Path
|
||
import sys
|
||
|
||
# 添加父目录到路径
|
||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||
from log import Logger
|
||
|
||
|
||
def init_mysql(project_root, pre_deploy_commands=None):
|
||
"""
|
||
初始化 MySQL
|
||
|
||
参数：
|
||
project_root: 项目根目录
|
||
pre_deploy_commands: 预部署命令列表（可选）
|
||
|
||
返回：
|
||
bool: 成功返回 True，失败返回 False
|
||
"""
|
||
try:
|
||
project_root = Path(project_root).resolve()
|
||
|
||
Logger.separator()
|
||
Logger.info("开始初始化 MySQL")
|
||
Logger.separator()
|
||
Logger.info(f"项目根目录: {project_root}")
|
||
|
||
# 定义路径
|
||
sql_dir = project_root / "sql"
|
||
target_dir = project_root / "docker" / "mysql" / "db"
|
||
docker_dir = project_root / "docker"
|
||
|
||
# 复制 SQL 脚本
|
||
Logger.info("复制 SQL 脚本到 MySQL 初始化目录")
|
||
Logger.info(f"源目录: {sql_dir}")
|
||
Logger.info(f"目标目录: {target_dir}")
|
||
|
||
target_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 如果提供了预部署命令，从中提取 SQL 文件名
|
||
if pre_deploy_commands:
|
||
for cmd in pre_deploy_commands:
|
||
# 解析命令：cp sql/xxx.sql docker/mysql/db/
|
||
if cmd.startswith('cp ') and '.sql' in cmd:
|
||
parts = cmd.split()
|
||
if len(parts) >= 2:
|
||
source_path = parts[1] # sql/xxx.sql
|
||
sql_file = Path(source_path).name
|
||
source_file = project_root / source_path
|
||
|
||
if source_file.exists():
|
||
Logger.info(f"复制文件: {sql_file}")
|
||
shutil.copy2(source_file, target_dir / sql_file)
|
||
else:
|
||
Logger.warn(f"SQL 文件不存在: {source_file}")
|
||
else:
|
||
# 默认的 SQL 文件列表（向后兼容）
|
||
sql_files = [
|
||
"ry_20250523.sql",
|
||
"ry_config_20250902.sql"
|
||
]
|
||
|
||
for sql_file in sql_files:
|
||
source_file = sql_dir / sql_file
|
||
if source_file.exists():
|
||
Logger.info(f"复制文件: {sql_file}")
|
||
shutil.copy2(source_file, target_dir / sql_file)
|
||
else:
|
||
Logger.warn(f"SQL 文件不存在: {source_file}")
|
||
|
||
Logger.info("目标目录内容:")
|
||
for item in target_dir.iterdir():
|
||
Logger.info(f" - {item.name}")
|
||
|
||
# 构建并启动 MySQL 容器
|
||
Logger.separator()
|
||
Logger.info("构建 MySQL 镜像")
|
||
Logger.separator()
|
||
Logger.info(f"执行目录: {docker_dir}")
|
||
Logger.info("执行命令: docker-compose build --no-cache ruoyi-mysql")
|
||
|
||
result = subprocess.run(
|
||
"docker-compose build --no-cache ruoyi-mysql",
|
||
shell=True,
|
||
cwd=docker_dir,
|
||
capture_output=True,
|
||
text=True
|
||
)
|
||
|
||
if result.returncode != 0:
|
||
Logger.error("MySQL 镜像构建失败")
|
||
if result.stderr:
|
||
Logger.error(f"错误信息: {result.stderr}")
|
||
return False
|
||
|
||
Logger.info("MySQL 镜像构建成功")
|
||
|
||
Logger.info("执行命令: docker-compose up -d ruoyi-mysql")
|
||
result = subprocess.run(
|
||
"docker-compose up -d ruoyi-mysql",
|
||
shell=True,
|
||
cwd=docker_dir,
|
||
capture_output=True,
|
||
text=True
|
||
)
|
||
|
||
if result.returncode != 0:
|
||
Logger.error("MySQL 容器启动失败")
|
||
if result.stderr:
|
||
Logger.error(f"错误信息: {result.stderr}")
|
||
return False
|
||
|
||
Logger.info("MySQL 容器启动成功")
|
||
Logger.info("MySQL 初始化完成")
|
||
return True
|
||
|
||
except Exception as e:
|
||
Logger.error(f"MySQL 初始化异常: {e}")
|
||
return False
|