python - 使用 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)