当前位置:首页 > 行业动态 > 正文

MySQL测试数据库创建成功且源数据库连接成功了吗?

MySQL测试数据库创建成功且源数据库连接成功了吗?  第1张

import mysql.connector
from mysql.connector import Error
def test_database_creation_and_connection(db_config):
    """
    测试数据库是否创建成功以及源数据库连接是否成功。
    :param db_config: 字典,包含数据库连接所需的配置信息,如host, database, user, password等。
    :return: 字典,包含测试结果和相应的信息。
    """
    try:
        # 尝试连接到源数据库
        connection = mysql.connector.connect(**db_config)
        if connection.is_connected():
            cursor = connection.cursor()
            cursor.execute("SHOW DATABASES LIKE %s", (db_config['database'],))
            database_exists = cursor.fetchone()
            if database_exists:
                print("数据库创建成功:", db_config['database'])
            else:
                print("数据库创建失败:", db_config['database'])
            cursor.close()
            connection.close()
            return {
                'database_creation': '成功' if database_exists else '失败',
                'source_connection': '成功'
            }
        else:
            print("无法连接到源数据库。")
            return {
                'database_creation': '未知',
                'source_connection': '失败'
            }
    except Error as e:
        print("数据库连接失败:", e)
        return {
            'database_creation': '未知',
            'source_connection': '失败'
        }
示例数据库配置信息
db_config = {
    'host': 'localhost',
    'database': 'test_db',
    'user': 'root',
    'password': 'your_password'
}
调用函数进行测试
test_results = test_database_creation_and_connection(db_config)
print("测试结果:")
print("数据库创建状态:", test_results['database_creation'])
print("源数据库连接状态:", test_results['source_connection'])

这段代码首先尝试连接到指定的数据库,然后检查该数据库是否存在于数据库列表中,根据这些检查,它会返回一个包含数据库创建状态和源数据库连接状态的字典,您需要替换db_config字典中的your_password为实际的数据库密码。

0