← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/text.py

  • Committer: Vincent Ladeuil
  • Date: 2010-04-28 10:33:44 UTC
  • mfrom: (5171.2.3 401599-strict-warnings)
  • mto: This revision was merged to the branch mainline in revision 5191.
  • Revision ID: v.ladeuil+lp@free.fr-20100428103344-e32qf3cn1avdd2cb
Don't mention --no-strict when we just issue the warning about unclean trees

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
 
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:
114
118
                password = password[:-1]
115
119
        return password
116
120
 
117
 
    def get_password(self, prompt=u'', **kwargs):
 
121
    def get_password(self, prompt='', **kwargs):
118
122
        """Prompt the user for a password.
119
123
 
120
124
        :param prompt: The prompt to present the user
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():
198
202
        :param kwargs: Dictionary of arguments to insert into the prompt,
199
203
            to allow UIs to reformat the prompt.
200
204
        """
201
 
        if type(prompt) != unicode:
202
 
            raise ValueError("prompt %r not a unicode string" % prompt)
203
205
        if kwargs:
204
206
            # See <https://launchpad.net/bugs/365891>
205
207
            prompt = prompt % kwargs
231
233
 
232
234
    def show_warning(self, msg):
233
235
        self.clear_term()
234
 
        if isinstance(msg, unicode):
235
 
            te = osutils.get_terminal_encoding()
236
 
            msg = msg.encode(te, 'replace')
237
236
        self.stderr.write("bzr: warning: %s\n" % msg)
238
237
 
239
238
    def _progress_updated(self, task):
302
301
        # correspond reliably to overall command progress
303
302
        self.enable_bar = False
304
303
 
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
304
    def _show_line(self, s):
314
305
        # sys.stderr.write("progress %r\n" % s)
315
 
        width = self._avail_width()
 
306
        width = osutils.terminal_width()
316
307
        if width is not None:
 
308
            # we need one extra space for terminals that wrap on last char
 
309
            width = width - 1
317
310
            s = '%-*.*s' % (width, width, s)
318
311
        self._term_file.write('\r' + s + '\r')
319
312
 
356
349
            return ''
357
350
 
358
351
    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
352
        if not task.show_count:
364
353
            s = ''
365
354
        elif task.current_cnt is not None and task.total_cnt is not None:
375
364
            t = t._parent_task
376
365
            if t.msg:
377
366
                m = t.msg + ':' + m
378
 
        return m, s
 
367
        return m + s
379
368
 
380
369
    def _render_line(self):
381
370
        bar_string = self._render_bar()
382
371
        if self._last_task:
383
 
            task_part, counter_part = self._format_task(self._last_task)
 
372
            task_msg = self._format_task(self._last_task)
384
373
        else:
385
 
            task_part = counter_part = ''
 
374
            task_msg = ''
386
375
        if self._last_task and not self._last_task.show_transport_activity:
387
376
            trans = ''
388
377
        else:
389
378
            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
 
379
            if trans:
 
380
                trans += ' | '
 
381
        return (bar_string + trans + task_msg)
411
382
 
412
383
    def _repaint(self):
413
384
        s = self._render_line()
469
440
            rate = (self._bytes_since_update
470
441
                    / (now - self._transport_update_time))
471
442
            # using base-10 units (see HACKING.txt).
472
 
            msg = ("%6dkB %5dkB/s " %
 
443
            msg = ("%6dkB %5dkB/s" %
473
444
                    (self._total_byte_count / 1000, int(rate) / 1000,))
474
445
            self._transport_update_time = now
475
446
            self._last_repaint = now