业务背景

• 近期用Python-Flask开发项目遇到一个多文件打包下载的业务接口，文件格式有Excel、图片文件、视频文件打包成zip下载。
• 前端通过接口发送ID，后端通过ID查询数据库相关数据并生成Excel文档、数据库查出图片和视频地址之后下载到服务器、打包成压缩包返回给前端。

实现思路

1. 后端根据ID创建文件夹到服务器目录
2. 查找数据库数据生成Excel文件，并存储到第一步生成的文件夹内
3. 根据数据库查找到的文件(图片、视频)URL地址下载文件，并存储到第一步生成的文件夹内
4. 将文件夹打包zip
5. 返回给前端

项目代码

• 项目基于Flask
• 篇幅原因只贴出本下载接口业务相关代码

# 添加下载API接口
# Download
api.add_resource(Download, "/activities/download/<int:activities_id>")

• 模型类：

# coding: utf-8
# extensions 为自定义模块
from extensions import db
import datetime

class Activities(db.Model):
    """
    活动
    """
    __tablename__ = 'activities'
    id = db.Column(db.Integer, primary_key=True)
    # 活动名称
    active = db.Column(db.String(64))
    # 活动类型
    active_type = db.Column(db.String(64))
    # 活动持续时间
    active_time = db.Column(db.Integer)
    # 活动对象
    active_object = db.Column(db.String(164))
    # 描述
    description = db.Column(db.String(164))
    # 图片 数据库存储格式为: ["url","url"]
    image = db.Column(db.String(1024))
    # 视频 数据库存储格式为: ["url","url"]
    video = db.Column(db.String(1024))

    # 创意
    idea_name = db.Column(db.String(164))
    # 学习
    learn_name = db.Column(db.String(164))
    # learn_id
    learn_id = db.Column(db.String(164))

    # 创建时间
    create_time = db.Column(db.DateTime, default=datetime.datetime.now)
    # 修改时间
    update_time = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now)
    # is_del
    is_delete = db.Column(db.Integer, default=0)
    # 状态
    status = db.Column(db.Integer, default=0)

• 核心代码：

from flask import make_response, send_from_directory
import os, zipfile, re, xlsxwriter, requests
# PS 数据库使用SQLAlchemy 导入自己的模型类
from models import Activities

class Download(Resource):
    def get(self, activities_id):
        # 1. 在项目目录内创建static文件夹
        path=os.path.abspath(os.path.dirname(os.path.dirname(__file__)))+"/static/"
        fileDownloadPath = path.replace('\\', '/')
        filePath = fileDownloadPath+ str(activities_id)+"/"
        if os.path.exists(filePath):
            shutil.rmtree(filePath)
        os.makedirs(filePath)
        
        #  2.数据库查找出数据存入excel
        object = Activities.query.filter(and_(Activities.id == activities_id, Activities.is_delete != 1)).first()
        # 自定义存储方法
        create_workbook(object, filePath+"Active.xlsx")
        
        # 3. 存储图片 存储视频
        if object.image is not None:
            imageUrl = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', str(object.image))
            if imageUrl is not None:
                for data in imageUrl:
                	# 自定义文件下载方法
                    getFile(filePath, data, "Image"+str(imageUrl.index(data)))
        if object.video is not None:
            videoUrl = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', str(object.video))
            if videoUrl is not None:
                for data in videoUrl:
                	# 自定义文件下载方法
                    getFile(filePath, data, "Video" + str(videoUrl.index(data)))
                    
        # 4. 自定义打包方法
        make_zip(filePath, fileDownloadPath+str(activities_id)+".zip")
        
        # 5. 返回zip
        response = make_response(
            send_from_directory(fileDownloadPath, str(activities_id)+".zip", as_attachment=True))
        return response

# 生成Excel
def create_workbook(object, filePath):
    # 创建Excel文件,不保存,直接输出
    workbook = xlsxwriter.Workbook(filePath)
    worksheet = workbook.add_worksheet("sheet")
    title = ["Activity Types", "Activity Details", "Duration Hours", "With Whom",  "Description", "Learnings", "Ideas"]
    worksheet.write_row('A1', title)
    worksheet.write_row('A2', [str(object.active),
                               str(object.active_type),
                               str(object.active_time),
                               str(object.active_object),
                               str(object.description),
                               str(object.learn_name),
                               str(object.idea_name)
                               ])
    workbook.close()

# 下载文件
def getFile(filePath, url, fileName):
    response = requests.get(url).content
    with open(filePath+fileName+url[-4:], 'wb') as f:
        f.write(response)
    print ("Sucessful to download "+fileName)

# 打包zip
def make_zip(filePath, source_dir):
  zipf = zipfile.ZipFile(source_dir, 'w')
  pre_len = len(os.path.dirname(filePath))
  for parent, dirnames, filenames in os.walk(filePath):
    for filename in filenames:
      pathfile = os.path.join(parent, filename)
      arcname = pathfile[pre_len:].strip(os.path.sep)
      zipf.write(pathfile, arcname)
  zipf.close()

简介

在做接口测试的过程中，我们会遇到需要处理文件上传和下载的情况，我们可以使用Jmeter来完成文件的上传下载处理，本文提供两个小示例来展示如何使用Jmeter完成文件上传和文件下载并保存的功能

首先来看看普通的Http接口请求的过程：

填写服务器环境信息

填写接口请求方法和接口路径

填写参数信息

发送请求

接受响应

文件处理上传

文件上传的处理完全遵循上述流程，只需要在步骤3中去申明请求的类型是multipart/form-data和相应的Content-Type

multipart/form-data是Http协议规定的一种表单类型,它会将表单的数据处理为一条消息，以标签为单元，用分隔符分开。既可以上传键值对，也可以上传文件。当上传的字段是文件时，会有Content-Type来表明文件类型

Content-Type: 互联网媒体类型；也叫做MIME类型，在Http协议消息头中，使用Content-Type来表示具体请求中的媒体类型信息，可以是:

image/gif ：gif图片格式

application/pdf ：pdf格式

application/msword：Word文档格式

application/octet-stream：二进制流

binary:相当于Content-Type:application/octet-stream,从字面意思得知，只可以上传二进制数据，通常用来上传文件

示例：上传图片到baidu实现baidu搜图

在HttpSampler里，正常填写请求参数

切换到File Upload 参数页，填写文件路径和MimeType（Content-Type）即可

运行脚本，查看请求数据，图片被以二进制形式的表单参数传递到服务器

成功返回响应结果

文件下载处理

文件下载的步骤也和普通的Http请求一样，只不过返回的是二进制的文件流，需要额外添加请求后置处理器去保存文件到本地电脑上，此处可以通过添加BeanshellPostProcessor来实现文件的保存和文件内容的解析。Beanshell是一种Java脚本语言，本示例展示如何把文件下载保存到本地：

发起图片下载请求

添加Beanshell Post Processor，使用内置变量prev获取接口响应数据，并通过Java代码把响应的二进制流保存到本地E盘

运行脚本，查看结果和本地文件

如果想解析文件内容可以使用Java来读取文件并进行值得校验

作　　者：Testfan Kitty

出　　处：微信公众号：自动化软件测试平台

版权说明：欢迎转载，但必须注明出处，并在文章页面明显位置给出文章链接

postman导出下载文件接口实例

场景

有一个导出报表的功能，页面上点击了导出，下载了报表

实例

接口上，点击send and download,选择存放路径，即可完成发送请求并下载