blob: 58bbe4509c3ddcac6630639356ad2a3ab3bb2e04 [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
38
39.. class:: Differ
40
41 This is a class for comparing sequences of lines of text, and producing
42 human-readable differences or deltas. Differ uses :class:`SequenceMatcher`
43 both to compare sequences of lines, and to compare sequences of characters
44 within similar (near-matching) lines.
45
46 Each line of a :class:`Differ` delta begins with a two-letter code:
47
48 +----------+-------------------------------------------+
49 | Code | Meaning |
50 +==========+===========================================+
51 | ``'- '`` | line unique to sequence 1 |
52 +----------+-------------------------------------------+
53 | ``'+ '`` | line unique to sequence 2 |
54 +----------+-------------------------------------------+
55 | ``' '`` | line common to both sequences |
56 +----------+-------------------------------------------+
57 | ``'? '`` | line not present in either input sequence |
58 +----------+-------------------------------------------+
59
60 Lines beginning with '``?``' attempt to guide the eye to intraline differences,
61 and were not present in either input sequence. These lines can be confusing if
62 the sequences contain tab characters.
63
64
65.. class:: HtmlDiff
66
67 This class can be used to create an HTML table (or a complete HTML file
68 containing the table) showing a side by side, line by line comparison of text
69 with inter-line and intra-line change highlights. The table can be generated in
70 either full or contextual difference mode.
71
72 The constructor for this class is:
73
74
Georg Brandlc2a4f4f2009-04-10 09:03:43 +000075 .. method:: __init__(tabsize=8, wrapcolumn=None, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +000076
77 Initializes instance of :class:`HtmlDiff`.
78
79 *tabsize* is an optional keyword argument to specify tab stop spacing and
80 defaults to ``8``.
81
82 *wrapcolumn* is an optional keyword to specify column number where lines are
83 broken and wrapped, defaults to ``None`` where lines are not wrapped.
84
85 *linejunk* and *charjunk* are optional keyword arguments passed into ``ndiff()``
86 (used by :class:`HtmlDiff` to generate the side by side HTML differences). See
87 ``ndiff()`` documentation for argument default values and descriptions.
88
89 The following methods are public:
90
Georg Brandlc2a4f4f2009-04-10 09:03:43 +000091 .. method:: make_file(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +000092
93 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
94 is a complete HTML file containing a table showing line by line differences with
95 inter-line and intra-line changes highlighted.
96
97 *fromdesc* and *todesc* are optional keyword arguments to specify from/to file
98 column header strings (both default to an empty string).
99
100 *context* and *numlines* are both optional keyword arguments. Set *context* to
101 ``True`` when contextual differences are to be shown, else the default is
102 ``False`` to show the full files. *numlines* defaults to ``5``. When *context*
103 is ``True`` *numlines* controls the number of context lines which surround the
104 difference highlights. When *context* is ``False`` *numlines* controls the
105 number of lines which are shown before a difference highlight when using the
106 "next" hyperlinks (setting to zero would cause the "next" hyperlinks to place
107 the next difference highlight at the top of the browser without any leading
108 context).
109
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000110 .. method:: make_table(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000111
112 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
113 is a complete HTML table showing line by line differences with inter-line and
114 intra-line changes highlighted.
115
116 The arguments for this method are the same as those for the :meth:`make_file`
117 method.
118
119 :file:`Tools/scripts/diff.py` is a command-line front-end to this class and
120 contains a good example of its use.
121
Georg Brandl116aa622007-08-15 14:28:22 +0000122
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000123.. function:: context_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000124
Georg Brandl9afde1c2007-11-01 20:32:30 +0000125 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
126 generating the delta lines) in context diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000127
128 Context diffs are a compact way of showing just the lines that have changed plus
129 a few lines of context. The changes are shown in a before/after style. The
130 number of context lines is set by *n* which defaults to three.
131
132 By default, the diff control lines (those with ``***`` or ``---``) are created
133 with a trailing newline. This is helpful so that inputs created from
134 :func:`file.readlines` result in diffs that are suitable for use with
135 :func:`file.writelines` since both the inputs and outputs have trailing
136 newlines.
137
138 For inputs that do not have trailing newlines, set the *lineterm* argument to
139 ``""`` so that the output will be uniformly newline free.
140
141 The context diff format normally has a header for filenames and modification
142 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murrayb2416e52010-04-12 16:58:02 +0000143 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
144 expressed in the ISO 8601 format. If not specified, the
Georg Brandl116aa622007-08-15 14:28:22 +0000145 strings default to blanks.
146
Christian Heimes8640e742008-02-23 16:23:06 +0000147 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
148 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
149 >>> for line in context_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000150 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000151 *** before.py
152 --- after.py
153 ***************
154 *** 1,4 ****
155 ! bacon
156 ! eggs
157 ! ham
158 guido
159 --- 1,4 ----
160 ! python
161 ! eggy
162 ! hamster
163 guido
164
165 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000166
Georg Brandl116aa622007-08-15 14:28:22 +0000167
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000168.. function:: get_close_matches(word, possibilities, n=3, cutoff=0.6)
Georg Brandl116aa622007-08-15 14:28:22 +0000169
170 Return a list of the best "good enough" matches. *word* is a sequence for which
171 close matches are desired (typically a string), and *possibilities* is a list of
172 sequences against which to match *word* (typically a list of strings).
173
174 Optional argument *n* (default ``3``) is the maximum number of close matches to
175 return; *n* must be greater than ``0``.
176
177 Optional argument *cutoff* (default ``0.6``) is a float in the range [0, 1].
178 Possibilities that don't score at least that similar to *word* are ignored.
179
180 The best (no more than *n*) matches among the possibilities are returned in a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000181 list, sorted by similarity score, most similar first.
Georg Brandl116aa622007-08-15 14:28:22 +0000182
183 >>> get_close_matches('appel', ['ape', 'apple', 'peach', 'puppy'])
184 ['apple', 'ape']
185 >>> import keyword
186 >>> get_close_matches('wheel', keyword.kwlist)
187 ['while']
188 >>> get_close_matches('apple', keyword.kwlist)
189 []
190 >>> get_close_matches('accept', keyword.kwlist)
191 ['except']
192
193
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000194.. function:: ndiff(a, b, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +0000195
Georg Brandl9afde1c2007-11-01 20:32:30 +0000196 Compare *a* and *b* (lists of strings); return a :class:`Differ`\ -style
197 delta (a :term:`generator` generating the delta lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000198
199 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
200 (or ``None``):
201
Georg Brandle6bcc912008-05-12 18:05:20 +0000202 *linejunk*: A function that accepts a single string argument, and returns
203 true if the string is junk, or false if not. The default is ``None``. There
204 is also a module-level function :func:`IS_LINE_JUNK`, which filters out lines
205 without visible characters, except for at most one pound character (``'#'``)
206 -- however the underlying :class:`SequenceMatcher` class does a dynamic
207 analysis of which lines are so frequent as to constitute noise, and this
208 usually works better than using this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000209
210 *charjunk*: A function that accepts a character (a string of length 1), and
211 returns if the character is junk, or false if not. The default is module-level
212 function :func:`IS_CHARACTER_JUNK`, which filters out whitespace characters (a
213 blank or tab; note: bad idea to include newline in this!).
214
Christian Heimesfe337bf2008-03-23 21:54:12 +0000215 :file:`Tools/scripts/ndiff.py` is a command-line front-end to this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000216
217 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
218 ... 'ore\ntree\nemu\n'.splitlines(1))
Georg Brandl6911e3c2007-09-04 07:15:32 +0000219 >>> print(''.join(diff), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000220 - one
221 ? ^
222 + ore
223 ? ^
224 - two
225 - three
226 ? -
227 + tree
228 + emu
229
230
231.. function:: restore(sequence, which)
232
233 Return one of the two sequences that generated a delta.
234
235 Given a *sequence* produced by :meth:`Differ.compare` or :func:`ndiff`, extract
236 lines originating from file 1 or 2 (parameter *which*), stripping off line
237 prefixes.
238
Christian Heimesfe337bf2008-03-23 21:54:12 +0000239 Example:
Georg Brandl116aa622007-08-15 14:28:22 +0000240
241 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
242 ... 'ore\ntree\nemu\n'.splitlines(1))
243 >>> diff = list(diff) # materialize the generated delta into a list
Georg Brandl6911e3c2007-09-04 07:15:32 +0000244 >>> print(''.join(restore(diff, 1)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000245 one
246 two
247 three
Georg Brandl6911e3c2007-09-04 07:15:32 +0000248 >>> print(''.join(restore(diff, 2)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000249 ore
250 tree
251 emu
252
253
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000254.. function:: unified_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000255
Georg Brandl9afde1c2007-11-01 20:32:30 +0000256 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
257 generating the delta lines) in unified diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000258
259 Unified diffs are a compact way of showing just the lines that have changed plus
260 a few lines of context. The changes are shown in a inline style (instead of
261 separate before/after blocks). The number of context lines is set by *n* which
262 defaults to three.
263
264 By default, the diff control lines (those with ``---``, ``+++``, or ``@@``) are
265 created with a trailing newline. This is helpful so that inputs created from
266 :func:`file.readlines` result in diffs that are suitable for use with
267 :func:`file.writelines` since both the inputs and outputs have trailing
268 newlines.
269
270 For inputs that do not have trailing newlines, set the *lineterm* argument to
271 ``""`` so that the output will be uniformly newline free.
272
273 The context diff format normally has a header for filenames and modification
274 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murrayb2416e52010-04-12 16:58:02 +0000275 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
276 expressed in the ISO 8601 format. If not specified, the
Georg Brandl116aa622007-08-15 14:28:22 +0000277 strings default to blanks.
278
Christian Heimes8640e742008-02-23 16:23:06 +0000279
280 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
281 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
282 >>> for line in unified_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000283 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000284 --- before.py
285 +++ after.py
286 @@ -1,4 +1,4 @@
287 -bacon
288 -eggs
289 -ham
290 +python
291 +eggy
292 +hamster
293 guido
294
295 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000296
Georg Brandl116aa622007-08-15 14:28:22 +0000297
298.. function:: IS_LINE_JUNK(line)
299
300 Return true for ignorable lines. The line *line* is ignorable if *line* is
301 blank or contains a single ``'#'``, otherwise it is not ignorable. Used as a
Georg Brandle6bcc912008-05-12 18:05:20 +0000302 default for parameter *linejunk* in :func:`ndiff` in older versions.
Georg Brandl116aa622007-08-15 14:28:22 +0000303
304
305.. function:: IS_CHARACTER_JUNK(ch)
306
307 Return true for ignorable characters. The character *ch* is ignorable if *ch*
308 is a space or tab, otherwise it is not ignorable. Used as a default for
309 parameter *charjunk* in :func:`ndiff`.
310
311
312.. seealso::
313
314 `Pattern Matching: The Gestalt Approach <http://www.ddj.com/184407970?pgno=5>`_
315 Discussion of a similar algorithm by John W. Ratcliff and D. E. Metzener. This
316 was published in `Dr. Dobb's Journal <http://www.ddj.com/>`_ in July, 1988.
317
318
319.. _sequence-matcher:
320
321SequenceMatcher Objects
322-----------------------
323
324The :class:`SequenceMatcher` class has this constructor:
325
326
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000327.. class:: SequenceMatcher(isjunk=None, a='', b='')
Georg Brandl116aa622007-08-15 14:28:22 +0000328
329 Optional argument *isjunk* must be ``None`` (the default) or a one-argument
330 function that takes a sequence element and returns true if and only if the
331 element is "junk" and should be ignored. Passing ``None`` for *isjunk* is
332 equivalent to passing ``lambda x: 0``; in other words, no elements are ignored.
333 For example, pass::
334
335 lambda x: x in " \t"
336
337 if you're comparing lines as sequences of characters, and don't want to synch up
338 on blanks or hard tabs.
339
340 The optional arguments *a* and *b* are sequences to be compared; both default to
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000341 empty strings. The elements of both sequences must be :term:`hashable`.
Georg Brandl116aa622007-08-15 14:28:22 +0000342
Benjamin Petersone41251e2008-04-25 01:59:09 +0000343 :class:`SequenceMatcher` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000344
345
Benjamin Petersone41251e2008-04-25 01:59:09 +0000346 .. method:: set_seqs(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000347
Benjamin Petersone41251e2008-04-25 01:59:09 +0000348 Set the two sequences to be compared.
Georg Brandl116aa622007-08-15 14:28:22 +0000349
Benjamin Petersone41251e2008-04-25 01:59:09 +0000350 :class:`SequenceMatcher` computes and caches detailed information about the
351 second sequence, so if you want to compare one sequence against many
352 sequences, use :meth:`set_seq2` to set the commonly used sequence once and
353 call :meth:`set_seq1` repeatedly, once for each of the other sequences.
Georg Brandl116aa622007-08-15 14:28:22 +0000354
355
Benjamin Petersone41251e2008-04-25 01:59:09 +0000356 .. method:: set_seq1(a)
Georg Brandl116aa622007-08-15 14:28:22 +0000357
Benjamin Petersone41251e2008-04-25 01:59:09 +0000358 Set the first sequence to be compared. The second sequence to be compared
359 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000360
361
Benjamin Petersone41251e2008-04-25 01:59:09 +0000362 .. method:: set_seq2(b)
Georg Brandl116aa622007-08-15 14:28:22 +0000363
Benjamin Petersone41251e2008-04-25 01:59:09 +0000364 Set the second sequence to be compared. The first sequence to be compared
365 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000366
367
Benjamin Petersone41251e2008-04-25 01:59:09 +0000368 .. method:: find_longest_match(alo, ahi, blo, bhi)
Georg Brandl116aa622007-08-15 14:28:22 +0000369
Benjamin Petersone41251e2008-04-25 01:59:09 +0000370 Find longest matching block in ``a[alo:ahi]`` and ``b[blo:bhi]``.
Georg Brandl116aa622007-08-15 14:28:22 +0000371
Benjamin Petersone41251e2008-04-25 01:59:09 +0000372 If *isjunk* was omitted or ``None``, :meth:`find_longest_match` returns
373 ``(i, j, k)`` such that ``a[i:i+k]`` is equal to ``b[j:j+k]``, where ``alo
374 <= i <= i+k <= ahi`` and ``blo <= j <= j+k <= bhi``. For all ``(i', j',
375 k')`` meeting those conditions, the additional conditions ``k >= k'``, ``i
376 <= i'``, and if ``i == i'``, ``j <= j'`` are also met. In other words, of
377 all maximal matching blocks, return one that starts earliest in *a*, and
378 of all those maximal matching blocks that start earliest in *a*, return
379 the one that starts earliest in *b*.
Georg Brandl116aa622007-08-15 14:28:22 +0000380
Benjamin Petersone41251e2008-04-25 01:59:09 +0000381 >>> s = SequenceMatcher(None, " abcd", "abcd abcd")
382 >>> s.find_longest_match(0, 5, 0, 9)
383 Match(a=0, b=4, size=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000384
Benjamin Petersone41251e2008-04-25 01:59:09 +0000385 If *isjunk* was provided, first the longest matching block is determined
386 as above, but with the additional restriction that no junk element appears
387 in the block. Then that block is extended as far as possible by matching
388 (only) junk elements on both sides. So the resulting block never matches
389 on junk except as identical junk happens to be adjacent to an interesting
390 match.
Georg Brandl116aa622007-08-15 14:28:22 +0000391
Benjamin Petersone41251e2008-04-25 01:59:09 +0000392 Here's the same example as before, but considering blanks to be junk. That
393 prevents ``' abcd'`` from matching the ``' abcd'`` at the tail end of the
394 second sequence directly. Instead only the ``'abcd'`` can match, and
395 matches the leftmost ``'abcd'`` in the second sequence:
Georg Brandl116aa622007-08-15 14:28:22 +0000396
Benjamin Petersone41251e2008-04-25 01:59:09 +0000397 >>> s = SequenceMatcher(lambda x: x==" ", " abcd", "abcd abcd")
398 >>> s.find_longest_match(0, 5, 0, 9)
399 Match(a=1, b=0, size=4)
Georg Brandl116aa622007-08-15 14:28:22 +0000400
Benjamin Petersone41251e2008-04-25 01:59:09 +0000401 If no blocks match, this returns ``(alo, blo, 0)``.
Georg Brandl116aa622007-08-15 14:28:22 +0000402
Benjamin Petersone41251e2008-04-25 01:59:09 +0000403 This method returns a :term:`named tuple` ``Match(a, b, size)``.
Christian Heimes25bb7832008-01-11 16:17:00 +0000404
Georg Brandl116aa622007-08-15 14:28:22 +0000405
Benjamin Petersone41251e2008-04-25 01:59:09 +0000406 .. method:: get_matching_blocks()
Georg Brandl116aa622007-08-15 14:28:22 +0000407
Benjamin Petersone41251e2008-04-25 01:59:09 +0000408 Return list of triples describing matching subsequences. Each triple is of
409 the form ``(i, j, n)``, and means that ``a[i:i+n] == b[j:j+n]``. The
410 triples are monotonically increasing in *i* and *j*.
Georg Brandl116aa622007-08-15 14:28:22 +0000411
Benjamin Petersone41251e2008-04-25 01:59:09 +0000412 The last triple is a dummy, and has the value ``(len(a), len(b), 0)``. It
413 is the only triple with ``n == 0``. If ``(i, j, n)`` and ``(i', j', n')``
414 are adjacent triples in the list, and the second is not the last triple in
415 the list, then ``i+n != i'`` or ``j+n != j'``; in other words, adjacent
416 triples always describe non-adjacent equal blocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000417
Benjamin Petersone41251e2008-04-25 01:59:09 +0000418 .. XXX Explain why a dummy is used!
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000419
Benjamin Petersone41251e2008-04-25 01:59:09 +0000420 .. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000421
Benjamin Petersone41251e2008-04-25 01:59:09 +0000422 >>> s = SequenceMatcher(None, "abxcd", "abcd")
423 >>> s.get_matching_blocks()
424 [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 +0000425
426
Benjamin Petersone41251e2008-04-25 01:59:09 +0000427 .. method:: get_opcodes()
Georg Brandl116aa622007-08-15 14:28:22 +0000428
Benjamin Petersone41251e2008-04-25 01:59:09 +0000429 Return list of 5-tuples describing how to turn *a* into *b*. Each tuple is
430 of the form ``(tag, i1, i2, j1, j2)``. The first tuple has ``i1 == j1 ==
431 0``, and remaining tuples have *i1* equal to the *i2* from the preceding
432 tuple, and, likewise, *j1* equal to the previous *j2*.
Georg Brandl116aa622007-08-15 14:28:22 +0000433
Benjamin Petersone41251e2008-04-25 01:59:09 +0000434 The *tag* values are strings, with these meanings:
Georg Brandl116aa622007-08-15 14:28:22 +0000435
Benjamin Petersone41251e2008-04-25 01:59:09 +0000436 +---------------+---------------------------------------------+
437 | Value | Meaning |
438 +===============+=============================================+
439 | ``'replace'`` | ``a[i1:i2]`` should be replaced by |
440 | | ``b[j1:j2]``. |
441 +---------------+---------------------------------------------+
442 | ``'delete'`` | ``a[i1:i2]`` should be deleted. Note that |
443 | | ``j1 == j2`` in this case. |
444 +---------------+---------------------------------------------+
445 | ``'insert'`` | ``b[j1:j2]`` should be inserted at |
446 | | ``a[i1:i1]``. Note that ``i1 == i2`` in |
447 | | this case. |
448 +---------------+---------------------------------------------+
449 | ``'equal'`` | ``a[i1:i2] == b[j1:j2]`` (the sub-sequences |
450 | | are equal). |
451 +---------------+---------------------------------------------+
Georg Brandl116aa622007-08-15 14:28:22 +0000452
Benjamin Petersone41251e2008-04-25 01:59:09 +0000453 For example:
Georg Brandl116aa622007-08-15 14:28:22 +0000454
Benjamin Petersone41251e2008-04-25 01:59:09 +0000455 >>> a = "qabxcd"
456 >>> b = "abycdf"
457 >>> s = SequenceMatcher(None, a, b)
458 >>> for tag, i1, i2, j1, j2 in s.get_opcodes():
459 ... print(("%7s a[%d:%d] (%s) b[%d:%d] (%s)" %
460 ... (tag, i1, i2, a[i1:i2], j1, j2, b[j1:j2])))
461 delete a[0:1] (q) b[0:0] ()
462 equal a[1:3] (ab) b[0:2] (ab)
463 replace a[3:4] (x) b[2:3] (y)
464 equal a[4:6] (cd) b[3:5] (cd)
465 insert a[6:6] () b[5:6] (f)
Georg Brandl116aa622007-08-15 14:28:22 +0000466
467
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000468 .. method:: get_grouped_opcodes(n=3)
Georg Brandl116aa622007-08-15 14:28:22 +0000469
Benjamin Petersone41251e2008-04-25 01:59:09 +0000470 Return a :term:`generator` of groups with up to *n* lines of context.
Georg Brandl116aa622007-08-15 14:28:22 +0000471
Benjamin Petersone41251e2008-04-25 01:59:09 +0000472 Starting with the groups returned by :meth:`get_opcodes`, this method
473 splits out smaller change clusters and eliminates intervening ranges which
474 have no changes.
Georg Brandl116aa622007-08-15 14:28:22 +0000475
Benjamin Petersone41251e2008-04-25 01:59:09 +0000476 The groups are returned in the same format as :meth:`get_opcodes`.
Georg Brandl116aa622007-08-15 14:28:22 +0000477
Georg Brandl116aa622007-08-15 14:28:22 +0000478
Benjamin Petersone41251e2008-04-25 01:59:09 +0000479 .. method:: ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000480
Benjamin Petersone41251e2008-04-25 01:59:09 +0000481 Return a measure of the sequences' similarity as a float in the range [0,
482 1].
Georg Brandl116aa622007-08-15 14:28:22 +0000483
Benjamin Petersone41251e2008-04-25 01:59:09 +0000484 Where T is the total number of elements in both sequences, and M is the
485 number of matches, this is 2.0\*M / T. Note that this is ``1.0`` if the
486 sequences are identical, and ``0.0`` if they have nothing in common.
Georg Brandl116aa622007-08-15 14:28:22 +0000487
Benjamin Petersone41251e2008-04-25 01:59:09 +0000488 This is expensive to compute if :meth:`get_matching_blocks` or
489 :meth:`get_opcodes` hasn't already been called, in which case you may want
490 to try :meth:`quick_ratio` or :meth:`real_quick_ratio` first to get an
491 upper bound.
Georg Brandl116aa622007-08-15 14:28:22 +0000492
493
Benjamin Petersone41251e2008-04-25 01:59:09 +0000494 .. method:: quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000495
Benjamin Petersone41251e2008-04-25 01:59:09 +0000496 Return an upper bound on :meth:`ratio` relatively quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000497
Georg Brandl116aa622007-08-15 14:28:22 +0000498
Benjamin Petersone41251e2008-04-25 01:59:09 +0000499 .. method:: real_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` very quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000502
Georg Brandl116aa622007-08-15 14:28:22 +0000503
504The three methods that return the ratio of matching to total characters can give
505different results due to differing levels of approximation, although
506:meth:`quick_ratio` and :meth:`real_quick_ratio` are always at least as large as
Christian Heimesfe337bf2008-03-23 21:54:12 +0000507:meth:`ratio`:
Georg Brandl116aa622007-08-15 14:28:22 +0000508
509 >>> s = SequenceMatcher(None, "abcd", "bcde")
510 >>> s.ratio()
511 0.75
512 >>> s.quick_ratio()
513 0.75
514 >>> s.real_quick_ratio()
515 1.0
516
517
518.. _sequencematcher-examples:
519
520SequenceMatcher Examples
521------------------------
522
Christian Heimesfe337bf2008-03-23 21:54:12 +0000523This example compares two strings, considering blanks to be "junk:"
Georg Brandl116aa622007-08-15 14:28:22 +0000524
525 >>> s = SequenceMatcher(lambda x: x == " ",
526 ... "private Thread currentThread;",
527 ... "private volatile Thread currentThread;")
528
529:meth:`ratio` returns a float in [0, 1], measuring the similarity of the
530sequences. As a rule of thumb, a :meth:`ratio` value over 0.6 means the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000531sequences are close matches:
Georg Brandl116aa622007-08-15 14:28:22 +0000532
Georg Brandl6911e3c2007-09-04 07:15:32 +0000533 >>> print(round(s.ratio(), 3))
Georg Brandl116aa622007-08-15 14:28:22 +0000534 0.866
535
536If you're only interested in where the sequences match,
Christian Heimesfe337bf2008-03-23 21:54:12 +0000537:meth:`get_matching_blocks` is handy:
Georg Brandl116aa622007-08-15 14:28:22 +0000538
539 >>> for block in s.get_matching_blocks():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000540 ... print("a[%d] and b[%d] match for %d elements" % block)
Georg Brandl116aa622007-08-15 14:28:22 +0000541 a[0] and b[0] match for 8 elements
Christian Heimesfe337bf2008-03-23 21:54:12 +0000542 a[8] and b[17] match for 21 elements
Georg Brandl116aa622007-08-15 14:28:22 +0000543 a[29] and b[38] match for 0 elements
544
545Note that the last tuple returned by :meth:`get_matching_blocks` is always a
546dummy, ``(len(a), len(b), 0)``, and this is the only case in which the last
547tuple element (number of elements matched) is ``0``.
548
549If you want to know how to change the first sequence into the second, use
Christian Heimesfe337bf2008-03-23 21:54:12 +0000550:meth:`get_opcodes`:
Georg Brandl116aa622007-08-15 14:28:22 +0000551
552 >>> for opcode in s.get_opcodes():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000553 ... print("%6s a[%d:%d] b[%d:%d]" % opcode)
Georg Brandl116aa622007-08-15 14:28:22 +0000554 equal a[0:8] b[0:8]
555 insert a[8:8] b[8:17]
Christian Heimesfe337bf2008-03-23 21:54:12 +0000556 equal a[8:29] b[17:38]
Georg Brandl116aa622007-08-15 14:28:22 +0000557
Raymond Hettinger58c8c262009-04-27 21:01:21 +0000558.. seealso::
559
560 * The :func:`get_close_matches` function in this module which shows how
561 simple code building on :class:`SequenceMatcher` can be used to do useful
562 work.
563
564 * `Simple version control recipe
565 <http://code.activestate.com/recipes/576729/>`_ for a small application
566 built with :class:`SequenceMatcher`.
Georg Brandl116aa622007-08-15 14:28:22 +0000567
568
569.. _differ-objects:
570
571Differ Objects
572--------------
573
574Note that :class:`Differ`\ -generated deltas make no claim to be **minimal**
575diffs. To the contrary, minimal diffs are often counter-intuitive, because they
576synch up anywhere possible, sometimes accidental matches 100 pages apart.
577Restricting synch points to contiguous matches preserves some notion of
578locality, at the occasional cost of producing a longer diff.
579
580The :class:`Differ` class has this constructor:
581
582
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000583.. class:: Differ(linejunk=None, charjunk=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000584
585 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
586 (or ``None``):
587
588 *linejunk*: A function that accepts a single string argument, and returns true
589 if the string is junk. The default is ``None``, meaning that no line is
590 considered junk.
591
592 *charjunk*: A function that accepts a single character argument (a string of
593 length 1), and returns true if the character is junk. The default is ``None``,
594 meaning that no character is considered junk.
595
Benjamin Petersone41251e2008-04-25 01:59:09 +0000596 :class:`Differ` objects are used (deltas generated) via a single method:
Georg Brandl116aa622007-08-15 14:28:22 +0000597
598
Benjamin Petersone41251e2008-04-25 01:59:09 +0000599 .. method:: Differ.compare(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000600
Benjamin Petersone41251e2008-04-25 01:59:09 +0000601 Compare two sequences of lines, and generate the delta (a sequence of lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000602
Benjamin Petersone41251e2008-04-25 01:59:09 +0000603 Each sequence must contain individual single-line strings ending with newlines.
604 Such sequences can be obtained from the :meth:`readlines` method of file-like
605 objects. The delta generated also consists of newline-terminated strings, ready
606 to be printed as-is via the :meth:`writelines` method of a file-like object.
Georg Brandl116aa622007-08-15 14:28:22 +0000607
608
609.. _differ-examples:
610
611Differ Example
612--------------
613
614This example compares two texts. First we set up the texts, sequences of
615individual single-line strings ending with newlines (such sequences can also be
Christian Heimesfe337bf2008-03-23 21:54:12 +0000616obtained from the :meth:`readlines` method of file-like objects):
Georg Brandl116aa622007-08-15 14:28:22 +0000617
618 >>> text1 = ''' 1. Beautiful is better than ugly.
619 ... 2. Explicit is better than implicit.
620 ... 3. Simple is better than complex.
621 ... 4. Complex is better than complicated.
622 ... '''.splitlines(1)
623 >>> len(text1)
624 4
625 >>> text1[0][-1]
626 '\n'
627 >>> text2 = ''' 1. Beautiful is better than ugly.
628 ... 3. Simple is better than complex.
629 ... 4. Complicated is better than complex.
630 ... 5. Flat is better than nested.
631 ... '''.splitlines(1)
632
Christian Heimesfe337bf2008-03-23 21:54:12 +0000633Next we instantiate a Differ object:
Georg Brandl116aa622007-08-15 14:28:22 +0000634
635 >>> d = Differ()
636
637Note that when instantiating a :class:`Differ` object we may pass functions to
638filter out line and character "junk." See the :meth:`Differ` constructor for
639details.
640
Christian Heimesfe337bf2008-03-23 21:54:12 +0000641Finally, we compare the two:
Georg Brandl116aa622007-08-15 14:28:22 +0000642
643 >>> result = list(d.compare(text1, text2))
644
Christian Heimesfe337bf2008-03-23 21:54:12 +0000645``result`` is a list of strings, so let's pretty-print it:
Georg Brandl116aa622007-08-15 14:28:22 +0000646
647 >>> from pprint import pprint
648 >>> pprint(result)
649 [' 1. Beautiful is better than ugly.\n',
650 '- 2. Explicit is better than implicit.\n',
651 '- 3. Simple is better than complex.\n',
652 '+ 3. Simple is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000653 '? ++\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000654 '- 4. Complex is better than complicated.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000655 '? ^ ---- ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000656 '+ 4. Complicated is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000657 '? ++++ ^ ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000658 '+ 5. Flat is better than nested.\n']
659
Christian Heimesfe337bf2008-03-23 21:54:12 +0000660As a single multi-line string it looks like this:
Georg Brandl116aa622007-08-15 14:28:22 +0000661
662 >>> import sys
663 >>> sys.stdout.writelines(result)
664 1. Beautiful is better than ugly.
665 - 2. Explicit is better than implicit.
666 - 3. Simple is better than complex.
667 + 3. Simple is better than complex.
668 ? ++
669 - 4. Complex is better than complicated.
670 ? ^ ---- ^
671 + 4. Complicated is better than complex.
672 ? ++++ ^ ^
673 + 5. Flat is better than nested.
674
Christian Heimes8640e742008-02-23 16:23:06 +0000675
676.. _difflib-interface:
677
678A command-line interface to difflib
679-----------------------------------
680
681This example shows how to use difflib to create a ``diff``-like utility.
682It is also contained in the Python source distribution, as
683:file:`Tools/scripts/diff.py`.
684
Christian Heimesfe337bf2008-03-23 21:54:12 +0000685.. testcode::
Christian Heimes8640e742008-02-23 16:23:06 +0000686
687 """ Command line interface to difflib.py providing diffs in four formats:
688
689 * ndiff: lists every line and highlights interline changes.
690 * context: highlights clusters of changes in a before/after format.
691 * unified: highlights clusters of changes in an inline format.
692 * html: generates side by side comparison with change highlights.
693
694 """
695
696 import sys, os, time, difflib, optparse
697
698 def main():
699 # Configure the option parser
700 usage = "usage: %prog [options] fromfile tofile"
701 parser = optparse.OptionParser(usage)
702 parser.add_option("-c", action="store_true", default=False,
703 help='Produce a context format diff (default)')
704 parser.add_option("-u", action="store_true", default=False,
705 help='Produce a unified format diff')
706 hlp = 'Produce HTML side by side diff (can use -c and -l in conjunction)'
707 parser.add_option("-m", action="store_true", default=False, help=hlp)
708 parser.add_option("-n", action="store_true", default=False,
709 help='Produce a ndiff format diff')
710 parser.add_option("-l", "--lines", type="int", default=3,
711 help='Set number of context lines (default 3)')
712 (options, args) = parser.parse_args()
713
714 if len(args) == 0:
715 parser.print_help()
716 sys.exit(1)
717 if len(args) != 2:
718 parser.error("need to specify both a fromfile and tofile")
719
720 n = options.lines
721 fromfile, tofile = args # as specified in the usage string
722
723 # we're passing these as arguments to the diff function
724 fromdate = time.ctime(os.stat(fromfile).st_mtime)
725 todate = time.ctime(os.stat(tofile).st_mtime)
726 fromlines = open(fromfile, 'U').readlines()
727 tolines = open(tofile, 'U').readlines()
728
729 if options.u:
730 diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
731 fromdate, todate, n=n)
732 elif options.n:
733 diff = difflib.ndiff(fromlines, tolines)
734 elif options.m:
735 diff = difflib.HtmlDiff().make_file(fromlines, tolines, fromfile,
736 tofile, context=options.c,
737 numlines=n)
738 else:
739 diff = difflib.context_diff(fromlines, tolines, fromfile, tofile,
740 fromdate, todate, n=n)
741
742 # we're using writelines because diff is a generator
743 sys.stdout.writelines(diff)
744
745 if __name__ == '__main__':
746 main()