堡垒机连接数据库

import pymysql

# 堡垒机连接配置
bastion_host = 'bastion.example.com'
bastion_port = 22
bastion_user = 'bastion_user'
bastion_password = 'bastion_password'

# 数据库连接配置
db_host = 'database.example.com'
db_port = 3306
db_user = 'db_user'
db_password = 'db_password'
db_name = 'db_name'

# 通过堡垒机连接数据库
def connect_to_database():
    # 连接堡垒机
    bastion_conn = pymysql.connect(
        host=bastion_host,
        port=bastion_port,
        user=bastion_user,
        password=bastion_password
    )

    # 通过堡垒机连接数据库
    db_conn = pymysql.connect(
        host=db_host,
        port=db_port,
        user=db_user,
        password=db_password,
        db=db_name,
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor
    )

    return db_conn

# 测试连接
try:
    db_conn = connect_to_database()
    print("成功连接到数据库")
except Exception as e:
    print(f"连接失败: {e}")
finally:
    if 'db_conn' in locals():
        db_conn.close()