mysql.connect()在with语句中使用时返回不同的类型

时间:2016-01-26 21:50:17

标签: python mysql flask

我在Flask-MySQL中使用Flask,之前正在执行以下操作:

db = mysql.connect()
cursor = db.cursor()
#do database stuff
cursor.close()
db.close()

但我认为我会让事情变得更清洁,更具惯用性,所以我尝试过:

with mysql.connect() as db, db.cursor() as cursor:
    #do database stuff

它给了我以下错误" AttributeError:' Cursor'对象没有属性'光标'"

经过一番调查后,当我对它自己执行mysql.connect()时,它返回一个连接对象,但当我在一个with语句中执行它时...它返回一个游标对象!所以这有效:

with mysql.connect() as cursor:
    #do database stuff

那么这里发生了什么?为什么在with语句中返回类型会发生变化?我查看了Flask-MySQL的源代码,但无法找到任何内容......

此外,这仍然可以正常关闭with块末尾的数据库连接吗?

相关版本信息:

  • Python == 2.7.3
  • Flask == 0.10.1
  • Flask-MySQL == 1.3

1 个答案:

答案 0 :(得分:2)

表达式:

with X as Y:
    ....

并不意味着Y“是”X。而是调用X.__enter__(),然后将其结果绑定到Y语句的主体with

with正文结束时,或者在发生异常时,会调用X.__exit__(...)来处理成功或失败。

在这种情况下,作为连接的mysql.connect()的结果公开了执行以下操作的上下文管理器:

  • __enter__创建游标并开始交易;
  • __exit__可以根据需要提交或回滚事务。