comparison mercurial/ui.py @ 40521:49746e53ac92

ui: simplify interface of low-level write() functions

_write() and _write_err() will be replaced with fout.write() and ferr.write()
respectively. This is the first step.

40520:fd60c2afb484

        else:
            msgs = args
            if self._colormode is not None:
                label = opts.get(r'label', '')
                msgs = [self.label(a, label) for a in args]
            write(*msgs, **opts)

    def _write(self, *msgs, **opts):
        # opencode timeblockedsection because this is a critical path
        starttime = util.timer()
        try:
            self.fout.write(''.join(msgs))
        except IOError as err:
            raise error.StdioError(err)
        finally:
            self._blockedtimes['stdio_blocked'] += \
                (util.timer() - starttime) * 1000

        if self._bufferstates and self._bufferstates[-1][0]:
            self.write(*args, **opts)
        else:
            self._writenobuf(self._write_err, *args, **opts)

    def _write_err(self, *msgs, **opts):
        try:
            with self.timeblockedsection('stdio'):
                if not getattr(self.fout, 'closed', False):
                    self.fout.flush()
                for a in msgs:
                    self.ferr.write(a)
                # stderr may be buffered under win32 when redirected to files,
                # including stdout.
                if not getattr(self.ferr, 'closed', False):
                    self.ferr.flush()
        except IOError as inst:

40521:49746e53ac92

        else:
            msgs = args
            if self._colormode is not None:
                label = opts.get(r'label', '')
                msgs = [self.label(a, label) for a in args]
            write(b''.join(msgs))

    def _write(self, data):
        # opencode timeblockedsection because this is a critical path
        starttime = util.timer()
        try:
            self.fout.write(data)
        except IOError as err:
            raise error.StdioError(err)
        finally:
            self._blockedtimes['stdio_blocked'] += \
                (util.timer() - starttime) * 1000

        if self._bufferstates and self._bufferstates[-1][0]:
            self.write(*args, **opts)
        else:
            self._writenobuf(self._write_err, *args, **opts)

    def _write_err(self, data):
        try:
            with self.timeblockedsection('stdio'):
                if not getattr(self.fout, 'closed', False):
                    self.fout.flush()
                self.ferr.write(data)
                # stderr may be buffered under win32 when redirected to files,
                # including stdout.
                if not getattr(self.ferr, 'closed', False):
                    self.ferr.flush()
        except IOError as inst: