~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/annotate.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-01-15 04:49:48 UTC
  • mfrom: (3984.5.22 switch-r-183559)
  • Revision ID: pqm@pqm.ubuntu.com-20100115044948-yxz5m3vchxapbq22
(andrew) Add --revision option to 'bzr switch'. (#184559)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 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
16
16
 
17
17
"""File annotate based on weave storage"""
18
18
 
19
 
from __future__ import absolute_import
20
 
 
21
19
# TODO: Choice of more or less verbose formats:
22
20
#
23
21
# interposed: show more details between blocks of modified lines
30
28
import sys
31
29
import time
32
30
 
33
 
from bzrlib.lazy_import import lazy_import
34
 
lazy_import(globals(), """
35
31
from bzrlib import (
 
32
    errors,
 
33
    osutils,
36
34
    patiencediff,
37
35
    tsort,
38
36
    )
39
 
""")
40
 
from bzrlib import (
41
 
    errors,
42
 
    osutils,
43
 
    )
44
37
from bzrlib.config import extract_email_address
45
38
from bzrlib.repository import _strip_NULL_ghosts
46
 
from bzrlib.revision import (
47
 
    CURRENT_REVISION,
48
 
    Revision,
49
 
    )
 
39
from bzrlib.revision import CURRENT_REVISION, Revision
 
40
 
 
41
 
 
42
def annotate_file(branch, rev_id, file_id, verbose=False, full=False,
 
43
                  to_file=None, show_ids=False):
 
44
    """Annotate file_id at revision rev_id in branch.
 
45
 
 
46
    The branch should already be read_locked() when annotate_file is called.
 
47
 
 
48
    :param branch: The branch to look for revision numbers and history from.
 
49
    :param rev_id: The revision id to annotate.
 
50
    :param file_id: The file_id to annotate.
 
51
    :param verbose: Show all details rather than truncating to ensure
 
52
        reasonable text width.
 
53
    :param full: XXXX Not sure what this does.
 
54
    :param to_file: The file to output the annotation to; if None stdout is
 
55
        used.
 
56
    :param show_ids: Show revision ids in the annotation output.
 
57
    """
 
58
    if to_file is None:
 
59
        to_file = sys.stdout
 
60
 
 
61
    # Handle the show_ids case
 
62
    annotations = _annotations(branch.repository, file_id, rev_id)
 
63
    if show_ids:
 
64
        return _show_id_annotations(annotations, to_file, full)
 
65
 
 
66
    # Calculate the lengths of the various columns
 
67
    annotation = list(_expand_annotations(annotations, branch))
 
68
    _print_annotations(annotation, verbose, to_file, full)
50
69
 
51
70
 
52
71
def annotate_file_tree(tree, file_id, to_file, verbose=False, full=False,
53
 
    show_ids=False, branch=None):
 
72
    show_ids=False):
54
73
    """Annotate file_id in a tree.
55
74
 
56
75
    The tree should already be read_locked() when annotate_file_tree is called.
62
81
        reasonable text width.
63
82
    :param full: XXXX Not sure what this does.
64
83
    :param show_ids: Show revision ids in the annotation output.
65
 
    :param branch: Branch to use for revision revno lookups
66
84
    """
67
 
    if branch is None:
68
 
        branch = tree.branch
69
 
    if to_file is None:
70
 
        to_file = sys.stdout
 
85
    rev_id = tree.last_revision()
 
86
    branch = tree.branch
71
87
 
72
88
    # Handle the show_ids case
73
89
    annotations = list(tree.annotate_iter(file_id))
74
90
    if show_ids:
75
91
        return _show_id_annotations(annotations, to_file, full)
76
92
 
77
 
    if not getattr(tree, "get_revision_id", False):
78
 
        # Create a virtual revision to represent the current tree state.
79
 
        # Should get some more pending commit attributes, like pending tags,
80
 
        # bugfixes etc.
81
 
        current_rev = Revision(CURRENT_REVISION)
82
 
        current_rev.parent_ids = tree.get_parent_ids()
83
 
        try:
84
 
            current_rev.committer = branch.get_config_stack().get('email')
85
 
        except errors.NoWhoami:
86
 
            current_rev.committer = 'local user'
87
 
        current_rev.message = "?"
88
 
        current_rev.timestamp = round(time.time(), 3)
89
 
        current_rev.timezone = osutils.local_time_offset()
90
 
    else:
91
 
        current_rev = None
92
 
    annotation = list(_expand_annotations(annotations, branch,
 
93
    # Create a virtual revision to represent the current tree state.
 
94
    # Should get some more pending commit attributes, like pending tags,
 
95
    # bugfixes etc.
 
96
    current_rev = Revision(CURRENT_REVISION)
 
97
    current_rev.parent_ids = tree.get_parent_ids()
 
98
    current_rev.committer = tree.branch.get_config().username()
 
99
    current_rev.message = "?"
 
100
    current_rev.timestamp = round(time.time(), 3)
 
101
    current_rev.timezone = osutils.local_time_offset()
 
102
    annotation = list(_expand_annotations(annotations, tree.branch,
93
103
        current_rev))
94
104
    _print_annotations(annotation, verbose, to_file, full)
95
105
 
155
165
    return
156
166
 
157
167
 
 
168
def _annotations(repo, file_id, rev_id):
 
169
    """Return the list of (origin_revision_id, line_text) for a revision of a file in a repository."""
 
170
    annotations = repo.texts.annotate((file_id, rev_id))
 
171
    #
 
172
    return [(key[-1], line) for (key, line) in annotations]
 
173
 
 
174
 
158
175
def _expand_annotations(annotations, branch, current_rev=None):
159
176
    """Expand a file's annotations into command line UI ready tuples.
160
177
 
167
184
    """
168
185
    repository = branch.repository
169
186
    if current_rev is not None:
170
 
        # This can probably become a function on MutableTree, get_revno_map
171
 
        # there, or something.
 
187
        # This can probably become a function on MutableTree, get_revno_map there,
 
188
        # or something.
172
189
        last_revision = current_rev.revision_id
173
190
        # XXX: Partially Cloned from branch, uses the old_get_graph, eep.
174
191
        # XXX: The main difficulty is that we need to inject a single new node
295
312
 
296
313
 
297
314
def _get_matching_blocks(old, new):
298
 
    matcher = patiencediff.PatienceSequenceMatcher(None, old, new)
 
315
    matcher = patiencediff.PatienceSequenceMatcher(None,
 
316
        old, new)
299
317
    return matcher.get_matching_blocks()
300
318
 
301
319