python - 运行 python 生成器清理代码的最佳方式

Question

我正在尝试编写一个生成器函数，从数据库中获取行并一次返回一个行。但是，我不确定下面标记为 ** 的清理代码是否像我认为的那样执行。如果没有，将清理代码放入在最后一个 yield 语句之后执行的生成器本身的最佳方法是什么？我查看了捕获 StopIteration 但这似乎是从调用者而不是在生成器中完成的。

def MYSQLSelectGenerator(stmt):
...
try:   
    myDB = MySQLdb.connect(host=..., port=..., user=..., passwd=..., db=...)   
    dbc=myDB.cursor()
    dbc.execute(stmt)
    d = "asdf"
    while d is not None:
        d = dbc.fetchone() #can also use fetchmany() to be more efficient
        yield d
    dbc.close() #** DOES THIS WORK AS I INTEND, MEANING AS SOON AS d = "None"
except MySQLdb.Error, msg:
    print("MYSQL ERROR!")
    print msg

Answer 1

只要 d 为 None，您的版本就会运行 dbc.close()，但如果引发异常则不会。你需要一个 finally clause .此版本保证运行dbc.close()，即使出现异常也是如此:

try:   
    myDB = MySQLdb.connect(host=..., port=..., user=..., passwd=..., db=...)   
    dbc = myDB.cursor()
    dbc.execute(stmt)
    d = "asdf"
    while d is not None:
        d = dbc.fetchone() #can also use fetchmany() to be more efficient
        yield d
except MySQLdb.Error, msg:
    print("MYSQL ERROR!")
    print msg
finally:
    dbc.close()