网络通信Socket模块实现文件传输

本篇给大家介绍网络通信Socket模块实现文件传输
首页 新闻资讯 行业资讯 网络通信Socket模块实现文件传输

 实现的效果如下的Gif所示,就是网络通信Socket模块实现文件下载传输。

服务端

首先需要获取本机ip,这里服务端采用多线程的方法,就是定义一个函数,然后用threading创建任务。客户端连接成功,接收客户端的请求信息,就是下载的文件名。所以需要判断,有输出文件字节数。然后在问用户是不是要下载,得到信息就使用  while True: 读文件的内容,再一个send。看代码是不是就是这么回事。

复制

import socket import os import threading  # 获取本机ip def get_host_ip():     try:         s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)         s.connect(('8.8.8.8', 80))         ip = s.getsockname()[0]     finally:         s.close()      return ip  # 处理客户端请求下载文件的操作(从主线程提出来的代码) def deal_client_request(ip_port, service_client_socket):     # 连接成功后,输出“客户端连接成功”和客户端的ip和端口     print("客户端连接成功", ip_port)     # 接收客户端的请求信息【recv】     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)         #发送和打印文件信息【send】         service_client_socket.send(senddata.encode("utf-8"))         print("请求文件名:%s  文件大小:%.2f MB"%(file_name_data,fmb))         #接受客户是否需要下载【recv】         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                     #progress = cnum/nums*100                      #print("当前已下载:%.2f%%"%progress,end = "\r")                     if file_data:                         # 只要读取到数据,就向客户端进行发送【send】                         service_client_socket.send(file_data)                     # 数据读完,退出循环                     else:                         print("请求的文件数据发送完成")                         break         else:             print("下载取消!")     else:         print("下载的文件不存在!")     # 关闭服务当前客户端的套接字【close】     service_client_socket.close()   if __name__ == '__main__':     # 获取本机ip     print("TCP文件传输服务器,本机IP:" + get_host_ip())          # 把工作目录切换到data目录下     os.chdir("./data")     # 创建套接字【socket】     tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)     # 绑定端口号【bind】     tcp_server_socket.bind(("", 3356))     # 设置监听,将主动套接字变为被动套接字【listen】     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()
  • 1.

  • 2.

  • 3.

  • 4.

  • 5.

  • 6.

  • 7.

  • 8.

  • 9.

  • 10.

  • 11.

  • 12.

  • 13.

  • 14.

  • 15.

  • 16.

  • 17.

  • 18.

  • 19.

  • 20.

  • 21.

  • 22.

  • 23.

  • 24.

  • 25.

  • 26.

  • 27.

  • 28.

  • 29.

  • 30.

  • 31.

  • 32.

  • 33.

  • 34.

  • 35.

  • 36.

  • 37.

  • 38.

  • 39.

  • 40.

  • 41.

  • 42.

  • 43.

  • 44.

  • 45.

  • 46.

  • 47.

  • 48.

  • 49.

  • 50.

  • 51.

  • 52.

  • 53.

  • 54.

  • 55.

  • 56.

  • 57.

  • 58.

  • 59.

  • 60.

  • 61.

  • 62.

  • 63.

  • 64.

  • 65.

  • 66.

  • 67.

  • 68.

  • 69.

  • 70.

  • 71.

  • 72.

  • 73.

  • 74.

  • 75.

  • 76.

  • 77.

  • 78.

  • 79.

  • 80.

  • 81.

  • 82.

  • 83.

  • 84.

  • 85.

  • 86.

  • 87.

  • 88.

客户端

客户端更简单,连接服务端,发送下载文件的请求,定义一个写入的文件夹,就是小儿科东西。不写了,看代码。

复制

# -*- coding:utf-8 -*- # 多任务文件下载器客户端 import socket import os  if __name__ == '__main__':     # 创建套接字【socket】     tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)     # 和服务端连接【connect】     server_ip = input("输入服务器IP:")     tcp_client_socket.connect((server_ip, 3356))     # 发送下载文件的请求     file_name = input("请输入要下载的文件名:")     # 编码     file_name_data = file_name.encode("utf-8")     # 发送文件下载请求数据【send】     tcp_client_socket.send(file_name_data)     # 接收要下载的文件信息【recv】     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     # 是否下载?输入y 确认 输入q 取消     opts = input("是否下载?(y 确认 q 取消)")     if opts == 'q':         print("下载取消!程序退出")     else:         print("正在下载 >>>>>>")         #向服务器确认正在下载【send】         tcp_client_socket.send(b'y')          recvpath = "./receive/"         if not os.path.exists(recvpath):             os.mkdir(recvpath)                   # 把数据写入到文件里         with open(recvpath + file_name, "wb") as file:             #目前接收到的数据包数目             cnum = 0              while True:                 # 循环接收文件数据【recv】                 file_data = tcp_client_socket.recv(1024)                 # 接收到数据                 if file_data:                     # 写入数据                     file.write(file_data)                     cnum = cnum+1                     #progress =cnum/fileszie2*100                     #print("当前已下载:%.2f%%"%progress,end = "\r")                 # 接收完成                 else:                     print("下载结束!")                     break     # 关闭套接字【close】     tcp_client_socket.close()
  • 1.

  • 2.

  • 3.

  • 4.

  • 5.

  • 6.

  • 7.

  • 8.

  • 9.

  • 10.

  • 11.

  • 12.

  • 13.

  • 14.

  • 15.

  • 16.

  • 17.

  • 18.

  • 19.

  • 20.

  • 21.

  • 22.

  • 23.

  • 24.

  • 25.

  • 26.

  • 27.

  • 28.

  • 29.

  • 30.

  • 31.

  • 32.

  • 33.

  • 34.

  • 35.

  • 36.

  • 37.

  • 38.

  • 39.

  • 40.

  • 41.

  • 42.

  • 43.

  • 44.

  • 45.

  • 46.

  • 47.

  • 48.

  • 49.

  • 50.

  • 51.

  • 52.

  • 53.

  • 54.

  • 55.

  • 56.

  • 57.

  • 58.

  • 59.

扩展

如果手机想要下载电脑上的文件(电脑和手机之间),需要确保电脑和手机连接的网络在同一个路由器下

通过ipconfig查看ip

命令行输入python -m http.server 8888来搭建局域网

手机访问192.168.0.101:8888即可下载文件

本文已收录 GitHub,传送门~[1] ,里面更有大厂面试完整考点,欢迎 Star。

Reference

[1]传送门~:

 

https://github.com/MaoliRUNsen/runsenlearnpy100

 

27    2020-11-13 08:30:57    Socket