使用 Python Flask 上传 CSV 文件并进行处理

Question

我有以下代码使用 Python FLASK 上传 CSV 文件。

from flask_restful import Resource
import pandas as pd

ROOT_PATH = os.path.dirname(os.path.abspath(__file__))

class UploadCSV(Resource):

    def post(self):
        files = request.files['file']
        files.save(os.path.join(ROOT_PATH,files.filename))
        data = pd.read_csv(os.path.join(ROOT_PATH,files.filename))
        print(data)

api.add_resource(UploadCSV, '/v1/upload')

if __name__ == '__main__':
    app.run(host='localhost', debug=True, port=5000)

这段代码运行良好，我可以成功上传 CSV 文件并使用 Pandas 数据框读取它。但我将 csv 保存在本地文件系统中并读取它。

我试过像下面这样阅读 -

files = request.files['file']
files.read()

获得的结果是字节格式，但我需要字典格式。因此，我使用 Pandas 数据框来读取它，然后将其转换为我的格式的自定义字典。

是否可以在不将其写入本地文件系统的情况下即时读取 CSV 文件？或者我们可以使用 Python Flask Restful 实现的任何等效方法？

Answer 1

（不太恰当地命名，因为它是一个文件）files变量包含一个werkzeug.datastructures.FileStorage对象。这是一个类似于对象的文件（包含一个read()方法），因此可以直接将其用作输入，pandas.read_csv()如熊猫文档中所见。

因此，不保存到磁盘的解决方案非常简单：

class UploadCSV(Resource):

    def post(self):
        file = request.files['file']
        data = pd.read_csv(file)
        print(data)