【python】后端文件流下载数据

后台实现下载接口

1.1 通过文件流下载

import os
import time
from django.http import StreamingHttpResponse


def download_file_blob(name, url):
    """
    :param name: 文件名称(带后缀)
    :param url: 文件路径
    :return: 解析后数据
    """
    # 文件读取
    # chunk_size 单次读写大小
    def file_iterator(file_name, chunk_size=5120):
        with open(file_name, 'rb') as file:
            while True:
                char_buffer = file.read(chunk_size)
                if char_buffer:
                    yield char_buffer
                else:
                    break
                # 下载限速
                # time.sleep(0.001)

    if os.path.exists(url):
        response = StreamingHttpResponse(file_iterator(url))
        response['Content-Type'] = 'application/octet-stream'
        response['Content-Disposition'] = f'attachment;filename={name}'
        # 设置下载文件总大小
        file_info = os.stat(url)
        response['Content-Length'] = file_info.st_size
        return response

    return None