python实现断文件点续传

#!/usr/bin/env python # --coding = utf-8 # Author Allen Lee import socket, os, json, sys import struct import time import sys class Myclient(): # 在构造方法中启动client的socket对象以及连接server端 def __init__(self): self.client = socket.socket() self.client.connect(('127.0.0.1', 9909)) # main为client的主方法，其中通过对用户输入的命令来判断具体操作，并通过反射将其引导到对应的方法 # !/usr/bin/python3 def clear(self): os.system('cls') def main(self): while True: res_data = self.client.recv(1024).strip() if not res_data: break print(str(res_data, encoding='utf-8')) print('请输入命令 put：发送文件\n') send_data = input('>>: ').strip() if not send_data: continue if hasattr(self, send_data): getattr(self, send_data)(send_data) else: self.client.close() # 当用户输入的不是指定的put方法时，退出 def put(self, send_data): pathh = input('Input the file’s path ：') # 首先判断文件是否存在，通过os.path.isfile来实现 ret = os.path.isfile(pathh) if not ret: print('The file is not exist') return False # 如果存在，则将操作类型、文件名、文件大小以字典形势格式化，再以json的方法进行格式化 else: # 通过os.stat方法来取文件的大小 file_size = os.stat(pathh).st_size file_name = os.path.basename(pathh) self.client.send(bytes(send_data,encoding='utf-8')) time.sleep(1) print(send_data) print(file_name) self.client.send(bytes(file_name, encoding='utf-8')) time.sleep(1) print(file_size) self.client.send(str(file_size).encode('utf-8')) res_tag = str(self.client.recv(1024).decode('utf-8')) print(res_tag) ####打印接受的ok或continue if not res_tag: return False if res_tag.startswith('ok'): # 上传文件只用r就够了 with open(pathh, 'rb') as f: new_size = 0 scale = file_size//1024 while True: try: for i in range(scale + 1): filedata = f.read(1024) # print(filedata) new_size += len(filedata) if not filedata: break self.client.send(filedata) self.progres(new_size, file_size) time.sleep(0.1) except KeyboardInterrupt: self.client.close() if res_tag.startswith('continue'): with open(pathh, 'rb+') as f: # 此处使用seek来进行文件指针的偏移，进而完成断点续传的功能 sendd_data = str(self.client.recv(1024),encoding='utf-8') print(sendd_data) sendd_data1 = int (sendd_data) scale = file_size//1024 print(scale) i= sendd_data1//1024 print(i) new_size = sendd_data1 #start = time.perf_counter() while True: f.seek(sendd_data1, 0) try: for i in range(scale + 1): filedata = f.read(1024) # print(filedata) new_size += len(filedata) if not filedata: break self.client.send(filedata) self.progres(new_size,file_size) except KeyboardInterrupt: self.client.close() #定义实现进度条的两个方法 def humanbytes(self, B): B = float(B) KB = float(1024) MB = float(KB ** 2) # 1,048,576 GB = float(KB ** 3) # 1,073,741,824 if B < KB: return '{0} {1}'.format(B, 'Bytes' if 0 == B > 1 else 'Byte') elif KB <= B < MB: return '{0:.2f} KB'.format(B / KB) elif MB <= B < GB: return '{0:.2f} MB'.format(B / MB) def progres(self, num, Sum): """ 显示上传进度条 num：已上传文件大小 Sum：文件总大小 """ bar_length = 50 # 定义进度条总大小 percent = float(num) / float(Sum) # print(percent) hashes = '=' * int(percent * bar_length) # 进度显示的数量长度百分比 spaces = ' ' * (bar_length - len(hashes)) # 空格的数量=总长度-显示长度 sys.stdout.write("\r传输中: [%s] %d%% %s/%s " % ( hashes + spaces, percent * 100, self.humanbytes(num), self.humanbytes(Sum))) # 输出显示进度条[进度，空白]百分比，已发送数据/文件总大小 sys.stdout.flush() # 强制刷新到屏幕 if __name__ == '__main__': myclient=Myclient() myclient.main()