pexpect
Pexpect 是 Don Libes 的 Expect 語言的一個 Python 實現,是一個用來啓動子程序,並使用正則表達式對程序輸出做出特定響應,以此實現與其自動交互的 Python 模塊。 Pexpect 的使用範圍很廣,可以用來實現與 ssh、ftp 、telnet 等程序的自動交互;可以用來自動複製軟件安裝包並在不同機器自動安裝;還可以用來實現軟件測試中與命令行交互的自動化。
import pexpect
import sys
child = pexpect.spawn('ssh [email protected]')
child.logfile = sys.stdout
#fout = file('mylog.txt', 'w')
#child.logfile = fout
child.expect('password:')
child.sendline('std20')
child.expect('std20.*')
child.sendline('ls /')
child.expect('std20.*')
child.sendline('exit')
#coding=utf_8
from pexpect import pxssh
import sys
s = pxssh.pxssh()
s.logfile = sys.stdout
hostname = '123.57.211.212'
username = 'std20'
password = 'std20'
s.login(hostname, username, password)
s.sendline('ls /')
s.prompt() #匹配系統提示符
s.sendline('whoami')
s.prompt()
s.logout()
fabric
項目發佈和運維的工作相當機械,頻率還蠻高,導致時間浪費在敲大量重複的命令上。
修復bug什麼的,測試,提交版本庫(2分鐘),ssh到測試環境pull部署(2分鐘),rsync到線上機器A,B,C,D,E(1分鐘),分別ssh到ABCDE五臺機器,逐一重啓(8-10分鐘) = 13-15分鐘
其中鬱悶的是,每次操作都是相同的,命令一樣,要命的是在多個機器上,很難在本機一個腳本搞定,主要時間都浪費在ssh,敲命令上了,寫成腳本,完全可以一鍵執行,花兩分鐘看下執行結果。
安裝
pip install fabric
入門示例
#fabfile.py
from fabric.api import run
def host_type():
run('uname -s')
啓動
itcast@ubuntu:~/tmp/fab$ fab -H 127.0.0.1 host_type
[127.0.0.1] Executing task 'host_type'
[127.0.0.1] run: uname -s
[127.0.0.1] Login password for 'itcast':
[127.0.0.1] out: Linux
[127.0.0.1] out:
Done.
Disconnecting from 127.0.0.1... done.
itcast@ubuntu:~/tmp/fab$ fab -H 127.0.0.1 host_type
[127.0.0.1] Executing task 'host_type'
[127.0.0.1] run: uname -s
[127.0.0.1] Login password for 'itcast':
[127.0.0.1] out: Linux
[127.0.0.1] out:
fabric常用參數
- -l : 顯示定義好的任務函數名
- -f : 指定fab入口文件,默認入口文件名爲fabfile.py
- -H : 指定目標主機,多臺主機用","號分割
fabric常用API
- local : 執行本地命令,如:local('uname -s')
- lcd : 切換本地目錄,如:lcd('/home')
- cd : 切換遠程目錄,如:cd('/etc')
- run : 執行遠程命令,如:run('free -m')
- sudo : sudo方式執行遠程命令,如:sudo('touch /abc')
- put : 上傳本地文件到遠程主機,如:put('/hello', '/home/itcast/hello')
- get : 從遠程主機下載文件到本地,如:get('/home/python/world', '/home/itcast/world')
- reboot : 重啓遠程主機,如:reboot()
- @task : 函數裝飾器,標識的函數爲fab可調用的,非標記的對fab不可見,純業務邏輯
- @runs_once : 函數裝飾器,標識的函數只會執行一次,不受多臺主機影響
fabric全局屬性設定
- env.host : 定義目標主機,如:env.host=['192.168.17.192', '192.168.17.193']
- env.user : 定義用戶名,如:env.user="root"
- env.port : 定義目標主機端口,默認爲22,如:env.port="22"
- env.password : 定義密碼,如:env.password="chuanzhi"
- env.passwords : 不同的主機不同的密碼,如:env.passwords={'[email protected]:22':'chuanzhi', '[email protected]:22':'python'}
示例1:動態獲取遠程目錄列表
from fabric.api import *
env.hosts=['192.168.17.192', '192.168.17.193']
#env.password='python'
env.passwords = {
'[email protected]:22':'python',
'[email protected]:22':'python',
}
@runs_once
def input_raw():
return prompt("please input directory name:", default="/home")
def workask(dirname):
run('ls -l ' + dirname)
@task
def go():
print('start ...')
getdirname = input_raw()
workask(getdirname)
print('end ...')
示例2:上傳文件並執行
from fabric.api import *
env.user = 'itcast'
env.hosts = ['192.168.17.192', '192.168.17.193']
env.password = 'python'
@task
@runs_once
def tar_task():
with lcd('/home/itcast/testdemo'):
local('tar zcvf demo.tar.gz demo.py')
@task
def put_task():
run('mkdir -p /home/itcast/testdemo')
with cd('/home/itcast/testdemo'):
put('/home/itcast/testdemo/demo.tar.gz', '/home/itcast/testdemo/demo.tar.gz')
@task
def check_task():
lmd5 = local('md5sum /home/itcast/testdemo/demo.tar.gz', capture=True).split(' ')[0]
rmd5 = run('md5sum /home/itcast/testdemo/demo.tar.gz').split(' ')[0]
if lmd5 == rmd5:
print('OK ...')
else:
print('ERROR ...')
@task
def run_task():
with cd('/home/itcast/testdemo'):
run('tar zxvf demo.tar.gz')
run('python demo.py')
@task
def go():
tar_task()
put_task()
check_task()
run_task()
代碼自動化部署
from fabric.api import *
env.user = 'itcast'
env.hosts = ['192.168.17.192', '192.168.17.193']
env.password = 'python'
@runs_once
@task
def local_update():
with lcd("/home/itcast/tmp/itcasthello"):
local("git add -A")
local("git commit -m 'update'")
local("git pull origin master")
local("git push origin master")
@task
def remote_update():
with cd("/home/itcast/tmp/itcasthello"):
run("git checkout master")
run("git pull origin master")
@task
def deploy():
local_update()
remote_update()