← Back to branch summary

~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: 2011-05-04 22:17:22 UTC
  • mfrom: (5815.3.15 use-tree-annotate)
  • Revision ID: pqm@pqm.ubuntu.com-20110504221722-fz5hr1xagchptyje
(jelmer) Avoid directly accessing VersionedFiles.annotate();
 rather, access it through RevisionTree.annotate_iter(). (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/annotate.py
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 CURRENT_REVISION, Revision
45
 
 
46
 
 
 
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)))
47
55
def annotate_file(branch, rev_id, file_id, verbose=False, full=False,
48
56
                  to_file=None, show_ids=False):
49
57
    """Annotate file_id at revision rev_id in branch.
60
68
        used.
61
69
    :param show_ids: Show revision ids in the annotation output.
62
70
    """
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)
 
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)
74
74
 
75
75
 
76
76
def annotate_file_tree(tree, file_id, to_file, verbose=False, full=False,
77
 
    show_ids=False):
 
77
    show_ids=False, branch=None):
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
89
90
    """
90
 
    rev_id = tree.last_revision()
91
 
    branch = tree.branch
 
91
    if branch is None:
 
92
        branch = tree.branch
 
93
    if to_file is None:
 
94
        to_file = sys.stdout
92
95
 
93
96
    # Handle the show_ids case
94
97
    annotations = list(tree.annotate_iter(file_id))
95
98
    if show_ids:
96
99
        return _show_id_annotations(annotations, to_file, full)
97
100
 
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,
 
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
        current_rev.committer = branch.get_config().username()
 
108
        current_rev.message = "?"
 
109
        current_rev.timestamp = round(time.time(), 3)
 
110
        current_rev.timezone = osutils.local_time_offset()
 
111
    else:
 
112
        current_rev = None
 
113
    annotation = list(_expand_annotations(annotations, branch,
108
114
        current_rev))
109
115
    _print_annotations(annotation, verbose, to_file, full)
110
116
 
170
176
    return
171
177
 
172
178
 
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
 
 
180
179
def _expand_annotations(annotations, branch, current_rev=None):
181
180
    """Expand a file's annotations into command line UI ready tuples.
182
181
 
189
188
    """
190
189
    repository = branch.repository
191
190
    if current_rev is not None:
192
 
        # This can probably become a function on MutableTree, get_revno_map there,
193
 
        # or something.
 
191
        # This can probably become a function on MutableTree, get_revno_map
 
192
        # there, or something.
194
193
        last_revision = current_rev.revision_id
195
194
        # XXX: Partially Cloned from branch, uses the old_get_graph, eep.
196
195
        # XXX: The main difficulty is that we need to inject a single new node