我如何压缩一个目录，然后返回结果字节数组，而不在磁盘上物理创建zip文件？

import os
import zipfile
from io import BytesIO

def compress_directory(directory_path):
    # 创建内存中的字节流对象
    buffer = BytesIO()

    # 创建ZipFile对象，将其与内存中的字节流对象关联
    with zipfile.ZipFile(buffer, 'w', zipfile.ZIP_DEFLATED) as zipf:
        # 遍历目录中的文件和子目录
        for root, dirs, files in os.walk(directory_path):
            for file in files:
                file_path = os.path.join(root, file)
                # 将文件添加到ZipFile对象中
                zipf.write(file_path, os.path.relpath(file_path, directory_path))

    # 关闭ZipFile对象，确保所有数据都被写入内存中的字节流对象
    buffer.seek(0)

    # 从内存中的字节流对象中获取压缩后的字节数组
    compressed_data = buffer.getvalue()

    return compressed_data