Python Flask服务同时接收单个、多个上传的文件

墨痕诉清风

已于 2022-08-25 15:17:55 修改

阅读量2.5k

点赞数

分类专栏： Python/算法文章标签： flask python

于 2022-03-16 10:46:07 首次发布

本文链接：https://blog.csdn.net/u012206617/article/details/123520672

版权

Python/算法专栏收录该内容

302 篇文章

订阅专栏

只接收单个文件上传

服务的

import zip

@app.route("/create/aaa", methods=["POST"])
def create_aaa():
    data = request.files["zip"]                                                                                                                         
    file_name = data.filename
    print(file_name)

    return jsonify(status="success")

客户端

def create_aaa():
    url = "http://11.22.33.44:55/create/aaa"
    header = {
        "aaa": bbb
    }

    upgrade_file_path = "C:/Users/aa/Desktop/rules-master/aaa/aaa.zip"
    with open(upgrade_file_path, "rb") as f:
        files = {"zip": ("aaa.zip", f)}
        res = requests.post(url=url, headers=header, files=files)

def main():
    create_aaa()

同时接收单个、多个文件上传

服务端

import yara
import os

@app.route("/import/aaa", methods=["POST"])
def import_aaa():
    uploaded_files = request.files.getlist("yar")

    for f_yar in uploaded_files:
        file_name = f_yar.filename
        _, extension = os.path.splitext(file_name)
        if not extension == ".yar":
            return json_error(403, "The {} no yar file.".format(file_name))

        file_path = os.path.join(dirpath, file_name)

        try:
            yara.compile(source=f_yar.read())
        except Exception as e:
            return json_error(403, "{} illegal yara rule - {}.".format(file_name, e))

        f_yar.save(file_path)

    return jsonify(status="success")

客户端

import requests

# 导入yara规则
def import_aaa():
    url = "http://11.22.33.44:55/import/aaa"
    header = {
        "77777": 8888888
    }

    res = requests.post(url=url, headers=header, files=[
        #("yar", open("C:/Users/aaa/Desktop/rules-master/cve_rules/aaa.yar", "rb")),
        ("yar", open("C:/Users/aaa/Desktop/rules-master/cve_rules/CVE-2010-0805.yar", "rb")),
        ("yar", open("C:/Users/aaa/Desktop/rules-master/cve_rules/CVE-2010-0887.yar", "rb"))
    ])

def main():
    import_aaa()