def get(self, user_input): ''' 客户端下载方法 :param user_input: 用户输入的指令 :return: 无 ''' import os if len(user_input) == 2: # 如果是get xxx,则表示下载到当前目录 dst_path = '.' elif len(user_input) == 3: # 如果是 get xxx xxx,则表示要下载到其他目录 dst_path = user_input[2] else: print(self.__code_list['401']) return None if os.path.isdir(dst_path): # 判断目标文件是否存在 filename = user_input[1] # 获取下载的文件名 tmp_filename = os.path.join(self.__tmp_path, '%s.ftp' % os.path.split(filename)[1]) # 定义临时文件 if os.path.isfile(tmp_filename): # 判断临时文件是否存在(也就是是否需要续传) tmp_file_size = os.path.getsize(tmp_filename) # 获取临时文件的大小 else: tmp_file_size = 0 self.__sk.send( mylib.s2b(r'get|{"filename":"%s", "sendsize": %s}' % (filename, tmp_file_size))) # 向服务端发送下载指令接文件名和已近下载的大小 Confirm = mylib.b2s(self.__sk.recv(100)).split('|') # 获取服务器反馈 if Confirm[0] == 'ready': # 如果是ready则准备下载文件 file_size = json.loads(Confirm[1])['filesize'] # 获取文件大小 md5 = json.loads(Confirm[1])['md5'] # 获取md5 recv_size = tmp_file_size f = open(tmp_filename, 'ab') while file_size != recv_size: # 只要已下载大小不等于文件大小则循环接受数据 try: data = self.__sk.recv(conf.FILE_PER_SIZE) recv_size += len(data) f.write(data) mylib.process_bar(recv_size, file_size) except socket.error as e: print(self.__code_list['306']) f.close() break except IOError as e: print(self.__code_list['305']) f.close() print('') print('Validating...') new_md5 = mylib.get_file_md5(tmp_filename) if new_md5 == md5: # 验证md5,如果验证通过修改成正式文件名,如果没验证过去删除临时文件 import shutil shutil.move(tmp_filename, os.path.join(dst_path, filename)) else: os.remove(tmp_filename) else: print(self.__code_list[Confirm[1]]) else: print(self.__code_list['304'])
def get(self, user_input): ''' 客户端下载方法 :param user_input: 用户输入的指令 :return: 无 ''' import os if len(user_input) == 2: # 如果是get xxx,则表示下载到当前目录 dst_path = '.' elif len(user_input) == 3: # 如果是 get xxx xxx,则表示要下载到其他目录 dst_path = user_input[2] else: print(self.__code_list['401']) return None if os.path.isdir(dst_path): # 判断目标文件是否存在 filename = user_input[1] # 获取下载的文件名 tmp_filename = os.path.join(self.__tmp_path,'%s.ftp' %os.path.split(filename)[1]) # 定义临时文件 if os.path.isfile(tmp_filename): # 判断临时文件是否存在(也就是是否需要续传) tmp_file_size = os.path.getsize(tmp_filename) # 获取临时文件的大小 else: tmp_file_size = 0 self.__sk.send(mylib.s2b(r'get|{"filename":"%s", "sendsize": %s}' %(filename, tmp_file_size))) # 向服务端发送下载指令接文件名和已近下载的大小 Confirm = mylib.b2s(self.__sk.recv(100)).split('|') # 获取服务器反馈 if Confirm[0] == 'ready': # 如果是ready则准备下载文件 file_size = json.loads(Confirm[1])['filesize'] # 获取文件大小 md5 = json.loads(Confirm[1])['md5'] # 获取md5 recv_size = tmp_file_size f = open(tmp_filename, 'ab') while file_size != recv_size: # 只要已下载大小不等于文件大小则循环接受数据 try: data = self.__sk.recv(conf.FILE_PER_SIZE) recv_size += len(data) f.write(data) mylib.process_bar(recv_size, file_size) except socket.error as e: print(self.__code_list['306']) f.close() break except IOError as e: print(self.__code_list['305']) f.close() print('') print('Validating...') new_md5 = mylib.get_file_md5(tmp_filename) if new_md5 == md5: # 验证md5,如果验证通过修改成正式文件名,如果没验证过去删除临时文件 import shutil shutil.move(tmp_filename, os.path.join(dst_path, filename)) else: os.remove(tmp_filename) else: print(self.__code_list[Confirm[1]]) else: print(self.__code_list['304'])
def put(self, instructions): ''' put方法 :param instructions: 客户端指令 :return: 无 ''' import os import json res = json.loads(instructions[1]) file_name = os.path.join(self.__current_path, res['filename']) # 获取上传的文件名 md5 = res['md5'] # 获取上传的文件的md5 tmp_file_name = os.path.join(self.__current_path, '%s.ftp' %res['filename']) # 定义临时文件 if os.path.isfile(tmp_file_name): # 判断临时文件,如果存在说明没传完,将续传 tmp_file_size = os.path.getsize(tmp_file_name) # 获取已经上传的大小 else: tmp_file_size = 0 self.request.send(mylib.s2b(r'ready|{"recv_size":%s}' %tmp_file_size)) # 通知客户端准备上传并返回已近上传的大小 recv_size = tmp_file_size file_size = res['file_size'] f = open(tmp_file_name, 'ab') self.__loger.info('Begin recv file %s from client [%s]' %(file_name, self.client_address)) while recv_size != file_size: # 只要已上传的大小不等于文件的大小就循环获取数据写入文件 try: data = self.request.recv(conf.FILE_PER_SIZE) recv_size += len(data) f.write(data) mylib.process_bar(recv_size, file_size) except socket.error as e: print(self.__code_list['306']) f.close() break except IOError as e: print(self.__code_list['305']) f.close() break f.close() self.__loger.info('End recv file %s from client [%s]' %(file_name, self.client_address)) new_md5 = mylib.get_file_md5(tmp_file_name) # 获取上传后的md5值 self.__loger.info('Begin validate md5 [%s]' %tmp_file_name) if new_md5 == md5: # 验证md5,成功则将临时文件名改为正式文件名 import shutil shutil.move(tmp_file_name, file_name) self.request.sendall(mylib.s2b('308')) self.__loger.info('%s %s' %(self.__code_list['308'], tmp_file_name)) else: os.remove(tmp_file_name) self.request.sendall(mylib.s2b('309')) self.__loger.info('%s %s' %(self.__code_list['309'], tmp_file_name)) self.__loger.info('End validate md5 [%s]' %tmp_file_name)
def put(self, user_input): ''' 客户端上传方法 :param user_input: 指令 :return: 无 ''' import json if len(user_input) == 2: file_name = user_input[1] if not os.path.isfile(file_name): print(self.__code_list['302']) return None self.__sk.sendall(mylib.s2b('checkspacesize')) res = json.loads(mylib.b2s(self.__sk.recv(100))) free_size = int(res['freesize']) * 1024 * 1024 # 获取剩余空间大小 file_size = os.path.getsize(file_name) if free_size > file_size: # 判断剩余空间是否够,够就下载,不够就直接返回 md5 = mylib.get_file_md5(file_name) # 获取上传文件的md5 self.__sk.send( mylib.s2b( r'put|{"filename":"%s", "file_size":%s, "md5":"%s"}' % (os.path.split(file_name)[1], file_size, md5))) # 发送上传指令,并通知服务端文件名,文件大小和md5 res = mylib.b2s(self.__sk.recv(200)).split('|') # 获取服务端确认信息 if res[0] == 'ready': send_size = json.loads(res[1])['recv_size'] # 获取已经上传的大小 f = open(file_name, 'rb') f.seek(send_size) # 设定文件的其实位置为已上传的大小 while file_size != send_size: # 只要文件大小不等于上传的大小则循环读文件并上传数据 if file_size - send_size > conf.FILE_PER_SIZE: data = f.read(conf.FILE_PER_SIZE) send_size += conf.FILE_PER_SIZE else: data = f.read(file_size - send_size) send_size += (file_size - send_size) self.__sk.send(data) mylib.process_bar(send_size, file_size) print(mylib.b2s(self.__sk.recv(200))) print("") else: print(self.__code_list['307']) else: print(self.__code_list['401'])
def put(self, user_input): ''' 客户端上传方法 :param user_input: 指令 :return: 无 ''' import json if len(user_input) == 2: file_name = user_input[1] if not os.path.isfile(file_name): print(self.__code_list['302']) return None self.__sk.sendall(mylib.s2b('checkspacesize')) res = json.loads(mylib.b2s(self.__sk.recv(100))) free_size = int(res['freesize']) * 1024 * 1024 # 获取剩余空间大小 file_size = os.path.getsize(file_name) if free_size > file_size: # 判断剩余空间是否够,够就下载,不够就直接返回 md5 = mylib.get_file_md5(file_name) # 获取上传文件的md5 self.__sk.send(mylib.s2b(r'put|{"filename":"%s", "file_size":%s, "md5":"%s"}' %(os.path.split(file_name)[1], file_size, md5))) # 发送上传指令,并通知服务端文件名,文件大小和md5 res = mylib.b2s(self.__sk.recv(200)).split('|') # 获取服务端确认信息 if res[0] == 'ready': send_size = json.loads(res[1])['recv_size'] # 获取已经上传的大小 f = open(file_name, 'rb') f.seek(send_size) # 设定文件的其实位置为已上传的大小 while file_size != send_size: # 只要文件大小不等于上传的大小则循环读文件并上传数据 if file_size - send_size > conf.FILE_PER_SIZE: data = f.read(conf.FILE_PER_SIZE) send_size += conf.FILE_PER_SIZE else: data = f.read(file_size - send_size) send_size += (file_size - send_size) self.__sk.send(data) mylib.process_bar(send_size, file_size) print(mylib.b2s(self.__sk.recv(200))) print("") else: print(self.__code_list['307']) else: print(self.__code_list['401'])