Python的pexpect
庫是一個用于自動化交互式命令行應用程序的工具,它允許你編寫腳本來自動執行命令、發送數據、接收輸出,并等待特定的輸出模式。以下是一些使用pexpect
庫的案例:
使用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
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
時,請確保你了解相關的安全風險,并采取適當的措施來保護你的系統和數據。