Python3使用TCP编写一个简易的文件下载器功能

Python  /  管理员 发布于 7年前   228

利用Python3来实现TCP协议，和UDP类似。UDP应用于及时通信，而TCP协议用来传送文件、命令等操作，因为这些数据不允许丢失，否则会造成文件错误或命令混乱。下面代码就是模拟客户端通过命令行操作服务器。客户端输入命令，服务器执行并且返回结果。

TCP（Transmission Control Protocol 传输控制协议）:是一种面向连接的、可靠的、基于字节流的传输层通信协议，由IETF的RFC 793定义。

使用TCP编写一个简易的文件下载器要求：需编写文件下载器服务端和文件下载器客户端

服务器端代码：

import socketimport osimport threading# 处理客户端请求下载文件的操作（从主线程提出来的代码）def deal_client_request(ip_port, service_client_socket):  # 连接成功后，输出“客户端连接成功”和客户端的ip和端口  print("客户端连接成功", ip_port)  # 接收客户端的请求信息  file_name = service_client_socket.recv(1024)  # 解码  file_name_data = file_name.decode("utf-8")  # 判断文件是否存在  if os.path.exists(file_name_data):    #输出文件字节数    fsize = os.path.getsize(file_name_data)    #转化为兆单位    fmb = fsize/float(1024*1024)    #要传输的文件信息    senddata = "文件名：%s 文件大小：%.2fMB"%(file_name_data,fmb)    #发送和打印文件信息    service_client_socket.send(senddata.encode("utf-8"))    print("请求文件名：%s 文件大小：%.2f MB"%(file_name_data,fmb))    #接受客户是否需要下载    options = service_client_socket.recv(1024)    if options.decode("utf-8") == "y":      # 打开文件      with open(file_name_data, "rb") as f:        #　计算总数据包数目        nums = fsize/1024         #　当前传输的数据包数目        cnum = 0        while True:          file_data = f.read(1024)          cnum = cnum + 1          jindu = cnum/nums*100          print("当前已下载：%.2f%%"%jindu,end = "\r")          if file_data:# 只要读取到数据，就向客户端进行发送service_client_socket.send(file_data)          # 数据读完，退出循环          else:print("请求的文件数据发送完成")break    else:      print("下载取消！")  else:    print("下载的文件不存在！")  # 关闭服务当前客户端的套接字  service_client_socket.close()if __name__ == '__main__':  # 把工作目录切换到data目录下  os.chdir("./data")  # 创建套接字  tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 绑定端口号  tcp_server_socket.bind(("", 3356))  # 设置监听，将主动套接字变为被动套接字  tcp_server_socket.listen(128)  # 循环调用accept，可以支持多个客户端同时连接，和多个客户端同时下载文件  while True:    service_client_socket, ip_port = tcp_server_socket.accept()    # 连接成功后打印套接字号    #print(id(service_client_socket))    # 创建子线程    sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))    # 启动子线程    sub_thread.start()

客户端代码：

# 多任务文件下载器客户端import socketif __name__ == '__main__':  # 创建套接字  tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 和服务端连接  server_ip = input("输入服务器IP：")  tcp_client_socket.connect((server_ip, 3356))  # 发送下载文件的请求  file_name = input("请输入要下载的文件名：")  # 编码  file_name_data = file_name.encode("utf-8")  # 发送文件下载请求数据  tcp_client_socket.send(file_name_data)  #　接收要下载的文件信息  file_info = tcp_client_socket.recv(1024)  # 文件信息解码  info_decode = file_info.decode("utf-8")  print(info_decode)  #获取文件大小  fileszie = float(info_decode.split('：')[2].split('MB')[0])  fileszie2 = fileszie*1024  # 是否下载？输入ｙ　确认　输入ｑ 取消  opts = input("是否下载？(y 确认　q 取消)")  if opts == 'q':    print("下载取消！程序退出")  else:    print("正在下载　》》》")    #向服务器确认正在下载    tcp_client_socket.send(b'y')    # 把数据写入到文件里    with open("./" + file_name, "wb") as file:      #目前接收到的数据包数目      cnum = 0      while True:        # 循环接收文件数据        file_data = tcp_client_socket.recv(1024)        # 接收到数据        if file_data:          # 写入数据          file.write(file_data)          cnum = cnum+1          jindu =cnum/fileszie2*100          print("当前已下载：%.2f%%"%jindu,end = "\r")        # 接收完成        else:          print("下载结束！")          break  # 关闭套接字  tcp_client_socket.close()

运行窗口如下：

１）服务器端

２）客户端

注意：客户端和服务器端不要运行在IDLE中，直接使用终端运行。

总结

以上所述是小编给大家介绍的Python3使用TCP编写一个简易的文件下载器功能,希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对站的支持！
如果你觉得本文对你有帮助，欢迎转载，烦请注明出处，谢谢！