コード例 #1
0
    def __check_for_command_completion(self, cmd):
        """Check for a delimiter at the end of user input.

    The end of the user input is scanned for a legal delimiter.
    If a delimiter is not found:
      - Input is not send to onecmd()
        - onecmd() is a method in Cmd which routes the user input to the
          appropriate method. An empty string results in a no-op.
      - Input is removed from history.
      - Input is appended to partial_cmd
    If a delimiter is found:
      - The contents of partial_cmd are put in history, as they represent
        a completed command.
      - The contents are passed to the appropriate method for execution.
      - partial_cmd is reset to an empty string.
    """
        if self.readline:
            current_history_len = self.readline.get_current_history_length()
        # Input is incomplete, store the contents and do nothing.
        if not cmd.endswith(ImpalaShell.CMD_DELIM):
            # The user input is incomplete, change the prompt to reflect this.
            if not self.partial_cmd and cmd:
                self.cached_prompt = self.prompt
                self.prompt = '> '.rjust(len(self.cached_prompt))
            # partial_cmd is already populated, add the current input after a newline.
            if self.partial_cmd and cmd:
                self.partial_cmd = "%s\n%s" % (self.partial_cmd, cmd)
            else:
                # If the input string is empty or partial_cmd is empty.
                self.partial_cmd = "%s%s" % (self.partial_cmd, cmd)
            # Remove the most recent item from history if:
            #   -- The current state of user input in incomplete.
            #   -- The most recent user input is not an empty string
            if self.readline and current_history_len > 0 and cmd:
                self.readline.remove_history_item(current_history_len - 1)
            # An empty string results in a no-op. Look at emptyline()
            return str()
        elif self.partial_cmd:  # input ends with a delimiter and partial_cmd is not empty
            if cmd != ImpalaShell.CMD_DELIM:
                completed_cmd = "%s\n%s" % (self.partial_cmd, cmd)
            else:
                completed_cmd = "%s%s" % (self.partial_cmd, cmd)
            # Reset partial_cmd to an empty string
            self.partial_cmd = str()
            # Replace the most recent history item with the completed command.
            # In order for it to be read from the history file, the \n has
            # to be escaped.
            if self.readline and current_history_len > 0:
                self.readline.replace_history_item(
                    current_history_len - 1,
                    completed_cmd.encode('string-escape'))
            # Revert the prompt to its earlier state
            self.prompt = self.cached_prompt
        else:  # Input has a delimiter and partial_cmd is empty
            completed_cmd = cmd
        return completed_cmd
コード例 #2
0
ファイル: impala_shell.py プロジェクト: SunGuo/impala
  def __check_for_command_completion(self, cmd):
    """Check for a delimiter at the end of user input.

    The end of the user input is scanned for a legal delimiter.
    If a delimiter is not found:
      - Input is not send to onecmd()
        - onecmd() is a method in Cmd which routes the user input to the
          appropriate method. An empty string results in a no-op.
      - Input is removed from history.
      - Input is appended to partial_cmd
    If a delimiter is found:
      - The contents of partial_cmd are put in history, as they represent
        a completed command.
      - The contents are passed to the appropriate method for execution.
      - partial_cmd is reset to an empty string.
    """
    if self.readline:
      current_history_len = self.readline.get_current_history_length()
    # Input is incomplete, store the contents and do nothing.
    if not cmd.endswith(ImpalaShell.CMD_DELIM):
      # The user input is incomplete, change the prompt to reflect this.
      if not self.partial_cmd and cmd:
        self.cached_prompt = self.prompt
        self.prompt = '> '.rjust(len(self.cached_prompt))
      # partial_cmd is already populated, add the current input after a newline.
      if self.partial_cmd and cmd:
        self.partial_cmd = "%s\n%s" % (self.partial_cmd, cmd)
      else:
        # If the input string is empty or partial_cmd is empty.
        self.partial_cmd = "%s%s" % (self.partial_cmd, cmd)
      # Remove the most recent item from history if:
      #   -- The current state of user input in incomplete.
      #   -- The most recent user input is not an empty string
      if self.readline and current_history_len > 0 and cmd:
        self.readline.remove_history_item(current_history_len - 1)
      # An empty string results in a no-op. Look at emptyline()
      return str()
    elif self.partial_cmd: # input ends with a delimiter and partial_cmd is not empty
      if cmd != ImpalaShell.CMD_DELIM:
        completed_cmd = "%s\n%s" % (self.partial_cmd, cmd)
      else:
        completed_cmd = "%s%s" % (self.partial_cmd, cmd)
      # Reset partial_cmd to an empty string
      self.partial_cmd = str()
      # Replace the most recent history item with the completed command.
      # In order for it to be read from the history file, the \n has
      # to be escaped.
      if self.readline and current_history_len > 0:
        self.readline.replace_history_item(current_history_len - 1,
                                           completed_cmd.encode('string-escape'))
      # Revert the prompt to its earlier state
      self.prompt = self.cached_prompt
    else: # Input has a delimiter and partial_cmd is empty
      completed_cmd = cmd
    return completed_cmd
コード例 #3
0
ファイル: shell.py プロジェクト: pombreda/conary-1
    def default(self, cmd):
        cmd = cmd.strip()

        if cmd == 'EOF':
            # EOF means exit.  print a new line to clean up
            print
            return True

        if not cmd.endswith(';'):
            cmd = self.multiline(cmd).strip()

        # if there are multiple statements on one line, split them up
        if ';' in cmd:
            # split up the command and execute each part
            complete, partial = cmd.split(';', 1)
            if partial:
                # if there are two or more commands, run the first
                self.default(complete + ';')
                return self.default(partial)

        if not cmd:
            # no sql, noop
            return False

        start = time.time()
        # execute the SQL command
        try:
            # a weak attempt to see if we have any %(keyword)s to expand
            if r'%(' in cmd:
                cmd = cmd % self.db.keywords
            self.cu.execute(cmd)
        except sqlerrors.DatabaseError, e:
            if len(e.args) > 1:
                print 'Error:', str(e.args[0])
            else:
                print 'Error:', str(e)
            return False
コード例 #4
0
ファイル: shell.py プロジェクト: fedora-conary/conary
    def default(self, cmd):
        cmd = cmd.strip()

        if cmd == 'EOF':
            # EOF means exit.  print a new line to clean up
            print
            return True

        if not cmd.endswith(';'):
            cmd = self.multiline(cmd).strip()

        # if there are multiple statements on one line, split them up
        if ';' in cmd:
            # split up the command and execute each part
            complete, partial = cmd.split(';', 1)
            if partial:
                # if there are two or more commands, run the first
                self.default(complete + ';')
                return self.default(partial)

        if not cmd:
            # no sql, noop
            return False

        start = time.time()
        # execute the SQL command
        try:
            # a weak attempt to see if we have any %(keyword)s to expand
            if r'%(' in cmd:
                cmd = cmd % self.db.keywords
            self.cu.execute(cmd)
        except sqlerrors.DatabaseError, e:
            if len(e.args) > 1:
                print 'Error:', str(e.args[0])
            else:
                print 'Error:', str(e)
            return False