def handle_cursor(cls, cursor: Cursor, query: Query, session: Session) -> None: tracking_url = cls.get_tracking_url(cursor) if tracking_url: query.tracking_url = tracking_url # Adds the executed query id to the extra payload so the query can be cancelled query.set_extra_json_key("cancel_query", cursor.stats["queryId"]) session.commit() BaseEngineSpec.handle_cursor(cursor=cursor, query=query, session=session)
def handle_cursor(cls, cursor: Any, query: Query, session: Session) -> None: """Updates progress information""" BaseEngineSpec.handle_cursor(cursor=cursor, query=query, session=session)