109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
MySQL数据库创建脚本
|
||
"""
|
||
|
||
import pymysql
|
||
import sys
|
||
|
||
def create_database():
|
||
"""创建MySQL数据库"""
|
||
print("=" * 50)
|
||
print("MySQL数据库创建")
|
||
print("=" * 50)
|
||
|
||
try:
|
||
# 连接MySQL服务器（不指定数据库）
|
||
connection = pymysql.connect(
|
||
host='localhost',
|
||
user='root',
|
||
password='123456',
|
||
charset='utf8mb4'
|
||
)
|
||
|
||
cursor = connection.cursor()
|
||
|
||
# 创建数据库
|
||
database_name = 'tsp_assistant'
|
||
cursor.execute(f"CREATE DATABASE IF NOT EXISTS {database_name} CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci")
|
||
print(f"✓ 数据库 {database_name} 创建成功")
|
||
|
||
# 创建用户（可选）
|
||
try:
|
||
cursor.execute("CREATE USER IF NOT EXISTS 'tsp_user'@'localhost' IDENTIFIED BY 'tsp_password'")
|
||
cursor.execute(f"GRANT ALL PRIVILEGES ON {database_name}.* TO 'tsp_user'@'localhost'")
|
||
cursor.execute("FLUSH PRIVILEGES")
|
||
print("✓ 用户 tsp_user 创建成功")
|
||
except Exception as e:
|
||
print(f"⚠ 用户创建失败（可能已存在）: {e}")
|
||
|
||
connection.commit()
|
||
cursor.close()
|
||
connection.close()
|
||
|
||
print("✓ MySQL数据库设置完成")
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"✗ MySQL数据库创建失败: {e}")
|
||
print("\n请检查：")
|
||
print("1. MySQL服务是否已启动")
|
||
print("2. 用户名和密码是否正确")
|
||
print("3. 是否有创建数据库的权限")
|
||
return False
|
||
|
||
def test_connection():
|
||
"""测试数据库连接"""
|
||
print("\n" + "=" * 50)
|
||
print("测试数据库连接")
|
||
print("=" * 50)
|
||
|
||
try:
|
||
connection = pymysql.connect(
|
||
host='localhost',
|
||
user='root',
|
||
password='123456',
|
||
database='tsp_assistant',
|
||
charset='utf8mb4'
|
||
)
|
||
|
||
cursor = connection.cursor()
|
||
cursor.execute("SELECT VERSION()")
|
||
version = cursor.fetchone()
|
||
print(f"✓ MySQL连接成功，版本: {version[0]}")
|
||
|
||
cursor.close()
|
||
connection.close()
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"✗ MySQL连接失败: {e}")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("TSP助手MySQL数据库设置工具")
|
||
print("=" * 50)
|
||
|
||
# 创建数据库
|
||
if create_database():
|
||
# 测试连接
|
||
if test_connection():
|
||
print("\n" + "=" * 50)
|
||
print("MySQL数据库设置成功！")
|
||
print("=" * 50)
|
||
print("现在您可以运行以下命令初始化数据库：")
|
||
print("python init_database.py")
|
||
else:
|
||
print("\n" + "=" * 50)
|
||
print("数据库连接测试失败！")
|
||
print("=" * 50)
|
||
else:
|
||
print("\n" + "=" * 50)
|
||
print("MySQL数据库设置失败！")
|
||
print("=" * 50)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|