在将自定义上下文管理器类与 mysql 连接器一起使用时如何提交或回滚数据库中的更改

问：

通过自定义上下文管理器类使用 mysql 连接器时，处理提交或回滚数据库中更改的最佳方法是什么？例如，假设自定义类采用以下形式：

class DatabaseConnection:
    def __init__(self, host, user, password, database_name)
        self.host = host
        self.user = user
        self.passwd = password
        self.database_name = database_name
        self.connection = None

    def __enter__(self) -> Union[mysql.cursor.MySQLCursor, None]:
       try:
          if self.connection is None or not sef.connection.is_connected():
              self.connection = mysql.connect(
                  host = self.host, user=self.user, psswd=self.psswd, db=self.database_name
              )
          cursor = self.connection.cursor()
          return cursor
      except Exception:
          return None

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.connection is not None: 
            self.connection.close()

    def commit(self):
        if self.connection is not None and self.connection.is_connected():
             try:
                 self.connection.commit()
                 return True
             except:
                 self.connection.rollback()

    def rollback(self):
        if self.connection is not None:
           self.connection.rollback()

然后你想使用它，使用语句，这样你就有了光标。您将如何管理提交或回滚更改，尤其是在引发错误时。例如：with

test = True
database_connection = DatabaseConnection(....)
try:
    with database_connection() as cursor:
        # perform some operations

        if test:
            database_connection.rollback()
        else:
            database_connection.commit()
except Exception as e:
     database_connection.rollback()

忽略错误处理方面的不良做法，提交和回滚是否按预期工作，或者您是否必须以不同的方式组织它？

编辑

上述模式似乎按预期工作（如果测试，则回滚，如果不是，则提交，如果异常，则回滚）。我不完全确定为什么，所以在这方面的任何启示都会有所帮助。谢谢！