~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/annotate.py

  • Committer: Andrew Bennetts
  • Date: 2010-07-29 11:17:57 UTC
  • mfrom: (5050.3.17 2.2)
  • mto: This revision was merged to the branch mainline in revision 5365.
  • Revision ID: andrew.bennetts@canonical.com-20100729111757-018h3pcefo7z0dnq
Merge lp:bzr/2.2 into lp:bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
41
41
    )
42
42
from bzrlib.config import extract_email_address
43
43
from bzrlib.repository import _strip_NULL_ghosts
44
 
from bzrlib.revision import (
45
 
    CURRENT_REVISION,
46
 
    Revision,
47
 
    )
48
 
from bzrlib.symbol_versioning import (
49
 
    deprecated_function,
50
 
    deprecated_in,
51
 
    )
52
 
 
53
 
 
54
 
@deprecated_function(deprecated_in((2, 4, 0)))
 
44
from bzrlib.revision import CURRENT_REVISION, Revision
 
45
 
 
46
 
55
47
def annotate_file(branch, rev_id, file_id, verbose=False, full=False,
56
48
                  to_file=None, show_ids=False):
57
49
    """Annotate file_id at revision rev_id in branch.
68
60
        used.
69
61
    :param show_ids: Show revision ids in the annotation output.
70
62
    """
71
 
    tree = branch.repository.revision_tree(rev_id)
72
 
    annotate_file_tree(tree, file_id, to_file, verbose=verbose,
73
 
        full=full, show_ids=show_ids, branch=branch)
 
63
    if to_file is None:
 
64
        to_file = sys.stdout
 
65
 
 
66
    # Handle the show_ids case
 
67
    annotations = _annotations(branch.repository, file_id, rev_id)
 
68
    if show_ids:
 
69
        return _show_id_annotations(annotations, to_file, full)
 
70
 
 
71
    # Calculate the lengths of the various columns
 
72
    annotation = list(_expand_annotations(annotations, branch))
 
73
    _print_annotations(annotation, verbose, to_file, full)
74
74
 
75
75
 
76
76
def annotate_file_tree(tree, file_id, to_file, verbose=False, full=False,
77
 
    show_ids=False, branch=None):
 
77
    show_ids=False):
78
78
    """Annotate file_id in a tree.
79
79
 
80
80
    The tree should already be read_locked() when annotate_file_tree is called.
86
86
        reasonable text width.
87
87
    :param full: XXXX Not sure what this does.
88
88
    :param show_ids: Show revision ids in the annotation output.
89
 
    :param branch: Branch to use for revision revno lookups
90
89
    """
91
 
    if branch is None:
92
 
        branch = tree.branch
93
 
    if to_file is None:
94
 
        to_file = sys.stdout
 
90
    rev_id = tree.last_revision()
 
91
    branch = tree.branch
95
92
 
96
93
    # Handle the show_ids case
97
94
    annotations = list(tree.annotate_iter(file_id))
98
95
    if show_ids:
99
96
        return _show_id_annotations(annotations, to_file, full)
100
97
 
101
 
    if not getattr(tree, "get_revision_id", False):
102
 
        # Create a virtual revision to represent the current tree state.
103
 
        # Should get some more pending commit attributes, like pending tags,
104
 
        # bugfixes etc.
105
 
        current_rev = Revision(CURRENT_REVISION)
106
 
        current_rev.parent_ids = tree.get_parent_ids()
107
 
        try:
108
 
            current_rev.committer = branch.get_config().username()
109
 
        except errors.NoWhoami:
110
 
            current_rev.committer = 'local user'
111
 
        current_rev.message = "?"
112
 
        current_rev.timestamp = round(time.time(), 3)
113
 
        current_rev.timezone = osutils.local_time_offset()
114
 
    else:
115
 
        current_rev = None
116
 
    annotation = list(_expand_annotations(annotations, branch,
 
98
    # Create a virtual revision to represent the current tree state.
 
99
    # Should get some more pending commit attributes, like pending tags,
 
100
    # bugfixes etc.
 
101
    current_rev = Revision(CURRENT_REVISION)
 
102
    current_rev.parent_ids = tree.get_parent_ids()
 
103
    current_rev.committer = tree.branch.get_config().username()
 
104
    current_rev.message = "?"
 
105
    current_rev.timestamp = round(time.time(), 3)
 
106
    current_rev.timezone = osutils.local_time_offset()
 
107
    annotation = list(_expand_annotations(annotations, tree.branch,
117
108
        current_rev))
118
109
    _print_annotations(annotation, verbose, to_file, full)
119
110
 
179
170
    return
180
171
 
181
172
 
 
173
def _annotations(repo, file_id, rev_id):
 
174
    """Return the list of (origin_revision_id, line_text) for a revision of a file in a repository."""
 
175
    annotations = repo.texts.annotate((file_id, rev_id))
 
176
    #
 
177
    return [(key[-1], line) for (key, line) in annotations]
 
178
 
 
179
 
182
180
def _expand_annotations(annotations, branch, current_rev=None):
183
181
    """Expand a file's annotations into command line UI ready tuples.
184
182
 
191
189
    """
192
190
    repository = branch.repository
193
191
    if current_rev is not None:
194
 
        # This can probably become a function on MutableTree, get_revno_map
195
 
        # there, or something.
 
192
        # This can probably become a function on MutableTree, get_revno_map there,
 
193
        # or something.
196
194
        last_revision = current_rev.revision_id
197
195
        # XXX: Partially Cloned from branch, uses the old_get_graph, eep.
198
196
        # XXX: The main difficulty is that we need to inject a single new node