blob: 3fd44a6422348cccd7f92940ebdd2d694e4521cc [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`difflib` --- Helpers for computing deltas
2===============================================
3
4.. module:: difflib
5 :synopsis: Helpers for computing differences between objects.
6.. moduleauthor:: Tim Peters <tim_one@users.sourceforge.net>
7.. sectionauthor:: Tim Peters <tim_one@users.sourceforge.net>
Christian Heimes5b5e81c2007-12-31 16:14:33 +00008.. Markup by Fred L. Drake, Jr. <fdrake@acm.org>
Georg Brandl116aa622007-08-15 14:28:22 +00009
Christian Heimesfe337bf2008-03-23 21:54:12 +000010.. testsetup::
Georg Brandl116aa622007-08-15 14:28:22 +000011
Christian Heimesfe337bf2008-03-23 21:54:12 +000012 import sys
13 from difflib import *
Georg Brandl116aa622007-08-15 14:28:22 +000014
Georg Brandl9afde1c2007-11-01 20:32:30 +000015This module provides classes and functions for comparing sequences. It
16can be used for example, for comparing files, and can produce difference
17information in various formats, including HTML and context and unified
18diffs. For comparing directories and files, see also, the :mod:`filecmp` module.
19
Georg Brandl116aa622007-08-15 14:28:22 +000020.. class:: SequenceMatcher
21
22 This is a flexible class for comparing pairs of sequences of any type, so long
Guido van Rossum2cc30da2007-11-02 23:46:40 +000023 as the sequence elements are :term:`hashable`. The basic algorithm predates, and is a
Georg Brandl116aa622007-08-15 14:28:22 +000024 little fancier than, an algorithm published in the late 1980's by Ratcliff and
25 Obershelp under the hyperbolic name "gestalt pattern matching." The idea is to
26 find the longest contiguous matching subsequence that contains no "junk"
27 elements (the Ratcliff and Obershelp algorithm doesn't address junk). The same
28 idea is then applied recursively to the pieces of the sequences to the left and
29 to the right of the matching subsequence. This does not yield minimal edit
30 sequences, but does tend to yield matches that "look right" to people.
31
32 **Timing:** The basic Ratcliff-Obershelp algorithm is cubic time in the worst
33 case and quadratic time in the expected case. :class:`SequenceMatcher` is
34 quadratic time for the worst case and has expected-case behavior dependent in a
35 complicated way on how many elements the sequences have in common; best case
36 time is linear.
37
Terry Reedyc40b25f2010-11-10 17:50:26 +000038 **Heuristic:** To speed-up matching, items whose duplicates appear more than 1% of
39 the time in sequences of at least 200 items are treated as junk. This has the
40 unfortunate side-effect of giving bad results for sequences constructed from
41 a small set of items. An option to turn off the heuristic will be added to
42 Python 3.2.
Georg Brandl116aa622007-08-15 14:28:22 +000043
44.. class:: Differ
45
46 This is a class for comparing sequences of lines of text, and producing
47 human-readable differences or deltas. Differ uses :class:`SequenceMatcher`
48 both to compare sequences of lines, and to compare sequences of characters
49 within similar (near-matching) lines.
50
51 Each line of a :class:`Differ` delta begins with a two-letter code:
52
53 +----------+-------------------------------------------+
54 | Code | Meaning |
55 +==========+===========================================+
56 | ``'- '`` | line unique to sequence 1 |
57 +----------+-------------------------------------------+
58 | ``'+ '`` | line unique to sequence 2 |
59 +----------+-------------------------------------------+
60 | ``' '`` | line common to both sequences |
61 +----------+-------------------------------------------+
62 | ``'? '`` | line not present in either input sequence |
63 +----------+-------------------------------------------+
64
65 Lines beginning with '``?``' attempt to guide the eye to intraline differences,
66 and were not present in either input sequence. These lines can be confusing if
67 the sequences contain tab characters.
68
69
70.. class:: HtmlDiff
71
72 This class can be used to create an HTML table (or a complete HTML file
73 containing the table) showing a side by side, line by line comparison of text
74 with inter-line and intra-line change highlights. The table can be generated in
75 either full or contextual difference mode.
76
77 The constructor for this class is:
78
79
Georg Brandlc2a4f4f2009-04-10 09:03:43 +000080 .. method:: __init__(tabsize=8, wrapcolumn=None, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +000081
82 Initializes instance of :class:`HtmlDiff`.
83
84 *tabsize* is an optional keyword argument to specify tab stop spacing and
85 defaults to ``8``.
86
87 *wrapcolumn* is an optional keyword to specify column number where lines are
88 broken and wrapped, defaults to ``None`` where lines are not wrapped.
89
90 *linejunk* and *charjunk* are optional keyword arguments passed into ``ndiff()``
91 (used by :class:`HtmlDiff` to generate the side by side HTML differences). See
92 ``ndiff()`` documentation for argument default values and descriptions.
93
94 The following methods are public:
95
Georg Brandlc2a4f4f2009-04-10 09:03:43 +000096 .. method:: make_file(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +000097
98 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
99 is a complete HTML file containing a table showing line by line differences with
100 inter-line and intra-line changes highlighted.
101
102 *fromdesc* and *todesc* are optional keyword arguments to specify from/to file
103 column header strings (both default to an empty string).
104
105 *context* and *numlines* are both optional keyword arguments. Set *context* to
106 ``True`` when contextual differences are to be shown, else the default is
107 ``False`` to show the full files. *numlines* defaults to ``5``. When *context*
108 is ``True`` *numlines* controls the number of context lines which surround the
109 difference highlights. When *context* is ``False`` *numlines* controls the
110 number of lines which are shown before a difference highlight when using the
111 "next" hyperlinks (setting to zero would cause the "next" hyperlinks to place
112 the next difference highlight at the top of the browser without any leading
113 context).
114
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000115 .. method:: make_table(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000116
117 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
118 is a complete HTML table showing line by line differences with inter-line and
119 intra-line changes highlighted.
120
121 The arguments for this method are the same as those for the :meth:`make_file`
122 method.
123
124 :file:`Tools/scripts/diff.py` is a command-line front-end to this class and
125 contains a good example of its use.
126
Georg Brandl116aa622007-08-15 14:28:22 +0000127
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000128.. function:: context_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000129
Georg Brandl9afde1c2007-11-01 20:32:30 +0000130 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
131 generating the delta lines) in context diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000132
133 Context diffs are a compact way of showing just the lines that have changed plus
134 a few lines of context. The changes are shown in a before/after style. The
135 number of context lines is set by *n* which defaults to three.
136
137 By default, the diff control lines (those with ``***`` or ``---``) are created
138 with a trailing newline. This is helpful so that inputs created from
139 :func:`file.readlines` result in diffs that are suitable for use with
140 :func:`file.writelines` since both the inputs and outputs have trailing
141 newlines.
142
143 For inputs that do not have trailing newlines, set the *lineterm* argument to
144 ``""`` so that the output will be uniformly newline free.
145
146 The context diff format normally has a header for filenames and modification
147 times. Any or all of these may be specified using strings for *fromfile*,
148 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
149 expressed in the format returned by :func:`time.ctime`. If not specified, the
150 strings default to blanks.
151
Christian Heimes8640e742008-02-23 16:23:06 +0000152 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
153 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
154 >>> for line in context_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000155 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000156 *** before.py
157 --- after.py
158 ***************
159 *** 1,4 ****
160 ! bacon
161 ! eggs
162 ! ham
163 guido
164 --- 1,4 ----
165 ! python
166 ! eggy
167 ! hamster
168 guido
169
170 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000171
Georg Brandl116aa622007-08-15 14:28:22 +0000172
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000173.. function:: get_close_matches(word, possibilities, n=3, cutoff=0.6)
Georg Brandl116aa622007-08-15 14:28:22 +0000174
175 Return a list of the best "good enough" matches. *word* is a sequence for which
176 close matches are desired (typically a string), and *possibilities* is a list of
177 sequences against which to match *word* (typically a list of strings).
178
179 Optional argument *n* (default ``3``) is the maximum number of close matches to
180 return; *n* must be greater than ``0``.
181
182 Optional argument *cutoff* (default ``0.6``) is a float in the range [0, 1].
183 Possibilities that don't score at least that similar to *word* are ignored.
184
185 The best (no more than *n*) matches among the possibilities are returned in a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000186 list, sorted by similarity score, most similar first.
Georg Brandl116aa622007-08-15 14:28:22 +0000187
188 >>> get_close_matches('appel', ['ape', 'apple', 'peach', 'puppy'])
189 ['apple', 'ape']
190 >>> import keyword
191 >>> get_close_matches('wheel', keyword.kwlist)
192 ['while']
193 >>> get_close_matches('apple', keyword.kwlist)
194 []
195 >>> get_close_matches('accept', keyword.kwlist)
196 ['except']
197
198
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000199.. function:: ndiff(a, b, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +0000200
Georg Brandl9afde1c2007-11-01 20:32:30 +0000201 Compare *a* and *b* (lists of strings); return a :class:`Differ`\ -style
202 delta (a :term:`generator` generating the delta lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000203
204 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
205 (or ``None``):
206
Georg Brandle6bcc912008-05-12 18:05:20 +0000207 *linejunk*: A function that accepts a single string argument, and returns
208 true if the string is junk, or false if not. The default is ``None``. There
209 is also a module-level function :func:`IS_LINE_JUNK`, which filters out lines
210 without visible characters, except for at most one pound character (``'#'``)
211 -- however the underlying :class:`SequenceMatcher` class does a dynamic
212 analysis of which lines are so frequent as to constitute noise, and this
213 usually works better than using this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000214
215 *charjunk*: A function that accepts a character (a string of length 1), and
216 returns if the character is junk, or false if not. The default is module-level
217 function :func:`IS_CHARACTER_JUNK`, which filters out whitespace characters (a
218 blank or tab; note: bad idea to include newline in this!).
219
Christian Heimesfe337bf2008-03-23 21:54:12 +0000220 :file:`Tools/scripts/ndiff.py` is a command-line front-end to this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000221
222 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
223 ... 'ore\ntree\nemu\n'.splitlines(1))
Georg Brandl6911e3c2007-09-04 07:15:32 +0000224 >>> print(''.join(diff), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000225 - one
226 ? ^
227 + ore
228 ? ^
229 - two
230 - three
231 ? -
232 + tree
233 + emu
234
235
236.. function:: restore(sequence, which)
237
238 Return one of the two sequences that generated a delta.
239
240 Given a *sequence* produced by :meth:`Differ.compare` or :func:`ndiff`, extract
241 lines originating from file 1 or 2 (parameter *which*), stripping off line
242 prefixes.
243
Christian Heimesfe337bf2008-03-23 21:54:12 +0000244 Example:
Georg Brandl116aa622007-08-15 14:28:22 +0000245
246 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
247 ... 'ore\ntree\nemu\n'.splitlines(1))
248 >>> diff = list(diff) # materialize the generated delta into a list
Georg Brandl6911e3c2007-09-04 07:15:32 +0000249 >>> print(''.join(restore(diff, 1)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000250 one
251 two
252 three
Georg Brandl6911e3c2007-09-04 07:15:32 +0000253 >>> print(''.join(restore(diff, 2)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000254 ore
255 tree
256 emu
257
258
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000259.. function:: unified_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000260
Georg Brandl9afde1c2007-11-01 20:32:30 +0000261 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
262 generating the delta lines) in unified diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000263
264 Unified diffs are a compact way of showing just the lines that have changed plus
265 a few lines of context. The changes are shown in a inline style (instead of
266 separate before/after blocks). The number of context lines is set by *n* which
267 defaults to three.
268
269 By default, the diff control lines (those with ``---``, ``+++``, or ``@@``) are
270 created with a trailing newline. This is helpful so that inputs created from
271 :func:`file.readlines` result in diffs that are suitable for use with
272 :func:`file.writelines` since both the inputs and outputs have trailing
273 newlines.
274
275 For inputs that do not have trailing newlines, set the *lineterm* argument to
276 ``""`` so that the output will be uniformly newline free.
277
278 The context diff format normally has a header for filenames and modification
279 times. Any or all of these may be specified using strings for *fromfile*,
280 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
281 expressed in the format returned by :func:`time.ctime`. If not specified, the
282 strings default to blanks.
283
Christian Heimes8640e742008-02-23 16:23:06 +0000284
285 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
286 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
287 >>> for line in unified_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000288 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000289 --- before.py
290 +++ after.py
291 @@ -1,4 +1,4 @@
292 -bacon
293 -eggs
294 -ham
295 +python
296 +eggy
297 +hamster
298 guido
299
300 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000301
Georg Brandl116aa622007-08-15 14:28:22 +0000302
303.. function:: IS_LINE_JUNK(line)
304
305 Return true for ignorable lines. The line *line* is ignorable if *line* is
306 blank or contains a single ``'#'``, otherwise it is not ignorable. Used as a
Georg Brandle6bcc912008-05-12 18:05:20 +0000307 default for parameter *linejunk* in :func:`ndiff` in older versions.
Georg Brandl116aa622007-08-15 14:28:22 +0000308
309
310.. function:: IS_CHARACTER_JUNK(ch)
311
312 Return true for ignorable characters. The character *ch* is ignorable if *ch*
313 is a space or tab, otherwise it is not ignorable. Used as a default for
314 parameter *charjunk* in :func:`ndiff`.
315
316
317.. seealso::
318
319 `Pattern Matching: The Gestalt Approach <http://www.ddj.com/184407970?pgno=5>`_
320 Discussion of a similar algorithm by John W. Ratcliff and D. E. Metzener. This
321 was published in `Dr. Dobb's Journal <http://www.ddj.com/>`_ in July, 1988.
322
323
324.. _sequence-matcher:
325
326SequenceMatcher Objects
327-----------------------
328
329The :class:`SequenceMatcher` class has this constructor:
330
331
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000332.. class:: SequenceMatcher(isjunk=None, a='', b='')
Georg Brandl116aa622007-08-15 14:28:22 +0000333
334 Optional argument *isjunk* must be ``None`` (the default) or a one-argument
335 function that takes a sequence element and returns true if and only if the
336 element is "junk" and should be ignored. Passing ``None`` for *isjunk* is
337 equivalent to passing ``lambda x: 0``; in other words, no elements are ignored.
338 For example, pass::
339
340 lambda x: x in " \t"
341
342 if you're comparing lines as sequences of characters, and don't want to synch up
343 on blanks or hard tabs.
344
345 The optional arguments *a* and *b* are sequences to be compared; both default to
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000346 empty strings. The elements of both sequences must be :term:`hashable`.
Georg Brandl116aa622007-08-15 14:28:22 +0000347
Benjamin Petersone41251e2008-04-25 01:59:09 +0000348 :class:`SequenceMatcher` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000349
350
Benjamin Petersone41251e2008-04-25 01:59:09 +0000351 .. method:: set_seqs(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000352
Benjamin Petersone41251e2008-04-25 01:59:09 +0000353 Set the two sequences to be compared.
Georg Brandl116aa622007-08-15 14:28:22 +0000354
Benjamin Petersone41251e2008-04-25 01:59:09 +0000355 :class:`SequenceMatcher` computes and caches detailed information about the
356 second sequence, so if you want to compare one sequence against many
357 sequences, use :meth:`set_seq2` to set the commonly used sequence once and
358 call :meth:`set_seq1` repeatedly, once for each of the other sequences.
Georg Brandl116aa622007-08-15 14:28:22 +0000359
360
Benjamin Petersone41251e2008-04-25 01:59:09 +0000361 .. method:: set_seq1(a)
Georg Brandl116aa622007-08-15 14:28:22 +0000362
Benjamin Petersone41251e2008-04-25 01:59:09 +0000363 Set the first sequence to be compared. The second sequence to be compared
364 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000365
366
Benjamin Petersone41251e2008-04-25 01:59:09 +0000367 .. method:: set_seq2(b)
Georg Brandl116aa622007-08-15 14:28:22 +0000368
Benjamin Petersone41251e2008-04-25 01:59:09 +0000369 Set the second sequence to be compared. The first sequence to be compared
370 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000371
372
Benjamin Petersone41251e2008-04-25 01:59:09 +0000373 .. method:: find_longest_match(alo, ahi, blo, bhi)
Georg Brandl116aa622007-08-15 14:28:22 +0000374
Benjamin Petersone41251e2008-04-25 01:59:09 +0000375 Find longest matching block in ``a[alo:ahi]`` and ``b[blo:bhi]``.
Georg Brandl116aa622007-08-15 14:28:22 +0000376
Benjamin Petersone41251e2008-04-25 01:59:09 +0000377 If *isjunk* was omitted or ``None``, :meth:`find_longest_match` returns
378 ``(i, j, k)`` such that ``a[i:i+k]`` is equal to ``b[j:j+k]``, where ``alo
379 <= i <= i+k <= ahi`` and ``blo <= j <= j+k <= bhi``. For all ``(i', j',
380 k')`` meeting those conditions, the additional conditions ``k >= k'``, ``i
381 <= i'``, and if ``i == i'``, ``j <= j'`` are also met. In other words, of
382 all maximal matching blocks, return one that starts earliest in *a*, and
383 of all those maximal matching blocks that start earliest in *a*, return
384 the one that starts earliest in *b*.
Georg Brandl116aa622007-08-15 14:28:22 +0000385
Benjamin Petersone41251e2008-04-25 01:59:09 +0000386 >>> s = SequenceMatcher(None, " abcd", "abcd abcd")
387 >>> s.find_longest_match(0, 5, 0, 9)
388 Match(a=0, b=4, size=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000389
Benjamin Petersone41251e2008-04-25 01:59:09 +0000390 If *isjunk* was provided, first the longest matching block is determined
391 as above, but with the additional restriction that no junk element appears
392 in the block. Then that block is extended as far as possible by matching
393 (only) junk elements on both sides. So the resulting block never matches
394 on junk except as identical junk happens to be adjacent to an interesting
395 match.
Georg Brandl116aa622007-08-15 14:28:22 +0000396
Benjamin Petersone41251e2008-04-25 01:59:09 +0000397 Here's the same example as before, but considering blanks to be junk. That
398 prevents ``' abcd'`` from matching the ``' abcd'`` at the tail end of the
399 second sequence directly. Instead only the ``'abcd'`` can match, and
400 matches the leftmost ``'abcd'`` in the second sequence:
Georg Brandl116aa622007-08-15 14:28:22 +0000401
Benjamin Petersone41251e2008-04-25 01:59:09 +0000402 >>> s = SequenceMatcher(lambda x: x==" ", " abcd", "abcd abcd")
403 >>> s.find_longest_match(0, 5, 0, 9)
404 Match(a=1, b=0, size=4)
Georg Brandl116aa622007-08-15 14:28:22 +0000405
Benjamin Petersone41251e2008-04-25 01:59:09 +0000406 If no blocks match, this returns ``(alo, blo, 0)``.
Georg Brandl116aa622007-08-15 14:28:22 +0000407
Benjamin Petersone41251e2008-04-25 01:59:09 +0000408 This method returns a :term:`named tuple` ``Match(a, b, size)``.
Christian Heimes25bb7832008-01-11 16:17:00 +0000409
Georg Brandl116aa622007-08-15 14:28:22 +0000410
Benjamin Petersone41251e2008-04-25 01:59:09 +0000411 .. method:: get_matching_blocks()
Georg Brandl116aa622007-08-15 14:28:22 +0000412
Benjamin Petersone41251e2008-04-25 01:59:09 +0000413 Return list of triples describing matching subsequences. Each triple is of
414 the form ``(i, j, n)``, and means that ``a[i:i+n] == b[j:j+n]``. The
415 triples are monotonically increasing in *i* and *j*.
Georg Brandl116aa622007-08-15 14:28:22 +0000416
Benjamin Petersone41251e2008-04-25 01:59:09 +0000417 The last triple is a dummy, and has the value ``(len(a), len(b), 0)``. It
418 is the only triple with ``n == 0``. If ``(i, j, n)`` and ``(i', j', n')``
419 are adjacent triples in the list, and the second is not the last triple in
420 the list, then ``i+n != i'`` or ``j+n != j'``; in other words, adjacent
421 triples always describe non-adjacent equal blocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000422
Benjamin Petersone41251e2008-04-25 01:59:09 +0000423 .. XXX Explain why a dummy is used!
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000424
Benjamin Petersone41251e2008-04-25 01:59:09 +0000425 .. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000426
Benjamin Petersone41251e2008-04-25 01:59:09 +0000427 >>> s = SequenceMatcher(None, "abxcd", "abcd")
428 >>> s.get_matching_blocks()
429 [Match(a=0, b=0, size=2), Match(a=3, b=2, size=2), Match(a=5, b=4, size=0)]
Georg Brandl116aa622007-08-15 14:28:22 +0000430
431
Benjamin Petersone41251e2008-04-25 01:59:09 +0000432 .. method:: get_opcodes()
Georg Brandl116aa622007-08-15 14:28:22 +0000433
Benjamin Petersone41251e2008-04-25 01:59:09 +0000434 Return list of 5-tuples describing how to turn *a* into *b*. Each tuple is
435 of the form ``(tag, i1, i2, j1, j2)``. The first tuple has ``i1 == j1 ==
436 0``, and remaining tuples have *i1* equal to the *i2* from the preceding
437 tuple, and, likewise, *j1* equal to the previous *j2*.
Georg Brandl116aa622007-08-15 14:28:22 +0000438
Benjamin Petersone41251e2008-04-25 01:59:09 +0000439 The *tag* values are strings, with these meanings:
Georg Brandl116aa622007-08-15 14:28:22 +0000440
Benjamin Petersone41251e2008-04-25 01:59:09 +0000441 +---------------+---------------------------------------------+
442 | Value | Meaning |
443 +===============+=============================================+
444 | ``'replace'`` | ``a[i1:i2]`` should be replaced by |
445 | | ``b[j1:j2]``. |
446 +---------------+---------------------------------------------+
447 | ``'delete'`` | ``a[i1:i2]`` should be deleted. Note that |
448 | | ``j1 == j2`` in this case. |
449 +---------------+---------------------------------------------+
450 | ``'insert'`` | ``b[j1:j2]`` should be inserted at |
451 | | ``a[i1:i1]``. Note that ``i1 == i2`` in |
452 | | this case. |
453 +---------------+---------------------------------------------+
454 | ``'equal'`` | ``a[i1:i2] == b[j1:j2]`` (the sub-sequences |
455 | | are equal). |
456 +---------------+---------------------------------------------+
Georg Brandl116aa622007-08-15 14:28:22 +0000457
Benjamin Petersone41251e2008-04-25 01:59:09 +0000458 For example:
Georg Brandl116aa622007-08-15 14:28:22 +0000459
Benjamin Petersone41251e2008-04-25 01:59:09 +0000460 >>> a = "qabxcd"
461 >>> b = "abycdf"
462 >>> s = SequenceMatcher(None, a, b)
463 >>> for tag, i1, i2, j1, j2 in s.get_opcodes():
464 ... print(("%7s a[%d:%d] (%s) b[%d:%d] (%s)" %
465 ... (tag, i1, i2, a[i1:i2], j1, j2, b[j1:j2])))
466 delete a[0:1] (q) b[0:0] ()
467 equal a[1:3] (ab) b[0:2] (ab)
468 replace a[3:4] (x) b[2:3] (y)
469 equal a[4:6] (cd) b[3:5] (cd)
470 insert a[6:6] () b[5:6] (f)
Georg Brandl116aa622007-08-15 14:28:22 +0000471
472
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000473 .. method:: get_grouped_opcodes(n=3)
Georg Brandl116aa622007-08-15 14:28:22 +0000474
Benjamin Petersone41251e2008-04-25 01:59:09 +0000475 Return a :term:`generator` of groups with up to *n* lines of context.
Georg Brandl116aa622007-08-15 14:28:22 +0000476
Benjamin Petersone41251e2008-04-25 01:59:09 +0000477 Starting with the groups returned by :meth:`get_opcodes`, this method
478 splits out smaller change clusters and eliminates intervening ranges which
479 have no changes.
Georg Brandl116aa622007-08-15 14:28:22 +0000480
Benjamin Petersone41251e2008-04-25 01:59:09 +0000481 The groups are returned in the same format as :meth:`get_opcodes`.
Georg Brandl116aa622007-08-15 14:28:22 +0000482
Georg Brandl116aa622007-08-15 14:28:22 +0000483
Benjamin Petersone41251e2008-04-25 01:59:09 +0000484 .. method:: ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000485
Benjamin Petersone41251e2008-04-25 01:59:09 +0000486 Return a measure of the sequences' similarity as a float in the range [0,
487 1].
Georg Brandl116aa622007-08-15 14:28:22 +0000488
Benjamin Petersone41251e2008-04-25 01:59:09 +0000489 Where T is the total number of elements in both sequences, and M is the
490 number of matches, this is 2.0\*M / T. Note that this is ``1.0`` if the
491 sequences are identical, and ``0.0`` if they have nothing in common.
Georg Brandl116aa622007-08-15 14:28:22 +0000492
Benjamin Petersone41251e2008-04-25 01:59:09 +0000493 This is expensive to compute if :meth:`get_matching_blocks` or
494 :meth:`get_opcodes` hasn't already been called, in which case you may want
495 to try :meth:`quick_ratio` or :meth:`real_quick_ratio` first to get an
496 upper bound.
Georg Brandl116aa622007-08-15 14:28:22 +0000497
498
Benjamin Petersone41251e2008-04-25 01:59:09 +0000499 .. method:: quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000500
Benjamin Petersone41251e2008-04-25 01:59:09 +0000501 Return an upper bound on :meth:`ratio` relatively quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000502
Georg Brandl116aa622007-08-15 14:28:22 +0000503
Benjamin Petersone41251e2008-04-25 01:59:09 +0000504 .. method:: real_quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000505
Benjamin Petersone41251e2008-04-25 01:59:09 +0000506 Return an upper bound on :meth:`ratio` very quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000507
Georg Brandl116aa622007-08-15 14:28:22 +0000508
509The three methods that return the ratio of matching to total characters can give
510different results due to differing levels of approximation, although
511:meth:`quick_ratio` and :meth:`real_quick_ratio` are always at least as large as
Christian Heimesfe337bf2008-03-23 21:54:12 +0000512:meth:`ratio`:
Georg Brandl116aa622007-08-15 14:28:22 +0000513
514 >>> s = SequenceMatcher(None, "abcd", "bcde")
515 >>> s.ratio()
516 0.75
517 >>> s.quick_ratio()
518 0.75
519 >>> s.real_quick_ratio()
520 1.0
521
522
523.. _sequencematcher-examples:
524
525SequenceMatcher Examples
526------------------------
527
Christian Heimesfe337bf2008-03-23 21:54:12 +0000528This example compares two strings, considering blanks to be "junk:"
Georg Brandl116aa622007-08-15 14:28:22 +0000529
530 >>> s = SequenceMatcher(lambda x: x == " ",
531 ... "private Thread currentThread;",
532 ... "private volatile Thread currentThread;")
533
534:meth:`ratio` returns a float in [0, 1], measuring the similarity of the
535sequences. As a rule of thumb, a :meth:`ratio` value over 0.6 means the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000536sequences are close matches:
Georg Brandl116aa622007-08-15 14:28:22 +0000537
Georg Brandl6911e3c2007-09-04 07:15:32 +0000538 >>> print(round(s.ratio(), 3))
Georg Brandl116aa622007-08-15 14:28:22 +0000539 0.866
540
541If you're only interested in where the sequences match,
Christian Heimesfe337bf2008-03-23 21:54:12 +0000542:meth:`get_matching_blocks` is handy:
Georg Brandl116aa622007-08-15 14:28:22 +0000543
544 >>> for block in s.get_matching_blocks():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000545 ... print("a[%d] and b[%d] match for %d elements" % block)
Georg Brandl116aa622007-08-15 14:28:22 +0000546 a[0] and b[0] match for 8 elements
Christian Heimesfe337bf2008-03-23 21:54:12 +0000547 a[8] and b[17] match for 21 elements
Georg Brandl116aa622007-08-15 14:28:22 +0000548 a[29] and b[38] match for 0 elements
549
550Note that the last tuple returned by :meth:`get_matching_blocks` is always a
551dummy, ``(len(a), len(b), 0)``, and this is the only case in which the last
552tuple element (number of elements matched) is ``0``.
553
554If you want to know how to change the first sequence into the second, use
Christian Heimesfe337bf2008-03-23 21:54:12 +0000555:meth:`get_opcodes`:
Georg Brandl116aa622007-08-15 14:28:22 +0000556
557 >>> for opcode in s.get_opcodes():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000558 ... print("%6s a[%d:%d] b[%d:%d]" % opcode)
Georg Brandl116aa622007-08-15 14:28:22 +0000559 equal a[0:8] b[0:8]
560 insert a[8:8] b[8:17]
Christian Heimesfe337bf2008-03-23 21:54:12 +0000561 equal a[8:29] b[17:38]
Georg Brandl116aa622007-08-15 14:28:22 +0000562
Raymond Hettinger58c8c262009-04-27 21:01:21 +0000563.. seealso::
564
565 * The :func:`get_close_matches` function in this module which shows how
566 simple code building on :class:`SequenceMatcher` can be used to do useful
567 work.
568
569 * `Simple version control recipe
570 <http://code.activestate.com/recipes/576729/>`_ for a small application
571 built with :class:`SequenceMatcher`.
Georg Brandl116aa622007-08-15 14:28:22 +0000572
573
574.. _differ-objects:
575
576Differ Objects
577--------------
578
579Note that :class:`Differ`\ -generated deltas make no claim to be **minimal**
580diffs. To the contrary, minimal diffs are often counter-intuitive, because they
581synch up anywhere possible, sometimes accidental matches 100 pages apart.
582Restricting synch points to contiguous matches preserves some notion of
583locality, at the occasional cost of producing a longer diff.
584
585The :class:`Differ` class has this constructor:
586
587
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000588.. class:: Differ(linejunk=None, charjunk=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000589
590 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
591 (or ``None``):
592
593 *linejunk*: A function that accepts a single string argument, and returns true
594 if the string is junk. The default is ``None``, meaning that no line is
595 considered junk.
596
597 *charjunk*: A function that accepts a single character argument (a string of
598 length 1), and returns true if the character is junk. The default is ``None``,
599 meaning that no character is considered junk.
600
Benjamin Petersone41251e2008-04-25 01:59:09 +0000601 :class:`Differ` objects are used (deltas generated) via a single method:
Georg Brandl116aa622007-08-15 14:28:22 +0000602
603
Benjamin Petersone41251e2008-04-25 01:59:09 +0000604 .. method:: Differ.compare(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000605
Benjamin Petersone41251e2008-04-25 01:59:09 +0000606 Compare two sequences of lines, and generate the delta (a sequence of lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000607
Benjamin Petersone41251e2008-04-25 01:59:09 +0000608 Each sequence must contain individual single-line strings ending with newlines.
609 Such sequences can be obtained from the :meth:`readlines` method of file-like
610 objects. The delta generated also consists of newline-terminated strings, ready
611 to be printed as-is via the :meth:`writelines` method of a file-like object.
Georg Brandl116aa622007-08-15 14:28:22 +0000612
613
614.. _differ-examples:
615
616Differ Example
617--------------
618
619This example compares two texts. First we set up the texts, sequences of
620individual single-line strings ending with newlines (such sequences can also be
Christian Heimesfe337bf2008-03-23 21:54:12 +0000621obtained from the :meth:`readlines` method of file-like objects):
Georg Brandl116aa622007-08-15 14:28:22 +0000622
623 >>> text1 = ''' 1. Beautiful is better than ugly.
624 ... 2. Explicit is better than implicit.
625 ... 3. Simple is better than complex.
626 ... 4. Complex is better than complicated.
627 ... '''.splitlines(1)
628 >>> len(text1)
629 4
630 >>> text1[0][-1]
631 '\n'
632 >>> text2 = ''' 1. Beautiful is better than ugly.
633 ... 3. Simple is better than complex.
634 ... 4. Complicated is better than complex.
635 ... 5. Flat is better than nested.
636 ... '''.splitlines(1)
637
Christian Heimesfe337bf2008-03-23 21:54:12 +0000638Next we instantiate a Differ object:
Georg Brandl116aa622007-08-15 14:28:22 +0000639
640 >>> d = Differ()
641
642Note that when instantiating a :class:`Differ` object we may pass functions to
643filter out line and character "junk." See the :meth:`Differ` constructor for
644details.
645
Christian Heimesfe337bf2008-03-23 21:54:12 +0000646Finally, we compare the two:
Georg Brandl116aa622007-08-15 14:28:22 +0000647
648 >>> result = list(d.compare(text1, text2))
649
Christian Heimesfe337bf2008-03-23 21:54:12 +0000650``result`` is a list of strings, so let's pretty-print it:
Georg Brandl116aa622007-08-15 14:28:22 +0000651
652 >>> from pprint import pprint
653 >>> pprint(result)
654 [' 1. Beautiful is better than ugly.\n',
655 '- 2. Explicit is better than implicit.\n',
656 '- 3. Simple is better than complex.\n',
657 '+ 3. Simple is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000658 '? ++\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000659 '- 4. Complex is better than complicated.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000660 '? ^ ---- ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000661 '+ 4. Complicated is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000662 '? ++++ ^ ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000663 '+ 5. Flat is better than nested.\n']
664
Christian Heimesfe337bf2008-03-23 21:54:12 +0000665As a single multi-line string it looks like this:
Georg Brandl116aa622007-08-15 14:28:22 +0000666
667 >>> import sys
668 >>> sys.stdout.writelines(result)
669 1. Beautiful is better than ugly.
670 - 2. Explicit is better than implicit.
671 - 3. Simple is better than complex.
672 + 3. Simple is better than complex.
673 ? ++
674 - 4. Complex is better than complicated.
675 ? ^ ---- ^
676 + 4. Complicated is better than complex.
677 ? ++++ ^ ^
678 + 5. Flat is better than nested.
679
Christian Heimes8640e742008-02-23 16:23:06 +0000680
681.. _difflib-interface:
682
683A command-line interface to difflib
684-----------------------------------
685
686This example shows how to use difflib to create a ``diff``-like utility.
687It is also contained in the Python source distribution, as
688:file:`Tools/scripts/diff.py`.
689
Christian Heimesfe337bf2008-03-23 21:54:12 +0000690.. testcode::
Christian Heimes8640e742008-02-23 16:23:06 +0000691
692 """ Command line interface to difflib.py providing diffs in four formats:
693
694 * ndiff: lists every line and highlights interline changes.
695 * context: highlights clusters of changes in a before/after format.
696 * unified: highlights clusters of changes in an inline format.
697 * html: generates side by side comparison with change highlights.
698
699 """
700
701 import sys, os, time, difflib, optparse
702
703 def main():
704 # Configure the option parser
705 usage = "usage: %prog [options] fromfile tofile"
706 parser = optparse.OptionParser(usage)
707 parser.add_option("-c", action="store_true", default=False,
708 help='Produce a context format diff (default)')
709 parser.add_option("-u", action="store_true", default=False,
710 help='Produce a unified format diff')
711 hlp = 'Produce HTML side by side diff (can use -c and -l in conjunction)'
712 parser.add_option("-m", action="store_true", default=False, help=hlp)
713 parser.add_option("-n", action="store_true", default=False,
714 help='Produce a ndiff format diff')
715 parser.add_option("-l", "--lines", type="int", default=3,
716 help='Set number of context lines (default 3)')
717 (options, args) = parser.parse_args()
718
719 if len(args) == 0:
720 parser.print_help()
721 sys.exit(1)
722 if len(args) != 2:
723 parser.error("need to specify both a fromfile and tofile")
724
725 n = options.lines
726 fromfile, tofile = args # as specified in the usage string
727
728 # we're passing these as arguments to the diff function
729 fromdate = time.ctime(os.stat(fromfile).st_mtime)
730 todate = time.ctime(os.stat(tofile).st_mtime)
731 fromlines = open(fromfile, 'U').readlines()
732 tolines = open(tofile, 'U').readlines()
733
734 if options.u:
735 diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
736 fromdate, todate, n=n)
737 elif options.n:
738 diff = difflib.ndiff(fromlines, tolines)
739 elif options.m:
740 diff = difflib.HtmlDiff().make_file(fromlines, tolines, fromfile,
741 tofile, context=options.c,
742 numlines=n)
743 else:
744 diff = difflib.context_diff(fromlines, tolines, fromfile, tofile,
745 fromdate, todate, n=n)
746
747 # we're using writelines because diff is a generator
748 sys.stdout.writelines(diff)
749
750 if __name__ == '__main__':
751 main()