Function run_impala_shell_interactive

tests/shell/test_shell_interactive.py:1352–1367

Runs a command in the Impala shell interactively.

(vector, input_lines, shell_args=None,
                                 wait_until_connected=True)

def run_impala_shell_interactive(vector, input_lines, shell_args=None,
                                 wait_until_connected=True):
  """Runs a command in the Impala shell interactively."""
  # if argument "input_lines" is a string, makes it into a list
  if type(input_lines) is str:
    input_lines = [input_lines]
  # workaround to make Popen environment 'utf-8' compatible
  # since piping defaults to ascii
  my_env = os.environ
  my_env['PYTHONIOENCODING'] = 'utf-8'
  p = ImpalaShell(vector, args=shell_args, env=my_env,
      wait_until_connected=wait_until_connected)
  for line in input_lines:
    LOG.info("Running %s" % line)
    p.send_cmd(line)
  return p.get_result()