Flask框架和PostgreSQL数据库的集成|极客笔记

Flask框架和PostgreSQL数据库的集成

在开发Web应用程序时，经常需要使用数据库来存储和检索数据。PostgreSQL是一个功能强大的开源关系型数据库管理系统，而Flask则是一个轻量级的Python Web框架，提供了方便的方式来构建Web应用。

本文将详细介绍如何在Flask应用中集成PostgreSQL数据库，以实现数据的增删改查操作。我们将通过一个示例来演示如何使用Flask和PostgreSQL来创建一个简单的学生信息管理系统。

环境准备

在开始之前，确保已经安装了Python和PostgreSQL数据库，并且安装了Flask和psycopg2库。可以使用以下命令来安装所需的库：

pip install Flask psycopg2-binary

创建Flask应用

首先，我们需要创建一个新的Flask应用，并初始化一个新的PostgreSQL数据库。可以按照以下步骤来完成：

创建一个名为app.py的Python文件，用于编写Flask应用的代码。

from flask import Flask

app = Flask(__name__)

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

初始化Flask应用的路由和视图函数。在app.py文件中添加以下代码：

from flask import Flask

app = Flask(__name__)

@app.route('/')
def index():
    return 'Hello, Flask!'

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

运行Flask应用，访问http://127.0.0.1:5000/，可以看到页面上显示Hello, Flask!。

连接PostgreSQL数据库

接下来，我们将连接Flask应用和PostgreSQL数据库，并实现数据的增删改查操作。首先需要创建一个数据库，可以使用pgAdmin等工具来创建数据库和表。

假设我们已经在PostgreSQL数据库中创建了一个名为students的表，包含id、name和age字段。现在我们来编写Flask应用的代码，连接数据库并执行操作。

首先，在app.py文件中添加以下代码来连接PostgreSQL数据库：

import psycopg2
from flask import Flask

app = Flask(__name__)

# 配置数据库连接信息
conn = psycopg2.connect(
    database="your_database_name",
    user="your_username",
    password="your_password",
    host="localhost",
    port="5432"
)

@app.route('/')
def index():
    return 'Hello, Flask!'

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

现在，我们可以在Flask应用中执行SQL查询来操作数据库。例如，我们可以编写一个查询所有学生信息的视图函数：

@app.route('/students')
def get_students():
    cur = conn.cursor()
    cur.execute("SELECT id, name, age FROM students")
    rows = cur.fetchall()
    students = []
    for row in rows:
        student = {
            'id': row[0],
            'name': row[1],
            'age': row[2]
        }
        students.append(student)
    return {'students': students}

运行Flask应用，访问http://127.0.0.1:5000/students，可以看到返回的学生信息数据。

实现增删改查操作

除了查询操作，我们还可以实现插入、更新和删除数据的操作。以下是一个示例代码，演示了如何实现增删改查操作：

# 插入数据
@app.route('/students', methods=['POST'])
def add_student():
    data = request.get_json()
    name = data['name']
    age = data['age']
    cur = conn.cursor()
    cur.execute("INSERT INTO students (name, age) VALUES (%s, %s)", (name, age))
    conn.commit()
    return {'message': 'Student added successfully'}

# 更新数据
@app.route('/students/<int:id>', methods=['PUT'])
def update_student(id):
    data = request.get_json()
    name = data['name']
    age = data['age']
    cur = conn.cursor()
    cur.execute("UPDATE students SET name=%s, age=%s WHERE id=%s", (name, age, id))
    conn.commit()
    return {'message': 'Student updated successfully'}

# 删除数据
@app.route('/students/<int:id>', methods=['DELETE'])
def delete_student(id):
    cur = conn.cursor()
    cur.execute("DELETE FROM students WHERE id=%s", (id,))
    conn.commit()
    return {'message': 'Student deleted successfully'}

通过以上代码，我们可以实现对学生信息的增删改查操作。可以使用Postman等工具来测试API的功能。