文件上传接口

最新推荐文章于 2024-08-14 16:23:05 发布

Xue丶

最新推荐文章于 2024-08-14 16:23:05 发布

阅读量279

点赞数

文章标签： python flask 开发语言

本文链接：https://blog.csdn.net/weixin_45783664/article/details/126852553

版权

文件上传接口

# -*- coding:utf-8 -*-
# Time:2022/8/16 16:55
# Author:Xue
# FileName:1.py
import os
import pandas as pd
import json
from flask import request, Flask


app = Flask(__name__)
upload_export_path = 'D:\\python\\bushu1\\jiekou\\file\\'
moban_path = 'D:\\python\\bushu1\\jiekou\\moban\\'


@app.route('/upload', methods=['POST'])
def upload_file():
    """

    :return:
    """
    file_buffer = request.files['file']  # 接收参数
    f_name = file_buffer.filename  # 取文件名字
    if os.path.exists(upload_export_path + f_name):
        return json.dumps({"code": 201, 'message': '文件已存在，请重新上传！'}, ensure_ascii=False)
    elif not os.path.exists(upload_export_path + f_name):
        file_buffer.save(upload_export_path + f_name)
        # 取模板中的标题字段名称
        data = pd.read_excel(moban_path + '测试用例模板.xlsx', header=None)
        # 取前13列
        df = pd.concat([data[data.columns[0]], data[data.columns[:12]]], axis=1)
        # 将dataframe类型转化为list类型
        t = df.values.tolist()
        # 打印第1行的内容,注意python中list从0开始计数
        list1 = t[0]
        # print(list1)

        # 取上传文件中的标题字段名称
        data = pd.read_excel(upload_export_path + f_name, header=None, sheet_name=None)
        for j in data.values():
            df = pd.concat([j[j.columns[0]], j[j.columns[:12]]], axis=1)
            j1 = df.values.tolist()
            list2 = j1[0]
            # print(list2)
            if list1 != list2:
                os.remove(upload_export_path + f_name)
                return json.dumps({"code": 202, 'message': '上传的文件标题与模板不符'}, ensure_ascii=False)
        return json.dumps({"code": 200, 'message': '上传成功！'}, ensure_ascii=False)


if __name__ == '__main__':
    app.run(host="0.0.0.0", port=5000)