python利用socket实现udp文件传输功能

本文实例为大家分享了UDP实现文件传输的具体代码，供大家参考，具体内容如下

tcp进行文件传输看这里–python实现TCP文件接发

这里实现的接收方一直接收，发送方每次发送一个文件，方便我在其他函数中调用发送文件。

使用udp 容易出现丢包现象需要处理

要注意 tcp 和udp的套接字不一样

# udp:
udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
# tcp
tcp_socketr = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

1、发送

# import socket
# import tqdm
# import os
# import threading
#
# # 由客户端向服务器传数据，文件

import threading
import socket
import tqdm
import os
import cv2
from time import ctime, sleep

def send(address, filename):

# 传输数据间隔符
   SEPARATOR = '<SEPARATOR>'
   # 服务器信息
   host, port = address

# 文件缓冲区
   Buffersize = 4096*10
   # 传输文件名字
   filename = filename
   # 文件大小)
   file_size = os.path.getsize(filename)
   # 创建socket链接
   s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
   print(f'服务器连接中{host}:{port}')
   s.connect((host, port))
   print('与服务器连接成功')

# 发送文件名字和文件大小，必须进行编码处理
   # s.sendto(f'{filename}{SEPARATOR}{file_size}'.encode(), ("127.0.0.1", 1234))
   s.send(f'{filename}{SEPARATOR}{file_size}'.encode('utf-8'))

# 文件传输
   progress = tqdm.tqdm(range(file_size), f'发送{filename}', unit='B', unit_divisor=1024)

with open(filename, 'rb') as f:
       # 读取文件
       for _ in progress:
           bytes_read = f.read(Buffersize)
           # print(bytes_read)
           if not bytes_read:
               print('exit退出传输，传输完毕！')
               s.sendall('file_download_exit'.encode('utf-8'))
               break
           # sendall 确保络忙碌的时候，数据仍然可以传输
           s.sendall(bytes_read)
           progress.update(len(bytes_read))
           sleep(0.001)

# 关闭资源
   s.close()

if __name__ == '__main__':
   address = ('127.0.0.1', 1234)
   # host = '127.0.0.1'
   # port = 1234
   filename = input('请输入文件名：')
   t = threading.Thread(target=send, args=(address, filename))
   t.start()
   # received(address, filename)

2、接收

import socket
import tqdm
import os
import threading

# 使用UDP传输视频，全双工，但只需一方接，一方收即可

# 设置服务器的ip和 port
# 服务器信息
# sever_host = '127.0.0.1'
# sever_port =1234

def recvived(address, port):

# 传输数据间隔符
   SEPARATOR = '<SEPARATOR>'
   # 文件缓冲区
   Buffersize = 4096*10

while True:
       print('准备接收新的文件...')

udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
       udp_socket.bind((address, port))
       recv_data = udp_socket.recvfrom(Buffersize)
       recv_file_info = recv_data[0].decode('utf-8')  # 存储接收到的数据,文件名
       print(f'接收到的文件信息{recv_file_info}')
       c_address = recv_data[1]  # 存储客户的地址信息
       # 打印客户端ip
       print(f'客户端{c_address}连接')
       # recv_data = udp_socket.recv()
       # 接收客户端信息
       # received = udp_socket.recvfrom(Buffersize).decode()
       filename ,file_size = recv_file_info.split(SEPARATOR)
       # 获取文件的名字,大小
       filename = os.path.basename(filename)
       file_size = int(file_size)

# 文件接收处理
       progress = tqdm.tqdm(range(file_size), f'接收{filename}', unit='B', unit_divisor=1024, unit_scale=True)

with open('8_18_'+filename,'wb') as f:
           for _ in progress:
               # 从客户端读取数据

bytes_read = udp_socket.recv(Buffersize)
               # 如果没有数据传输内容
               # print(bytes_read)
               if bytes_read == b'file_download_exit':
                   print('完成传输！')
                   print(bytes_read)
                   break
               # 读取写入
               f.write(bytes_read)
               # 更新进度条
               progress.update(len(bytes_read))
       udp_socket.close()

if __name__ == '__main__':

# address = ("127.0.0.1", 1234)
   port = 1234
   address = "127.0.0.1"
   t = threading.Thread(target=recvived, args=(address, port))
   t.start()
   # send(address)

来源：https://blog.csdn.net/JSSQSC/article/details/120041114