Added support for the asynchronous context manager protocol on the

AsyncCursor class as a convenience.

Author: anthony-tuininga

doc/src/release_notes.rst

@@ -15,6 +15,8 @@ Thin Mode Changes
 
 #)  Added support for using alternative event loop implementations like uvloop
     (`issue 276 <https://github.com/oracle/python-oracledb/issues/276>`__).
+#)  Added support for the asynchronous context manager protocol on the
+    AsyncCursor class as a convenience.
 #)  Fixed bug with intermittent hang on some versions of Oracle Database when
     using asyncio and the database raises an error and output variables are
     present

src/oracledb/cursor.py

@@ -1,5 +1,5 @@
 # -----------------------------------------------------------------------------
-# Copyright (c) 2021, 2023, Oracle and/or its affiliates.
+# Copyright (c) 2021, 2024, Oracle and/or its affiliates.
 #
 # This software is dual-licensed to you under the Universal Permissive License
 # (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl and Apache License
@@ -911,6 +911,15 @@ def scroll(self, value: int = 0, mode: str = "relative") -> None:
 class AsyncCursor(BaseCursor):
     __module__ = MODULE_NAME
 
+    async def __aenter__(self):
+        self._verify_open()
+        return self
+
+    async def __aexit__(self, *exc_info):
+        self._verify_open()
+        self._impl.close(in_del=True)
+        self._impl = None
+
     def __aiter__(self):
         return self
 

tests/test_5400_cursor_execute_async.py

@@ -1,5 +1,5 @@
 # -----------------------------------------------------------------------------
-# Copyright (c) 2023, Oracle and/or its affiliates.
+# Copyright (c) 2023, 2024, Oracle and/or its affiliates.
 #
 # This software is dual-licensed to you under the Universal Permissive License
 # (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl and Apache License
@@ -583,6 +583,16 @@ async def test_5433_fetch_info_slice(self):
         (fetch_info,) = self.cursor.description
         self.assertEqual(fetch_info[1:3], (oracledb.DB_TYPE_NUMBER, 10))
 
+    async def test_5434_async_context_manager(self):
+        "5434 - test async context manager"
+        expected_value = test_env.get_main_user().upper()
+        with self.conn.cursor() as cursor:
+            await cursor.execute("select user from dual")
+            self.assertEqual(await cursor.fetchone(), (expected_value,))
+        async with self.conn.cursor() as cursor:
+            await cursor.execute("select user from dual")
+            self.assertEqual(await cursor.fetchone(), (expected_value,))
+
 
 if __name__ == "__main__":
     test_env.run_test_cases()