如何在Django的save()方法中使用自定义psycopg3数据库连接对象？

要让Django Model的save()方法和你自定义的Database对象共享事务（确保两个操作都成功才提交），核心是让Django使用你自定义的psycopg3连接，从而把两个操作纳入同一个事务上下文。

步骤1：确认自定义Database类的结构 #

假设你的Database类持有psycopg3连接实例，大致结构如下：

import psycopg3

class Database:
    def __init__(self, conn_data):
        self.conn = psycopg3.connect(**conn_data)
        self.cur = self.conn.cursor()

    def commit(self):
        self.conn.commit()

    def rollback(self):
        self.conn.rollback()

    # 其他自定义方法...

步骤2：修改第二个类的insert方法 #

在调用save()前，将Django的数据库连接替换为你自定义的psycopg3连接，禁用Django的自动提交机制：

from django.db import connection

def insert(self, insert_data, db):
    # 确保Django使用自定义的psycopg3连接
    if connection.connection != db.conn:
        # 关闭Django默认连接（若已打开）
        connection.close()
        # 替换为自定义连接
        connection.connection = db.conn
        # 重置Django连接状态，禁用自动提交
        connection._cursor = None
        connection._commit_on_exit = False
    
    # 执行Django Model的save()
    self.save()

步骤3：保持原有事务逻辑 #

调用时仍使用你原来的流程，异常时触发回滚：

try:
    db = Database(postgres_connection_data)
    # 自定义类的插入操作（带异常处理）
    class_one_instance.insert(insert_data, db)
    # Django Model的插入操作
    class_two_instance.insert(insert_data, db)
    # 两个操作都成功才提交
    db.commit()
except Exception as e:
    # 任何一步失败都回滚
    db.rollback()
    # 处理异常（如打印日志、抛出上层异常等）
    raise

关键注意事项 #

• 自定义类的insert方法需保留异常捕获逻辑，确保出错时触发回滚，阻止后续commit执行。
• 替换Django连接后，若后续有其他Django数据库操作，记得恢复默认连接（可在事务结束后调用connection.close()让Django重新建立连接）。
• 禁止Django自动提交是核心，否则save()会直接提交事务，破坏你需要的原子性。

内容的提问来源于stack exchange，提问作者concrete_boi