KifdZddlmZddlmZddlmZddlmZededeefdZ 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)MySQLCursorAbstractconnreturnc#K|j}|}d} |rd|_|V|n7#t$r*d} |n#t$rYnwxYwwxYw ||_ |dS#t$r|sYdSwxYw#||_ |w#t$r|sYwwxYwxYw)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_raiseds X/var/www/html/analyses/venv/lib/python3.11/site-packages/mysql/ai/utils/atomic_cursor.pyatomic_transactionr*sR._N [[]]F  $#DO       MMOOOO    D    )  LLNNNNN   #      )  LLNNNN   #    sw!AB- A7A%$A7% A2/A71A22A77B-B B*)B*-C6C  C CCCCN) __doc__ contextlibrtypingrmysql.connector.abstractsrmysql.connector.cursorrrrrs: &%%%%%======6666663 !3 !"333333r