python expect案例有哪些

Python的pexpect库是一个用于自动化交互式命令行应用程序的工具，它允许你编写脚本来自动执行命令、发送数据、接收输出，并等待特定的输出模式。以下是一些使用pexpect库的案例：

SSH自动化登录

使用pexpect可以自动登录到远程服务器，执行命令，而无需每次手动输入密码。

import pexpect

def ssh_login(host, user, password):
    ssh = pexpect.spawn(f'ssh {user}@{host}')
    i = ssh.expect(['password:', 'continue connecting (yes/no) ?'])
    if i == 0:
        ssh.sendline('yes')
        ssh.expect('password:')
        ssh.sendline(password)
    ssh.expect('#')  # 假设登录后的提示符是#
    print('登录成功！')
    return ssh

FTP文件传输

pexpect也可以用于自动化FTP文件传输，包括上传和下载文件。

import pexpect

def scp_file(file_path, remote_path, password):
    scp = pexpect.spawn(f'scp {file_path} {remote_path}')
    scp.expect('password:')
    scp.sendline(password)
    scp.expect(pexpect.EOF)  # 等待传输完成
    print('文件传输成功！')

自动化软件安装

在安装软件时，经常需要回答yes/no或输入密码，pexpect可以帮助自动化这个过程。

import pexpect

def install_software(setup_script, password):
    child = pexpect.spawn(setup_script)
    child.expect('Do you agree to the license?')
    child.sendline('yes')
    child.expect('Enter installation path:')
    child.sendline('/opt/software')
    child.expect('Installation complete.')
    child.close()

监控命令行输出

pexpect允许你捕获命令行的输出，这对于监控脚本执行或调试非常有用。

import pexpect

def monitor_command():
    child = pexpect.spawn('some_command')
    index = child.expect(['Step 1 completed.', pexpect.EOF])
    if index == 0:
        print("Before:", child.before.decode())
        print("After:", child.after.decode())
    else:
        print('命令执行完成！')

这些案例展示了pexpect库在自动化日常任务和简化复杂流程方面的强大能力。通过这些案例，你可以看到pexpect如何帮助开发人员提高工作效率，减少重复性工作。

在使用pexpect时，请确保你了解相关的安全风险，并采取适当的措施来保护你的系统和数据。