十年网站开发经验 + 多家企业客户 + 靠谱的建站团队
量身定制 + 运营维护+专业推广+无忧售后,网站问题一站解决
写python脚本的初衷,每次在windows编辑完文件后,想同步到linux上去,只能够登录服务器,然后再利用网络copy,重复性很大,就想着能不能写一个小脚本帮我同步。
专注于为中小企业提供成都做网站、网站设计服务,电脑端+手机端+微信端的三站合一,更高效的管理,为中小企业元宝免费做网站提供优质的服务。我们立足成都,凝聚了一批互联网行业人才,有力地推动了数千家企业的稳健成长,帮助中小企业通过网站建设实现规模扩充和转变。逻辑:比对本地和服务器文件的md5,如果md5不一致,则备份服务器上的文件,将本地的给上传上去。
代码分为windows端和服务器端,有些东西,都让python一个做了,写着有点累,就想着,能否服务端提供一个端口,windows去调用这个接口,来完成一部分工作。
python代码如下:
需要额外安装的包为是paramiko,安装方法:pip install paramiko即可,代码如下,利用paramiko的SSHClient来调用服务器自己编写的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()