← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/text.py

  • Committer: Robert Collins
  • Date: 2010-06-28 02:41:22 UTC
  • mto: This revision was merged to the branch mainline in revision 5324.
  • Revision ID: robertc@robertcollins.net-20100628024122-g951fzp74f3u6wst
Sanity check that new_trace_file in pop_log_file is valid, and also fix a test that monkey patched get_terminal_encoding.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/ui/text.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
18
18
"""Text UI, write output to the console.
19
19
"""
20
20
 
 
21
import codecs
 
22
import getpass
21
23
import os
22
24
import sys
23
25
import time
 
26
import warnings
24
27
 
25
28
from bzrlib.lazy_import import lazy_import
26
29
lazy_import(globals(), """
27
 
import codecs
28
 
import getpass
29
 
import warnings
30
 
 
31
30
from bzrlib import (
32
31
    debug,
33
32
    progress,
34
33
    osutils,
 
34
    symbol_versioning,
35
35
    trace,
36
36
    )
37
37
 
114
114
                password = password[:-1]
115
115
        return password
116
116
 
117
 
    def get_password(self, prompt=u'', **kwargs):
 
117
    def get_password(self, prompt='', **kwargs):
118
118
        """Prompt the user for a password.
119
119
 
120
120
        :param prompt: The prompt to present the user
198
198
        :param kwargs: Dictionary of arguments to insert into the prompt,
199
199
            to allow UIs to reformat the prompt.
200
200
        """
201
 
        if type(prompt) != unicode:
202
 
            raise ValueError("prompt %r not a unicode string" % prompt)
203
201
        if kwargs:
204
202
            # See <https://launchpad.net/bugs/365891>
205
203
            prompt = prompt % kwargs
302
300
        # correspond reliably to overall command progress
303
301
        self.enable_bar = False
304
302
 
305
 
    def _avail_width(self):
306
 
        # we need one extra space for terminals that wrap on last char
307
 
        w = osutils.terminal_width() 
308
 
        if w is None:
309
 
            return None
310
 
        else:
311
 
            return w - 1
312
 
 
313
303
    def _show_line(self, s):
314
304
        # sys.stderr.write("progress %r\n" % s)
315
 
        width = self._avail_width()
 
305
        width = osutils.terminal_width()
316
306
        if width is not None:
 
307
            # we need one extra space for terminals that wrap on last char
 
308
            width = width - 1
317
309
            s = '%-*.*s' % (width, width, s)
318
310
        self._term_file.write('\r' + s + '\r')
319
311
 
356
348
            return ''
357
349
 
358
350
    def _format_task(self, task):
359
 
        """Format task-specific parts of progress bar.
360
 
 
361
 
        :returns: (text_part, counter_part) both unicode strings.
362
 
        """
363
351
        if not task.show_count:
364
352
            s = ''
365
353
        elif task.current_cnt is not None and task.total_cnt is not None:
375
363
            t = t._parent_task
376
364
            if t.msg:
377
365
                m = t.msg + ':' + m
378
 
        return m, s
 
366
        return m + s
379
367
 
380
368
    def _render_line(self):
381
369
        bar_string = self._render_bar()
382
370
        if self._last_task:
383
 
            task_part, counter_part = self._format_task(self._last_task)
 
371
            task_msg = self._format_task(self._last_task)
384
372
        else:
385
 
            task_part = counter_part = ''
 
373
            task_msg = ''
386
374
        if self._last_task and not self._last_task.show_transport_activity:
387
375
            trans = ''
388
376
        else:
389
377
            trans = self._last_transport_msg
390
 
        # the bar separates the transport activity from the message, so even
391
 
        # if there's no bar or spinner, we must show something if both those
392
 
        # fields are present
393
 
        if (task_part or trans) and not bar_string:
394
 
            bar_string = '| '
395
 
        # preferentially truncate the task message if we don't have enough
396
 
        # space
397
 
        avail_width = self._avail_width()
398
 
        if avail_width is not None:
399
 
            # if terminal avail_width is unknown, don't truncate
400
 
            current_len = len(bar_string) + len(trans) + len(task_part) + len(counter_part)
401
 
            gap = current_len - avail_width
402
 
            if gap > 0:
403
 
                task_part = task_part[:-gap-2] + '..'
404
 
        s = trans + bar_string + task_part + counter_part
405
 
        if avail_width is not None:
406
 
            if len(s) < avail_width:
407
 
                s = s.ljust(avail_width)
408
 
            elif len(s) > avail_width:
409
 
                s = s[:avail_width]
410
 
        return s
 
378
            if trans:
 
379
                trans += ' | '
 
380
        return (bar_string + trans + task_msg)
411
381
 
412
382
    def _repaint(self):
413
383
        s = self._render_line()
469
439
            rate = (self._bytes_since_update
470
440
                    / (now - self._transport_update_time))
471
441
            # using base-10 units (see HACKING.txt).
472
 
            msg = ("%6dkB %5dkB/s " %
 
442
            msg = ("%6dkB %5dkB/s" %
473
443
                    (self._total_byte_count / 1000, int(rate) / 1000,))
474
444
            self._transport_update_time = now
475
445
            self._last_repaint = now