Python MySQL和上下文管理器:__ exit__属性错误

时间:2016-12-06 11:44:25

标签: python mysql python-2.7 contextmanager

使用Python 2.7.9,我试图分解mysql连接和游标的打开和关闭

代码是

class MySQLCursor:
    def __init__(self, commit=False):
        self.commit = commit

    def __enter__(self):
        self.conn = MySQLdb.connect(host=_MYMYSQLHOST,
                                    user=_MYMYSQLUSER,
                                    passwd=_MYMYSQLPASSWD,
                                    db=_MYMYSQLDB)
        self.cursor = self.conn.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.commit:
            self.conn.commit()
        self.cursor.close()
        self.conn.close()
        return

将其用作

with MySQLCursor as cursor:
    cursor.execute("SELECT VERSION()")
    row = cursor.fetchone()
    print "server version:", row[0]

我收到错误消息

AttributeError: __exit__ 

这是mysql问题还是上下文管理器问题?

1 个答案:

答案 0 :(得分:4)

只做with MySQLCursor() as cursor: - With语句需要具有__enter____exit__方法的类的实例,而不是类本身。