Python记11(网络传输大文件

客户端

import socket, tqdm, os

# 传输数据分隔符
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 = ""

# 服务器信息
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()









你可能感兴趣的:(Python,python,网络,服务器)