blob: 836e240b83061fe966f5e51cfc1adbce5ae1fc16 [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
Terry Reedy99f96372010-11-25 06:12:34 +000020
Georg Brandl116aa622007-08-15 14:28:22 +000021.. class:: SequenceMatcher
22
23 This is a flexible class for comparing pairs of sequences of any type, so long
Guido van Rossum2cc30da2007-11-02 23:46:40 +000024 as the sequence elements are :term:`hashable`. The basic algorithm predates, and is a
Georg Brandl116aa622007-08-15 14:28:22 +000025 little fancier than, an algorithm published in the late 1980's by Ratcliff and
26 Obershelp under the hyperbolic name "gestalt pattern matching." The idea is to
27 find the longest contiguous matching subsequence that contains no "junk"
28 elements (the Ratcliff and Obershelp algorithm doesn't address junk). The same
29 idea is then applied recursively to the pieces of the sequences to the left and
30 to the right of the matching subsequence. This does not yield minimal edit
31 sequences, but does tend to yield matches that "look right" to people.
32
33 **Timing:** The basic Ratcliff-Obershelp algorithm is cubic time in the worst
34 case and quadratic time in the expected case. :class:`SequenceMatcher` is
35 quadratic time for the worst case and has expected-case behavior dependent in a
36 complicated way on how many elements the sequences have in common; best case
37 time is linear.
38
Terry Reedy99f96372010-11-25 06:12:34 +000039 **Automatic junk heuristic:** :class:`SequenceMatcher` supports a heuristic that
40 automatically treats certain sequence items as junk. The heuristic counts how many
41 times each individual item appears in the sequence. If an item's duplicates (after
42 the first one) account for more than 1% of the sequence and the sequence is at least
43 200 items long, this item is marked as "popular" and is treated as junk for
44 the purpose of sequence matching. This heuristic can be turned off by setting
45 the ``autojunk`` argument to ``False`` when creating the :class:`SequenceMatcher`.
46
Terry Reedydc9b17d2010-11-27 20:52:14 +000047 .. versionadded:: 3.2
48 The *autojunk* parameter.
49
Georg Brandl116aa622007-08-15 14:28:22 +000050
51.. class:: Differ
52
53 This is a class for comparing sequences of lines of text, and producing
54 human-readable differences or deltas. Differ uses :class:`SequenceMatcher`
55 both to compare sequences of lines, and to compare sequences of characters
56 within similar (near-matching) lines.
57
58 Each line of a :class:`Differ` delta begins with a two-letter code:
59
60 +----------+-------------------------------------------+
61 | Code | Meaning |
62 +==========+===========================================+
63 | ``'- '`` | line unique to sequence 1 |
64 +----------+-------------------------------------------+
65 | ``'+ '`` | line unique to sequence 2 |
66 +----------+-------------------------------------------+
67 | ``' '`` | line common to both sequences |
68 +----------+-------------------------------------------+
69 | ``'? '`` | line not present in either input sequence |
70 +----------+-------------------------------------------+
71
72 Lines beginning with '``?``' attempt to guide the eye to intraline differences,
73 and were not present in either input sequence. These lines can be confusing if
74 the sequences contain tab characters.
75
76
77.. class:: HtmlDiff
78
79 This class can be used to create an HTML table (or a complete HTML file
80 containing the table) showing a side by side, line by line comparison of text
81 with inter-line and intra-line change highlights. The table can be generated in
82 either full or contextual difference mode.
83
84 The constructor for this class is:
85
86
Georg Brandlc2a4f4f2009-04-10 09:03:43 +000087 .. method:: __init__(tabsize=8, wrapcolumn=None, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +000088
89 Initializes instance of :class:`HtmlDiff`.
90
91 *tabsize* is an optional keyword argument to specify tab stop spacing and
92 defaults to ``8``.
93
94 *wrapcolumn* is an optional keyword to specify column number where lines are
95 broken and wrapped, defaults to ``None`` where lines are not wrapped.
96
97 *linejunk* and *charjunk* are optional keyword arguments passed into ``ndiff()``
98 (used by :class:`HtmlDiff` to generate the side by side HTML differences). See
99 ``ndiff()`` documentation for argument default values and descriptions.
100
101 The following methods are public:
102
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000103 .. method:: make_file(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000104
105 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
106 is a complete HTML file containing a table showing line by line differences with
107 inter-line and intra-line changes highlighted.
108
109 *fromdesc* and *todesc* are optional keyword arguments to specify from/to file
110 column header strings (both default to an empty string).
111
112 *context* and *numlines* are both optional keyword arguments. Set *context* to
113 ``True`` when contextual differences are to be shown, else the default is
114 ``False`` to show the full files. *numlines* defaults to ``5``. When *context*
115 is ``True`` *numlines* controls the number of context lines which surround the
116 difference highlights. When *context* is ``False`` *numlines* controls the
117 number of lines which are shown before a difference highlight when using the
118 "next" hyperlinks (setting to zero would cause the "next" hyperlinks to place
119 the next difference highlight at the top of the browser without any leading
120 context).
121
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000122 .. method:: make_table(fromlines, tolines, fromdesc='', todesc='', context=False, numlines=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000123
124 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
125 is a complete HTML table showing line by line differences with inter-line and
126 intra-line changes highlighted.
127
128 The arguments for this method are the same as those for the :meth:`make_file`
129 method.
130
131 :file:`Tools/scripts/diff.py` is a command-line front-end to this class and
132 contains a good example of its use.
133
Georg Brandl116aa622007-08-15 14:28:22 +0000134
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000135.. function:: context_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000136
Georg Brandl9afde1c2007-11-01 20:32:30 +0000137 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
138 generating the delta lines) in context diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000139
140 Context diffs are a compact way of showing just the lines that have changed plus
141 a few lines of context. The changes are shown in a before/after style. The
142 number of context lines is set by *n* which defaults to three.
143
144 By default, the diff control lines (those with ``***`` or ``---``) are created
145 with a trailing newline. This is helpful so that inputs created from
146 :func:`file.readlines` result in diffs that are suitable for use with
147 :func:`file.writelines` since both the inputs and outputs have trailing
148 newlines.
149
150 For inputs that do not have trailing newlines, set the *lineterm* argument to
151 ``""`` so that the output will be uniformly newline free.
152
153 The context diff format normally has a header for filenames and modification
154 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murrayb2416e52010-04-12 16:58:02 +0000155 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
156 expressed in the ISO 8601 format. If not specified, the
Georg Brandl116aa622007-08-15 14:28:22 +0000157 strings default to blanks.
158
Christian Heimes8640e742008-02-23 16:23:06 +0000159 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
160 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
161 >>> for line in context_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000162 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000163 *** before.py
164 --- after.py
165 ***************
166 *** 1,4 ****
167 ! bacon
168 ! eggs
169 ! ham
170 guido
171 --- 1,4 ----
172 ! python
173 ! eggy
174 ! hamster
175 guido
176
177 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000178
Georg Brandl116aa622007-08-15 14:28:22 +0000179
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000180.. function:: get_close_matches(word, possibilities, n=3, cutoff=0.6)
Georg Brandl116aa622007-08-15 14:28:22 +0000181
182 Return a list of the best "good enough" matches. *word* is a sequence for which
183 close matches are desired (typically a string), and *possibilities* is a list of
184 sequences against which to match *word* (typically a list of strings).
185
186 Optional argument *n* (default ``3``) is the maximum number of close matches to
187 return; *n* must be greater than ``0``.
188
189 Optional argument *cutoff* (default ``0.6``) is a float in the range [0, 1].
190 Possibilities that don't score at least that similar to *word* are ignored.
191
192 The best (no more than *n*) matches among the possibilities are returned in a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000193 list, sorted by similarity score, most similar first.
Georg Brandl116aa622007-08-15 14:28:22 +0000194
195 >>> get_close_matches('appel', ['ape', 'apple', 'peach', 'puppy'])
196 ['apple', 'ape']
197 >>> import keyword
198 >>> get_close_matches('wheel', keyword.kwlist)
199 ['while']
200 >>> get_close_matches('apple', keyword.kwlist)
201 []
202 >>> get_close_matches('accept', keyword.kwlist)
203 ['except']
204
205
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000206.. function:: ndiff(a, b, linejunk=None, charjunk=IS_CHARACTER_JUNK)
Georg Brandl116aa622007-08-15 14:28:22 +0000207
Georg Brandl9afde1c2007-11-01 20:32:30 +0000208 Compare *a* and *b* (lists of strings); return a :class:`Differ`\ -style
209 delta (a :term:`generator` generating the delta lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000210
211 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
212 (or ``None``):
213
Georg Brandle6bcc912008-05-12 18:05:20 +0000214 *linejunk*: A function that accepts a single string argument, and returns
215 true if the string is junk, or false if not. The default is ``None``. There
216 is also a module-level function :func:`IS_LINE_JUNK`, which filters out lines
217 without visible characters, except for at most one pound character (``'#'``)
218 -- however the underlying :class:`SequenceMatcher` class does a dynamic
219 analysis of which lines are so frequent as to constitute noise, and this
220 usually works better than using this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000221
222 *charjunk*: A function that accepts a character (a string of length 1), and
223 returns if the character is junk, or false if not. The default is module-level
224 function :func:`IS_CHARACTER_JUNK`, which filters out whitespace characters (a
225 blank or tab; note: bad idea to include newline in this!).
226
Christian Heimesfe337bf2008-03-23 21:54:12 +0000227 :file:`Tools/scripts/ndiff.py` is a command-line front-end to this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000228
229 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
230 ... 'ore\ntree\nemu\n'.splitlines(1))
Georg Brandl6911e3c2007-09-04 07:15:32 +0000231 >>> print(''.join(diff), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000232 - one
233 ? ^
234 + ore
235 ? ^
236 - two
237 - three
238 ? -
239 + tree
240 + emu
241
242
243.. function:: restore(sequence, which)
244
245 Return one of the two sequences that generated a delta.
246
247 Given a *sequence* produced by :meth:`Differ.compare` or :func:`ndiff`, extract
248 lines originating from file 1 or 2 (parameter *which*), stripping off line
249 prefixes.
250
Christian Heimesfe337bf2008-03-23 21:54:12 +0000251 Example:
Georg Brandl116aa622007-08-15 14:28:22 +0000252
253 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
254 ... 'ore\ntree\nemu\n'.splitlines(1))
255 >>> diff = list(diff) # materialize the generated delta into a list
Georg Brandl6911e3c2007-09-04 07:15:32 +0000256 >>> print(''.join(restore(diff, 1)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000257 one
258 two
259 three
Georg Brandl6911e3c2007-09-04 07:15:32 +0000260 >>> print(''.join(restore(diff, 2)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000261 ore
262 tree
263 emu
264
265
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000266.. function:: unified_diff(a, b, fromfile='', tofile='', fromfiledate='', tofiledate='', n=3, lineterm='\\n')
Georg Brandl116aa622007-08-15 14:28:22 +0000267
Georg Brandl9afde1c2007-11-01 20:32:30 +0000268 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
269 generating the delta lines) in unified diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000270
271 Unified diffs are a compact way of showing just the lines that have changed plus
272 a few lines of context. The changes are shown in a inline style (instead of
273 separate before/after blocks). The number of context lines is set by *n* which
274 defaults to three.
275
276 By default, the diff control lines (those with ``---``, ``+++``, or ``@@``) are
277 created with a trailing newline. This is helpful so that inputs created from
278 :func:`file.readlines` result in diffs that are suitable for use with
279 :func:`file.writelines` since both the inputs and outputs have trailing
280 newlines.
281
282 For inputs that do not have trailing newlines, set the *lineterm* argument to
283 ``""`` so that the output will be uniformly newline free.
284
285 The context diff format normally has a header for filenames and modification
286 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murrayb2416e52010-04-12 16:58:02 +0000287 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
288 expressed in the ISO 8601 format. If not specified, the
Georg Brandl116aa622007-08-15 14:28:22 +0000289 strings default to blanks.
290
Christian Heimes8640e742008-02-23 16:23:06 +0000291
292 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
293 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
294 >>> for line in unified_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000295 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000296 --- before.py
297 +++ after.py
298 @@ -1,4 +1,4 @@
299 -bacon
300 -eggs
301 -ham
302 +python
303 +eggy
304 +hamster
305 guido
306
307 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000308
Georg Brandl116aa622007-08-15 14:28:22 +0000309
310.. function:: IS_LINE_JUNK(line)
311
312 Return true for ignorable lines. The line *line* is ignorable if *line* is
313 blank or contains a single ``'#'``, otherwise it is not ignorable. Used as a
Georg Brandle6bcc912008-05-12 18:05:20 +0000314 default for parameter *linejunk* in :func:`ndiff` in older versions.
Georg Brandl116aa622007-08-15 14:28:22 +0000315
316
317.. function:: IS_CHARACTER_JUNK(ch)
318
319 Return true for ignorable characters. The character *ch* is ignorable if *ch*
320 is a space or tab, otherwise it is not ignorable. Used as a default for
321 parameter *charjunk* in :func:`ndiff`.
322
323
324.. seealso::
325
326 `Pattern Matching: The Gestalt Approach <http://www.ddj.com/184407970?pgno=5>`_
327 Discussion of a similar algorithm by John W. Ratcliff and D. E. Metzener. This
328 was published in `Dr. Dobb's Journal <http://www.ddj.com/>`_ in July, 1988.
329
330
331.. _sequence-matcher:
332
333SequenceMatcher Objects
334-----------------------
335
336The :class:`SequenceMatcher` class has this constructor:
337
338
Terry Reedy99f96372010-11-25 06:12:34 +0000339.. class:: SequenceMatcher(isjunk=None, a='', b='', autojunk=True)
Georg Brandl116aa622007-08-15 14:28:22 +0000340
341 Optional argument *isjunk* must be ``None`` (the default) or a one-argument
342 function that takes a sequence element and returns true if and only if the
343 element is "junk" and should be ignored. Passing ``None`` for *isjunk* is
344 equivalent to passing ``lambda x: 0``; in other words, no elements are ignored.
345 For example, pass::
346
347 lambda x: x in " \t"
348
349 if you're comparing lines as sequences of characters, and don't want to synch up
350 on blanks or hard tabs.
351
352 The optional arguments *a* and *b* are sequences to be compared; both default to
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000353 empty strings. The elements of both sequences must be :term:`hashable`.
Georg Brandl116aa622007-08-15 14:28:22 +0000354
Terry Reedy99f96372010-11-25 06:12:34 +0000355 The optional argument *autojunk* can be used to disable the automatic junk
356 heuristic.
357
Terry Reedydc9b17d2010-11-27 20:52:14 +0000358 .. versionadded:: 3.2
359 The *autojunk* parameter.
360
Terry Reedy74a7c672010-12-03 18:57:42 +0000361 SequenceMatcher objects get three data attributes: *bjunk* is the
Terry Reedy17a59252010-12-15 20:18:10 +0000362 set of elements of *b* for which *isjunk* is True; *bpopular* is the set of
363 non-junk elements considered popular by the heuristic (if it is not
364 disabled); *b2j* is a dict mapping the remaining elements of *b* to a list
365 of positions where they occur. All three are reset whenever *b* is reset
366 with :meth:`set_seqs` or :meth:`set_seq2`.
Terry Reedy74a7c672010-12-03 18:57:42 +0000367
Georg Brandl500be242010-12-03 19:56:42 +0000368 .. versionadded:: 3.2
Terry Reedy74a7c672010-12-03 18:57:42 +0000369 The *bjunk* and *bpopular* attributes.
370
Benjamin Petersone41251e2008-04-25 01:59:09 +0000371 :class:`SequenceMatcher` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000372
Benjamin Petersone41251e2008-04-25 01:59:09 +0000373 .. method:: set_seqs(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000374
Benjamin Petersone41251e2008-04-25 01:59:09 +0000375 Set the two sequences to be compared.
Georg Brandl116aa622007-08-15 14:28:22 +0000376
Benjamin Petersone41251e2008-04-25 01:59:09 +0000377 :class:`SequenceMatcher` computes and caches detailed information about the
378 second sequence, so if you want to compare one sequence against many
379 sequences, use :meth:`set_seq2` to set the commonly used sequence once and
380 call :meth:`set_seq1` repeatedly, once for each of the other sequences.
Georg Brandl116aa622007-08-15 14:28:22 +0000381
382
Benjamin Petersone41251e2008-04-25 01:59:09 +0000383 .. method:: set_seq1(a)
Georg Brandl116aa622007-08-15 14:28:22 +0000384
Benjamin Petersone41251e2008-04-25 01:59:09 +0000385 Set the first sequence to be compared. The second sequence to be compared
386 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000387
388
Benjamin Petersone41251e2008-04-25 01:59:09 +0000389 .. method:: set_seq2(b)
Georg Brandl116aa622007-08-15 14:28:22 +0000390
Benjamin Petersone41251e2008-04-25 01:59:09 +0000391 Set the second sequence to be compared. The first sequence to be compared
392 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000393
394
Benjamin Petersone41251e2008-04-25 01:59:09 +0000395 .. method:: find_longest_match(alo, ahi, blo, bhi)
Georg Brandl116aa622007-08-15 14:28:22 +0000396
Benjamin Petersone41251e2008-04-25 01:59:09 +0000397 Find longest matching block in ``a[alo:ahi]`` and ``b[blo:bhi]``.
Georg Brandl116aa622007-08-15 14:28:22 +0000398
Benjamin Petersone41251e2008-04-25 01:59:09 +0000399 If *isjunk* was omitted or ``None``, :meth:`find_longest_match` returns
400 ``(i, j, k)`` such that ``a[i:i+k]`` is equal to ``b[j:j+k]``, where ``alo
401 <= i <= i+k <= ahi`` and ``blo <= j <= j+k <= bhi``. For all ``(i', j',
402 k')`` meeting those conditions, the additional conditions ``k >= k'``, ``i
403 <= i'``, and if ``i == i'``, ``j <= j'`` are also met. In other words, of
404 all maximal matching blocks, return one that starts earliest in *a*, and
405 of all those maximal matching blocks that start earliest in *a*, return
406 the one that starts earliest in *b*.
Georg Brandl116aa622007-08-15 14:28:22 +0000407
Benjamin Petersone41251e2008-04-25 01:59:09 +0000408 >>> s = SequenceMatcher(None, " abcd", "abcd abcd")
409 >>> s.find_longest_match(0, 5, 0, 9)
410 Match(a=0, b=4, size=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000411
Benjamin Petersone41251e2008-04-25 01:59:09 +0000412 If *isjunk* was provided, first the longest matching block is determined
413 as above, but with the additional restriction that no junk element appears
414 in the block. Then that block is extended as far as possible by matching
415 (only) junk elements on both sides. So the resulting block never matches
416 on junk except as identical junk happens to be adjacent to an interesting
417 match.
Georg Brandl116aa622007-08-15 14:28:22 +0000418
Benjamin Petersone41251e2008-04-25 01:59:09 +0000419 Here's the same example as before, but considering blanks to be junk. That
420 prevents ``' abcd'`` from matching the ``' abcd'`` at the tail end of the
421 second sequence directly. Instead only the ``'abcd'`` can match, and
422 matches the leftmost ``'abcd'`` in the second sequence:
Georg Brandl116aa622007-08-15 14:28:22 +0000423
Benjamin Petersone41251e2008-04-25 01:59:09 +0000424 >>> s = SequenceMatcher(lambda x: x==" ", " abcd", "abcd abcd")
425 >>> s.find_longest_match(0, 5, 0, 9)
426 Match(a=1, b=0, size=4)
Georg Brandl116aa622007-08-15 14:28:22 +0000427
Benjamin Petersone41251e2008-04-25 01:59:09 +0000428 If no blocks match, this returns ``(alo, blo, 0)``.
Georg Brandl116aa622007-08-15 14:28:22 +0000429
Benjamin Petersone41251e2008-04-25 01:59:09 +0000430 This method returns a :term:`named tuple` ``Match(a, b, size)``.
Christian Heimes25bb7832008-01-11 16:17:00 +0000431
Georg Brandl116aa622007-08-15 14:28:22 +0000432
Benjamin Petersone41251e2008-04-25 01:59:09 +0000433 .. method:: get_matching_blocks()
Georg Brandl116aa622007-08-15 14:28:22 +0000434
Benjamin Petersone41251e2008-04-25 01:59:09 +0000435 Return list of triples describing matching subsequences. Each triple is of
436 the form ``(i, j, n)``, and means that ``a[i:i+n] == b[j:j+n]``. The
437 triples are monotonically increasing in *i* and *j*.
Georg Brandl116aa622007-08-15 14:28:22 +0000438
Benjamin Petersone41251e2008-04-25 01:59:09 +0000439 The last triple is a dummy, and has the value ``(len(a), len(b), 0)``. It
440 is the only triple with ``n == 0``. If ``(i, j, n)`` and ``(i', j', n')``
441 are adjacent triples in the list, and the second is not the last triple in
442 the list, then ``i+n != i'`` or ``j+n != j'``; in other words, adjacent
443 triples always describe non-adjacent equal blocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000444
Benjamin Petersone41251e2008-04-25 01:59:09 +0000445 .. XXX Explain why a dummy is used!
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000446
Benjamin Petersone41251e2008-04-25 01:59:09 +0000447 .. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000448
Benjamin Petersone41251e2008-04-25 01:59:09 +0000449 >>> s = SequenceMatcher(None, "abxcd", "abcd")
450 >>> s.get_matching_blocks()
451 [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 +0000452
453
Benjamin Petersone41251e2008-04-25 01:59:09 +0000454 .. method:: get_opcodes()
Georg Brandl116aa622007-08-15 14:28:22 +0000455
Benjamin Petersone41251e2008-04-25 01:59:09 +0000456 Return list of 5-tuples describing how to turn *a* into *b*. Each tuple is
457 of the form ``(tag, i1, i2, j1, j2)``. The first tuple has ``i1 == j1 ==
458 0``, and remaining tuples have *i1* equal to the *i2* from the preceding
459 tuple, and, likewise, *j1* equal to the previous *j2*.
Georg Brandl116aa622007-08-15 14:28:22 +0000460
Benjamin Petersone41251e2008-04-25 01:59:09 +0000461 The *tag* values are strings, with these meanings:
Georg Brandl116aa622007-08-15 14:28:22 +0000462
Benjamin Petersone41251e2008-04-25 01:59:09 +0000463 +---------------+---------------------------------------------+
464 | Value | Meaning |
465 +===============+=============================================+
466 | ``'replace'`` | ``a[i1:i2]`` should be replaced by |
467 | | ``b[j1:j2]``. |
468 +---------------+---------------------------------------------+
469 | ``'delete'`` | ``a[i1:i2]`` should be deleted. Note that |
470 | | ``j1 == j2`` in this case. |
471 +---------------+---------------------------------------------+
472 | ``'insert'`` | ``b[j1:j2]`` should be inserted at |
473 | | ``a[i1:i1]``. Note that ``i1 == i2`` in |
474 | | this case. |
475 +---------------+---------------------------------------------+
476 | ``'equal'`` | ``a[i1:i2] == b[j1:j2]`` (the sub-sequences |
477 | | are equal). |
478 +---------------+---------------------------------------------+
Georg Brandl116aa622007-08-15 14:28:22 +0000479
Benjamin Petersone41251e2008-04-25 01:59:09 +0000480 For example:
Georg Brandl116aa622007-08-15 14:28:22 +0000481
Benjamin Petersone41251e2008-04-25 01:59:09 +0000482 >>> a = "qabxcd"
483 >>> b = "abycdf"
484 >>> s = SequenceMatcher(None, a, b)
485 >>> for tag, i1, i2, j1, j2 in s.get_opcodes():
Raymond Hettingerdbb677a2011-04-09 19:41:00 -0700486 print('{:7} a[{}:{}] --> b[{}:{}] {!r:>8} --> {!r}'.format(
487 tag, i1, i2, j1, j2, a[i1:i2], b[j1:j2]))
488
489
490 delete a[0:1] --> b[0:0] 'q' --> ''
491 equal a[1:3] --> b[0:2] 'ab' --> 'ab'
492 replace a[3:4] --> b[2:3] 'x' --> 'y'
493 equal a[4:6] --> b[3:5] 'cd' --> 'cd'
494 insert a[6:6] --> b[5:6] '' --> 'f'
Georg Brandl116aa622007-08-15 14:28:22 +0000495
496
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000497 .. method:: get_grouped_opcodes(n=3)
Georg Brandl116aa622007-08-15 14:28:22 +0000498
Benjamin Petersone41251e2008-04-25 01:59:09 +0000499 Return a :term:`generator` of groups with up to *n* lines of context.
Georg Brandl116aa622007-08-15 14:28:22 +0000500
Benjamin Petersone41251e2008-04-25 01:59:09 +0000501 Starting with the groups returned by :meth:`get_opcodes`, this method
502 splits out smaller change clusters and eliminates intervening ranges which
503 have no changes.
Georg Brandl116aa622007-08-15 14:28:22 +0000504
Benjamin Petersone41251e2008-04-25 01:59:09 +0000505 The groups are returned in the same format as :meth:`get_opcodes`.
Georg Brandl116aa622007-08-15 14:28:22 +0000506
Georg Brandl116aa622007-08-15 14:28:22 +0000507
Benjamin Petersone41251e2008-04-25 01:59:09 +0000508 .. method:: ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000509
Benjamin Petersone41251e2008-04-25 01:59:09 +0000510 Return a measure of the sequences' similarity as a float in the range [0,
511 1].
Georg Brandl116aa622007-08-15 14:28:22 +0000512
Benjamin Petersone41251e2008-04-25 01:59:09 +0000513 Where T is the total number of elements in both sequences, and M is the
514 number of matches, this is 2.0\*M / T. Note that this is ``1.0`` if the
515 sequences are identical, and ``0.0`` if they have nothing in common.
Georg Brandl116aa622007-08-15 14:28:22 +0000516
Benjamin Petersone41251e2008-04-25 01:59:09 +0000517 This is expensive to compute if :meth:`get_matching_blocks` or
518 :meth:`get_opcodes` hasn't already been called, in which case you may want
519 to try :meth:`quick_ratio` or :meth:`real_quick_ratio` first to get an
520 upper bound.
Georg Brandl116aa622007-08-15 14:28:22 +0000521
522
Benjamin Petersone41251e2008-04-25 01:59:09 +0000523 .. method:: quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000524
Benjamin Petersone41251e2008-04-25 01:59:09 +0000525 Return an upper bound on :meth:`ratio` relatively quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000526
Georg Brandl116aa622007-08-15 14:28:22 +0000527
Benjamin Petersone41251e2008-04-25 01:59:09 +0000528 .. method:: real_quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000529
Benjamin Petersone41251e2008-04-25 01:59:09 +0000530 Return an upper bound on :meth:`ratio` very quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000531
Georg Brandl116aa622007-08-15 14:28:22 +0000532
533The three methods that return the ratio of matching to total characters can give
534different results due to differing levels of approximation, although
535:meth:`quick_ratio` and :meth:`real_quick_ratio` are always at least as large as
Christian Heimesfe337bf2008-03-23 21:54:12 +0000536:meth:`ratio`:
Georg Brandl116aa622007-08-15 14:28:22 +0000537
538 >>> s = SequenceMatcher(None, "abcd", "bcde")
539 >>> s.ratio()
540 0.75
541 >>> s.quick_ratio()
542 0.75
543 >>> s.real_quick_ratio()
544 1.0
545
546
547.. _sequencematcher-examples:
548
549SequenceMatcher Examples
550------------------------
551
Terry Reedy74a7c672010-12-03 18:57:42 +0000552This example compares two strings, considering blanks to be "junk":
Georg Brandl116aa622007-08-15 14:28:22 +0000553
554 >>> s = SequenceMatcher(lambda x: x == " ",
555 ... "private Thread currentThread;",
556 ... "private volatile Thread currentThread;")
557
558:meth:`ratio` returns a float in [0, 1], measuring the similarity of the
559sequences. As a rule of thumb, a :meth:`ratio` value over 0.6 means the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000560sequences are close matches:
Georg Brandl116aa622007-08-15 14:28:22 +0000561
Georg Brandl6911e3c2007-09-04 07:15:32 +0000562 >>> print(round(s.ratio(), 3))
Georg Brandl116aa622007-08-15 14:28:22 +0000563 0.866
564
565If you're only interested in where the sequences match,
Christian Heimesfe337bf2008-03-23 21:54:12 +0000566:meth:`get_matching_blocks` is handy:
Georg Brandl116aa622007-08-15 14:28:22 +0000567
568 >>> for block in s.get_matching_blocks():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000569 ... print("a[%d] and b[%d] match for %d elements" % block)
Georg Brandl116aa622007-08-15 14:28:22 +0000570 a[0] and b[0] match for 8 elements
Christian Heimesfe337bf2008-03-23 21:54:12 +0000571 a[8] and b[17] match for 21 elements
Georg Brandl116aa622007-08-15 14:28:22 +0000572 a[29] and b[38] match for 0 elements
573
574Note that the last tuple returned by :meth:`get_matching_blocks` is always a
575dummy, ``(len(a), len(b), 0)``, and this is the only case in which the last
576tuple element (number of elements matched) is ``0``.
577
578If you want to know how to change the first sequence into the second, use
Christian Heimesfe337bf2008-03-23 21:54:12 +0000579:meth:`get_opcodes`:
Georg Brandl116aa622007-08-15 14:28:22 +0000580
581 >>> for opcode in s.get_opcodes():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000582 ... print("%6s a[%d:%d] b[%d:%d]" % opcode)
Georg Brandl116aa622007-08-15 14:28:22 +0000583 equal a[0:8] b[0:8]
584 insert a[8:8] b[8:17]
Christian Heimesfe337bf2008-03-23 21:54:12 +0000585 equal a[8:29] b[17:38]
Georg Brandl116aa622007-08-15 14:28:22 +0000586
Raymond Hettinger58c8c262009-04-27 21:01:21 +0000587.. seealso::
588
589 * The :func:`get_close_matches` function in this module which shows how
590 simple code building on :class:`SequenceMatcher` can be used to do useful
591 work.
592
593 * `Simple version control recipe
594 <http://code.activestate.com/recipes/576729/>`_ for a small application
595 built with :class:`SequenceMatcher`.
Georg Brandl116aa622007-08-15 14:28:22 +0000596
597
598.. _differ-objects:
599
600Differ Objects
601--------------
602
603Note that :class:`Differ`\ -generated deltas make no claim to be **minimal**
604diffs. To the contrary, minimal diffs are often counter-intuitive, because they
605synch up anywhere possible, sometimes accidental matches 100 pages apart.
606Restricting synch points to contiguous matches preserves some notion of
607locality, at the occasional cost of producing a longer diff.
608
609The :class:`Differ` class has this constructor:
610
611
Georg Brandlc2a4f4f2009-04-10 09:03:43 +0000612.. class:: Differ(linejunk=None, charjunk=None)
Georg Brandl116aa622007-08-15 14:28:22 +0000613
614 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
615 (or ``None``):
616
617 *linejunk*: A function that accepts a single string argument, and returns true
618 if the string is junk. The default is ``None``, meaning that no line is
619 considered junk.
620
621 *charjunk*: A function that accepts a single character argument (a string of
622 length 1), and returns true if the character is junk. The default is ``None``,
623 meaning that no character is considered junk.
624
Benjamin Petersone41251e2008-04-25 01:59:09 +0000625 :class:`Differ` objects are used (deltas generated) via a single method:
Georg Brandl116aa622007-08-15 14:28:22 +0000626
627
Benjamin Petersone41251e2008-04-25 01:59:09 +0000628 .. method:: Differ.compare(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000629
Benjamin Petersone41251e2008-04-25 01:59:09 +0000630 Compare two sequences of lines, and generate the delta (a sequence of lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000631
Benjamin Petersone41251e2008-04-25 01:59:09 +0000632 Each sequence must contain individual single-line strings ending with newlines.
633 Such sequences can be obtained from the :meth:`readlines` method of file-like
634 objects. The delta generated also consists of newline-terminated strings, ready
635 to be printed as-is via the :meth:`writelines` method of a file-like object.
Georg Brandl116aa622007-08-15 14:28:22 +0000636
637
638.. _differ-examples:
639
640Differ Example
641--------------
642
643This example compares two texts. First we set up the texts, sequences of
644individual single-line strings ending with newlines (such sequences can also be
Christian Heimesfe337bf2008-03-23 21:54:12 +0000645obtained from the :meth:`readlines` method of file-like objects):
Georg Brandl116aa622007-08-15 14:28:22 +0000646
647 >>> text1 = ''' 1. Beautiful is better than ugly.
648 ... 2. Explicit is better than implicit.
649 ... 3. Simple is better than complex.
650 ... 4. Complex is better than complicated.
651 ... '''.splitlines(1)
652 >>> len(text1)
653 4
654 >>> text1[0][-1]
655 '\n'
656 >>> text2 = ''' 1. Beautiful is better than ugly.
657 ... 3. Simple is better than complex.
658 ... 4. Complicated is better than complex.
659 ... 5. Flat is better than nested.
660 ... '''.splitlines(1)
661
Christian Heimesfe337bf2008-03-23 21:54:12 +0000662Next we instantiate a Differ object:
Georg Brandl116aa622007-08-15 14:28:22 +0000663
664 >>> d = Differ()
665
666Note that when instantiating a :class:`Differ` object we may pass functions to
667filter out line and character "junk." See the :meth:`Differ` constructor for
668details.
669
Christian Heimesfe337bf2008-03-23 21:54:12 +0000670Finally, we compare the two:
Georg Brandl116aa622007-08-15 14:28:22 +0000671
672 >>> result = list(d.compare(text1, text2))
673
Christian Heimesfe337bf2008-03-23 21:54:12 +0000674``result`` is a list of strings, so let's pretty-print it:
Georg Brandl116aa622007-08-15 14:28:22 +0000675
676 >>> from pprint import pprint
677 >>> pprint(result)
678 [' 1. Beautiful is better than ugly.\n',
679 '- 2. Explicit is better than implicit.\n',
680 '- 3. Simple is better than complex.\n',
681 '+ 3. Simple is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000682 '? ++\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000683 '- 4. Complex is better than complicated.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000684 '? ^ ---- ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000685 '+ 4. Complicated is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000686 '? ++++ ^ ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000687 '+ 5. Flat is better than nested.\n']
688
Christian Heimesfe337bf2008-03-23 21:54:12 +0000689As a single multi-line string it looks like this:
Georg Brandl116aa622007-08-15 14:28:22 +0000690
691 >>> import sys
692 >>> sys.stdout.writelines(result)
693 1. Beautiful is better than ugly.
694 - 2. Explicit is better than implicit.
695 - 3. Simple is better than complex.
696 + 3. Simple is better than complex.
697 ? ++
698 - 4. Complex is better than complicated.
699 ? ^ ---- ^
700 + 4. Complicated is better than complex.
701 ? ++++ ^ ^
702 + 5. Flat is better than nested.
703
Christian Heimes8640e742008-02-23 16:23:06 +0000704
705.. _difflib-interface:
706
707A command-line interface to difflib
708-----------------------------------
709
710This example shows how to use difflib to create a ``diff``-like utility.
711It is also contained in the Python source distribution, as
712:file:`Tools/scripts/diff.py`.
713
Christian Heimesfe337bf2008-03-23 21:54:12 +0000714.. testcode::
Christian Heimes8640e742008-02-23 16:23:06 +0000715
716 """ Command line interface to difflib.py providing diffs in four formats:
717
718 * ndiff: lists every line and highlights interline changes.
719 * context: highlights clusters of changes in a before/after format.
720 * unified: highlights clusters of changes in an inline format.
721 * html: generates side by side comparison with change highlights.
722
723 """
724
725 import sys, os, time, difflib, optparse
726
727 def main():
728 # Configure the option parser
729 usage = "usage: %prog [options] fromfile tofile"
730 parser = optparse.OptionParser(usage)
731 parser.add_option("-c", action="store_true", default=False,
732 help='Produce a context format diff (default)')
733 parser.add_option("-u", action="store_true", default=False,
734 help='Produce a unified format diff')
735 hlp = 'Produce HTML side by side diff (can use -c and -l in conjunction)'
736 parser.add_option("-m", action="store_true", default=False, help=hlp)
737 parser.add_option("-n", action="store_true", default=False,
738 help='Produce a ndiff format diff')
739 parser.add_option("-l", "--lines", type="int", default=3,
740 help='Set number of context lines (default 3)')
741 (options, args) = parser.parse_args()
742
743 if len(args) == 0:
744 parser.print_help()
745 sys.exit(1)
746 if len(args) != 2:
747 parser.error("need to specify both a fromfile and tofile")
748
749 n = options.lines
750 fromfile, tofile = args # as specified in the usage string
751
752 # we're passing these as arguments to the diff function
753 fromdate = time.ctime(os.stat(fromfile).st_mtime)
754 todate = time.ctime(os.stat(tofile).st_mtime)
Éric Araujoa3dd56b2011-03-11 17:42:48 +0100755 with open(fromlines) as fromf, open(tofile) as tof:
756 fromlines, tolines = list(fromf), list(tof)
Christian Heimes8640e742008-02-23 16:23:06 +0000757
758 if options.u:
759 diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
760 fromdate, todate, n=n)
761 elif options.n:
762 diff = difflib.ndiff(fromlines, tolines)
763 elif options.m:
764 diff = difflib.HtmlDiff().make_file(fromlines, tolines, fromfile,
765 tofile, context=options.c,
766 numlines=n)
767 else:
768 diff = difflib.context_diff(fromlines, tolines, fromfile, tofile,
769 fromdate, todate, n=n)
770
771 # we're using writelines because diff is a generator
772 sys.stdout.writelines(diff)
773
774 if __name__ == '__main__':
775 main()