Mysql
 sql >> Base de données >  >> RDS >> Mysql

Gestionnaire de contexte pour MySQLdb de Python

Auparavant , les connexions MySQLdb étaient des gestionnaires de contexte. Depuis ce commit le 2018-12-04 , cependant, les connexions MySQLdb ne sont plus des gestionnaires de contexte, et les utilisateurs doivent appeler explicitement conn.commit() ou conn.rollback(), ou écrire leur propre gestionnaire de contexte, comme celui ci-dessous.

Vous pourriez utiliser quelque chose comme ceci :

import config
import MySQLdb
import MySQLdb.cursors as mc
import _mysql_exceptions
import contextlib
DictCursor = mc.DictCursor
SSCursor = mc.SSCursor
SSDictCursor = mc.SSDictCursor
Cursor = mc.Cursor

@contextlib.contextmanager
def connection(cursorclass=Cursor,
               host=config.HOST, user=config.USER,
               passwd=config.PASS, dbname=config.MYDB,
               driver=MySQLdb):
    connection = driver.connect(
            host=host, user=user, passwd=passwd, db=dbname,
            cursorclass=cursorclass)
    try:
        yield connection
    except Exception:
        connection.rollback()
        raise
    else:
        connection.commit()
    finally:
        connection.close()

@contextlib.contextmanager
def cursor(cursorclass=Cursor, host=config.HOST, user=config.USER,
           passwd=config.PASS, dbname=config.MYDB):
    with connection(cursorclass, host, user, passwd, dbname) as conn:
        cursor = conn.cursor()
        try:
            yield cursor
        finally:
            cursor.close()


with cursor(SSDictCursor) as cur:
    print(cur)
    connection = cur.connection
    print(connection)
    sql = 'select * from table'
    cur.execute(sql)
    for row in cur:
        print(row)

Pour l'utiliser, vous placeriez config.py dans votre PYTHONPATH et définissez-y les variables HOST, USER, PASS, MYDB.