Method do_with

shell/impala_shell/impala_shell.py:1640–1652

Executes a query with a WITH clause, fetching all rows

(self, args)

        self._build_query_string(self.last_leading_comment, self.orig_cmd, args))

  def do_with(self, args):
    """Executes a query with a WITH clause, fetching all rows"""
    query = self._build_query_string(self.last_leading_comment, self.orig_cmd, args)
    # Parse the query with sqlparse to identify if it is a DML query or not.
    # Because the WITH clause may precede DML or SELECT queries, checking the
    # first string token is insufficient.
    parsed = sqlparse.parse(query)[0]
    query_type = sqlparse.sql.Statement(parsed.tokens).get_type()
    try:
      is_dml = self.DML_REGEX.match(query_type.lower())
      return self._execute_stmt(query, bool(is_dml), print_web_link=True)
    except ValueError:
      return self._execute_stmt(query, print_web_link=True)

  def do_use(self, args):
    """Executes a USE... query"""