blob: 01c9c85f8fb7c3e60330d51932bdbdab6778e0cf [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
75 .. function:: __init__([tabsize][, wrapcolumn][, linejunk][, charjunk])
76
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
91
92 .. function:: make_file(fromlines, tolines [, fromdesc][, todesc][, context][, numlines])
93
94 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
95 is a complete HTML file containing a table showing line by line differences with
96 inter-line and intra-line changes highlighted.
97
98 *fromdesc* and *todesc* are optional keyword arguments to specify from/to file
99 column header strings (both default to an empty string).
100
101 *context* and *numlines* are both optional keyword arguments. Set *context* to
102 ``True`` when contextual differences are to be shown, else the default is
103 ``False`` to show the full files. *numlines* defaults to ``5``. When *context*
104 is ``True`` *numlines* controls the number of context lines which surround the
105 difference highlights. When *context* is ``False`` *numlines* controls the
106 number of lines which are shown before a difference highlight when using the
107 "next" hyperlinks (setting to zero would cause the "next" hyperlinks to place
108 the next difference highlight at the top of the browser without any leading
109 context).
110
111
112 .. function:: make_table(fromlines, tolines [, fromdesc][, todesc][, context][, numlines])
113
114 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
115 is a complete HTML table showing line by line differences with inter-line and
116 intra-line changes highlighted.
117
118 The arguments for this method are the same as those for the :meth:`make_file`
119 method.
120
121 :file:`Tools/scripts/diff.py` is a command-line front-end to this class and
122 contains a good example of its use.
123
Georg Brandl116aa622007-08-15 14:28:22 +0000124
125.. function:: context_diff(a, b[, fromfile][, tofile][, fromfiledate][, tofiledate][, n][, lineterm])
126
Georg Brandl9afde1c2007-11-01 20:32:30 +0000127 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
128 generating the delta lines) in context diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000129
130 Context diffs are a compact way of showing just the lines that have changed plus
131 a few lines of context. The changes are shown in a before/after style. The
132 number of context lines is set by *n* which defaults to three.
133
134 By default, the diff control lines (those with ``***`` or ``---``) are created
135 with a trailing newline. This is helpful so that inputs created from
136 :func:`file.readlines` result in diffs that are suitable for use with
137 :func:`file.writelines` since both the inputs and outputs have trailing
138 newlines.
139
140 For inputs that do not have trailing newlines, set the *lineterm* argument to
141 ``""`` so that the output will be uniformly newline free.
142
143 The context diff format normally has a header for filenames and modification
144 times. Any or all of these may be specified using strings for *fromfile*,
145 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
146 expressed in the format returned by :func:`time.ctime`. If not specified, the
147 strings default to blanks.
148
Christian Heimes8640e742008-02-23 16:23:06 +0000149 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
150 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
151 >>> for line in context_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000152 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000153 *** before.py
154 --- after.py
155 ***************
156 *** 1,4 ****
157 ! bacon
158 ! eggs
159 ! ham
160 guido
161 --- 1,4 ----
162 ! python
163 ! eggy
164 ! hamster
165 guido
166
167 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000168
Georg Brandl116aa622007-08-15 14:28:22 +0000169
170.. function:: get_close_matches(word, possibilities[, n][, cutoff])
171
172 Return a list of the best "good enough" matches. *word* is a sequence for which
173 close matches are desired (typically a string), and *possibilities* is a list of
174 sequences against which to match *word* (typically a list of strings).
175
176 Optional argument *n* (default ``3``) is the maximum number of close matches to
177 return; *n* must be greater than ``0``.
178
179 Optional argument *cutoff* (default ``0.6``) is a float in the range [0, 1].
180 Possibilities that don't score at least that similar to *word* are ignored.
181
182 The best (no more than *n*) matches among the possibilities are returned in a
Christian Heimesfe337bf2008-03-23 21:54:12 +0000183 list, sorted by similarity score, most similar first.
Georg Brandl116aa622007-08-15 14:28:22 +0000184
185 >>> get_close_matches('appel', ['ape', 'apple', 'peach', 'puppy'])
186 ['apple', 'ape']
187 >>> import keyword
188 >>> get_close_matches('wheel', keyword.kwlist)
189 ['while']
190 >>> get_close_matches('apple', keyword.kwlist)
191 []
192 >>> get_close_matches('accept', keyword.kwlist)
193 ['except']
194
195
196.. function:: ndiff(a, b[, linejunk][, charjunk])
197
Georg Brandl9afde1c2007-11-01 20:32:30 +0000198 Compare *a* and *b* (lists of strings); return a :class:`Differ`\ -style
199 delta (a :term:`generator` generating the delta lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000200
201 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
202 (or ``None``):
203
204 *linejunk*: A function that accepts a single string argument, and returns true
205 if the string is junk, or false if not. The default is (``None``), starting with
206 Python 2.3. Before then, the default was the module-level function
207 :func:`IS_LINE_JUNK`, which filters out lines without visible characters, except
208 for at most one pound character (``'#'``). As of Python 2.3, the underlying
209 :class:`SequenceMatcher` class does a dynamic analysis of which lines are so
210 frequent as to constitute noise, and this usually works better than the pre-2.3
211 default.
212
213 *charjunk*: A function that accepts a character (a string of length 1), and
214 returns if the character is junk, or false if not. The default is module-level
215 function :func:`IS_CHARACTER_JUNK`, which filters out whitespace characters (a
216 blank or tab; note: bad idea to include newline in this!).
217
Christian Heimesfe337bf2008-03-23 21:54:12 +0000218 :file:`Tools/scripts/ndiff.py` is a command-line front-end to this function.
Georg Brandl116aa622007-08-15 14:28:22 +0000219
220 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
221 ... 'ore\ntree\nemu\n'.splitlines(1))
Georg Brandl6911e3c2007-09-04 07:15:32 +0000222 >>> print(''.join(diff), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000223 - one
224 ? ^
225 + ore
226 ? ^
227 - two
228 - three
229 ? -
230 + tree
231 + emu
232
233
234.. function:: restore(sequence, which)
235
236 Return one of the two sequences that generated a delta.
237
238 Given a *sequence* produced by :meth:`Differ.compare` or :func:`ndiff`, extract
239 lines originating from file 1 or 2 (parameter *which*), stripping off line
240 prefixes.
241
Christian Heimesfe337bf2008-03-23 21:54:12 +0000242 Example:
Georg Brandl116aa622007-08-15 14:28:22 +0000243
244 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
245 ... 'ore\ntree\nemu\n'.splitlines(1))
246 >>> diff = list(diff) # materialize the generated delta into a list
Georg Brandl6911e3c2007-09-04 07:15:32 +0000247 >>> print(''.join(restore(diff, 1)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000248 one
249 two
250 three
Georg Brandl6911e3c2007-09-04 07:15:32 +0000251 >>> print(''.join(restore(diff, 2)), end="")
Georg Brandl116aa622007-08-15 14:28:22 +0000252 ore
253 tree
254 emu
255
256
257.. function:: unified_diff(a, b[, fromfile][, tofile][, fromfiledate][, tofiledate][, n][, lineterm])
258
Georg Brandl9afde1c2007-11-01 20:32:30 +0000259 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
260 generating the delta lines) in unified diff format.
Georg Brandl116aa622007-08-15 14:28:22 +0000261
262 Unified diffs are a compact way of showing just the lines that have changed plus
263 a few lines of context. The changes are shown in a inline style (instead of
264 separate before/after blocks). The number of context lines is set by *n* which
265 defaults to three.
266
267 By default, the diff control lines (those with ``---``, ``+++``, or ``@@``) are
268 created with a trailing newline. This is helpful so that inputs created from
269 :func:`file.readlines` result in diffs that are suitable for use with
270 :func:`file.writelines` since both the inputs and outputs have trailing
271 newlines.
272
273 For inputs that do not have trailing newlines, set the *lineterm* argument to
274 ``""`` so that the output will be uniformly newline free.
275
276 The context diff format normally has a header for filenames and modification
277 times. Any or all of these may be specified using strings for *fromfile*,
278 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
279 expressed in the format returned by :func:`time.ctime`. If not specified, the
280 strings default to blanks.
281
Christian Heimes8640e742008-02-23 16:23:06 +0000282
283 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
284 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
285 >>> for line in unified_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Christian Heimesfe337bf2008-03-23 21:54:12 +0000286 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Christian Heimes8640e742008-02-23 16:23:06 +0000287 --- before.py
288 +++ after.py
289 @@ -1,4 +1,4 @@
290 -bacon
291 -eggs
292 -ham
293 +python
294 +eggy
295 +hamster
296 guido
297
298 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl116aa622007-08-15 14:28:22 +0000299
Georg Brandl116aa622007-08-15 14:28:22 +0000300
301.. function:: IS_LINE_JUNK(line)
302
303 Return true for ignorable lines. The line *line* is ignorable if *line* is
304 blank or contains a single ``'#'``, otherwise it is not ignorable. Used as a
305 default for parameter *linejunk* in :func:`ndiff` before Python 2.3.
306
307
308.. function:: IS_CHARACTER_JUNK(ch)
309
310 Return true for ignorable characters. The character *ch* is ignorable if *ch*
311 is a space or tab, otherwise it is not ignorable. Used as a default for
312 parameter *charjunk* in :func:`ndiff`.
313
314
315.. seealso::
316
317 `Pattern Matching: The Gestalt Approach <http://www.ddj.com/184407970?pgno=5>`_
318 Discussion of a similar algorithm by John W. Ratcliff and D. E. Metzener. This
319 was published in `Dr. Dobb's Journal <http://www.ddj.com/>`_ in July, 1988.
320
321
322.. _sequence-matcher:
323
324SequenceMatcher Objects
325-----------------------
326
327The :class:`SequenceMatcher` class has this constructor:
328
329
330.. class:: SequenceMatcher([isjunk[, a[, b]]])
331
332 Optional argument *isjunk* must be ``None`` (the default) or a one-argument
333 function that takes a sequence element and returns true if and only if the
334 element is "junk" and should be ignored. Passing ``None`` for *isjunk* is
335 equivalent to passing ``lambda x: 0``; in other words, no elements are ignored.
336 For example, pass::
337
338 lambda x: x in " \t"
339
340 if you're comparing lines as sequences of characters, and don't want to synch up
341 on blanks or hard tabs.
342
343 The optional arguments *a* and *b* are sequences to be compared; both default to
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000344 empty strings. The elements of both sequences must be :term:`hashable`.
Georg Brandl116aa622007-08-15 14:28:22 +0000345
Benjamin Petersone41251e2008-04-25 01:59:09 +0000346 :class:`SequenceMatcher` objects have the following methods:
Georg Brandl116aa622007-08-15 14:28:22 +0000347
348
Benjamin Petersone41251e2008-04-25 01:59:09 +0000349 .. method:: set_seqs(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000350
Benjamin Petersone41251e2008-04-25 01:59:09 +0000351 Set the two sequences to be compared.
Georg Brandl116aa622007-08-15 14:28:22 +0000352
Benjamin Petersone41251e2008-04-25 01:59:09 +0000353 :class:`SequenceMatcher` computes and caches detailed information about the
354 second sequence, so if you want to compare one sequence against many
355 sequences, use :meth:`set_seq2` to set the commonly used sequence once and
356 call :meth:`set_seq1` repeatedly, once for each of the other sequences.
Georg Brandl116aa622007-08-15 14:28:22 +0000357
358
Benjamin Petersone41251e2008-04-25 01:59:09 +0000359 .. method:: set_seq1(a)
Georg Brandl116aa622007-08-15 14:28:22 +0000360
Benjamin Petersone41251e2008-04-25 01:59:09 +0000361 Set the first sequence to be compared. The second sequence to be compared
362 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000363
364
Benjamin Petersone41251e2008-04-25 01:59:09 +0000365 .. method:: set_seq2(b)
Georg Brandl116aa622007-08-15 14:28:22 +0000366
Benjamin Petersone41251e2008-04-25 01:59:09 +0000367 Set the second sequence to be compared. The first sequence to be compared
368 is not changed.
Georg Brandl116aa622007-08-15 14:28:22 +0000369
370
Benjamin Petersone41251e2008-04-25 01:59:09 +0000371 .. method:: find_longest_match(alo, ahi, blo, bhi)
Georg Brandl116aa622007-08-15 14:28:22 +0000372
Benjamin Petersone41251e2008-04-25 01:59:09 +0000373 Find longest matching block in ``a[alo:ahi]`` and ``b[blo:bhi]``.
Georg Brandl116aa622007-08-15 14:28:22 +0000374
Benjamin Petersone41251e2008-04-25 01:59:09 +0000375 If *isjunk* was omitted or ``None``, :meth:`find_longest_match` returns
376 ``(i, j, k)`` such that ``a[i:i+k]`` is equal to ``b[j:j+k]``, where ``alo
377 <= i <= i+k <= ahi`` and ``blo <= j <= j+k <= bhi``. For all ``(i', j',
378 k')`` meeting those conditions, the additional conditions ``k >= k'``, ``i
379 <= i'``, and if ``i == i'``, ``j <= j'`` are also met. In other words, of
380 all maximal matching blocks, return one that starts earliest in *a*, and
381 of all those maximal matching blocks that start earliest in *a*, return
382 the one that starts earliest in *b*.
Georg Brandl116aa622007-08-15 14:28:22 +0000383
Benjamin Petersone41251e2008-04-25 01:59:09 +0000384 >>> s = SequenceMatcher(None, " abcd", "abcd abcd")
385 >>> s.find_longest_match(0, 5, 0, 9)
386 Match(a=0, b=4, size=5)
Georg Brandl116aa622007-08-15 14:28:22 +0000387
Benjamin Petersone41251e2008-04-25 01:59:09 +0000388 If *isjunk* was provided, first the longest matching block is determined
389 as above, but with the additional restriction that no junk element appears
390 in the block. Then that block is extended as far as possible by matching
391 (only) junk elements on both sides. So the resulting block never matches
392 on junk except as identical junk happens to be adjacent to an interesting
393 match.
Georg Brandl116aa622007-08-15 14:28:22 +0000394
Benjamin Petersone41251e2008-04-25 01:59:09 +0000395 Here's the same example as before, but considering blanks to be junk. That
396 prevents ``' abcd'`` from matching the ``' abcd'`` at the tail end of the
397 second sequence directly. Instead only the ``'abcd'`` can match, and
398 matches the leftmost ``'abcd'`` in the second sequence:
Georg Brandl116aa622007-08-15 14:28:22 +0000399
Benjamin Petersone41251e2008-04-25 01:59:09 +0000400 >>> s = SequenceMatcher(lambda x: x==" ", " abcd", "abcd abcd")
401 >>> s.find_longest_match(0, 5, 0, 9)
402 Match(a=1, b=0, size=4)
Georg Brandl116aa622007-08-15 14:28:22 +0000403
Benjamin Petersone41251e2008-04-25 01:59:09 +0000404 If no blocks match, this returns ``(alo, blo, 0)``.
Georg Brandl116aa622007-08-15 14:28:22 +0000405
Benjamin Petersone41251e2008-04-25 01:59:09 +0000406 This method returns a :term:`named tuple` ``Match(a, b, size)``.
Christian Heimes25bb7832008-01-11 16:17:00 +0000407
Georg Brandl116aa622007-08-15 14:28:22 +0000408
Benjamin Petersone41251e2008-04-25 01:59:09 +0000409 .. method:: get_matching_blocks()
Georg Brandl116aa622007-08-15 14:28:22 +0000410
Benjamin Petersone41251e2008-04-25 01:59:09 +0000411 Return list of triples describing matching subsequences. Each triple is of
412 the form ``(i, j, n)``, and means that ``a[i:i+n] == b[j:j+n]``. The
413 triples are monotonically increasing in *i* and *j*.
Georg Brandl116aa622007-08-15 14:28:22 +0000414
Benjamin Petersone41251e2008-04-25 01:59:09 +0000415 The last triple is a dummy, and has the value ``(len(a), len(b), 0)``. It
416 is the only triple with ``n == 0``. If ``(i, j, n)`` and ``(i', j', n')``
417 are adjacent triples in the list, and the second is not the last triple in
418 the list, then ``i+n != i'`` or ``j+n != j'``; in other words, adjacent
419 triples always describe non-adjacent equal blocks.
Georg Brandl116aa622007-08-15 14:28:22 +0000420
Benjamin Petersone41251e2008-04-25 01:59:09 +0000421 .. XXX Explain why a dummy is used!
Christian Heimes5b5e81c2007-12-31 16:14:33 +0000422
Benjamin Petersone41251e2008-04-25 01:59:09 +0000423 .. doctest::
Georg Brandl116aa622007-08-15 14:28:22 +0000424
Benjamin Petersone41251e2008-04-25 01:59:09 +0000425 >>> s = SequenceMatcher(None, "abxcd", "abcd")
426 >>> s.get_matching_blocks()
427 [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 +0000428
429
Benjamin Petersone41251e2008-04-25 01:59:09 +0000430 .. method:: get_opcodes()
Georg Brandl116aa622007-08-15 14:28:22 +0000431
Benjamin Petersone41251e2008-04-25 01:59:09 +0000432 Return list of 5-tuples describing how to turn *a* into *b*. Each tuple is
433 of the form ``(tag, i1, i2, j1, j2)``. The first tuple has ``i1 == j1 ==
434 0``, and remaining tuples have *i1* equal to the *i2* from the preceding
435 tuple, and, likewise, *j1* equal to the previous *j2*.
Georg Brandl116aa622007-08-15 14:28:22 +0000436
Benjamin Petersone41251e2008-04-25 01:59:09 +0000437 The *tag* values are strings, with these meanings:
Georg Brandl116aa622007-08-15 14:28:22 +0000438
Benjamin Petersone41251e2008-04-25 01:59:09 +0000439 +---------------+---------------------------------------------+
440 | Value | Meaning |
441 +===============+=============================================+
442 | ``'replace'`` | ``a[i1:i2]`` should be replaced by |
443 | | ``b[j1:j2]``. |
444 +---------------+---------------------------------------------+
445 | ``'delete'`` | ``a[i1:i2]`` should be deleted. Note that |
446 | | ``j1 == j2`` in this case. |
447 +---------------+---------------------------------------------+
448 | ``'insert'`` | ``b[j1:j2]`` should be inserted at |
449 | | ``a[i1:i1]``. Note that ``i1 == i2`` in |
450 | | this case. |
451 +---------------+---------------------------------------------+
452 | ``'equal'`` | ``a[i1:i2] == b[j1:j2]`` (the sub-sequences |
453 | | are equal). |
454 +---------------+---------------------------------------------+
Georg Brandl116aa622007-08-15 14:28:22 +0000455
Benjamin Petersone41251e2008-04-25 01:59:09 +0000456 For example:
Georg Brandl116aa622007-08-15 14:28:22 +0000457
Benjamin Petersone41251e2008-04-25 01:59:09 +0000458 >>> a = "qabxcd"
459 >>> b = "abycdf"
460 >>> s = SequenceMatcher(None, a, b)
461 >>> for tag, i1, i2, j1, j2 in s.get_opcodes():
462 ... print(("%7s a[%d:%d] (%s) b[%d:%d] (%s)" %
463 ... (tag, i1, i2, a[i1:i2], j1, j2, b[j1:j2])))
464 delete a[0:1] (q) b[0:0] ()
465 equal a[1:3] (ab) b[0:2] (ab)
466 replace a[3:4] (x) b[2:3] (y)
467 equal a[4:6] (cd) b[3:5] (cd)
468 insert a[6:6] () b[5:6] (f)
Georg Brandl116aa622007-08-15 14:28:22 +0000469
470
Benjamin Petersone41251e2008-04-25 01:59:09 +0000471 .. method:: get_grouped_opcodes([n])
Georg Brandl116aa622007-08-15 14:28:22 +0000472
Benjamin Petersone41251e2008-04-25 01:59:09 +0000473 Return a :term:`generator` of groups with up to *n* lines of context.
Georg Brandl116aa622007-08-15 14:28:22 +0000474
Benjamin Petersone41251e2008-04-25 01:59:09 +0000475 Starting with the groups returned by :meth:`get_opcodes`, this method
476 splits out smaller change clusters and eliminates intervening ranges which
477 have no changes.
Georg Brandl116aa622007-08-15 14:28:22 +0000478
Benjamin Petersone41251e2008-04-25 01:59:09 +0000479 The groups are returned in the same format as :meth:`get_opcodes`.
Georg Brandl116aa622007-08-15 14:28:22 +0000480
Georg Brandl116aa622007-08-15 14:28:22 +0000481
Benjamin Petersone41251e2008-04-25 01:59:09 +0000482 .. method:: ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000483
Benjamin Petersone41251e2008-04-25 01:59:09 +0000484 Return a measure of the sequences' similarity as a float in the range [0,
485 1].
Georg Brandl116aa622007-08-15 14:28:22 +0000486
Benjamin Petersone41251e2008-04-25 01:59:09 +0000487 Where T is the total number of elements in both sequences, and M is the
488 number of matches, this is 2.0\*M / T. Note that this is ``1.0`` if the
489 sequences are identical, and ``0.0`` if they have nothing in common.
Georg Brandl116aa622007-08-15 14:28:22 +0000490
Benjamin Petersone41251e2008-04-25 01:59:09 +0000491 This is expensive to compute if :meth:`get_matching_blocks` or
492 :meth:`get_opcodes` hasn't already been called, in which case you may want
493 to try :meth:`quick_ratio` or :meth:`real_quick_ratio` first to get an
494 upper bound.
Georg Brandl116aa622007-08-15 14:28:22 +0000495
496
Benjamin Petersone41251e2008-04-25 01:59:09 +0000497 .. method:: quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000498
Benjamin Petersone41251e2008-04-25 01:59:09 +0000499 Return an upper bound on :meth:`ratio` relatively quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000500
Benjamin Petersone41251e2008-04-25 01:59:09 +0000501 This isn't defined beyond that it is an upper bound on :meth:`ratio`, and
502 is faster to compute.
Georg Brandl116aa622007-08-15 14:28:22 +0000503
504
Benjamin Petersone41251e2008-04-25 01:59:09 +0000505 .. method:: real_quick_ratio()
Georg Brandl116aa622007-08-15 14:28:22 +0000506
Benjamin Petersone41251e2008-04-25 01:59:09 +0000507 Return an upper bound on :meth:`ratio` very quickly.
Georg Brandl116aa622007-08-15 14:28:22 +0000508
Benjamin Petersone41251e2008-04-25 01:59:09 +0000509 This isn't defined beyond that it is an upper bound on :meth:`ratio`, and
510 is faster to compute than either :meth:`ratio` or :meth:`quick_ratio`.
Georg Brandl116aa622007-08-15 14:28:22 +0000511
512The three methods that return the ratio of matching to total characters can give
513different results due to differing levels of approximation, although
514:meth:`quick_ratio` and :meth:`real_quick_ratio` are always at least as large as
Christian Heimesfe337bf2008-03-23 21:54:12 +0000515:meth:`ratio`:
Georg Brandl116aa622007-08-15 14:28:22 +0000516
517 >>> s = SequenceMatcher(None, "abcd", "bcde")
518 >>> s.ratio()
519 0.75
520 >>> s.quick_ratio()
521 0.75
522 >>> s.real_quick_ratio()
523 1.0
524
525
526.. _sequencematcher-examples:
527
528SequenceMatcher Examples
529------------------------
530
Christian Heimesfe337bf2008-03-23 21:54:12 +0000531This example compares two strings, considering blanks to be "junk:"
Georg Brandl116aa622007-08-15 14:28:22 +0000532
533 >>> s = SequenceMatcher(lambda x: x == " ",
534 ... "private Thread currentThread;",
535 ... "private volatile Thread currentThread;")
536
537:meth:`ratio` returns a float in [0, 1], measuring the similarity of the
538sequences. As a rule of thumb, a :meth:`ratio` value over 0.6 means the
Christian Heimesfe337bf2008-03-23 21:54:12 +0000539sequences are close matches:
Georg Brandl116aa622007-08-15 14:28:22 +0000540
Georg Brandl6911e3c2007-09-04 07:15:32 +0000541 >>> print(round(s.ratio(), 3))
Georg Brandl116aa622007-08-15 14:28:22 +0000542 0.866
543
544If you're only interested in where the sequences match,
Christian Heimesfe337bf2008-03-23 21:54:12 +0000545:meth:`get_matching_blocks` is handy:
Georg Brandl116aa622007-08-15 14:28:22 +0000546
547 >>> for block in s.get_matching_blocks():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000548 ... print("a[%d] and b[%d] match for %d elements" % block)
Georg Brandl116aa622007-08-15 14:28:22 +0000549 a[0] and b[0] match for 8 elements
Christian Heimesfe337bf2008-03-23 21:54:12 +0000550 a[8] and b[17] match for 21 elements
Georg Brandl116aa622007-08-15 14:28:22 +0000551 a[29] and b[38] match for 0 elements
552
553Note that the last tuple returned by :meth:`get_matching_blocks` is always a
554dummy, ``(len(a), len(b), 0)``, and this is the only case in which the last
555tuple element (number of elements matched) is ``0``.
556
557If you want to know how to change the first sequence into the second, use
Christian Heimesfe337bf2008-03-23 21:54:12 +0000558:meth:`get_opcodes`:
Georg Brandl116aa622007-08-15 14:28:22 +0000559
560 >>> for opcode in s.get_opcodes():
Georg Brandl6911e3c2007-09-04 07:15:32 +0000561 ... print("%6s a[%d:%d] b[%d:%d]" % opcode)
Georg Brandl116aa622007-08-15 14:28:22 +0000562 equal a[0:8] b[0:8]
563 insert a[8:8] b[8:17]
Christian Heimesfe337bf2008-03-23 21:54:12 +0000564 equal a[8:29] b[17:38]
Georg Brandl116aa622007-08-15 14:28:22 +0000565
566See also the function :func:`get_close_matches` in this module, which shows how
567simple code building on :class:`SequenceMatcher` can be used to do useful work.
568
569
570.. _differ-objects:
571
572Differ Objects
573--------------
574
575Note that :class:`Differ`\ -generated deltas make no claim to be **minimal**
576diffs. To the contrary, minimal diffs are often counter-intuitive, because they
577synch up anywhere possible, sometimes accidental matches 100 pages apart.
578Restricting synch points to contiguous matches preserves some notion of
579locality, at the occasional cost of producing a longer diff.
580
581The :class:`Differ` class has this constructor:
582
583
584.. class:: Differ([linejunk[, charjunk]])
585
586 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
587 (or ``None``):
588
589 *linejunk*: A function that accepts a single string argument, and returns true
590 if the string is junk. The default is ``None``, meaning that no line is
591 considered junk.
592
593 *charjunk*: A function that accepts a single character argument (a string of
594 length 1), and returns true if the character is junk. The default is ``None``,
595 meaning that no character is considered junk.
596
Benjamin Petersone41251e2008-04-25 01:59:09 +0000597 :class:`Differ` objects are used (deltas generated) via a single method:
Georg Brandl116aa622007-08-15 14:28:22 +0000598
599
Benjamin Petersone41251e2008-04-25 01:59:09 +0000600 .. method:: Differ.compare(a, b)
Georg Brandl116aa622007-08-15 14:28:22 +0000601
Benjamin Petersone41251e2008-04-25 01:59:09 +0000602 Compare two sequences of lines, and generate the delta (a sequence of lines).
Georg Brandl116aa622007-08-15 14:28:22 +0000603
Benjamin Petersone41251e2008-04-25 01:59:09 +0000604 Each sequence must contain individual single-line strings ending with newlines.
605 Such sequences can be obtained from the :meth:`readlines` method of file-like
606 objects. The delta generated also consists of newline-terminated strings, ready
607 to be printed as-is via the :meth:`writelines` method of a file-like object.
Georg Brandl116aa622007-08-15 14:28:22 +0000608
609
610.. _differ-examples:
611
612Differ Example
613--------------
614
615This example compares two texts. First we set up the texts, sequences of
616individual single-line strings ending with newlines (such sequences can also be
Christian Heimesfe337bf2008-03-23 21:54:12 +0000617obtained from the :meth:`readlines` method of file-like objects):
Georg Brandl116aa622007-08-15 14:28:22 +0000618
619 >>> text1 = ''' 1. Beautiful is better than ugly.
620 ... 2. Explicit is better than implicit.
621 ... 3. Simple is better than complex.
622 ... 4. Complex is better than complicated.
623 ... '''.splitlines(1)
624 >>> len(text1)
625 4
626 >>> text1[0][-1]
627 '\n'
628 >>> text2 = ''' 1. Beautiful is better than ugly.
629 ... 3. Simple is better than complex.
630 ... 4. Complicated is better than complex.
631 ... 5. Flat is better than nested.
632 ... '''.splitlines(1)
633
Christian Heimesfe337bf2008-03-23 21:54:12 +0000634Next we instantiate a Differ object:
Georg Brandl116aa622007-08-15 14:28:22 +0000635
636 >>> d = Differ()
637
638Note that when instantiating a :class:`Differ` object we may pass functions to
639filter out line and character "junk." See the :meth:`Differ` constructor for
640details.
641
Christian Heimesfe337bf2008-03-23 21:54:12 +0000642Finally, we compare the two:
Georg Brandl116aa622007-08-15 14:28:22 +0000643
644 >>> result = list(d.compare(text1, text2))
645
Christian Heimesfe337bf2008-03-23 21:54:12 +0000646``result`` is a list of strings, so let's pretty-print it:
Georg Brandl116aa622007-08-15 14:28:22 +0000647
648 >>> from pprint import pprint
649 >>> pprint(result)
650 [' 1. Beautiful is better than ugly.\n',
651 '- 2. Explicit is better than implicit.\n',
652 '- 3. Simple is better than complex.\n',
653 '+ 3. Simple is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000654 '? ++\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000655 '- 4. Complex is better than complicated.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000656 '? ^ ---- ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000657 '+ 4. Complicated is better than complex.\n',
Christian Heimesfe337bf2008-03-23 21:54:12 +0000658 '? ++++ ^ ^\n',
Georg Brandl116aa622007-08-15 14:28:22 +0000659 '+ 5. Flat is better than nested.\n']
660
Christian Heimesfe337bf2008-03-23 21:54:12 +0000661As a single multi-line string it looks like this:
Georg Brandl116aa622007-08-15 14:28:22 +0000662
663 >>> import sys
664 >>> sys.stdout.writelines(result)
665 1. Beautiful is better than ugly.
666 - 2. Explicit is better than implicit.
667 - 3. Simple is better than complex.
668 + 3. Simple is better than complex.
669 ? ++
670 - 4. Complex is better than complicated.
671 ? ^ ---- ^
672 + 4. Complicated is better than complex.
673 ? ++++ ^ ^
674 + 5. Flat is better than nested.
675
Christian Heimes8640e742008-02-23 16:23:06 +0000676
677.. _difflib-interface:
678
679A command-line interface to difflib
680-----------------------------------
681
682This example shows how to use difflib to create a ``diff``-like utility.
683It is also contained in the Python source distribution, as
684:file:`Tools/scripts/diff.py`.
685
Christian Heimesfe337bf2008-03-23 21:54:12 +0000686.. testcode::
Christian Heimes8640e742008-02-23 16:23:06 +0000687
688 """ Command line interface to difflib.py providing diffs in four formats:
689
690 * ndiff: lists every line and highlights interline changes.
691 * context: highlights clusters of changes in a before/after format.
692 * unified: highlights clusters of changes in an inline format.
693 * html: generates side by side comparison with change highlights.
694
695 """
696
697 import sys, os, time, difflib, optparse
698
699 def main():
700 # Configure the option parser
701 usage = "usage: %prog [options] fromfile tofile"
702 parser = optparse.OptionParser(usage)
703 parser.add_option("-c", action="store_true", default=False,
704 help='Produce a context format diff (default)')
705 parser.add_option("-u", action="store_true", default=False,
706 help='Produce a unified format diff')
707 hlp = 'Produce HTML side by side diff (can use -c and -l in conjunction)'
708 parser.add_option("-m", action="store_true", default=False, help=hlp)
709 parser.add_option("-n", action="store_true", default=False,
710 help='Produce a ndiff format diff')
711 parser.add_option("-l", "--lines", type="int", default=3,
712 help='Set number of context lines (default 3)')
713 (options, args) = parser.parse_args()
714
715 if len(args) == 0:
716 parser.print_help()
717 sys.exit(1)
718 if len(args) != 2:
719 parser.error("need to specify both a fromfile and tofile")
720
721 n = options.lines
722 fromfile, tofile = args # as specified in the usage string
723
724 # we're passing these as arguments to the diff function
725 fromdate = time.ctime(os.stat(fromfile).st_mtime)
726 todate = time.ctime(os.stat(tofile).st_mtime)
727 fromlines = open(fromfile, 'U').readlines()
728 tolines = open(tofile, 'U').readlines()
729
730 if options.u:
731 diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
732 fromdate, todate, n=n)
733 elif options.n:
734 diff = difflib.ndiff(fromlines, tolines)
735 elif options.m:
736 diff = difflib.HtmlDiff().make_file(fromlines, tolines, fromfile,
737 tofile, context=options.c,
738 numlines=n)
739 else:
740 diff = difflib.context_diff(fromlines, tolines, fromfile, tofile,
741 fromdate, todate, n=n)
742
743 # we're using writelines because diff is a generator
744 sys.stdout.writelines(diff)
745
746 if __name__ == '__main__':
747 main()