python通过堡垒机连接数据库

import psycopg2
from sshtunnel import SSHTunnelForwarder

# 堡垒机配置
ssh_host = 'your_bastion_host'
ssh_port = 22
ssh_username = 'your_ssh_username'
ssh_password = 'your_ssh_password'

# 数据库配置
db_host = 'your_database_host'
db_port = 5432
db_name = 'your_database_name'
db_user = 'your_database_user'
db_password = 'your_database_password'

# 创建SSH隧道
with SSHTunnelForwarder(
    (ssh_host, ssh_port),
    ssh_username=ssh_username,
    ssh_password=ssh_password,
    remote_bind_address=(db_host, db_port)
) as tunnel:
    # 数据库连接配置
    db_config = {
        'host': '127.0.0.1',
        'port': tunnel.local_bind_port,
        'database': db_name,
        'user': db_user,
        'password': db_password
    }

    # 连接数据库
    conn = psycopg2.connect(**db_config)
    cursor = conn.cursor()

    # 执行查询
    cursor.execute("SELECT * FROM your_table")
    results = cursor.fetchall()

    # 打印结果
    for row in results:
        print(row)

    # 关闭连接
    cursor.close()
    conn.close()