comparison mercurial/ui.py @ 42119:2d428b859282

readline: provide styled prompt to readline (issue6070)

Differential Revision: https://phab.mercurial-scm.org/D6168

42118:967c098eed33

            # usually those are non-interactive
            return self._isatty(self._fout)

        return i

    def _readline(self):
        # Replacing stdin/stdout temporarily is a hard problem on Python 3
        # because they have to be text streams with *no buffering*. Instead,
        # we use rawinput() only if call_readline() will be invoked by
        # PyOS_Readline(), so no I/O will be made at Python layer.
        usereadline = (self._isatty(self._fin) and self._isatty(self._fout)

                readline.read_history_file
                # windows sometimes raises something other than ImportError
            except Exception:
                usereadline = False

        # prompt ' ' must exist; otherwise readline may delete entire line
        # - http://bugs.python.org/issue12833
        with self.timeblockedsection('stdio'):
            if usereadline:
                line = encoding.strtolocal(pycompat.rawinput(r' '))
                # When stdin is in binary mode on Windows, it can cause
                # raw_input() to emit an extra trailing carriage return
                if pycompat.oslinesep == b'\r\n' and line.endswith(b'\r'):
                    line = line[:-1]
            else:
                self._fout.write(b' ')
                self._fout.flush()
                line = self._fin.readline()
                if not line:
                    raise EOFError
                line = line.rstrip(pycompat.oslinesep)

        if not self.interactive():
            self._writemsg(self._fmsgout, msg, ' ', type='prompt', **opts)
            self._writemsg(self._fmsgout, default or '', "\n",
                           type='promptecho')
            return default
        self._writemsgnobuf(self._fmsgout, msg, type='prompt', **opts)
        self.flush()
        try:
            r = self._readline()
            if not r:
                r = default
            if self.configbool('ui', 'promptecho'):
                self._writemsg(self._fmsgout, r, "\n", type='promptecho')
            return r

42119:2d428b859282

            # usually those are non-interactive
            return self._isatty(self._fout)

        return i

    def _readline(self, prompt=' ', promptopts=None):
        # Replacing stdin/stdout temporarily is a hard problem on Python 3
        # because they have to be text streams with *no buffering*. Instead,
        # we use rawinput() only if call_readline() will be invoked by
        # PyOS_Readline(), so no I/O will be made at Python layer.
        usereadline = (self._isatty(self._fin) and self._isatty(self._fout)

                readline.read_history_file
                # windows sometimes raises something other than ImportError
            except Exception:
                usereadline = False

        if self._colormode == 'win32' or not usereadline:
            if not promptopts:
                promptopts = {}
            self._writemsgnobuf(self._fmsgout, prompt, type='prompt',
                                **promptopts)
            self.flush()
            prompt = ' '
        else:
            prompt = self.label(prompt, 'ui.prompt') + ' '

        # prompt ' ' must exist; otherwise readline may delete entire line
        # - http://bugs.python.org/issue12833
        with self.timeblockedsection('stdio'):
            if usereadline:
                line = encoding.strtolocal(pycompat.rawinput(prompt))
                # When stdin is in binary mode on Windows, it can cause
                # raw_input() to emit an extra trailing carriage return
                if pycompat.oslinesep == b'\r\n' and line.endswith(b'\r'):
                    line = line[:-1]
            else:
                self._fout.write(pycompat.bytestr(prompt))
                self._fout.flush()
                line = self._fin.readline()
                if not line:
                    raise EOFError
                line = line.rstrip(pycompat.oslinesep)

        if not self.interactive():
            self._writemsg(self._fmsgout, msg, ' ', type='prompt', **opts)
            self._writemsg(self._fmsgout, default or '', "\n",
                           type='promptecho')
            return default
        try:
            r = self._readline(prompt=msg, promptopts=opts)
            if not r:
                r = default
            if self.configbool('ui', 'promptecho'):
                self._writemsg(self._fmsgout, r, "\n", type='promptecho')
            return r