文件打包为tar包

发布时间 2023-10-16 21:14:38作者: 守望人间

文件打包为tar包

后端需要将传入的文件和本地的文件一同打包为 tar 包,然后下载;一个示例:

# coding: utf-8
import tarfile
import json
import datetime
from io import BytesIO


students = {
    'students': [
        {
            'name': '张三',
            '性别': '男'
        },
        {
            'name': '李四',
            '性别': '男'
        },
    ]
}

bio = BytesIO()  # 创建一个 BytesIO 用来储存 tar 包数据在内存中
tarfile_obj = tarfile.open(fileobj=bio, mode='w')
# 当打包内存中的数据时,需要使用 TarInfo 类来包装
tarinfo = tarfile.TarInfo(name='students.json')
students_file = BytesIO(json.dumps(students, indent=4, ensure_ascii=False).encode(encoding='utf-8'))
tarinfo.size = len(students_file.read())
tarinfo.mtime = int(datetime.datetime.now().timestamp())  # 文件修改时间
students_file.seek(0)
tarfile_obj.addfile(tarinfo, students_file)
# 打包本地文件
tarfile_obj.add('./test.txt', arcname='test.txt')

# 将 bio 保存到本地文件,实际下载应该直接构建 Response 对象返回
with open('test.tar', 'wb') as f:
    f.write(bio.getvalue())