/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/annotate.py

  • Committer: Jelmer Vernooij
  • Date: 2020-04-05 19:11:34 UTC
  • mto: (7490.7.16 work)
  • mto: This revision was merged to the branch mainline in revision 7501.
  • Revision ID: jelmer@jelmer.uk-20200405191134-0aebh8ikiwygxma5
Populate the .gitignore file.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/annotate.py
1
 
# Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 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
16
16
 
17
17
"""File annotate based on weave storage"""
18
18
 
 
19
from __future__ import absolute_import
 
20
 
19
21
# TODO: Choice of more or less verbose formats:
20
22
#
21
23
# interposed: show more details between blocks of modified lines
28
30
import sys
29
31
import time
30
32
 
31
 
from bzrlib import (
 
33
from .lazy_import import lazy_import
 
34
lazy_import(globals(), """
 
35
 
 
36
import patiencediff
 
37
 
 
38
from breezy import (
 
39
    tsort,
 
40
    )
 
41
""")
 
42
from . import (
 
43
    config,
32
44
    errors,
33
45
    osutils,
34
 
    patiencediff,
35
 
    tsort,
36
 
    )
37
 
from bzrlib.config import extract_email_address
38
 
from bzrlib.repository import _strip_NULL_ghosts
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.
 
46
    )
 
47
from .repository import _strip_NULL_ghosts
 
48
from .revision import (
 
49
    CURRENT_REVISION,
 
50
    Revision,
 
51
    )
 
52
 
 
53
 
 
54
def annotate_file_tree(tree, path, to_file, verbose=False, full=False,
 
55
                       show_ids=False, branch=None):
 
56
    """Annotate path in a tree.
 
57
 
 
58
    The tree should already be read_locked() when annotate_file_tree is called.
 
59
 
 
60
    :param tree: The tree to look for revision numbers and history from.
 
61
    :param path: The path to annotate
 
62
    :param to_file: The file to output the annotation to.
51
63
    :param verbose: Show all details rather than truncating to ensure
52
64
        reasonable text width.
53
65
    :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
66
    :param show_ids: Show revision ids in the annotation output.
 
67
    :param branch: Branch to use for revision revno lookups
57
68
    """
 
69
    if branch is None:
 
70
        branch = tree.branch
58
71
    if to_file is None:
59
72
        to_file = sys.stdout
60
73
 
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)
69
 
 
70
 
 
71
 
def annotate_file_tree(tree, file_id, to_file, verbose=False, full=False,
72
 
    show_ids=False):
73
 
    """Annotate file_id in a tree.
74
 
 
75
 
    The tree should already be read_locked() when annotate_file_tree is called.
76
 
 
77
 
    :param tree: The tree to look for revision numbers and history from.
78
 
    :param file_id: The file_id to annotate.
79
 
    :param to_file: The file to output the annotation to.
80
 
    :param verbose: Show all details rather than truncating to ensure
81
 
        reasonable text width.
82
 
    :param full: XXXX Not sure what this does.
83
 
    :param show_ids: Show revision ids in the annotation output.
84
 
    """
85
 
    rev_id = tree.last_revision()
86
 
    branch = tree.branch
87
 
 
88
 
    # Handle the show_ids case
89
 
    annotations = list(tree.annotate_iter(file_id))
90
 
    if show_ids:
91
 
        return _show_id_annotations(annotations, to_file, full)
92
 
 
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,
103
 
        current_rev))
104
 
    _print_annotations(annotation, verbose, to_file, full)
105
 
 
106
 
 
107
 
def _print_annotations(annotation, verbose, to_file, full):
 
74
    encoding = osutils.get_terminal_encoding()
 
75
    # Handle the show_ids case
 
76
    annotations = list(tree.annotate_iter(path))
 
77
    if show_ids:
 
78
        return _show_id_annotations(annotations, to_file, full, encoding)
 
79
 
 
80
    if not getattr(tree, "get_revision_id", False):
 
81
        # Create a virtual revision to represent the current tree state.
 
82
        # Should get some more pending commit attributes, like pending tags,
 
83
        # bugfixes etc.
 
84
        current_rev = Revision(CURRENT_REVISION)
 
85
        current_rev.parent_ids = tree.get_parent_ids()
 
86
        try:
 
87
            current_rev.committer = branch.get_config_stack().get('email')
 
88
        except errors.NoWhoami:
 
89
            current_rev.committer = 'local user'
 
90
        current_rev.message = "?"
 
91
        current_rev.timestamp = round(time.time(), 3)
 
92
        current_rev.timezone = osutils.local_time_offset()
 
93
    else:
 
94
        current_rev = None
 
95
    annotation = list(_expand_annotations(
 
96
        annotations, branch, current_rev))
 
97
    _print_annotations(annotation, verbose, to_file, full, encoding)
 
98
 
 
99
 
 
100
def _print_annotations(annotation, verbose, to_file, full, encoding):
108
101
    """Print annotations to to_file.
109
102
 
110
103
    :param to_file: The file to output the annotation to.
113
106
    :param full: XXXX Not sure what this does.
114
107
    """
115
108
    if len(annotation) == 0:
116
 
        max_origin_len = max_revno_len = max_revid_len = 0
 
109
        max_origin_len = max_revno_len = 0
117
110
    else:
118
111
        max_origin_len = max(len(x[1]) for x in annotation)
119
112
        max_revno_len = max(len(x[0]) for x in annotation)
120
 
        max_revid_len = max(len(x[3]) for x in annotation)
121
113
    if not verbose:
122
114
        max_revno_len = min(max_revno_len, 12)
123
115
    max_revno_len = max(max_revno_len, 3)
124
116
 
125
117
    # Output the annotations
126
118
    prevanno = ''
127
 
    encoding = getattr(to_file, 'encoding', None) or \
128
 
            osutils.get_terminal_encoding()
129
119
    for (revno_str, author, date_str, line_rev_id, text) in annotation:
130
120
        if verbose:
131
121
            anno = '%-*s %-*s %8s ' % (max_revno_len, revno_str,
132
122
                                       max_origin_len, author, date_str)
133
123
        else:
134
124
            if len(revno_str) > max_revno_len:
135
 
                revno_str = revno_str[:max_revno_len-1] + '>'
 
125
                revno_str = revno_str[:max_revno_len - 1] + '>'
136
126
            anno = "%-*s %-7s " % (max_revno_len, revno_str, author[:7])
137
127
        if anno.lstrip() == "" and full:
138
128
            anno = prevanno
139
 
        try:
140
 
            to_file.write(anno)
141
 
        except UnicodeEncodeError:
142
 
            # cmd_annotate should be passing in an 'exact' object, which means
143
 
            # we have a direct handle to sys.stdout or equivalent. It may not
144
 
            # be able to handle the exact Unicode characters, but 'annotate' is
145
 
            # a user function (non-scripting), so shouldn't die because of
146
 
            # unrepresentable annotation characters. So encode using 'replace',
147
 
            # and write them again.
148
 
            to_file.write(anno.encode(encoding, 'replace'))
149
 
        to_file.write('| %s\n' % (text,))
 
129
        # GZ 2017-05-21: Writing both unicode annotation and bytes from file
 
130
        # which the given to_file must cope with.
 
131
        to_file.write(anno)
 
132
        to_file.write('| %s\n' % (text.decode(encoding),))
150
133
        prevanno = anno
151
134
 
152
135
 
153
 
def _show_id_annotations(annotations, to_file, full):
 
136
def _show_id_annotations(annotations, to_file, full, encoding):
154
137
    if not annotations:
155
138
        return
156
139
    last_rev_id = None
159
142
        if full or last_rev_id != origin:
160
143
            this = origin
161
144
        else:
162
 
            this = ''
163
 
        to_file.write('%*s | %s' % (max_origin_len, this, text))
 
145
            this = b''
 
146
        to_file.write('%*s | %s' % (
 
147
            max_origin_len, this.decode('utf-8'), text.decode(encoding)))
164
148
        last_rev_id = origin
165
149
    return
166
150
 
