使用 Python 通过 SSH 隧道连接到远程 PostgreSQL 数据库

Question

我在使用 SSH 隧道连接到远程数据库时遇到问题（现在我正在尝试使用 Paramiko）。这是我的代码：

#!/usr/bin/env python3
import psycopg2
import paramiko
import time

#ssh = paramiko.SSHClient()
#ssh.load_system_host_keys()
#ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#ssh.connect('pluton.kt.agh.edu.pl', 22, username='aburban', password='pass')

t = paramiko.Transport(('pluton.kt.agh.edu.pl', 22))
t.connect(username="aburban", password='pass') 
c = paramiko.Channel(t)
conn = psycopg2.connect(database="dotest")
curs = conn.cursor()
sql = "select * from tabelka"
curs.execute(sql)
rows = curs.fetchall()
print(rows)

问题是该程序总是尝试连接到本地数据库。我尝试了其他 SSH 隧道，也出现了同样的情况。远程服务器上的数据库存在，并且通过终端使用“经典”SSH 连接可以正常工作。

Answer 1

您可以尝试使用sshtunnel使用 Paramiko 的模块，并且它与 Python 3 兼容。

希望它对你有帮助......我也挠头了一段时间，在 Python 代码中做到这一点并避免 SSH 外部隧道，我们需要感谢将复杂的库包装成简单代码的开发人员！

很简单，从本地端口生成到远程服务器的本地主机中的端口 5432 的隧道，然后使用它通过本地主机连接到远程数据库。

这将是满足您需求的示例工作代码：

#!/usr/bin/env python3
import psycopg2
from sshtunnel import SSHTunnelForwarder
import time

with SSHTunnelForwarder(
         ('pluton.kt.agh.edu.pl', 22),
         ssh_password="password",
         ssh_username="aburban",
         remote_bind_address=('127.0.0.1', 5432)) as server:

    conn = psycopg2.connect(database="dotest",port=server.local_bind_port)
    curs = conn.cursor()
    sql = "select * from tabelka"
    curs.execute(sql)
    rows = curs.fetchall()
    print(rows)