ext/python: Add support for Context Manager

bindings/python/src/lib.rs

@@ -266,6 +266,19 @@ impl Connection {
             "Transactions are not supported in this version",
         ))
     }
+
+    fn __enter__(&self) -> PyResult<Self> {
+        Ok(self.clone())
+    }
+
+    fn __exit__(
+        &self,
+        _exc_type: Option<&Bound<'_, PyAny>>,
+        _exc_val: Option<&Bound<'_, PyAny>>,
+        _exc_tb: Option<&Bound<'_, PyAny>>,
+    ) -> PyResult<()> {
+        self.close()
+    }
 }
 
 #[allow(clippy::arc_with_non_send_sync)]

bindings/python/tests/test_database.py

@@ -144,6 +144,17 @@ def test_commit(provider):
     conn.close()
     assert record
 
+@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
+def test_with_statement(provider):
+    with connect(provider, "tests/database.db") as conn:
+        conn = connect(provider, "tests/database.db")
+        cursor = conn.cursor()
+        cursor.execute("SELECT MAX(id) FROM users")
+    
+        max_id = cursor.fetchone()
+    
+        assert max_id
+        assert max_id == (2,)
 
 def connect(provider, database):
     if provider == "limbo":