本文实例为大家分享了UDP实现文件传输的具体代码，供大家参考，具体内容如下
tcp进行文件传输看这里–python实现TCP文件接发
这里实现的接收方一直接收，发送方每次发送一个文件，方便我在其他函数中调用发送文件。
使用udp 容易出现丢包现象需要处理
要注意 tcp 和udp的套接字不一样
|
1
2
3
4
|
# udp:udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)# tcptcp_socketr = socket.socket(socket.AF_INET,socket.SOCK_STREAM) |
1、发送
|
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
|
# import socket# import tqdm# import os# import threading## # 由客户端向服务器传数据，文件import threadingimport socketimport tqdmimport osimport cv2from time import ctime, sleepdef 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、接收
|
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
|
import socketimport tqdmimport osimport threading# 使用UDP传输视频，全双工，但只需一方接，一方收即可# 设置服务器的ip和 port# 服务器信息# sever_host = '127.0.0.1'# sever_port =1234def 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
