#!/usr/bin/env python
 
import os
import hashlib
import paramiko
import time
import sys
 
def getmd5(filename):
  filehash = hashlib.md5()
  f = open(filename,'rb')
  while True:
    b=f.read(8096)
    if not b:
      break
    filehash.update(b)
  f.close()
  return filehash.hexdigest()
 
def send_files(sftp,ssh,filename,abspathfile,server_dir):
  try:
    print ("删除文件",end=" ")
    print (filename)
     
    ssh.exec_command('bash /root/put_api.sh %s move' % filename)
     
    print ("删除文件成功")
    print ("开始上传文件")
    server_file=server_dir + filename
 
    sftp.put(abspathfile,server_file)
    print ("文件上传成功")
 
  except:
      print ("文件上传失败")
   
def search_file(dir,path,ssh,server_dir):
  #定义排除以md结尾的字符串
  suffix='md'
  #定义循环
 
  print ("开启sftp")
  t = paramiko.Transport(('myhostname',22))
  t.connect(username='root',password='mypassword')
  sftp = paramiko.SFTPClient.from_transport(t)
   
  for filename in os.listdir(path):
 
    print (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
    print ("开始处理 %s" %(filename))
     
    #如果以md结尾，就continue
    if filename.endswith(suffix):
      print ("不处理以md结尾")
      continue
     
    #定义绝对路径
    abspathfile=dir + '\\' + filename
 
    #获取文件的md5
    localfilemd5=getmd5(abspathfile)+'\n'
 
    byte_localfilemd5=localfilemd5.encode(encoding="gb2312")
    print ("%s md5值 %s" %(filename,byte_localfilemd5))
     
    serverfilemd5=get_server_md5(ssh,filename)
 
    print ("%s 服务器md5值 %s" %(filename,serverfilemd5))
     
    if byte_localfilemd5 == serverfilemd5 :
      print ("md5值一致，不需要上传")
    else:
      print ("md5值不一致，上传文件")
 
#    send_files(sftp,ssh,filename,abspathfile,server_dir)
      try:
#        send_files(ssh,filename,abspathfile,server_dir)
        send_files(sftp,ssh,filename,abspathfile,server_dir)
      except:
        print ("上传失败...")
 
  print ("关闭sftp")
  t.close()
 
 
 
def get_server_md5(ssh,filename):
  stdin,stdout,stderr = ssh.exec_command('bash /root/put_api.sh %s' % filename)
  result = stdout.read()
   
  return result
 
def main():
  starttime=time.time()
   
  print ("脚本开始同步")
  print (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
  try:
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(hostname='myhostname',port='22',username='root',password='mypassword')
  except:
    print ("服务器连接失败，异常退出")
    sys.exit(-1)
     
  #定义路径
  local_dir = 'E:\\TEMP\\note_html'
  server_dir= '/root/nginx_02/'
 
  #利用函数排除文件
  search_file(local_dir,local_dir,ssh,server_dir)
 
#  print (get_server_md5(ssh,'d0180727_install_rabbitmt_png_06.png'))
   
  ssh.close()
  endtime=time.time()
  print (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
  print ("服务器断开连接,本次文件更新成功")
  print ("本次更新时间为:%.2f s" %(endtime - starttime))
  print ("更精确的时间:",end=" ")
  usedtime = endtime - starttime
  print (usedtime,end=" ")
  print ("s")
   
   
if __name__ == "__main__" :
  main()

#!/bin/bash 
 
#判断参数的个数是否大于等于1
if [ $# -ge 1 ];then
   
  #定义文件夹
  root_dir="/root/nginx_02/"
  backdir="/root/.backup/"
  abs_filename=$root_dir$1
 
  #判断是否有这个文件
  if [ -f $abs_filename ];then
 
    #利用md5sum工具获取文件的md5值
    getfilemd5=`md5sum $abs_filename | awk '{print $1}'`
    echo "$getfilemd5"
 
    #判断第二个参数是否是move
    if [ 'move' == $2 ];then
      #判断是否存在备份目录，没有则存在，有则挪至备份目录
      if [ -d $backdir ];then
        nowtime=`date +"%F_%H:%M:%S"`
         
        mv $abs_filename $backdir$1_$nowtime
         
        if [ 0 -eq $? ];then
          echo "move successful"
        else
          echo "move failed"
        fi
         
      else
        mkdir -p $backdir
      fi
    fi    
 
  else
    #报错，没有这个文件
    echo "$1 No such file or directory"
  fi
 
else
  #报错，参数错误
  echo "Parameter error"
fi