Python:通过 SSH 隧道连接到 Azure PostgreSQL 实例

Question

我正在尝试使用 Python 连接到 PostgreSQL 实例，该实例通过 SSH 隧道位于 Azure 上。我可以使用 DBeaver 连接到数据库，没有任何问题。这是我正在使用的代码。

from sshtunnel import SSHTunnelForwarder

server = SSHTunnelForwarder(
    ('160.**.**.**', 22),
    ssh_username="*******",
    ssh_password="*******",
    remote_bind_address=('localhost', 5432))

server.start()
print("server connected")

params = {
    'database': '*******',
    'user': '*****postgresadmin@*****dev-postgres',
    'password': '************',
    'host': '**********-postgres.postgres.database.azure.com',
    'port': server.local_bind_port
}

conn = psycopg2.connect(**params)
cur = conn.cursor()

text = "select * from table"
cur.execute(text)

但是我收到以下错误:

conn = _connect(dsn, connection_factory=connection_factory, **kwasync) psycopg2.OperationalError:无法翻译主机名“************-postgres.postgres.database.azure.com “地址:未知主机

我还尝试使用 this 进行 sqlalchemy得到相同的结果。知道我做错了什么吗？我是否需要主机的 IP 地址而不是域？

Answer 1

如果你想在远程服务器上做一些事情，可以使用 SSHTunnelForwarder。

如果您需要使用远程服务器作为连接到另一台服务器的桥梁，则需要另一个代码块:

import sshtunnel

with sshtunnel.open_tunnel(
    (REMOTE_SERVER_IP, 443),
    ssh_username="",
    ssh_password="*******",
    remote_bind_address=(AZURE_SERVER_HOST, 22),
    local_bind_address=('0.0.0.0', 10022)
) as tunnel:
    params = {
        'database': '*******',
        'user': '*****postgresadmin@*****dev-postgres',
        'password': '************',
        'host': '127.0.0.1',
        'port': 10022
    }

    conn = psycopg2.connect(**params)
    cur = conn.cursor()

    text = "select * from table"
    cur.execute(text)