Python实战：批量上传文件信息到服务器

一、背景

在进行软件测试的过程中，经常会需要准备一批数据，这不，今天就遇到了：后端开发好了文件信息上传接口，可是前端并未开发出相关界面，而其他同学正在调试其他接口，需要用到文件信息，那么这个时候就需要我上场了

简而言之就是需要通过接口上传一批文件信息到服务器

二、准备工作

1.Python环境：家中常备，这个就不多说了

2.requests第三方库：用于向服务器发送请求，未安装的同学移步控制台输入以下命令

pip install requests

3.源文件：将不同类型的文件（excel、word、pdf）放到一个文件夹下

4.文件云存储地址：文件已经批量上传到了云服务器，文件云存储地址示例：https://your_file_head_url/file_name.pdf

5.上传接口：从接口文档查看接口地址、传参、响应等，没有接口文档的情况请同学们另辟蹊径

三、编写脚本

获取文件名和类型

源文件夹是这样的：

源文件名称是正常的，不包含“.”

源文件后缀也是正常的，与文件类型匹配

这里简单快速处理，使用split方法通过“.”将文件分割出名称和文件类型

dir_path = 'C:\\Users\\Administrator\\Desktop\\test_file'

# 获取文件名和类型
file_list = os.listdir(dir_path)  # 文件列表
file_name_list = []  # 定义文件名列表
file_type_list = []  # 定义文件类型列表
for i in file_list:
    file_name = i.split('.')[0]  # 文件名
    file_type = i.split('.')[-1]  # 文件类型
    file_name_list.append(file_name)
    file_type_list.append(file_type)
print(file_name_list)
print(file_type_list)

运行结果：

拼接文件信息

根据接口文档，上传的文件信息需要包含类型、名称、状态、url四个参数，这就需要对每个文件信息进行拼接

这里使用了enumerate函数来遍历文件名列表，可以枚举出列表的下标和值，灰常好用

注意：文件类型存在大写字母的情况（PDF，DOC等），所以使用lower函数将文件名转换为小写后再判断对应的参数值

# 拼接文件信息
file_url_head = 'https://your_file_head_url/'  # 云存储地址
type_para_list = [1, 2, 3]  # 文件类型列表：1--excel, 2--word, 3--pdf
file_msg_li = []  # 文件信息列表
for i, v in enumerate(file_name_list):  # 遍历文件名列表
    if file_type_list[i].lower() in ['xlsx', 'xls']:
        type_para = 1
    elif file_type_list[i].lower() in ['docx', 'doc']:
        type_para = 2
    elif file_type_list[i].lower() == 'pdf':
        type_para = 3
    else:
        type_para = 0
    file_msg = {
        "file_type": type_para,
        "name": v,
        "status": random.choice(['true', 'false']),  # 随机状态参数
        "file_url": file_url_head + file_list[i]  # 拼接文件url
    }
    file_msg_li.append(file_msg)
pprint(file_msg_li)

运行结果：

文件信息批量上传

使用requests.post方法请求接口即可

# 文件信息批量上传
for i in file_msg_li:
    res = requests.post(url=server_url, json=i)
    print(res.json())

运行结果：

下面是控制台打印的接口请求返回结果：

下图是数据库的查询结果：

四、优化

1.写到一个方法中，传入文件路径和地址等参数

2.使用列表推导式简化代码

3.文件类型列表改为字典，根据类型快速查找对应的参数

4添加接口响应异常情况处理

优化后完整代码：

import os
import random

import requests


def file_msg_to_server(dir_path, file_url_head, server_url):
    # 获取文件名和类型
    file_list = os.listdir(dir_path)
    file_name_list = [i.split('.')[0] for i in file_list]
    file_type_list = [i.split('.')[-1] for i in file_list]

    # 拼接文件信息
    type_para_dict = {'xlsx': 1, 'xls': 1, 'docx': 2, 'doc': 2, 'pdf': 3}  # 文件类型字典：1--excel, 2--word, 3--pdf
    file_msg_li = [{
        "file_type": type_para_dict.get(file_type_list[i].lower(), 0),  # 根据文件类型获取对应的参数值，若不存在则返回0
        "name": v,
        "status": random.choice(['true', 'false']),  # 随机状态参数
        "file_url": file_url_head + file_list[i]  # 拼接文件url
    } for i, v in enumerate(file_name_list)]

    # 文件信息批量上传

    try:
        for i in file_msg_li:
            res = requests.post(url=server_url, json=i)
            res.raise_for_status()
            print(res.json())
    except requests.exceptions.HTTPError as err:
        print(err)
    except requests.exceptions.Timeout as err:
        print(err)
    # 下面的代码有问题，在循环内return了，只会请求一次
    # for i in file_msg_li:
    #     try:
    #         res = requests.post(url=server_url, json=i)
    #         res.raise_for_status()
    #         return res.json()
    #     except requests.exceptions.HTTPError as err:
    #         return err
    #     except requests.exceptions.Timeout as err:
    #         return err


if __name__ == "__main__":
    path = 'C:\\Users\\Administrator\\Desktop\\test_file'
    url_head = 'https://your_file_head_url/'  # 云存储地址
    host = "http://your_server_url"
    file_msg_to_server(path, url_head, host)