python使用socket高效传输视频数据帧(连续发送图片)

# 服务器的地址
server_address = ('127.0.0.1', 8000)


def send(dir_name, data_format, file_name):
    # 与接收端建立socket通信
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # AF_INET（TCP/IP – IPv4）协议
    sock.connect(server_address)

    # 每次通信都带一个通信头，表明数据源的类型(红外还是可见光)，要保存数据帧的文件夹名file_name
    # 你可以不要数据格式，这里可以定义成你自己的形式，也算是一种安全机制
    sock.send('{}|{}'.format(data_format, file_name).encode())  # 默认编码 utf-8,发送文件长度和文件名
    reply = sock.recv(1024)

    # 按照文件名排序，0.png,1.png
    file_list = os.listdir(dir_name)
    file_list.sort(key=lambda x: int(x[:-4]))
    if 'ok' == reply.decode():  # 确认一下服务器get到文件长度和文件名数据
        i = 0
        print(len(file_list))
        for file_name in file_list:
            data = file_deal(os.path.join(dir_name, file_name))
            sock.send('{}|{}'.format(len(data), file_name).encode())
            sock.recv(1024)
            go = 0
            total = len(data)
            while go < total:  # 发送文件
                data_to_send = data[go:go + total//2]
                sock.send(data_to_send)
                go += len(data_to_send)
            sock.recv(1024).decode()
            i += 1
            if i < len(file_list):
                sock.send(b'continue')
        sock.send(b'over')
        sock.close()
        sys.exit(0)


def file_deal(file_path):  # 读取文件的方法
    mes = b''
    try:
        file = open(file_path, 'rb')
        mes = file.read()
    except:
        print('error{}'.format(file_path))
    else:
        file.close()
        return mes

LOCAL_IP = '127.0.0.1'  # 本机测试使用ip，局域网中使用需更换ip
PORT = 8000  # 随意指定一个端口

def server():
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # socket.AF_INET 指ipv4  socket.SOCK_STREAM 使用tcp协议
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 设置端口
    sock.bind((LOCAL_IP, PORT))  # 绑定端口
    sock.listen(3)  # 监听端口
    while True:
        sc, sc_name = sock.accept()  # 当有请求到指定端口是 accept()会返回一个新的socket和对方主机的（ip,port）
        print('收到{}机器请求'.format(sc_name))
        info = sc.recv(1024)  # 接受客户端发来的协议头，区分数据源
        # 安全处理：如果不是以这个协议头开始，认为是非法接入，就直接断掉。这里可以自己定义一些安全消息机制
        print(info)
        try:
            data_format, directory_name = info.decode().split("|")
            sc.send(b'ok')  # 表示收到文件长度和文件名
        except:
            print('协议头不对，自动断开连接')
            sc.close()
            continue

        if not os.path.exists(directory_name):
            os.mkdir(directory_name)
        # 协议头正确之后，不断接收发来的数据帧
        while True:
            head_info = sc.recv(1024)
            # print(data_info)
            length, file_name = head_info.decode().split('|')
            sc.send(b'ok')
            if length and file_name:
                print(file_name)
                newfile = open(os.path.join(directory_name, file_name), 'wb')  # 这里可以使用从客户端解析出来的文件名
                file = b''
                total = int(length)
                get = 0
                while get < total:  # 接收文件
                    data = sc.recv(total//2)
                    file += data
                    get = get + len(data)
                sc.send(b'ok')
                print('应该接收{},实际接收{}'.format(length, len(file)))
                if file:
                    print('actually length:{}'.format(len(file)))
                    newfile.write(file[:])
                    newfile.close()
            reply = sc.recv(1024)
            if reply.decode() == "over":
                break

server()