如何使用mozilla/zipwriter将文件压缩为带有密码zip

import zipfile
import os
from io import BytesIO
from getpass import getpass

def compress_with_password(source_file, password, output_file):
    # 创建一个内存中的zip文件
    zip_buffer = BytesIO()

    # 创建一个ZipFile对象，指定压缩文件对象和模式为写入
    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED, True) as zip_file:
        # 将源文件添加到zip文件中，并设置密码
        zip_file.setpassword(password.encode())
        zip_file.write(source_file, os.path.basename(source_file))

    # 将压缩后的zip文件写入到输出文件中
    with open(output_file, "wb") as file:
        file.write(zip_buffer.getvalue())

source_file = input("请输入要压缩的文件路径：")
password = getpass("请输入要设置的密码：")
output_file = input("请输入输出的zip文件路径：")

compress_with_password(source_file, password, output_file)