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

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

# 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)

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)