如何使用 Python 通过 SSL 连接到远程 PostgreSQL 数据库

Question

我想通过 Python 连接到远程 PostgreSQL 数据库以进行一些基本的数据分析。该数据库需要 SSL (verify-ca)，以及三个文件（我有）：

服务器根证书文件
客户端证书文件
客户端密钥文件

我找不到描述如何与 Python 建立这种联系的教程。感谢任何帮助。

Answer 1

使用psycopg2模块。

您需要在连接字符串中使用 ssl 选项，或将它们添加为关键字参数：

import psycopg2

conn = psycopg2.connect(dbname='yourdb', user='dbuser', password='abcd1234', host='server', port='5432', sslmode='require')

在这种情况下 sslmode 指定需要 SSL。

要执行服务器证书验证，您可以将 sslmode 设置为 verify-full 或 verify-ca。您需要在 sslrootcert 中提供服务器证书的路径。同时将 sslcert 和 sslkey 值分别设置为您的客户端证书和密钥。

在PostgreSQL中有详细的解释Connection Strings documentation (see also Parameter Key Words) and in SSL Support。

Answer 2

您也可以使用带有 paramiko 和 sshtunnel 的 ssh 隧道：

import psycopg2
import paramiko
from sshtunnel import SSHTunnelForwarder

mypkey = paramiko.RSAKey.from_private_key_file('/path/to/private/key')

tunnel =  SSHTunnelForwarder(
        (host_ip, 22),
        ssh_username=username,
        ssh_pkey=mypkey,
        remote_bind_address=('localhost', psql_port))

tunnel.start()
conn = psycopg2.connect(dbname='gisdata', user=psql_username, password=psql_password, host='127.0.0.1', port=tunnel.local_bind_port)

Answer 3

添加这个是为了完整性，因为我在 SO 上的其他任何地方都找不到它。正如@mhawke 所说，您可以使用 psycopg2，但您也可以使用允许您手动指定数据库 postgresql URI (postgresql://[user[:password]@][netloc][:port][/dbname][?param1=value1&...]) 的任何其他 Python 数据库模块（ORM 等）连接到，因为 psycopg2.connect 用于强制 ssl 连接的 sslmode="require" 参数只是您用来连接到数据库的 postgresql:// URI 的一部分（请参阅 33.1.2. Parameter Key Words）。因此，如果您想使用 sqlalchemy 或其他 ORM 而不是普通的 psycopg2，您可以将所需的 sslmode 添加到数据库 URI 的末尾并以这种方式连接。

import sqlalchemy

DATABASE_URI = "postgresql://postgres:postgres@localhost:5432/dbname"
# sqlalchemy 1.4+ uses postgresql:// instead of postgres://
ssl_mode = "?sslmode=require"
DATABASE_URI += ssl_mode

engine = sqlalchemy.create_engine(URI)
Session = sqlalchemy.orm.sessionmaker(bind=engine)

关于 SSL 支持的 postgres 文档中有一个漂亮的数字 (Table 33.1)，它分解了您可以提供的不同选项。如果您想使用任何需要您指定特定证书路径的高级选项，您可以将其与格式字符串一起放入。

如何使用 Python 通过 SSL 连接到远程 PostgreSQL 数据库

How to connect to a remote PostgreSQL database through SSL with Python

python

postgresql

python-2.7

ssl

python-3.6