使用 Paramiko 设置 SSH 隧道以访问 PostgreSQL

Question

我目前使用 Paramiko 访问 SFTP 服务器并连接到同一服务器上的 PostgreSQL。我发现许多使用 sshtunnel 登录 PostgreSQL 的示例。但是我不知道用纯Paramiko怎么做。

目前我的代码类似于：

# establish SSH tunnel
self.ssh = paramiko.SSHClient()
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh.connect(hostname=host, username=user, password=password)

# setup SFTP server
self.sftp = self.ssh.open_sftp()

# connect to datebase
self.engine = create_engine('postgres+psycopg2://{}:{}@{}:{}/{}'.format(user, password, host, port, db))

感谢任何建议！

Answer 1

使用 SSH 端口转发。

修改中的数据库隧道代码，您将得到如下代码：

# establish SSH tunnel
self.ssh = paramiko.SSHClient()
# ...
self.ssh.connect(hostname=ssh_host, username=ssh_user, password=ssh_password)

transport = ssh_client.get_transport()
dest_addr = (db_host, db_port)
local_unique_port = 4000 # any unused local port
local_host = 'localhost'
local_addr = (local_host, local_unique_port)
vmchannel = vmtransport.open_channel("direct-tcpip", dest_addr, local_addr)

self.engine = create_engine(
    'postgres+psycopg2://{}:{}@{}:{}/{}'.format(
        db_user, db_password, local_host, local_unique_port, db))

如果 PostgreSQL 数据库在 SSH 服务器本身上运行，那么它通常只会在环回接口上侦听。在那种情况下 db_host 应该设置为 localhost.

不过请注意，sshtunnel 只是 Paramiko 的包装器。所以一般来说，你可以用它来简化代码，除非你有一些限制阻止你安装额外的包。

例如：Connecting to PostgreSQL database through SSH tunneling in Python

基于关于 MongoDB 的相同问题：
Connect and query Mongo database over SSH with private key in Python.

强制性警告：请勿使用 AutoAddPolicy - 您正在失去针对 MITM attacks by doing so. For a correct solution, see Paramiko "Unknown Server".

的保护

使用 Paramiko 设置 SSH 隧道以访问 PostgreSQL

Setup SSH tunnel with Paramiko to access PostgreSQL

python

postgresql

ssh

paramiko

ssh-tunnel