← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/text.py

Merge sftp-leaks into catch-them-all

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/ui/text.py
300
300
        # correspond reliably to overall command progress
301
301
        self.enable_bar = False
302
302
 
 
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
 
303
311
    def _show_line(self, s):
304
312
        # sys.stderr.write("progress %r\n" % s)
305
 
        width = osutils.terminal_width()
 
313
        width = self._avail_width()
306
314
        if width is not None:
307
 
            # we need one extra space for terminals that wrap on last char
308
 
            width = width - 1
309
315
            s = '%-*.*s' % (width, width, s)
310
316
        self._term_file.write('\r' + s + '\r')
311
317
 
348
354
            return ''
349
355
 
350
356
    def _format_task(self, task):
 
357
        """Format task-specific parts of progress bar.
 
358
 
 
359
        :returns: (text_part, counter_part) both unicode strings.
 
360
        """
351
361
        if not task.show_count:
352
362
            s = ''
353
363
        elif task.current_cnt is not None and task.total_cnt is not None:
363
373
            t = t._parent_task
364
374
            if t.msg:
365
375
                m = t.msg + ':' + m
366
 
        return m + s
 
376
        return m, s
367
377
 
368
378
    def _render_line(self):
369
379
        bar_string = self._render_bar()
370
380
        if self._last_task:
371
 
            task_msg = self._format_task(self._last_task)
 
381
            task_part, counter_part = self._format_task(self._last_task)
372
382
        else:
373
 
            task_msg = ''
 
383
            task_part = counter_part = ''
374
384
        if self._last_task and not self._last_task.show_transport_activity:
375
385
            trans = ''
376
386
        else:
377
387
            trans = self._last_transport_msg
378
 
            if trans:
379
 
                trans += ' | '
380
 
        return (bar_string + trans + task_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
381
409
 
382
410
    def _repaint(self):
383
411
        s = self._render_line()
439
467
            rate = (self._bytes_since_update
440
468
                    / (now - self._transport_update_time))
441
469
            # using base-10 units (see HACKING.txt).
442
 
            msg = ("%6dkB %5dkB/s" %
 
470
            msg = ("%6dkB %5dkB/s " %
443
471
                    (self._total_byte_count / 1000, int(rate) / 1000,))
444
472
            self._transport_update_time = now
445
473
            self._last_repaint = now