如何在Python中调用pg_restore时自动传递密码？

我来帮你搞定这个困扰！你之前尝试通过stdin传递密码却失败，核心原因是pg_restore（以及PostgreSQL的其他客户端工具，比如psql、pg_dump）默认不会从标准输入读取密码——它的密码提示是直接和终端（tty）交互的，绕过了你通过Popen创建的stdin管道，所以不管你往里面写\r还是\n都没用。

下面给你两种官方推荐的可靠解决方案：

方法一：使用PGPASSWORD环境变量 #

PostgreSQL客户端工具会自动读取PGPASSWORD环境变量的值作为密码，完全不需要手动输入。在Python里可以这样实现：

import os
from subprocess import PIPE, Popen
import shlex

# 临时设置环境变量（仅当前进程有效）
os.environ['PGPASSWORD'] = password

# 构造pg_restore命令
command = f'pg_restore ' \
          f'-h {host} ' \
          f'-d {database} ' \
          f'-U {username} ' \
          f'{file_path}'
command = shlex.split(command)

# 执行命令
p = Popen(command, shell=False, stdin=PIPE, stdout=PIPE, stderr=PIPE)
stdout, stderr = p.communicate()

# 可选：执行完后清除环境变量，避免密码泄露
if 'PGPASSWORD' in os.environ:
    del os.environ['PGPASSWORD']

这种方法简单直接，适合临时场景，密码只会在当前Python进程的环境中存在，不会暴露在命令行里。

方法二：使用.pgpass配置文件（更安全） #

如果你需要长期复用密码，或者不想让密码出现在环境变量中，推荐使用PostgreSQL的.pgpass配置文件。这个文件可以存储不同主机、数据库、用户对应的密码，且必须设置严格的权限（仅自己可读），否则PostgreSQL会忽略它。

import os
from subprocess import PIPE, Popen
import shlex

# 配置.pgpass文件路径（默认在用户根目录下）
pgpass_path = os.path.expanduser('~/.pgpass')

# 创建或更新.pgpass文件
with open(pgpass_path, 'a') as f:
    # 格式：hostname:port:database:username:password
    # 注意：如果已有相同条目，会重复添加，你可以先检查再写入
    f.write(f'{host}:5432:{database}:{username}:{password}\n')

# 必须设置权限为600（只有所有者可读可写）
os.chmod(pgpass_path, 0o600)

# 构造并执行pg_restore命令
command = f'pg_restore ' \
          f'-h {host} ' \
          f'-d {database} ' \
          f'-U {username} ' \
          f'{file_path}'
command = shlex.split(command)

p = Popen(command, shell=False, stdin=PIPE, stdout=PIPE, stderr=PIPE)
stdout, stderr = p.communicate()

这种方法更安全，因为密码存储在文件中，且权限严格，不会在进程环境或命令行中暴露。

为什么你之前的方法无效？ #

你尝试的shell=True报错，是因为shlex.split已经把命令拆分成了列表，再用shell=True会导致命令解析错误（把整个列表当成一个命令参数）。而通过stdin传递密码的思路本身就行不通，因为pg_restore的密码输入是直接读取终端输入，而非标准输入管道。

内容的提问来源于stack exchange，提问作者Nathan