Как да активирам автоматично повторно свързване на MySQL клиент с MySQLdb?

Реших този проблем, като създадох функция, която обвива cursor.execute() метод, тъй като това хвърляше MySQLdb.OperationalError изключение. Другият пример по-горе предполага, че това е conn.cursor() метод, който хвърля това изключение.

import MySQLdb

class DB:
  conn = None

  def connect(self):
    self.conn = MySQLdb.connect()

  def query(self, sql):
    try:
      cursor = self.conn.cursor()
      cursor.execute(sql)
    except (AttributeError, MySQLdb.OperationalError):
      self.connect()
      cursor = self.conn.cursor()
      cursor.execute(sql)
    return cursor

db = DB()
sql = "SELECT * FROM foo"
cur = db.query(sql)
# wait a long time for the Mysql connection to timeout
cur = db.query(sql)
# still works