python是個很好玩的東西?好吧我隨口說的,反正因為各種原因(其實到底是啥我也不知道),簡單的學習了下python,然后寫了一個上傳文件上服務器的小玩具練手。
大概功能是這樣:
1、獲取本地文件列表(包括文件夾)
2、檢查服務器上是否存在,不存在直接上傳,存在的話,文件夾無視,文件比較大小,大小不一致則覆蓋,最后檢查服務器上是否存在本地沒有的文件,存在則刪除
3、之后增加了忽略列表,忽略文件類型
4、然后增加了重啟tomcat,但是這個功能未進行測試
大概就是這個樣子,哦了,丟代碼丟代碼
#!/usr/bin/env python3# -*- coding: utf-8 -*-import os import os.path import paramikoimport datetimeimport re# 配置屬性config = { #本地項目路徑 'local_path' : '', # 服務器項目路徑 'ssh_path' : '', # 項目名 'project_name' : '', # 忽視列表 'ignore_list' : [], # ssh地址、端口、用戶名、密碼 'hostname' : '', 'port' : 22, 'username' : '', 'password' : '', # 是否強制更新 'mandatory_update' : False, # 更新完成后是否重啟tomcat 'restart_tomcat' : False, # tomcat bin地址 'tomcat_path' : '', # 被忽略的文件類型 'ignore_file_type_list' : []}# 檢查文件夾是否存在,不存在則創建def check_folder(path): stdin, stdout, stderr = ssh.exec_command('find ' + path) result = stdout.read().decode('utf-8') if len(result) == 0 : print('目錄 %s 不存在,創建目錄' % path) ssh.exec_command('mkdir ' + path) print('%s 創建成功' % path) return 1 else: print('目錄 %s 已存在' % path) return 0# 檢查文件是否存在,不存在直接上傳,存在檢查大小是否一樣,不一樣則上傳def check_file(local_path, ssh_path): # 檢查文件是否存在,不存在直接上傳 stdin, stdout, stderr = ssh.exec_command('find ' + ssh_path) result = stdout.read().decode('utf-8') if len(result) == 0 : sftp.put(local_path,ssh_path) print('%s 上傳成功' % (ssh_path)) return 1 else: # 存在則比較文件大小 # 本地文件大小 lf_size = os.path.getsize(local_path) # 目標文件大小 stdin, stdout, stderr = ssh.exec_command('du -b ' + ssh_path) result = stdout.read().decode('utf-8') tf_size = int(result.split('/t')[0]) print('本地文件大小為:%s,遠程文件大小為:%s' % (lf_size, tf_size)) if lf_size == tf_size: print('%s 大小與本地文件相同,不更新' % (ssh_path)) return 0 else: sftp.put(local_path,ssh_path) print('%s 更新成功' % (ssh_path)) return 1# 上傳流程開始print('上傳開始')begin = datetime.datetime.now()# 文件夾列表folder_list = []# 文件列表file_list = []# ssh上文件列表ssh_file_list = []for parent,dirnames,filenames in os.walk(config['local_path']+config['project_name']): #初始化文件夾列表 for dirname in dirnames: p = os.path.join(parent,dirname) folder_list.append(p[p.find(config['project_name']):]) #初始化文件列表 for filename in filenames: if config['ignore_list'].count(filename) == 0: p = os.path.join(parent,filename) file_list.append(p[p.find(config['project_name']):])print('共有文件夾%s個,文件%s個' % (len(folder_list),len(file_list)))# ssh控制臺ssh = paramiko.SSHClient()ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())ssh.connect(hostname=config['hostname'], port=config['port'], username=config['username'], password=config['password'])# ssh傳輸transport = paramiko.Transport((config['hostname'],config['port']))transport.connect(username=config['username'],password=config['password'])sftp = paramiko.SFTPClient.from_transport(transport)# 檢查根目錄是否存在root_path = config['ssh_path'] + config['project_name']stdin, stdout, stderr = ssh.exec_command('find ' + root_path)result = stdout.read().decode('utf-8')if len(result) == 0 : print('目錄 %s 不存在,創建目錄' % root_path) ssh.exec_command('mkdir ' + root_path) print('%s 創建成功' % root_path)else: print('目錄 %s 已存在,獲取所有文件' % root_path) ssh_file_list = re.split('/n',result)# 檢查文件夾create_folder_num = 0for item in folder_list: target_folder_path = config['ssh_path'] + item create_folder_num = create_folder_num + check_folder(target_folder_path)# 檢查文件update_file_num = 0for item in file_list: if config['ignore_file_type_list'].count(os.path.splitext(item)[1]) == 0: local_file_path = config['local_path'] + item target_file_path = config['ssh_path'] + item if config['mandatory_update']: sftp.put(local_file_path,target_file_path) print('%s 強制更新成功' % (target_file_path)) update_file_num = update_file_num + 1 else: update_file_num = update_file_num + check_file(local_file_path, target_file_path) else: print('%s 在被忽略文件類型中,所以被忽略' % item)# 檢查ssh是否有需要刪除的文件delete_file_num = 0for item in ssh_file_list: temp = item[item.find(config['project_name']):] if folder_list.count(temp) == 0 and file_list.count(temp) == 0 and temp != config['project_name'] and temp != '': print('%s 在本地不存在,刪除' % item) ssh.exec_command('rm -rf ' + item) delete_file_num = delete_file_num + 1end = datetime.datetime.now()print('本次上傳結束:創建文件夾%s個,更新文件%s個,刪除文件%s個,耗時:%s' % (create_folder_num, update_file_num, delete_file_num, end-begin))if config['restart_tomcat']: print('關閉tomcat') ssh.exec_command('sh ' + config['tomcat_path'] + 'shutdown.sh') print('啟動tomcat') ssh.exec_command('sh ' + config['tomcat_path'] + 'startup.sh')# 關閉連接sftp.close()ssh.close()
新聞熱點
疑難解答