python+rsync正確な同期指定フォーマットファイル

4097 ワード

       :http://blog.csdn.net/gcogle/article/details/52767113


# coding: utf-8
#!/usr/bin/env python
'''
updatedb           /home/upload/de locate.dbz   
  locate                     locate.src
  paramiko                        locate.dst
         locate.diff,  rsync    locate.diff      
      
'''
import paramiko
import commands
import os
import logging
logging.basicConfig(level=logging.DEBUG,
                    format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                    datefmt='%Y-%m-%d %H:%M:%S',
                    filename='/var/log/upload_to_chengdu.log',
                    filemode='a')
logging.info("###################################################################")
hostname = '8.8.8.8'
username = 'username'
password = 'password'
port = 22
locate_lists = ['/tmp/locate.src', '/tmp/locate.dst', '/tmp/locate.diff']
dict_path = {'locate_lists': locate_lists, 'dst_db': '/tmp/locate.db',
             'src_db': '/tmp/locate.db', 'dst_path': '/logs/rsync_log/',
             'src_path': '/home/upload/', 'passwd_file': '/etc/rsync.pas'}
def check_file():
    for l in dict_path['locate_lists']:
        if os.path.exists(l):
            commands.getstatusoutput('sudo rm %s' % l)
def get_dst_list():
    s = paramiko.SSHClient()
    s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    s.connect(hostname=hostname, port=port, username=username, password=password)
    cmd_create = "sudo updatedb -U %s -o %s && sudo locate -d %s --regex '.*[0-9]{4}([-]{0,1}[0-9]{2}){2}\..*gz$' >>%s" % (
        dict_path['dst_path'], dict_path['dst_db'], dict_path['dst_db'], dict_path['locate_lists'][1])
    cmd_delete = 'sudo rm ' + dict_path['locate_lists'][1]
    stdin, stdout, stderr = s.exec_command(cmd_create)
    # print stdout.read()
    try:
        t = paramiko.Transport((hostname, port))
        t.connect(username=username, password=password)
        sftp = paramiko.SFTPClient.from_transport(t)
        sftp.get(dict_path['locate_lists'][1], dict_path['locate_lists'][1])
        t.close()
    except Exception, e:
        print e
    stdin, stdout, stderr = s.exec_command(cmd_delete)
    s.close()
def get_src_list():
    cmd = "sudo updatedb -U %s -o %s && sudo locate -d %s --regex '.*[0-9]{4}([-]{0,1}[0-9]{2}){2}\..*gz$' >>%s" % (
        dict_path['src_path'], dict_path['src_db'], dict_path['src_db'], dict_path['locate_lists'][0])
    commands.getstatusoutput(cmd)
def cmp_diff():
    f1 = open(dict_path['locate_lists'][0], 'r')
    f2 = open(dict_path['locate_lists'][1], 'r')
    f3 = open(dict_path['locate_lists'][2], 'a')
    x = f1.readlines()
    y = f2.readlines()
    f1.close()
    f2.close()
    for i in x:
        j = dict_path['dst_path'] + i[13:]
        if j not in y:
            f3.writelines(i)
    f3.close()
def rsync_file():
    f = open(dict_path['locate_lists'][2], 'r')
    for i in f.readlines():
        cmd = 'cd %s && sudo rsync -Rvz %s upload@%s::aliyun_log --password-file=%s' % (
            dict_path['src_path'], i.strip('
')[13:], hostname, dict_path['passwd_file']) logging.info(cmd) commands.getstatusoutput(cmd) f.close() def del_locate_file(): cmd = 'sudo rm {%s,%s,%s}' % (dict_path['locate_lists'][1], dict_path['locate_lists'][0], dict_path['locate_lists'][2]) commands.getstatusoutput(cmd) if __name__ == '__main__': check_file() get_dst_list() get_src_list() cmp_diff() rsync_file() del_locate_file()