• 利用python同步windows和linux文件


    写python脚本的初衷,每次在windows编辑完文件后,想同步到linux上去,只能够登录服务器,然后再利用网络copy,重复性很大,就想着能不能写一个小脚本帮我同步

    逻辑:比对本地和服务器文件的md5,如果md5不一致,则备份服务器上的文件,将本地的给上传上去

    代码分为windows端和服务器端,有些东西,都让python一个做了,写着有点累,就想着,能否服务端提供一个端口,windows去调用这个接口,来完成一部分工作

    python代码如下:

    需要额外安装的包为是paramiko,安装方法:pip install paramiko即可,代码如下,利用paramikoSSHClient来调用服务器自己编写的API(嘿嘿,姑且叫做API)吧,Transport来实现sftp文件的上传,

    #!/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()

    Linux 自己编写的API:

    #!/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
        

    最终实现的功能如下:

  • 相关阅读:
    代码对比软件——code compare
    IAR调试和keil调试的一点小区别
    谷访问歌助手
    X86架构的寄存器
    ant-design-pro
    js 闭包 作用域
    《三体》总结
    如何通过SQL注入盗取数据库信息
    《看见》总结
    《欲望的演化》总结
  • 原文地址:https://www.cnblogs.com/NoneID/p/9434210.html
Copyright © 2020-2023  润新知