python 自定义报头 实现大文件传输

import socket
import json
import struct


buffer = 4096
# 每次接收长度
sk = socket.socket()
sk.bind(("127.0.0.1",8080))
sk.listen()
conn, addr = sk.accept()

head_len = conn.recv(4)
# 获取包装后的报头长度
head_len = struct.unpack("i",head_len)[0]
# 解包报头长度
json_head = conn.recv(head_len)
# 接收报头
head = json.loads(json_head)
print(head)
# 反序列化报头
filesize = head["filesize"]
# 获取文件大小
n_filesize = filesize
# 现在剩余未传输文件大小
filename = head["filename"]
# 获取文件名称
with open(filename,"wb") as f:
    # 打开文件
    while True:
        print(str(round(100 - n_filesize/filesize * 100,2)) + "%")
        # 显示进度
        if n_filesize >= buffer:
            # 文件剩余大小大于等于每次传输的大小
            content = conn.recv(buffer)
            # 接收buffer长度的数据
            f.write(content)
            # 将接收的数据写入
            n_filesize -= buffer
            # 文件大小减去本次传输的大小
        else:
            # 文件剩余大小不足buffer
            # 将剩余文件一次性传完
            content = conn.recv(n_filesize)
            # 接收剩余文件
            f.write(content)
            # 将接收的数据写入文件
            print("100%")
            break
conn.close()
sk.close()

import socket
import struct
import json
import os


buffer = 4096
# 每次send发送的大小
sk = socket.socket()
sk.connect(("127.0.0.1",8080))

head = {"filepath":"F:工具",
        "filename":"nox_setup_v6.2.7.1_full.exe",
        "filesize":None
        }
# 自定义报头
filepath = os.path.join(head["filepath"],head["filename"])
# 拼接文件路径
filesize = os.path.getsize(filepath)
head["filesize"] = filesize
n_filesize = filesize
# 获取文件大小
json_head = json.dumps(head).encode("utf8")
# 将报头序列化并转码成utf8
head_len = len(json_head)
# 获取报头文件长度
head_len = struct.pack("i",head_len)
# 将报头长度包装为四个字节
# 参数"i"代表要包装的是数字int类型
sk.send(head_len)
sk.send(json_head)
with open(filepath,"rb") as f:
    # 打开文件
    while True:
        print(str(round(100 - n_filesize/filesize * 100,2)) + "%")
        # 显示进度
        if n_filesize >= buffer:
            # 如果文件还大于buffer，则按buffer大小传输
            content = f.read(buffer)
            # 读取buffer大小的内容
            sk.send(content)
            # 发送文件
            n_filesize -= buffer
            # 文件总大小减去本次发送的大小
        else:
            content = f.read(n_filesize)
            # 读取剩余文件
            sk.send(content)
            # 发送剩余文件
            print("100%")
            break
sk.close()