DBcm.py

import mysql.connector

class ConnectionError(Exception):
    pass

class CredentialsError(Exception):
    pass

class SQLError(Exception):
    pass

class UseDatabase(object):
          def __init__(self, config: dict):
              self.configuration = config

          def __enter__(self) ->"cursor":
              try:
                  self.conn = mysql.connector.connect(**self.configuration)
                  self.cursor = self.conn.cursor()
                  return self.cursor
              except mysql.connector.errors.InterfaceError as err:
                  raise ConnectionError(err)
              except mysql.connector.errors.ProgrammingError as err:
                  raise CredentialsError(err)


          def __exit__(self, exc_type, exc_value, exc_traceback):
                    self.conn.commit()
                    self.cursor.close()
                    self.conn.close()
                    if exc_type is mysql.connector.errors.ProgrammingError:
                        raise SQLError(exc_value)
                    elif exc_type:
                        raise exc_type(exc_value)