# -*- coding:UTF-8 -*-
'''
實(shí)現(xiàn)文件打包、上傳與校驗(yàn)
Created on 2018年1月12日
@author: liuyazhuang
'''
 
from fabric.api import *
from fabric.context_managers import *
from fabric.contrib.console import confirm
 
env.user = 'root'
env.hosts = ['10.2.2.2']
env.password = 'cardio-2017'
 
@task
@runs_once
def tar_task():  #本地打包任務(wù)函數(shù)，只限執(zhí)行一次
  with lcd("/data/logs"):
    local("tar -czf access.tar.gz access.log")
    
 
@task
def put_task():   #上傳文件任務(wù)函數(shù)
  run("mkdir -p /nginx/logs")
  with cd("/nginx/logs"):
    #put(上傳操作)出現(xiàn)異常時(shí)，繼續(xù)執(zhí)行，非終止
    with settings(warn_only = True):
      result = put("/data/logs/access.tar.gz", "/nginx/logs/access.tar.gz")
    if result.failed and not confirm("put file failed, Contiunue[Y/N]?"):
      #出現(xiàn)異常時(shí)，確認(rèn)用戶是否繼續(xù)，（Y繼續(xù)）
      abort("Aborting file put task!")
 
@task
def check_task():  #校驗(yàn)文件任務(wù)函數(shù)
  with settings(warn_only = True):
    #本地local命令需要配置capture=True才能捕獲返回值
    lmd5 = local("md5sum /data/logs/access.tar.gz", capture=True).split(' ')[0]
    rmd5 = run("md5sum /nginx/logs/access.tar.gz").split(' ')[0]
    #對比本地與遠(yuǎn)程文件的md5信息
    if lmd5 == rmd5:
      print "OK";
    else:
      print "ERROR"
 
@task
def execute():   #統(tǒng)一執(zhí)行tar_task()、put_task()、check_task()
  tar_task()
  put_task()
  check_task()

fab -f file_handler.py tar_task  #文件打包操作
fab -f file_handler.py put_task  #文件上傳操作
fab -f file_handler.py check_task #文件校驗(yàn)操作

fab -f file_handler.py execute