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

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

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

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

服务器端代码：

import socket
import os
import 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 socket
if __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编写一个简易的文件下载器功能,网站的支持！
如果你觉得本文对你有帮助，欢迎转载，烦请注明出处，谢谢！

来源：https://www.linuxidc.com/Linux/2019-05/158530.htm