无法从本地 python 脚本在远程服务器上运行 psql 命令

Question

我正在编写一个在本地调用以在远程服务器上运行 psql 查询的脚本。我正在使用 subprocess 模块在 Python 2.7 中编写脚本。我正在使用 subprocess.Popen ssh 进入远程服务器并直接运行 psql 命令。我的本地机器是osx，我认为服务器是CentOS。

当我在本地调用我的脚本时，我收到一条错误消息 psql: command not found。如果我在远程服务器上运行相同的 psql 命令，那么我会得到正确的查询结果。

我怀疑我的 ssh 代码可能有问题，所以我没有将 psql 命令发送到远程服务器，而是尝试发送简单的命令，例如 ls、cd、mkdir，甚至是 @987654326 @。所有这些都运行良好，所以我不认为我的代码存在问题，即 ssh 对远程服务器的命令。

有人了解我的代码出了什么问题以及为什么我要返回psql: command not found？ 我研究并发现了这个较早的 SO 问题，但 psql 已安装在远程服务器上因为我可以在那里手动运行 psql 命令。
Postgresql -bash: psql: command not found


从外部类文件：

def db_cmd(self, query):
    # check that query ends in semicolon
    if query[-1] != ';':
        query = query + ';'

    formatted_query = 'psql ' + self.db_string + \
        ' -c "begin; ' + query + ' ' + self.db_mode + ';"'
    print formatted_query

    ssh = subprocess.Popen(['ssh', self.host, formatted_query],
                           shell=False,
                           stdout=subprocess.PIPE,
                           stderr=subprocess.PIPE)
    result = ssh.stdout.readlines()
    if not result:
        error = ssh.stderr.readlines()
        print >>sys.stderr
        for e in error:
            print 'ERROR: %s' % e,
    else:
        for r in result:
            print r,

本地脚本的摘录，该脚本导入上述类并将 psql 命令发送到远程服务器：

s = Jump(env, db_mode)
s.db_cmd('select * from student.students limit 5;')

在本地运行我的脚本。 请注意，该脚本会打印出 psql 命令以进行调试。如果我复制并粘贴相同的 psql 命令并在远程服务器上运行它，我会得到正确的查询结果。

$ ./script.py 1 PROD  ROLLBACK
psql -h <server_host> -d <db_name> -U <db_user> -p <port> -c "begin; select * from student.students limit 5; ROLLBACK;"

ERROR: bash: psql: command not found

谢谢

Answer 1

当您在远程系统上为交互式会话运行 ssh 时，ssh 会为远程系统请求 PTY（伪 TTY）。在远程系统上运行 ssh 运行命令时，ssh 默认不分配 PTY。

拥有一个 TTY/PTY 会改变你的 shell 初始化自己的方式。您的实际问题是您需要执行以下任何或所有操作才能在远程系统上运行 psql：

• 将一个或多个目录添加到您的命令路径。
• 向您的环境添加一个或多个环境变量。
• 定义一个或多个别名。

您在 shell 启动文件（例如 .bash_profile）中执行这些操作，并且只发生在交互式会话中。当您使用 ssh 远程运行 psql 时，您的 shell 会跳过允许 psql 运行的初始化。

有两种简单的方法可以解决这个问题：

1. 使用“-t”或“-tt”选项运行 ssh，这会导致 ssh 在远程系统上为 psql 分配一个 TTY。
2. 更改远程系统上的 shell 启动文件，以对非交互式会话执行必要的初始化。