使用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连接正常工作。

- user3690132

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)

- jsjc