Merged PR 5213: Adding autocommit function

Adding autocommit function

----
#### AI description  (iteration 1)
#### PR Classification
New feature

#### PR Summary
This pull request adds an autocommit function to the database connection class.
- `mssql_python/connection.py`: Added `autocommit` property to get the current autocommit mode and `setautocommit` method to set the autocommit mode. Updated the constructor to accept an `autocommit` parameter and initialize it.

Related work items: #33433

Author: jahnvi480

mssql_python/connection.py

@@ -32,7 +32,7 @@ class Connection:
         close() -> None:
     """
 
-    def __init__(self, connection_str: str) -> None:
+    def __init__(self, connection_str: str, autocommit: bool = True) -> None:
         """
         Initialize the connection object with the specified connection string.
 
@@ -52,6 +52,8 @@ def __init__(self, connection_str: str) -> None:
         self.hdbc = ctypes.c_void_p()
         self.connection_str = add_driver_to_connection_str(connection_str)
         self._initializer()
+        self._autocommit = autocommit
+        self.setautocommit(autocommit)
 
     def _initializer(self) -> None:
         """
@@ -126,6 +128,49 @@ def _connect_to_db(self) -> None:
             logging.error("An error occurred while connecting to the database: %s", e)
             raise
 
+    @property
+    def autocommit(self) -> bool:
+        """
+        Return the current autocommit mode of the connection.
+        Returns:
+            bool: True if autocommit is enabled, False otherwise.
+        """
+        try:
+            autocommit_mode = ddbc_bindings.DDBCSQLGetConnectionAttr(
+                self.hdbc.value,  # Connection handle
+                odbc_sql_const.SQL_ATTR_AUTOCOMMIT.value,  # Attribute
+            )
+            check_error(odbc_sql_const.SQL_HANDLE_DBC.value, self.hdbc.value, autocommit_mode)
+            return autocommit_mode == odbc_sql_const.SQL_AUTOCOMMIT_ON.value
+        except Exception as e:
+            logging.error("An error occurred while getting autocommit mode: %s", e)
+            raise Exception("DatabaseError: Failed to get autocommit mode") from e
+        
+    def setautocommit(self, value: bool) -> None:
+        """
+        Set the autocommit mode of the connection.
+        Args:
+            value (bool): True to enable autocommit, False to disable it.
+        Returns:
+            None
+        Raises:
+            DatabaseError: If there is an error while setting the autocommit mode.
+        """
+        try:
+            ret = ddbc_bindings.DDBCSQLSetConnectAttr(
+                self.hdbc.value,  # Connection handle
+                odbc_sql_const.SQL_ATTR_AUTOCOMMIT.value,  # Attribute
+                odbc_sql_const.SQL_AUTOCOMMIT_ON.value if value else odbc_sql_const.SQL_AUTOCOMMIT_OFF.value,  # Value
+                0  # String length
+            )
+            check_error(odbc_sql_const.SQL_HANDLE_DBC.value, self.hdbc.value, ret)
+            self._autocommit = value
+            logging.info("Autocommit mode set to %s.", value)
+        except Exception as e:
+            logging.error("An error occurred while setting autocommit mode: %s", e)
+            raise Exception("DatabaseError: Failed to set autocommit mode") from e
+
+
     def cursor(self) -> Cursor:
         """
         Return a new Cursor object using the connection.

mssql_python/constants.py

@@ -23,3 +23,6 @@ class ConstantsODBC(Enum):
     SQL_OV_ODBC3 = 3
     SQL_COMMIT = 0
     SQL_ROLLBACK = 1
+    SQL_ATTR_AUTOCOMMIT = 102
+    SQL_AUTOCOMMIT_ON = 1
+    SQL_AUTOCOMMIT_OFF = 0

tests/test_connection.py

@@ -40,6 +40,11 @@ def test_commit(db_connection):
 #     finally:
 #         cursor.execute("DROP TABLE test_commit;")
 
+def test_autocommit(db_connection):
+    assert db_connection.autocommit is True, "Autocommit should be True by default"
+    db_connection.setautocommit(False)
+    assert db_connection.autocommit is False, "Autocommit failed: Autocommit should be False"
+
 def test_rollback(db_connection):
     db_connection.rollback()
 #     # Make a transaction and rollback