Fabric实现文件打包、上传与校验

一 点睛

我们时常做一些文件包分发的工作，实施步骤一般是先压缩打包，再批量上传至目标服务器，最后做一致性校验。

本实战通过put（）方法实现文件的上传，通过对比本地与远程主机文件的md5，最终实现文件一致性校验。

二 代码

#coding=utf-8
#!/usr/bin/env python
from fabric.api import *
from fabric.context_managers import *
from fabric.contrib.console import confirm

env.user='root'
env.hosts=['192.168.0.120','192.168.0.101']
env.password='123456'


@task                           # 本地打包任务函数，只限执行一次
@runs_once
def tar_task():
    with lcd("/data"):
        local("tar -czf test.tar.gz test.txt")
@task
def put_task():                 # 上传文件任务函数
    run("mkdir -p /data/logs")
    with cd("/data/logs"):
        with settings(warn_only=True):           #put（上传）出现异常时继续执行，不终止
            result = put("/data/test.tar.gz", "/data/logs/test.tar.gz")
        if result.failed and not confirm("put file failed, Continue[Y/N]?"):         #出现异常时，确认用户是否继续，（Y继续）
            abort("Aborting file put task!")

@task
def check_task():            #校验文件任务函数
    with settings(warn_only=True):
        #本地local命令需要配置capture=True才能捕获返回值
        lmd5=local("md5sum /data/test.tar.gz",capture=True).split(' ')[0]
        rmd5=run("md5sum /data/logs/test.tar.gz").split(' ')[0]
    if lmd5==rmd5:
        print "OK"
    else:
        print "ERROR"

@task
def go():
    tar_task()
    put_task()
    check_task()

三 运行结果

[[email protected] pymaintain]# fab -f 7_3_6.py go
[192.168.0.120] Executing task 'go'
[localhost] local: tar -czf test.tar.gz test.txt
[192.168.0.120] run: mkdir -p /data/logs
[192.168.0.120] put: /data/test.tar.gz -> /data/logs/test.tar.gz
[localhost] local: md5sum /data/test.tar.gz
[192.168.0.120] run: md5sum /data/logs/test.tar.gz
[192.168.0.120] out: b93452066c856027f371dd8e9b396187  /data/logs/test.tar.gz
[192.168.0.120] out:

OK
[192.168.0.101] Executing task 'go'
[192.168.0.101] run: mkdir -p /data/logs
[192.168.0.101] put: /data/test.tar.gz -> /data/logs/test.tar.gz
[localhost] local: md5sum /data/test.tar.gz
[192.168.0.101] run: md5sum /data/logs/test.tar.gz
[192.168.0.101] out: b93452066c856027f371dd8e9b396187  /data/logs/test.tar.gz
[192.168.0.101] out:

OK

Done.
Disconnecting from 192.168.0.120... done.
Disconnecting from 192.168.0.101... done.