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

时间:2022-05-23 11:32:17

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

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

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

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

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

?
1
2
3
4
# udp:
udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
# tcp
tcp_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 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、接收

?
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
 
# 使用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