Method cursor

ddserver/db.py:56–88

(self, config)

  @contextlib.contextmanager
  @require(config = 'ddserver.config:Config')
  def cursor(self, config):
    # Ensure we have a connection for this thread
    if not hasattr(self.thread_local, 'connection'):
      connection = MySQLdb.connect(host = config.db.host,
                                   port = config.db.port,
                                   user = config.db.username,
                                   passwd = config.db.password,
                                   db = config.db.name,
                                   cursorclass = MySQLdb.cursors.DictCursor,
                                   charset = 'utf8')
      setattr(self.thread_local, 'connection', connection)

    else:
      # Use existing connection
      connection = getattr(self.thread_local, 'connection')

      # Reconnect if connection is down
      connection.ping(True)

    cursor = connection.cursor()

    try:
      yield cursor

    except:
      connection.rollback()
      raise

    else:
      connection.commit()

    finally:
      cursor.close()