python利用socket實(shí)現(xiàn)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
#
# # 由客戶(hù)端向服務(wù)器傳數(shù)據(jù)，文件

import threading
import socket
import tqdm
import os
import cv2
from time import ctime, sleep

def send(address, filename):

    # 傳輸數(shù)據(jù)間隔符
    SEPARATOR = 'SEPARATOR>'
    # 服務(wù)器信息
    host, port = address

    # 文件緩沖區(qū)
    Buffersize = 4096*10
    # 傳輸文件名字
    filename = filename
    # 文件大小)
    file_size = os.path.getsize(filename)
    # 創(chuàng)建socket鏈接
    s = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    print(f'服務(wù)器連接中{host}:{port}')
    s.connect((host, port))
    print('與服務(wù)器連接成功')

    # 發(fā)送文件名字和文件大小，必須進(jìn)行編碼處理
    # 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'發(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 確保絡(luò)忙碌的時(shí)候，數(shù)據(jù)仍然可以傳輸
            s.sendall(bytes_read)
            progress.update(len(bytes_read))
            sleep(0.001)

    # 關(guān)閉資源
    s.close()


if __name__ == '__main__':
    address = ('127.0.0.1', 1234)
    # host = '127.0.0.1'
    # port = 1234
    filename = input('請(qǐng)輸入文件名：')
    t = threading.Thread(target=send, args=(address, filename))
    t.start()
    # received(address, filename)

import socket
import tqdm
import os
import threading

# 使用UDP傳輸視頻，全雙工，但只需一方接，一方收即可

# 設(shè)置服務(wù)器的ip和 port
# 服務(wù)器信息
# sever_host = '127.0.0.1'
# sever_port =1234

def recvived(address, port):

    # 傳輸數(shù)據(jù)間隔符
    SEPARATOR = 'SEPARATOR>'
    # 文件緩沖區(qū)
    Buffersize = 4096*10

    while True:
        print('準(zhǔn)備接收新的文件...')

        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')  # 存儲(chǔ)接收到的數(shù)據(jù),文件名
        print(f'接收到的文件信息{recv_file_info}')
        c_address = recv_data[1]  # 存儲(chǔ)客戶(hù)的地址信息
        # 打印客戶(hù)端ip
        print(f'客戶(hù)端{(lán)c_address}連接')
        # recv_data = udp_socket.recv()
        # 接收客戶(hù)端信息
        # 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'接收{(diào)filename}', unit='B', unit_divisor=1024, unit_scale=True)

        with open('8_18_'+filename,'wb') as f:
            for _ in progress:
                # 從客戶(hù)端讀取數(shù)據(jù)

                bytes_read = udp_socket.recv(Buffersize)
                # 如果沒(méi)有數(shù)據(jù)傳輸內(nèi)容
                # print(bytes_read)
                if bytes_read == b'file_download_exit':
                    print('完成傳輸！')
                    print(bytes_read)
                    break
                # 讀取寫(xiě)入
                f.write(bytes_read)
                # 更新進(jìn)度條
                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)