使用flask将csv文件保存到postgres数据库中

Flask 是一个使用 Python 编写的轻量级 Web 应用框架，它可以帮助我们快速构建 Web 应用程序。PostgreSQL 是一个强大的开源关系型数据库管理系统，它具有高度的可扩展性和稳定性。

要使用 Flask 将 CSV 文件保存到 PostgreSQL 数据库中，可以按照以下步骤进行操作：

安装必要的库：首先，需要确保已经安装了 Flask 和 psycopg2（用于连接 PostgreSQL 数据库）这两个库。
创建 Flask 应用：创建一个 Python 文件，导入 Flask 库，并使用 Flask(__name__) 创建一个 Flask 应用实例。

from flask import Flask

app = Flask(__name__)

定义路由：使用 @app.route 装饰器定义一个路由，指定要访问的 URL 和请求方法。

@app.route('/upload', methods=['POST'])
def upload_csv():
    # 处理上传的 CSV 文件
    return 'CSV 文件上传成功'

处理上传的 CSV 文件：在 upload_csv 函数中，可以使用 Flask 提供的 request 对象来获取上传的文件，并将其保存到指定的位置。

from flask import request

@app.route('/upload', methods=['POST'])
def upload_csv():
    if 'file' not in request.files:
        return '没有上传文件'

    file = request.files['file']
    if file.filename == '':
        return '没有选择文件'

    if file:
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
        return 'CSV 文件上传成功'

将 CSV 数据保存到 PostgreSQL 数据库：在保存文件后，可以使用 psycopg2 连接到 PostgreSQL 数据库，并将 CSV 数据导入到相应的表中。

import psycopg2

def save_to_postgres(csv_path):
    conn = psycopg2.connect(
        host='数据库主机名',
        port='数据库端口',
        database='数据库名',
        user='数据库用户名',
        password='数据库密码'
    )
    cursor = conn.cursor()

    with open(csv_path, 'r') as f:
        next(f)  # 跳过 CSV 文件的标题行
        cursor.copy_from(f, '表名', sep=',')

    conn.commit()
    cursor.close()
    conn.close()

请注意，上述代码中的数据库连接参数需要根据实际情况进行修改。

调用保存函数：在 upload_csv 函数中，调用 save_to_postgres 函数，将上传的 CSV 文件的路径作为参数传递给它。

@app.route('/upload', methods=['POST'])
def upload_csv():
    if 'file' not in request.files:
        return '没有上传文件'

    file = request.files['file']
    if file.filename == '':
        return '没有选择文件'

    if file:
        filename = secure_filename(file.filename)
        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
        save_to_postgres(os.path.join(app.config['UPLOAD_FOLDER'], filename))
        return 'CSV 文件上传成功并保存到 PostgreSQL 数据库'

这样，当有用户访问 /upload URL 并以 POST 方法上传 CSV 文件时，Flask 应用将保存文件并将数据导入到 PostgreSQL 数据库中。

使用flask将csv文件保存到postgres数据库中

社区

活动

资源

关于

腾讯云开发者

热门产品

热门推荐

更多推荐