如何在Python (Flask)应用程序中捕获psycopg2.errors.UniqueViolation错误？

在Python的Flask应用程序中捕获psycopg2.errors.UniqueViolation错误通常涉及到数据库操作和异常处理。psycopg2是一个流行的PostgreSQL数据库适配器，而UniqueViolation错误表示试图插入的数据违反了数据库中的唯一性约束。

基础概念

psycopg2: 一个用于Python连接PostgreSQL数据库的库。
UniqueViolation: 当试图插入重复数据到具有唯一性约束的字段时，PostgreSQL会抛出此错误。

捕获UniqueViolation错误

要在Flask应用程序中捕获此错误，你需要使用try-except块来捕获psycopg2.errors.UniqueViolation异常。

示例代码

from flask import Flask, jsonify
import psycopg2
from psycopg2 import errors

app = Flask(__name__)

def insert_data(data):
    conn = psycopg2.connect(database="yourdb", user="youruser", password="yourpass", host="yourhost", port="yourport")
    cur = conn.cursor()
    
    try:
        cur.execute("INSERT INTO yourtable (unique_field) VALUES (%s)", (data,))
        conn.commit()
    except errors.UniqueViolation as e:
        conn.rollback()
        return jsonify({"error": "Duplicate entry for unique field: {}".format(data)}), 409
    finally:
        cur.close()
        conn.close()

@app.route('/insert', methods=['POST'])
def insert():
    data = request.json.get('data')
    if data:
        return insert_data(data)
    else:
        return jsonify({"error": "No data provided"}), 400

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

解释

数据库连接: 使用psycopg2.connect()连接到PostgreSQL数据库。
执行插入操作: 在try块中执行INSERT语句。
捕获异常: 如果发生UniqueViolation错误，except块会捕获它，执行rollback()来回滚事务，并返回一个409状态码（表示冲突）。
关闭连接: 在finally块中确保数据库连接被关闭。