167
151
 
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
 
 
175
152
def _expand_annotations(annotations, branch, current_rev=None):
176
153
    """Expand a file's annotations into command line UI ready tuples.
177
154
 
183
160
    :param branch: A locked branch to query for revision details.
184
161
    """
185
162
    repository = branch.repository
 
163
    revision_ids = set(o for o, t in annotations)
186
164
    if current_rev is not None:
187
 
        # This can probably become a function on MutableTree, get_revno_map there,
188
 
        # or something.
 
165
        # This can probably become a function on MutableTree, get_revno_map
 
166
        # there, or something.
189
167
        last_revision = current_rev.revision_id
190
168
        # XXX: Partially Cloned from branch, uses the old_get_graph, eep.
191
169
        # XXX: The main difficulty is that we need to inject a single new node
193
171
        #      Once KnownGraph gets an 'add_node()' function, we can use
194
172
        #      VF.get_known_graph_ancestry().
195
173
        graph = repository.get_graph()
196
 
        revision_graph = dict(((key, value) for key, value in
197
 
            graph.iter_ancestry(current_rev.parent_ids) if value is not None))
 
174
        revision_graph = {
 
175
            key: value for key, value in
 
176
            graph.iter_ancestry(current_rev.parent_ids) if value is not None}
198
177
        revision_graph = _strip_NULL_ghosts(revision_graph)
199
178
        revision_graph[last_revision] = current_rev.parent_ids
200
179
        merge_sorted_revisions = tsort.merge_sort(
202
181
            last_revision,
203
182
            None,
204
183
            generate_revno=True)
205
 
        revision_id_to_revno = dict((rev_id, revno)
 
184
        revision_id_to_revno = {
 
185
            rev_id: revno
206
186
            for seq_num, rev_id, depth, revno, end_of_merge in
207
 
                merge_sorted_revisions)
 
187
            merge_sorted_revisions}
208
188
    else:
 
189
        # TODO(jelmer): Only look up the revision ids that we need (i.e. those
 
190
        # in revision_ids). Possibly add a HPSS call that can look those up
 
191
        # in bulk over HPSS.
209
192
        revision_id_to_revno = branch.get_revision_id_to_revno_map()
210
193
    last_origin = None
211
 
    revision_ids = set(o for o, t in annotations)
212
194
    revisions = {}
213
195
    if CURRENT_REVISION in revision_ids:
214
196
        revision_id_to_revno[CURRENT_REVISION] = (
215
197
            "%d?" % (branch.revno() + 1),)
216
198
        revisions[CURRENT_REVISION] = current_rev
217
 
    revision_ids = [o for o in revision_ids if
218
 
                    repository.has_revision(o)]
219
 
    revisions.update((r.revision_id, r) for r in
220
 
                     repository.get_revisions(revision_ids))
 
199
    revisions.update(
 
200
        entry for entry in
 
201
        repository.iter_revisions(revision_ids)
 
202
        if entry[1] is not None)
221
203
    for origin, text in annotations:
222
 
        text = text.rstrip('\r\n')
 
204
        text = text.rstrip(b'\r\n')
223
205
        if origin == last_origin:
224
 
            (revno_str, author, date_str) = ('','','')
 
206
            (revno_str, author, date_str) = ('', '', '')
225
207
        else:
226
208
            last_origin = origin
227
209
            if origin not in revisions:
228
 
                (revno_str, author, date_str) = ('?','?','?')
 
210
                (revno_str, author, date_str) = ('?', '?', '?')
229
211
            else:
230
 
                revno_str = '.'.join(str(i) for i in
231
 
                                            revision_id_to_revno[origin])
 
212
                revno_str = '.'.join(
 
213
                    str(i) for i in revision_id_to_revno[origin])
232
214
            rev = revisions[origin]
233
215
            tz = rev.timezone or 0
