From 6dabf79eba1d9a2bcee77687785280cb9bdd38ac Mon Sep 17 00:00:00 2001 From: Willem Renes Date: Wed, 12 May 2021 12:07:27 +0200 Subject: updated code --- DBcm.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 DBcm.py (limited to 'DBcm.py') diff --git a/DBcm.py b/DBcm.py new file mode 100644 index 0000000..afe2391 --- /dev/null +++ b/DBcm.py @@ -0,0 +1,17 @@ +import mysql.connector + + +class UseDatabase: + + def __init__(self, config: dict) -> None: + self.configuration = config + + def __enter__(self) -> 'cursor': + self.conn = mysql.connector.connect(**self.configuration) + self.cursor = self.conn.cursor() + return self.cursor + + def __exit__(self, exc_type, exc_value, exc_trace) -> None: + self.conn.commit() + self.cursor.close() + self.conn.close() -- cgit v1.2.3-70-g09d2