diff options
Diffstat (limited to 'DBcm.py')
| -rw-r--r-- | DBcm.py | 17 |
1 files changed, 17 insertions, 0 deletions
@@ -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() |
