pexpect

Pexpect 是 Don Libes 的 Expect 語言的一個 Python 實(shí)現(xiàn)，是一個用來啟動子程序，并使用正則表達(dá)式對程序輸出做出特定響應(yīng)，以此實(shí)現(xiàn)與其自動交互的 Python 模塊。 Pexpect 的使用范圍很廣，可以用來實(shí)現(xiàn)與 ssh、ftp 、telnet 等程序的自動交互；可以用來自動復(fù)制軟件安裝包并在不同機(jī)器自動安裝；還可以用來實(shí)現(xiàn)軟件測試中與命令行交互的自動化。

import pexpect
import sys

child = pexpect.spawn('ssh std20@123.57.211.212')
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()  #匹配系統(tǒng)提示符
s.sendline('whoami')
s.prompt()  
s.logout()

fabric

項(xiàng)目發(fā)布和運(yùn)維的工作相當(dāng)機(jī)械，頻率還蠻高，導(dǎo)致時間浪費(fèi)在敲大量重復(fù)的命令上。

修復(fù)bug什么的，測試，提交版本庫(2分鐘)，ssh到測試環(huán)境pull部署（2分鐘），rsync到線上機(jī)器A,B,C,D,E（1分鐘），分別ssh到ABCDE五臺機(jī)器，逐一重啟(8-10分鐘) = 13-15分鐘

其中郁悶的是，每次操作都是相同的，命令一樣，要命的是在多個機(jī)器上，很難在本機(jī)一個腳本搞定，主要時間都浪費(fèi)在ssh，敲命令上了，寫成腳本，完全可以一鍵執(zhí)行，花兩分鐘看下執(zhí)行結(jié)果。

安裝

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常用參數(shù)

• -l : 顯示定義好的任務(wù)函數(shù)名
• -f : 指定fab入口文件，默認(rèn)入口文件名為fabfile.py
• -H : 指定目標(biāo)主機(jī)，多臺主機(jī)用","號分割

fabric常用API

• local : 執(zhí)行本地命令，如：local('uname -s')
• lcd : 切換本地目錄，如：lcd('/home')
• cd : 切換遠(yuǎn)程目錄，如：cd('/etc')
• run : 執(zhí)行遠(yuǎn)程命令，如：run('free -m')
• sudo : sudo方式執(zhí)行遠(yuǎn)程命令，如：sudo('touch /abc')
• put : 上傳本地文件到遠(yuǎn)程主機(jī)，如：put('/hello', '/home/itcast/hello')
• get : 從遠(yuǎn)程主機(jī)下載文件到本地，如：get('/home/python/world', '/home/itcast/world')
• reboot : 重啟遠(yuǎn)程主機(jī)，如：reboot()
• @task : 函數(shù)裝飾器，標(biāo)識的函數(shù)為fab可調(diào)用的，非標(biāo)記的對fab不可見，純業(yè)務(wù)邏輯
• @runs_once : 函數(shù)裝飾器，標(biāo)識的函數(shù)只會執(zhí)行一次，不受多臺主機(jī)影響

fabric全局屬性設(shè)定

• env.host : 定義目標(biāo)主機(jī)，如：env.host=['192.168.17.192', '192.168.17.193']
• env.user : 定義用戶名，如：env.user="root"
• env.port : 定義目標(biāo)主機(jī)端口，默認(rèn)為22，如：env.port="22"
• env.password : 定義密碼，如：env.password="chuanzhi"
• env.passwords : 不同的主機(jī)不同的密碼，如：env.passwords={'itcast@192.168.17.192:22':'chuanzhi', 'itcast@192.168.17.193:22':'python'}

示例1：動態(tài)獲取遠(yuǎn)程目錄列表

from fabric.api import *

env.hosts=['192.168.17.192', '192.168.17.193']
#env.password='python'
env.passwords = {
    'itcast@192.168.17.192:22':'python',
    'itcast@192.168.17.193: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：上傳文件并執(zhí)行

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()