Flask 使用 Flask 框架编写 CSV 文件

在本文中，我们将介绍如何在 Flask 框架中编写 CSV 文件。CSV（逗号分隔值）是一种常用的文件格式，用于存储表格数据。Flask 是一个轻量级的 Python Web 框架，它提供了许多功能，使得编写网页应用程序变得更加简单和高效。在本文中，我们将使用 Flask 框架来生成一个简单的网页，并将数据以 CSV 格式写入文件。

阅读更多：Flask 教程

准备工作

首先，我们需要安装 Flask。可以通过 pip 工具在命令行中使用以下命令来安装 Flask：

pip install flask

安装完成后，我们就可以开始编写代码了。

编写 Flask 应用程序

首先，我们需要导入 Flask 模块，并创建一个 Flask 应用程序的实例。在此之后，我们将创建一个简单的网页，其中包含一个按钮，点击按钮后将触发写入 CSV 文件的动作。

from flask import Flask

app = Flask(__name__)

@app.route('/')
def index():
    return '''
        <html>
            <body>
                <h1>写入 CSV 文件</h1>
                <button onClick="window.location.href='/write_csv'">点击这里</button>
            </body>
        </html>
    '''

@app.route('/write_csv')
def write_csv():
    # 在这里编写将数据写入 CSV 文件的代码
    return '数据已成功写入 CSV 文件！'

if __name__ == '__main__':
    app.run()

上述代码中，我们定义了两个路由。/ 路由对应的函数 index() 返回一个包含按钮的简单网页。/write_csv 路由对应的函数 write_csv() 将在点击按钮后触发，这个函数将用于将数据写入 CSV 文件。

编写写入 CSV 文件的代码

现在，我们需要在 write_csv() 函数中编写将数据写入 CSV 文件的代码。以下是一个示例代码：

import csv

# CSV 文件路径
csv_file = 'data.csv'

@app.route('/write_csv')
def write_csv():
    # 数据
    data = [
        ['Name', 'Age', 'Country'],
        ['Tom', 25, 'USA'],
        ['Mary', 28, 'Canada'],
        ['John', 30, 'UK']
    ]

    # 写入 CSV 文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(data)

    return '数据已成功写入 CSV 文件！'

在上述示例代码中，我们首先定义了一个包含数据的列表。每个内部列表表示一行数据，第一个元素是标题，后续元素是对应的数据。接下来，我们使用 open() 函数打开一个 CSV 文件，并使用 csv.writer 对象的 writerows() 方法将数据写入文件中。

请注意，在运行此代码之前，请确保已创建一个名为 data.csv 的空白文件，并将其保存在与 Python 脚本相同的目录中。