使用pexpect检查SSH上的文件是否存在

使用 pexpect 模块可以在 Python 中执行命令并检查其输出。你可以使用 ssh 命令连接到远程服务器，并执行 ls 命令检查文件是否存在。下面我就列举几个我经常遇到的几个错误并做个详细的解决方案。

1、问题背景

用户需要编写一个 Python 脚本，以检查一个文件是否存在于另一台计算机上，该计算机可以通过 SSH 访问。用户已经使用 pexpect 库编写了大部分代码，但需要捕获文件存在与否的值，以便断言文件是否存在。

2、解决方案

提出了以下三种解决方案：

方案 1：检查 SSH 命令的返回码

• 使用 SSH 命令检查文件是否存在，并检查返回码。如果返回码为 0，则文件存在；如果返回码为 1，则文件不存在；如果返回码为 255，则 SSH 连接超时或主机不存在。

方案 2：使用 Paramiko SSH2 模块

• 使用 Paramiko SSH2 模块与远程服务器建立 SFTP 连接，然后使用 stat() 方法检查文件是否存在。如果 stat() 方法引发 IOError，则文件不存在；否则，文件存在。

方案 3：使用 pexpect 库

• 在 pexpect 库的 expect() 方法中，使用 \r\n 换行符来确保命令执行的一致性。

• 定义一个函数 hostFileExists() 或 hostExpect() 来检查文件是否存在，并返回一个值来指示文件是否存在。

任何一种方案都能够解决用户的问题，即检查一个文件是否存在于另一台计算机上，该计算机可以通过 SSH 访问。用户可以选择一种最适合自己情况的方案。

代码例子

方案 1 的代码示例：

import subprocess

def check_file_exists(host, file_path):
    command = f"ssh {host} test -f {file_path}"
    result = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    if result.returncode == 0:
        return True
    elif result.returncode == 1:
        return False
    else:
        raise Exception("SSH connection error or host not found.")

if __name__ == "__main__":
    host = "10.10.0.0"
    file_path = "/opt/ad/bin/email_tidyup.sh"

    if check_file_exists(host, file_path):
        print("File exists.")
    else:
        print("File does not exist.")

方案 2 的代码示例：

import paramiko

def check_file_exists(host, username, password, file_path):
    try:
        transport = paramiko.Transport((host, 22))
        transport.connect(username=username, password=password)
        sftp = paramiko.SFTPClient.from_transport(transport)

        sftp.stat(file_path)
        return True
    except IOError:
        return False
    finally:
        transport.close()

if __name__ == "__main__":
    host = "10.10.0.0"
    username = "service"
    password = "word"
    file_path = "/opt/ad/bin/email_tidyup.sh"

    if check_file_exists(host, username, password, file_path):
        print("File exists.")
    else:
        print("File does not exist.")

方案 3 的代码示例：

import pexpect

def check_file_exists(host, username, password, file_path):
    ssh = pexpect.spawn('ssh %s@%s' % (username, host))
    ssh.expect('password:')
    ssh.sendline(password)
    ssh.expect('> ')
    ssh.sendline('[ ! -e %s ] && echo NO || echo YES' % file_path)
    ssh.expect(['NO', 'YES'])

    if ssh.after == 'YES':
        return True
    else:
        return False


if __name__ == "__main__":
    host = "10.10.0.0"
    username = "service"
    password = "word"
    file_path = "/opt/ad/bin/email_tidyup.sh"

    if check_file_exists(host, username, password, file_path):
        print("File exists.")
    else:
        print("File does not exist.")

请确保替换示例代码中的 hostname、username、password 和 file_path 为实际的值。这段代码会通过 SSH 连接到远程服务器，并执行 ls 命令来检查文件是否存在

如果有啥问题可以这里留言讨论。