95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
"""
|
||
使用 Python 执行数据库初始化脚本
|
||
"""
|
||
import pymysql
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 数据库配置
|
||
DB_CONFIG = {
|
||
'host': '10.100.51.51',
|
||
'port': 3306,
|
||
'user': 'root',
|
||
'password': 'Unis@123',
|
||
'charset': 'utf8mb4',
|
||
}
|
||
|
||
def execute_sql_file(sql_file_path):
|
||
"""执行 SQL 文件"""
|
||
try:
|
||
# 读取 SQL 文件
|
||
with open(sql_file_path, 'r', encoding='utf-8') as f:
|
||
sql_content = f.read()
|
||
|
||
# 连接数据库
|
||
print("正在连接数据库...")
|
||
connection = pymysql.connect(**DB_CONFIG)
|
||
|
||
try:
|
||
with connection.cursor() as cursor:
|
||
# 分割SQL语句（简单分割，按分号）
|
||
statements = []
|
||
current_statement = []
|
||
|
||
for line in sql_content.split('\n'):
|
||
# 跳过注释
|
||
line = line.strip()
|
||
if line.startswith('--') or not line:
|
||
continue
|
||
|
||
current_statement.append(line)
|
||
|
||
# 如果行以分号结尾，表示一条完整的语句
|
||
if line.endswith(';'):
|
||
statement = ' '.join(current_statement)
|
||
if statement.strip():
|
||
statements.append(statement)
|
||
current_statement = []
|
||
|
||
# 执行所有语句
|
||
print(f"共 {len(statements)} 条SQL语句...")
|
||
for i, statement in enumerate(statements, 1):
|
||
try:
|
||
cursor.execute(statement)
|
||
# 如果是SELECT语句，获取结果
|
||
if statement.strip().upper().startswith('SELECT'):
|
||
result = cursor.fetchall()
|
||
if result:
|
||
print(f" [{i}] {result}")
|
||
else:
|
||
print(f" [{i}] 执行成功")
|
||
except Exception as e:
|
||
print(f" [{i}] 执行失败: {e}")
|
||
print(f" SQL: {statement[:100]}...")
|
||
|
||
# 提交事务
|
||
connection.commit()
|
||
print("\n✅ 数据库初始化完成！")
|
||
|
||
finally:
|
||
connection.close()
|
||
|
||
except FileNotFoundError:
|
||
print(f"❌ SQL 文件不存在: {sql_file_path}")
|
||
sys.exit(1)
|
||
except pymysql.Error as e:
|
||
print(f"❌ 数据库错误: {e}")
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(f"❌ 未知错误: {e}")
|
||
sys.exit(1)
|
||
|
||
if __name__ == "__main__":
|
||
script_dir = Path(__file__).parent
|
||
sql_file = script_dir / "init_database.sql"
|
||
|
||
print("=" * 60)
|
||
print("NEX Docus 数据库初始化")
|
||
print("=" * 60)
|
||
print(f"SQL 文件: {sql_file}")
|
||
print(f"数据库地址: {DB_CONFIG['host']}:{DB_CONFIG['port']}")
|
||
print("=" * 60)
|
||
print()
|
||
|
||
execute_sql_file(sql_file)
|