summaryrefslogtreecommitdiff
path: root/DBcm.py
diff options
context:
space:
mode:
Diffstat (limited to 'DBcm.py')
-rw-r--r--DBcm.py17
1 files changed, 17 insertions, 0 deletions
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()