from duckdb import DuckDBPyConnection
class DuckDBAtomicTransaction:
 def __init__(self, conn: DuckDBPyConnection):
 self.conn = conn
 def __enter__(self):
 self.conn.begin()
 return self.conn
 def __exit__(self, exc_type, exc_value, traceback):
 if exc_type is None:
 self.conn.commit()
 else:
 self.conn.rollback()
with connect() as conn:
 with DuckDBAtomicTransaction(conn):
 pass