如何在Python中通过YAML配置文件检查与数据库的连接？

import yaml
import pymysql

def check_database_connection(config_file):
    # 读取YAML配置文件
    with open(config_file, 'r') as f:
        config = yaml.safe_load(f)

    # 解析配置文件中的数据库连接信息
    db_host = config['database']['host']
    db_port = config['database']['port']
    db_user = config['database']['user']
    db_password = config['database']['password']
    db_name = config['database']['name']

    # 建立与数据库的连接
    try:
        conn = pymysql.connect(
            host=db_host,
            port=db_port,
            user=db_user,
            password=db_password,
            database=db_name
        )
        print("数据库连接成功！")
        conn.close()
    except pymysql.Error as e:
        print(f"数据库连接失败：{e}")

# 使用示例
check_database_connection('config.yaml')