Python记11（网络传输大文件

客户端

import socket, tqdm, os

# 传输数据分隔符
separator = "<separator>"

# 服务器信息
host = "192.168.1.103"
port = 5002   # 1~1024多数会被系统占用，不建议用

#文件传输的缓冲区（传输不是一个字节一个字节传，而是一整个buffer）
buffer_size = 1024

# 传输文件
filename = r"C:\Users\ZHUIAO\Desktop\test.xml"

# 文件大小
file_size = os.path.getsize(filename)

# 创建socket连接
s = socket.socket()

# 连接服务器
print(f"服务器连接中{host}:{port}")
s.connect((host, port))
print("与服务器连接成功!")

# 发送文件名字和文件大小，必须进行编码处理encode()
s.send(f"{filename}{separator}{file_size}".encode())

# 文件传输，用tqdm提示发送进度，传输单位Byte（1024表示1Byte=1024bit）
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(buffer_size)    # 一次读取buffer_size大小的
        if not bytes_read:      # 读取不到就退出
            break
        # sendall()可以确保即使网络拥堵，数据仍然可以传输
        s.sendall(bytes_read)
        progress.update(len(bytes_read))    # 按照读的大小 更新进度条

# 关闭资源
s.close()

服务器

import socket, tqdm, os

# 传输数据分隔符
separator = "<separator>"

# 服务器信息
server_host = "192.168.1.103"
server_port = 5002   # 1~1024多数会被系统占用，不建议用

# 文件传输的缓冲区（传输不是一个字节一个字节传，而是一整个buffer）
buffer_size = 4096

# 创建的Server
s = socket.socket()
s.bind((server_host, server_port))  # 服务器绑定端口

# 设置连接监听数
s.listen(5)
print(f"服务器端监听{server_host}:{server_port}")

# 接受客户端连接
client_socket, address = s.accept()     # 哪一个客户的socket, ip地址
print(f"客户端{address}连接。")# 打印客户端ip

# 接收客户端信息
received = client_socket.recv(buffer_size).decode()     # 解码
filename, file_size = received.split(separator)         # 客户端用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(filename, "wb") as f:     # 写入
    for _ in progress:
        # 从客户端读取数据
        bytes_read = client_socket.recv(buffer_size)
        if not bytes_read:      # 读取结束
            break

        f.write(bytes_read)
        progress.update(len(bytes_read))

# 关闭资源，先关闭客户端，再关闭服务器
client_socket.close()
s.close()