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

Posted 2023-03-23

【中文标题】使用 Python 通过 SSH 隧道连接到远程 PostgreSQL 数据库

【英文标题】：Connecting to remote PostgreSQL database over SSH tunnel using Python

【发布时间】：2015-03-01 01:08:29

【问题描述】：

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

【参考方案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)