当前位置 博文首页 > 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)
