From 579f18c5ec6929e589f963bb8e3ff6c74688ff17 Mon Sep 17 00:00:00 2001 From: mike Date: Wed, 1 Jan 2020 02:07:29 +0800 Subject: [PATCH] implement context manager mplement context manager at cursor base --- lib/mysql/connector/abstracts.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lib/mysql/connector/abstracts.py b/lib/mysql/connector/abstracts.py index b8d0573..e341bc9 100644 --- a/lib/mysql/connector/abstracts.py +++ b/lib/mysql/connector/abstracts.py @@ -1225,3 +1225,11 @@ def lastrowid(self): def fetchwarnings(self): """Returns Warnings.""" return self._warnings + + + def __enter__(self): + return self + + + def __exit__(self, exc_type, exc_value, traceback): + self.close() \ No newline at end of file