48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
创建数据库脚本
|
|||
|
|
"""
|
|||
|
|
import pymysql
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 添加项目路径
|
|||
|
|
sys.path.insert(0, '/app')
|
|||
|
|
|
|||
|
|
def create_database():
|
|||
|
|
"""创建数据库"""
|
|||
|
|
# 数据库连接信息（从环境变量或直接配置）
|
|||
|
|
host = 'gz-cynosdbmysql-grp-d26pzce5.sql.tencentcdb.com'
|
|||
|
|
port = 24936
|
|||
|
|
user = 'root'
|
|||
|
|
password = '!Rjb12191'
|
|||
|
|
db_name = 'agent_db'
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 连接到MySQL服务器（不指定数据库）
|
|||
|
|
conn = pymysql.connect(
|
|||
|
|
host=host,
|
|||
|
|
port=port,
|
|||
|
|
user=user,
|
|||
|
|
password=password
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
cursor = conn.cursor()
|
|||
|
|
|
|||
|
|
# 创建数据库
|
|||
|
|
cursor.execute(f"CREATE DATABASE IF NOT EXISTS {db_name} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
|
|||
|
|
print(f"✅ 数据库 {db_name} 创建成功")
|
|||
|
|
|
|||
|
|
conn.commit()
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 创建数据库失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
success = create_database()
|
|||
|
|
sys.exit(0 if success else 1)
|