o =Di@sRdZddlmZddlmZddlmZddlmZededeefdd Z d S) zAtomic transaction context manager utilities for MySQL Connector/Python. Provides context manager atomic_transaction() that ensures commit on success and rollback on error without obscuring the original exception. )contextmanager)Iterator)MySQLConnectionAbstract)MySQLCursorAbstractconnreturnccs|j}|}d}zAz|rd|_|V|Wnty2d}z|Wty1YwwW||_z|WdStyK|sHYdSw||_z|Wwtya|s_Yww)a Context manager that wraps a MySQL database cursor and ensures transaction rollback in case of exception. NOTE: DDL statements such as CREATE TABLE cause implicit commits. These cannot be managed by a cursor object. Changes made at or before a DDL statement will be committed and not rolled back. Callers are responsible for any cleanup of this type. This class acts as a robust, PEP 343-compliant context manager for handling database cursor operations on a MySQL connection. It ensures that all operations executed within the context block are part of the same transaction, and automatically calls `connection.rollback()` if an exception occurs, helping to maintain database integrity. On normal completion (no exception), it simply closes the cursor after use. Exceptions are always propagated to the caller. Args: conn: A MySQLConnectionAbstract instance. FTN) autocommitcursorcommit Exceptionrollbackclose)rold_autocommitr exception_raisedrU/var/www/Datamplify/venv/lib/python3.10/site-packages/mysql/ai/utils/atomic_cursor.pyatomic_transaction*sH        rN) __doc__ contextlibrtypingrmysql.connector.abstractsrmysql.connector.cursorrrrrrrs