/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 bzrlib/patiencediff.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/patiencediff.py
15
15
# along with this program; if not, write to the Free Software
16
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
 
18
 
from __future__ import absolute_import
19
18
 
20
 
from .lazy_import import lazy_import
 
19
from bzrlib.lazy_import import lazy_import
21
20
lazy_import(globals(), """
22
21
import os
23
22
import sys
26
25
""")
27
26
 
28
27
 
29
 
__all__ = ['PatienceSequenceMatcher', 'unified_diff', 'unified_diff_bytes',
30
 
           'unified_diff_files']
 
28
__all__ = ['PatienceSequenceMatcher', 'unified_diff', 'unified_diff_files']
31
29
 
32
30
 
33
31
# This is a version of unified_diff which only adds a factory parameter
75
73
     four
76
74
    """
77
75
    if sequencematcher is None:
 
76
        import difflib
78
77
        sequencematcher = difflib.SequenceMatcher
79
78
 
80
79
    if fromfiledate:
83
82
        tofiledate = '\t' + str(tofiledate)
84
83
 
85
84
    started = False
86
 
    for group in sequencematcher(None, a, b).get_grouped_opcodes(n):
 
85
    for group in sequencematcher(None,a,b).get_grouped_opcodes(n):
87
86
        if not started:
88
87
            yield '--- %s%s%s' % (fromfile, fromfiledate, lineterm)
89
88
            yield '+++ %s%s%s' % (tofile, tofiledate, lineterm)
90
89
            started = True
91
90
        i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
92
 
        yield "@@ -%d,%d +%d,%d @@%s" % (i1 + 1, i2 - i1, j1 + 1, j2 - j1, lineterm)
 
91
        yield "@@ -%d,%d +%d,%d @@%s" % (i1+1, i2-i1, j1+1, j2-j1, lineterm)
93
92
        for tag, i1, i2, j1, j2 in group:
94
93
            if tag == 'equal':
95
94
                for line in a[i1:i2]:
103
102
                    yield '+' + line
104
103
 
105
104
 
106
 
def unified_diff_bytes(a, b, fromfile=b'', tofile=b'', fromfiledate=b'',
107
 
                       tofiledate=b'', n=3, lineterm=b'\n', sequencematcher=None):
108
 
    r"""
109
 
    Compare two sequences of lines; generate the delta as a unified diff.
110
 
 
111
 
    Unified diffs are a compact way of showing line changes and a few
112
 
    lines of context.  The number of context lines is set by 'n' which
113
 
    defaults to three.
114
 
 
115
 
    By default, the diff control lines (those with ---, +++, or @@) are
116
 
    created with a trailing newline.  This is helpful so that inputs
117
 
    created from file.readlines() result in diffs that are suitable for
118
 
    file.writelines() since both the inputs and outputs have trailing
119
 
    newlines.
120
 
 
121
 
    For inputs that do not have trailing newlines, set the lineterm
122
 
    argument to "" so that the output will be uniformly newline free.
123
 
 
124
 
    The unidiff format normally has a header for filenames and modification
125
 
    times.  Any or all of these may be specified using strings for
126
 
    'fromfile', 'tofile', 'fromfiledate', and 'tofiledate'.  The modification
127
 
    times are normally expressed in the format returned by time.ctime().
128
 
 
129
 
    Example:
130
 
 
131
 
    >>> for line in bytes_unified_diff(b'one two three four'.split(),
132
 
    ...             b'zero one tree four'.split(), b'Original', b'Current',
133
 
    ...             b'Sat Jan 26 23:30:50 1991', b'Fri Jun 06 10:20:52 2003',
134
 
    ...             lineterm=b''):
135
 
    ...     print line
136
 
    --- Original Sat Jan 26 23:30:50 1991
137
 
    +++ Current Fri Jun 06 10:20:52 2003
138
 
    @@ -1,4 +1,4 @@
139
 
    +zero
140
 
     one
141
 
    -two
142
 
    -three
143
 
    +tree
144
 
     four
145
 
    """
146
 
    if sequencematcher is None:
147
 
        sequencematcher = difflib.SequenceMatcher
148
 
 
149
 
    if fromfiledate:
150
 
        fromfiledate = b'\t' + bytes(fromfiledate)
151
 
    if tofiledate:
152
 
        tofiledate = b'\t' + bytes(tofiledate)
153
 
 
154
 
    started = False
155
 
    for group in sequencematcher(None, a, b).get_grouped_opcodes(n):
156
 
        if not started:
157
 
            yield b'--- %s%s%s' % (fromfile, fromfiledate, lineterm)
158
 
            yield b'+++ %s%s%s' % (tofile, tofiledate, lineterm)
159
 
            started = True
160
 
        i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
161
 
        yield b"@@ -%d,%d +%d,%d @@%s" % (i1 + 1, i2 - i1, j1 + 1, j2 - j1, lineterm)
162
 
        for tag, i1, i2, j1, j2 in group:
163
 
            if tag == 'equal':
164
 
                for line in a[i1:i2]:
165
 
                    yield b' ' + line
166
 
                continue
167
 
            if tag == 'replace' or tag == 'delete':
168
 
                for line in a[i1:i2]:
169
 
                    yield b'-' + line
170
 
            if tag == 'replace' or tag == 'insert':
171
 
                for line in b[j1:j2]:
172
 
                    yield b'+' + line
173
 
 
174
 
 
175
105
def unified_diff_files(a, b, sequencematcher=None):
176
106
    """Generate the diff for two files.
177
107
    """
193
123
        time_b = os.stat(b).st_mtime
194
124
 
195
125
    # TODO: Include fromfiledate and tofiledate
196
 
    return unified_diff_bytes(file_a.readlines(), file_b.readlines(),
197
 
                              fromfile=a, tofile=b,
198
 
                              sequencematcher=sequencematcher)
 
126
    return unified_diff(file_a.readlines(), file_b.readlines(),
 
127
                        fromfile=a, tofile=b,
 
128
                        sequencematcher=sequencematcher)
199
129
 
200
130
 
201
131
try:
202
 
    from ._patiencediff_c import (
 
132
    from bzrlib._patiencediff_c import (
203
133
        unique_lcs_c as unique_lcs,
204
134
        recurse_matches_c as recurse_matches,
205
135
        PatienceSequenceMatcher_c as PatienceSequenceMatcher
206
136
        )
207
137
except ImportError:
208
 
    from ._patiencediff_py import (
 
138
    from bzrlib._patiencediff_py import (
209
139
        unique_lcs_py as unique_lcs,
210
140
        recurse_matches_py as recurse_matches,
211
141
        PatienceSequenceMatcher_py as PatienceSequenceMatcher
212
 
        )  # noqa: F401
 
142
        )
213
143
 
214
144
 
215
145
def main(args):
221
151
    p.add_option('--difflib', dest='matcher', action='store_const', const='difflib',
222
152
                 default='patience', help='Use python\'s difflib algorithm')
223
153
 
224
 
    algorithms = {'patience': PatienceSequenceMatcher,
225
 
                  'difflib': difflib.SequenceMatcher}
 
154
    algorithms = {'patience':PatienceSequenceMatcher, 'difflib':difflib.SequenceMatcher}
226
155
 
227
156
    (opts, args) = p.parse_args(args)
228
157
    matcher = algorithms[opts.matcher]
229
158
 
230
159
    if len(args) != 2:
231
 
        print('You must supply 2 filenames to diff')
 
160
        print 'You must supply 2 filenames to diff'
232
161
        return -1
233
162
 
234
163
    for line in unified_diff_files(args[0], args[1], sequencematcher=matcher):