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

本文实例为大家分享了UDP实现文件传输的具体代码,供大家参考,具体内容如下
tcp进行文件传输看这里–python实现TCP文件接发
这里实现的接收方一直接收,发送方每次发送一个文件,方便我在其他函数中调用发送文件。
使用udp 容易出现丢包现象需要处理
要注意 tcp 和udp的套接字不一样

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

【python利用socket实现udp文件传输功能】1、发送
# import socket# import tqdm# import os# import threading## # 由客户端向服务器传数据,文件import threadingimport socketimport tqdmimport osimport cv2from time import ctime, sleepdef send(address, filename):# 传输数据间隔符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 = 1234filename = input('请输入文件名:')t = threading.Thread(target=send, args=(address, filename))t.start()# received(address, filename)

2、接收
import socketimport tqdmimport osimport threading# 使用UDP传输视频,全双工,但只需一方接,一方收即可# 设置服务器的ip和 port# 服务器信息# sever_host = '127.0.0.1'# sever_port =1234def recvived(address, port):# 传输数据间隔符SEPARATOR = ''# 文件缓冲区Buffersize = 4096*10while True:print('准备接收新的文件...')udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)udp_socket.bind((address, port))recv_data = https://www.it610.com/article/udp_socket.recvfrom(Buffersize)recv_file_info = recv_data[0].decode('utf-8')# 存储接收到的数据,文件名print(f'接收到的文件信息{recv_file_info}')c_address = recv_data[1]# 存储客户的地址信息# 打印客户端ipprint(f'客户端{c_address}连接')# recv_data = https://www.it610.com/article/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 = 1234address = "127.0.0.1"t = threading.Thread(target=recvived, args=(address, port))t.start()# send(address)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

    推荐阅读