лучший способ запустить код очистки генератора 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

Вы можете использовать менеджер контекста и with заявление. contextlib обеспечивает closing:

from contextlib import closing 

myDB = MySQLdb.connect(host=..., port=..., user=..., passwd=..., db=...) 
with closing(myDB.cursor()) as dbc: 
    dbc.execute(stmt) 
    d = "asdf" 
    while d is not None: 
     d = dbc.fetchone() #can also use fetchmany() to be more efficient 
     yield d 

Это автоматически вызовет close() на dbc в конце with блока, даже если исключение было возбуждено.

Answer 2

Ваша версия будет работать dbc.close() как только d is None, но не исключение, если получает поднятый. Вам нужен 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() 

Answer 3

Одна вещь, которую вы могли бы сделать, это использовать пункт finally. Другой вариант (который может быть излишним здесь, но это полезная вещь, чтобы знать о), чтобы сделать класс, который работает с with заявлением:

class DatabaseConnection: 
    def __init__(self, statement): 
     self.statemet = statement 
    def __enter__(self): 
     self.myDB = MySQLdb.connect(host=..., port=...,user=...,passwd=...,db=...) 
     self.dbc = myDB.cursor() 
     self.dbc.execute(self.statement) 
     self.d = "asdf" 
    def __exit__(self, exc_type, exc_value, traceback): 
     self.dbc.close() 

    def __iter__(self): 
     while self.d is not None: 
      self.d = self.dbc.fetchone() 
      yield self.d 


with DatabaseConnection(stmnt) as dbconnection: 
    for i in dbconnection: 
     print(i)