python基于FTP实现文件传输相关功能代码实例

import socketimport jsonimport osimport hashlibCODE = {  '1001':'重新上传文件'}def file_md5(file_path):  obj = open(file_path,'rb')  m = hashlib.md5()  for line in obj:    m.update(line)  obj.close()  return m.hexdigest()def jdt(size,total_size):  '''  显示进度条  '''  val = int(size/total_size * 100)  print('/r%s%%|%s' %(val,'#'*val,),end='')def send_file(exist_size,file_total_size):  '''  发送文件  :param exist_size: 开始读取字节的位置  :param file_total_size: 文件大小  :return:  '''  f = open(file_path, 'rb')  f.seek(exist_size)  send_size = exist_size  while send_size < file_total_size:    data = f.read(1024)    sk.sendall(data)    send_size += len(data)    jdt(send_size,file_total_size)  f.close()  print('上传成功')def upload(file_path):  '''文件上传（含断点）'''  file_path = '111.jpg'  file_md5_val = file_md5(file_path)  file_name = os.path.basename(file_path)  file_size = os.stat(file_path).st_size  cmd_dict = {'cmd': 'upload', 'file_name': file_name, 'size': file_size, 'md5': file_md5_val}  upload_cmd_bytes = json.dumps(cmd_dict).encode('utf8')  sk.sendall(upload_cmd_bytes)  # 2.等待服务端相应  response = json.loads(sk.recv(8096).decode('utf8'))  if response['code'] == 1001:    send_file(0,file_size)  else: # 断点续传    exist_size = response('size')    send_file(exist_size,file_size)sk = socket.socket()sk.connect(('127.0.0.1',8001))while 1:  #upload|文件路径  user_input = input('命令>>>')  # 1、自定义协议{'cmd':'upload','file_name':'...'}  cmd,file_path = user_input.split('|',maxsplit=1)  if cmd == 'upload':    upload(file_path)  elif cmd == 'download':    pass

import socketserverimport jsonimport osimport shutilCODE = {  '1001':'重新上传文件'}def upload(cmd_dict,conn,username):  '''服务端完成上传文件（含断点续传）'''  # 2.获取文件信息  file_md5 = cmd_dict['md5']  file_name = cmd_dict['file_name']  file_md5_path = os.path.join('home', username, file_md5)  file_name_path = os.path.join('home', username, file_name)  upload_file_size = cmd_dict['size']  # 3.判断文件是否存在  exist = os.path.exists(file_md5_path)  if not exist: # 不续传    # 3.1.1通知可以开始上传，已经准备好    response = {'code': 1001}    conn.sendall(json.dumps(response).encode('utf8'))    # 3.1.2接收上传的文件内容    f = open(file_md5_path, 'wb')    recv_size = 0    while recv_size < upload_file_size:      data = conn.recv(1024)      f.write(data) # 写到内存      f.flush() # 刷到硬盘      recv_size += len(data)    f.close()    # 3.1.3改文件名    shutil.move(file_md5_path, file_name_path)  else: # 存在就续传    # 3.2续传 + 大小    exist_size = os.stat(file_md5_path).st_size    response = {'code': 1002, 'size': exist_size}    conn.sendall(json.dumps(response).encode('utf8'))    f = open(file_md5_path, 'ab')    recv_size = exist_size    while recv_size < upload_file_size:      data = conn.recv(1024)      f.write(data) # 写到内存      f.flush() # 刷到硬盘      recv_size += len(data)    f.close()    # 3.1.3改文件名    shutil.move(file_md5_path, file_name_path)class NBServer(socketserver.BaseRequestHandler):  def handle(self):    '''self.request是请求客户端的socket对象'''    #1.接受命令    upload_cmd_bytes = self.request.recv(8096)    cmd_dict = json.loads(upload_cmd_bytes.decode('utf8'))    if cmd_dict['cmd'] == 'upload':      upload(cmd_dict,self.request,'zhangzijian')    elif cmd_dict['cmd'] == 'download':      passif __name__ == '__main__':  server = socketserver.ThreadingTCPServer(('127.0.0.1',8001),NBServer)  server.serve_forever()