234
216
            date_str = time.strftime('%Y%m%d',
236
218
            # a lazy way to get something like the email address
237
219
            # TODO: Get real email address
238
220
            author = rev.get_apparent_authors()[0]
239
 
            try:
240
 
                author = extract_email_address(author)
241
 
            except errors.NoEmailInUsername:
242
 
                pass        # use the whole name
 
221
            _, email = config.parse_username(author)
 
222
            if email:
 
223
                author = email
243
224
        yield (revno_str, author, date_str, origin, text)
244
225
 
245
226
 
299
280
    new_cur = 0
300
281
    if matching_blocks is None:
301
282
        plain_parent_lines = [l for r, l in parent_lines]
302
 
        matcher = patiencediff.PatienceSequenceMatcher(None,
303
 
            plain_parent_lines, new_lines)
 
283
        matcher = patiencediff.PatienceSequenceMatcher(
 
284
            None, plain_parent_lines, new_lines)
304
285
        matching_blocks = matcher.get_matching_blocks()
305
286
    lines = []
306
287
    for i, j, n in matching_blocks:
307
288
        for line in new_lines[new_cur:j]:
308
289
            lines.append((new_revision_id, line))
309
 
        lines.extend(parent_lines[i:i+n])
 
290
        lines.extend(parent_lines[i:i + n])
310
291
        new_cur = j + n
311
292
    return lines
312
293
 
313
294
 
314
295
def _get_matching_blocks(old, new):
315
 
    matcher = patiencediff.PatienceSequenceMatcher(None,
316
 
        old, new)
 
296
    matcher = patiencediff.PatienceSequenceMatcher(None, old, new)
317
297
    return matcher.get_matching_blocks()
318
298
 
319
299
 
320
300
_break_annotation_tie = None
321
301
 
 
302
 
322
303
def _old_break_annotation_tie(annotated_lines):
323
304
    """Chose an attribution between several possible ones.
324
305
 
366
347
    output_extend = output_lines.extend
367
348
    output_append = output_lines.append
368
349
    # We need to see if any of the unannotated lines match
369
 
    plain_right_subset = [l for a,l in right_lines[start_right:end_right]]
 
350
    plain_right_subset = [l for a, l in right_lines[start_right:end_right]]
370
351
    plain_child_subset = plain_child_lines[start_child:end_child]
371
352
    match_blocks = _get_matching_blocks(plain_right_subset, plain_child_subset)
372
353
 
375
356
    for right_idx, child_idx, match_len in match_blocks:
376
357
        # All the lines that don't match are just passed along
377
358
        if child_idx > last_child_idx:
378
 
            output_extend(child_lines[start_child + last_child_idx
379
 
                                      :start_child + child_idx])
380
 
        for offset in xrange(match_len):
381
 
            left = child_lines[start_child+child_idx+offset]
382
 
            right = right_lines[start_right+right_idx+offset]
 
359
            output_extend(child_lines[start_child + last_child_idx:
 
360
                                      start_child + child_idx])
 
361
        for offset in range(match_len):
 
362
            left = child_lines[start_child + child_idx + offset]
 
363
            right = right_lines[start_right + right_idx + offset]
383
364
            if left[0] == right[0]:
384
365
                # The annotations match, just return the left one
385
366
                output_append(left)
394
375
                else:
395
376
                    heads = heads_provider.heads((left[0], right[0]))
396
377
                    if len(heads) == 1:
397
 
                        output_append((iter(heads).next(), left[1]))
 
378
                        output_append((next(iter(heads)), left[1]))
398
379
                    else:
399
380
                        # Both claim different origins, get a stable result.
400
381
                        # If the result is not stable, there is a risk a
429
410
    # be the bulk of the lines, and they will need no further processing.
430
411
    lines = []
431
412
    lines_extend = lines.extend
432
 
    last_right_idx = 0 # The line just after the last match from the right side
 
413
    # The line just after the last match from the right side
 
414
    last_right_idx = 0
433
415
    last_left_idx = 0
434
416
    matching_left_and_right = _get_matching_blocks(right_parent_lines,
435
417
                                                   annotated_lines)
457
439
 
458
440
 
459
441
try:
460
 
    from bzrlib._annotator_pyx import Annotator
461
 
except ImportError, e:
 
442
    from breezy._annotator_pyx import Annotator
 
443
except ImportError as e:
462
444
    osutils.failed_to_load_extension(e)
463
 
    from bzrlib._annotator_py import Annotator
 
445
    from breezy._annotator_py import Annotator  # noqa: F401