← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/text.py

  • Committer: Martin Pool
  • Date: 2010-04-01 04:41:18 UTC
  • mto: This revision was merged to the branch mainline in revision 5128.
  • Revision ID: mbp@sourcefrog.net-20100401044118-shyctqc02ob08ngz
ignore .testrepository

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/ui/text.py
37
37
 
38
38
""")
39
39
 
 
40
from bzrlib.osutils import watch_sigwinch
 
41
 
40
42
from bzrlib.ui import (
41
43
    UIFactory,
42
44
    NullProgressView,
60
62
        self.stderr = stderr
61
63
        # paints progress, network activity, etc
62
64
        self._progress_view = self.make_progress_view()
 
65
        # hook up the signals to watch for terminal size changes
 
66
        watch_sigwinch()
63
67
 
64
68
    def be_quiet(self, state):
65
69
        if state and not self._quiet:
153
157
        """Construct and return a new ProgressView subclass for this UI.
154
158
        """
155
159
        # with --quiet, never any progress view
156
 
        # <https://bugs.launchpad.net/bzr/+bug/320035>.  Otherwise if the
 
160
        # <https://bugs.edge.launchpad.net/bzr/+bug/320035>.  Otherwise if the
157
161
        # user specifically requests either text or no progress bars, always
158
162
        # do that.  otherwise, guess based on $TERM and tty presence.
159
163
        if self.is_quiet():
229
233
 
230
234
    def show_warning(self, msg):
231
235
        self.clear_term()
232
 
        if isinstance(msg, unicode):
233
 
            te = osutils.get_terminal_encoding()
234
 
            msg = msg.encode(te, 'replace')
235
236
        self.stderr.write("bzr: warning: %s\n" % msg)
236
237
 
237
238
    def _progress_updated(self, task):
300
301
        # correspond reliably to overall command progress
301
302
        self.enable_bar = False
302
303
 
303
 
    def _avail_width(self):
304
 
        # we need one extra space for terminals that wrap on last char
305
 
        w = osutils.terminal_width() 
306
 
        if w is None:
307
 
            return None
308
 
        else:
309
 
            return w - 1
310
 
 
311
304
    def _show_line(self, s):
312
305
        # sys.stderr.write("progress %r\n" % s)
313
 
        width = self._avail_width()
 
306
        width = osutils.terminal_width()
314
307
        if width is not None:
 
308
            # we need one extra space for terminals that wrap on last char
 
309
            width = width - 1
315
310
            s = '%-*.*s' % (width, width, s)
316
311
        self._term_file.write('\r' + s + '\r')
317
312
 
354
349
            return ''
355
350
 
356
351
    def _format_task(self, task):
357
 
        """Format task-specific parts of progress bar.
358
 
 
359
 
        :returns: (text_part, counter_part) both unicode strings.
360
 
        """
361
352
        if not task.show_count:
362
353
            s = ''
363
354
        elif task.current_cnt is not None and task.total_cnt is not None:
373
364
            t = t._parent_task
374
365
            if t.msg:
375
366
                m = t.msg + ':' + m
376
 
        return m, s
 
367
        return m + s
377
368
 
378
369
    def _render_line(self):
379
370
        bar_string = self._render_bar()
380
371
        if self._last_task:
381
 
            task_part, counter_part = self._format_task(self._last_task)
 
372
            task_msg = self._format_task(self._last_task)
382
373
        else:
383
 
            task_part = counter_part = ''
 
374
            task_msg = ''
384
375
        if self._last_task and not self._last_task.show_transport_activity:
385
376
            trans = ''
386
377
        else:
387
378
            trans = self._last_transport_msg
388
 
        # the bar separates the transport activity from the message, so even
389
 
        # if there's no bar or spinner, we must show something if both those
390
 
        # fields are present
391
 
        if (task_part or trans) and not bar_string:
392
 
            bar_string = '| '
393
 
        # preferentially truncate the task message if we don't have enough
394
 
        # space
395
 
        avail_width = self._avail_width()
396
 
        if avail_width is not None:
397
 
            # if terminal avail_width is unknown, don't truncate
398
 
            current_len = len(bar_string) + len(trans) + len(task_part) + len(counter_part)
399
 
            gap = current_len - avail_width
400
 
            if gap > 0:
401
 
                task_part = task_part[:-gap-2] + '..'
402
 
        s = trans + bar_string + task_part + counter_part
403
 
        if avail_width is not None:
404
 
            if len(s) < avail_width:
405
 
                s = s.ljust(avail_width)
406
 
            elif len(s) > avail_width:
407
 
                s = s[:avail_width]
408
 
        return s
 
379
            if trans:
 
380
                trans += ' | '
 
381
        return (bar_string + trans + task_msg)
409
382
 
410
383
    def _repaint(self):
411
384
        s = self._render_line()
467
440
            rate = (self._bytes_since_update
468
441
                    / (now - self._transport_update_time))
469
442
            # using base-10 units (see HACKING.txt).
470
 
            msg = ("%6dkB %5dkB/s " %
 
443
            msg = ("%6dkB %5dkB/s" %
471
444
                    (self._total_byte_count / 1000, int(rate) / 1000,))
472
445
            self._transport_update_time = now
473
446
            self._last_repaint = now