blob: d51e230f6965e6d609822b704a0d3bd694a0d68d [file] [log] [blame]
Georg Brandl8ec7f652007-08-15 14:28:01 +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>
Georg Brandlb19be572007-12-29 10:57:00 +00008.. Markup by Fred L. Drake, Jr. <fdrake@acm.org>
Georg Brandl8ec7f652007-08-15 14:28:01 +00009
Georg Brandle8f1b002008-03-22 22:04:10 +000010.. testsetup::
Georg Brandl8ec7f652007-08-15 14:28:01 +000011
Georg Brandle8f1b002008-03-22 22:04:10 +000012 import sys
13 from difflib import *
Georg Brandl8ec7f652007-08-15 14:28:01 +000014
15.. versionadded:: 2.1
16
Mark Summerfield0752d202007-10-19 12:48:17 +000017This module provides classes and functions for comparing sequences. It
18can be used for example, for comparing files, and can produce difference
19information in various formats, including HTML and context and unified
20diffs. For comparing directories and files, see also, the :mod:`filecmp` module.
Georg Brandl8ec7f652007-08-15 14:28:01 +000021
22.. class:: SequenceMatcher
23
24 This is a flexible class for comparing pairs of sequences of any type, so long
Georg Brandl7c3e79f2007-11-02 20:06:17 +000025 as the sequence elements are :term:`hashable`. The basic algorithm predates, and is a
Georg Brandl8ec7f652007-08-15 14:28:01 +000026 little fancier than, an algorithm published in the late 1980's by Ratcliff and
27 Obershelp under the hyperbolic name "gestalt pattern matching." The idea is to
28 find the longest contiguous matching subsequence that contains no "junk"
29 elements (the Ratcliff and Obershelp algorithm doesn't address junk). The same
30 idea is then applied recursively to the pieces of the sequences to the left and
31 to the right of the matching subsequence. This does not yield minimal edit
32 sequences, but does tend to yield matches that "look right" to people.
33
34 **Timing:** The basic Ratcliff-Obershelp algorithm is cubic time in the worst
35 case and quadratic time in the expected case. :class:`SequenceMatcher` is
36 quadratic time for the worst case and has expected-case behavior dependent in a
37 complicated way on how many elements the sequences have in common; best case
38 time is linear.
39
Terry Reedyd2d2ae92010-11-11 23:22:19 +000040 **Automatic junk heuristic:** :class:`SequenceMatcher` supports a heuristic that
41 automatically treats certain sequence items as junk. The heuristic counts how many
42 times each individual item appears in the sequence. If an item's duplicates (after
43 the first one) account for more than 1% of the sequence and the sequence is at least
44 200 items long, this item is marked as "popular" and is treated as junk for
45 the purpose of sequence matching. This heuristic can be turned off by setting
46 the ``autojunk`` argument to ``False`` when creating the :class:`SequenceMatcher`.
47
Terry Reedyfc0280f2010-11-12 21:08:56 +000048 .. versionadded:: 2.7.1
Terry Reedyd2d2ae92010-11-11 23:22:19 +000049 The *autojunk* parameter.
Georg Brandl8ec7f652007-08-15 14:28:01 +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
87 .. function:: __init__([tabsize][, wrapcolumn][, linejunk][, charjunk])
88
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
103
104 .. function:: make_file(fromlines, tolines [, fromdesc][, todesc][, context][, numlines])
105
106 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
107 is a complete HTML file containing a table showing line by line differences with
108 inter-line and intra-line changes highlighted.
109
110 *fromdesc* and *todesc* are optional keyword arguments to specify from/to file
111 column header strings (both default to an empty string).
112
113 *context* and *numlines* are both optional keyword arguments. Set *context* to
114 ``True`` when contextual differences are to be shown, else the default is
115 ``False`` to show the full files. *numlines* defaults to ``5``. When *context*
116 is ``True`` *numlines* controls the number of context lines which surround the
117 difference highlights. When *context* is ``False`` *numlines* controls the
118 number of lines which are shown before a difference highlight when using the
119 "next" hyperlinks (setting to zero would cause the "next" hyperlinks to place
120 the next difference highlight at the top of the browser without any leading
121 context).
122
123
124 .. function:: make_table(fromlines, tolines [, fromdesc][, todesc][, context][, numlines])
125
126 Compares *fromlines* and *tolines* (lists of strings) and returns a string which
127 is a complete HTML table showing line by line differences with inter-line and
128 intra-line changes highlighted.
129
130 The arguments for this method are the same as those for the :meth:`make_file`
131 method.
132
133 :file:`Tools/scripts/diff.py` is a command-line front-end to this class and
134 contains a good example of its use.
135
136 .. versionadded:: 2.4
137
138
139.. function:: context_diff(a, b[, fromfile][, tofile][, fromfiledate][, tofiledate][, n][, lineterm])
140
Georg Brandlcf3fb252007-10-21 10:52:38 +0000141 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
142 generating the delta lines) in context diff format.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000143
144 Context diffs are a compact way of showing just the lines that have changed plus
145 a few lines of context. The changes are shown in a before/after style. The
146 number of context lines is set by *n* which defaults to three.
147
148 By default, the diff control lines (those with ``***`` or ``---``) are created
149 with a trailing newline. This is helpful so that inputs created from
150 :func:`file.readlines` result in diffs that are suitable for use with
151 :func:`file.writelines` since both the inputs and outputs have trailing
152 newlines.
153
154 For inputs that do not have trailing newlines, set the *lineterm* argument to
155 ``""`` so that the output will be uniformly newline free.
156
157 The context diff format normally has a header for filenames and modification
158 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murray1a14d3d2010-04-12 16:35:19 +0000159 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
160 expressed in the ISO 8601 format. If not specified, the
Georg Brandl8ec7f652007-08-15 14:28:01 +0000161 strings default to blanks.
162
Georg Brandl080b0942008-02-23 15:19:54 +0000163 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
164 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
165 >>> for line in context_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Georg Brandle8f1b002008-03-22 22:04:10 +0000166 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Georg Brandl080b0942008-02-23 15:19:54 +0000167 *** before.py
168 --- after.py
169 ***************
170 *** 1,4 ****
171 ! bacon
172 ! eggs
173 ! ham
174 guido
175 --- 1,4 ----
176 ! python
177 ! eggy
178 ! hamster
179 guido
180
181 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000182
183 .. versionadded:: 2.3
184
185
186.. function:: get_close_matches(word, possibilities[, n][, cutoff])
187
188 Return a list of the best "good enough" matches. *word* is a sequence for which
189 close matches are desired (typically a string), and *possibilities* is a list of
190 sequences against which to match *word* (typically a list of strings).
191
192 Optional argument *n* (default ``3``) is the maximum number of close matches to
193 return; *n* must be greater than ``0``.
194
195 Optional argument *cutoff* (default ``0.6``) is a float in the range [0, 1].
196 Possibilities that don't score at least that similar to *word* are ignored.
197
198 The best (no more than *n*) matches among the possibilities are returned in a
Georg Brandle8f1b002008-03-22 22:04:10 +0000199 list, sorted by similarity score, most similar first.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000200
201 >>> get_close_matches('appel', ['ape', 'apple', 'peach', 'puppy'])
202 ['apple', 'ape']
203 >>> import keyword
204 >>> get_close_matches('wheel', keyword.kwlist)
205 ['while']
206 >>> get_close_matches('apple', keyword.kwlist)
207 []
208 >>> get_close_matches('accept', keyword.kwlist)
209 ['except']
210
211
212.. function:: ndiff(a, b[, linejunk][, charjunk])
213
Georg Brandlcf3fb252007-10-21 10:52:38 +0000214 Compare *a* and *b* (lists of strings); return a :class:`Differ`\ -style
215 delta (a :term:`generator` generating the delta lines).
Georg Brandl8ec7f652007-08-15 14:28:01 +0000216
217 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
218 (or ``None``):
219
220 *linejunk*: A function that accepts a single string argument, and returns true
221 if the string is junk, or false if not. The default is (``None``), starting with
222 Python 2.3. Before then, the default was the module-level function
223 :func:`IS_LINE_JUNK`, which filters out lines without visible characters, except
224 for at most one pound character (``'#'``). As of Python 2.3, the underlying
225 :class:`SequenceMatcher` class does a dynamic analysis of which lines are so
226 frequent as to constitute noise, and this usually works better than the pre-2.3
227 default.
228
229 *charjunk*: A function that accepts a character (a string of length 1), and
230 returns if the character is junk, or false if not. The default is module-level
231 function :func:`IS_CHARACTER_JUNK`, which filters out whitespace characters (a
232 blank or tab; note: bad idea to include newline in this!).
233
Georg Brandle8f1b002008-03-22 22:04:10 +0000234 :file:`Tools/scripts/ndiff.py` is a command-line front-end to this function.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000235
236 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
237 ... 'ore\ntree\nemu\n'.splitlines(1))
238 >>> print ''.join(diff),
239 - one
240 ? ^
241 + ore
242 ? ^
243 - two
244 - three
245 ? -
246 + tree
247 + emu
248
249
250.. function:: restore(sequence, which)
251
252 Return one of the two sequences that generated a delta.
253
254 Given a *sequence* produced by :meth:`Differ.compare` or :func:`ndiff`, extract
255 lines originating from file 1 or 2 (parameter *which*), stripping off line
256 prefixes.
257
Georg Brandle8f1b002008-03-22 22:04:10 +0000258 Example:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000259
260 >>> diff = ndiff('one\ntwo\nthree\n'.splitlines(1),
261 ... 'ore\ntree\nemu\n'.splitlines(1))
262 >>> diff = list(diff) # materialize the generated delta into a list
263 >>> print ''.join(restore(diff, 1)),
264 one
265 two
266 three
267 >>> print ''.join(restore(diff, 2)),
268 ore
269 tree
270 emu
271
272
273.. function:: unified_diff(a, b[, fromfile][, tofile][, fromfiledate][, tofiledate][, n][, lineterm])
274
Georg Brandlcf3fb252007-10-21 10:52:38 +0000275 Compare *a* and *b* (lists of strings); return a delta (a :term:`generator`
276 generating the delta lines) in unified diff format.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000277
278 Unified diffs are a compact way of showing just the lines that have changed plus
279 a few lines of context. The changes are shown in a inline style (instead of
280 separate before/after blocks). The number of context lines is set by *n* which
281 defaults to three.
282
283 By default, the diff control lines (those with ``---``, ``+++``, or ``@@``) are
284 created with a trailing newline. This is helpful so that inputs created from
285 :func:`file.readlines` result in diffs that are suitable for use with
286 :func:`file.writelines` since both the inputs and outputs have trailing
287 newlines.
288
289 For inputs that do not have trailing newlines, set the *lineterm* argument to
290 ``""`` so that the output will be uniformly newline free.
291
292 The context diff format normally has a header for filenames and modification
293 times. Any or all of these may be specified using strings for *fromfile*,
R. David Murray1a14d3d2010-04-12 16:35:19 +0000294 *tofile*, *fromfiledate*, and *tofiledate*. The modification times are normally
295 expressed in the ISO 8601 format. If not specified, the
Georg Brandl8ec7f652007-08-15 14:28:01 +0000296 strings default to blanks.
297
Georg Brandl080b0942008-02-23 15:19:54 +0000298 >>> s1 = ['bacon\n', 'eggs\n', 'ham\n', 'guido\n']
299 >>> s2 = ['python\n', 'eggy\n', 'hamster\n', 'guido\n']
300 >>> for line in unified_diff(s1, s2, fromfile='before.py', tofile='after.py'):
Georg Brandle8f1b002008-03-22 22:04:10 +0000301 ... sys.stdout.write(line) # doctest: +NORMALIZE_WHITESPACE
Georg Brandl080b0942008-02-23 15:19:54 +0000302 --- before.py
303 +++ after.py
304 @@ -1,4 +1,4 @@
305 -bacon
306 -eggs
307 -ham
308 +python
309 +eggy
310 +hamster
311 guido
312
313 See :ref:`difflib-interface` for a more detailed example.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000314
315 .. versionadded:: 2.3
316
317
318.. function:: IS_LINE_JUNK(line)
319
320 Return true for ignorable lines. The line *line* is ignorable if *line* is
321 blank or contains a single ``'#'``, otherwise it is not ignorable. Used as a
322 default for parameter *linejunk* in :func:`ndiff` before Python 2.3.
323
324
325.. function:: IS_CHARACTER_JUNK(ch)
326
327 Return true for ignorable characters. The character *ch* is ignorable if *ch*
328 is a space or tab, otherwise it is not ignorable. Used as a default for
329 parameter *charjunk* in :func:`ndiff`.
330
331
332.. seealso::
333
334 `Pattern Matching: The Gestalt Approach <http://www.ddj.com/184407970?pgno=5>`_
335 Discussion of a similar algorithm by John W. Ratcliff and D. E. Metzener. This
336 was published in `Dr. Dobb's Journal <http://www.ddj.com/>`_ in July, 1988.
337
338
339.. _sequence-matcher:
340
341SequenceMatcher Objects
342-----------------------
343
344The :class:`SequenceMatcher` class has this constructor:
345
346
Terry Reedyd2d2ae92010-11-11 23:22:19 +0000347.. class:: SequenceMatcher([isjunk[, a[, b[, autojunk=True]]]])
Georg Brandl8ec7f652007-08-15 14:28:01 +0000348
349 Optional argument *isjunk* must be ``None`` (the default) or a one-argument
350 function that takes a sequence element and returns true if and only if the
351 element is "junk" and should be ignored. Passing ``None`` for *isjunk* is
352 equivalent to passing ``lambda x: 0``; in other words, no elements are ignored.
353 For example, pass::
354
355 lambda x: x in " \t"
356
357 if you're comparing lines as sequences of characters, and don't want to synch up
358 on blanks or hard tabs.
359
360 The optional arguments *a* and *b* are sequences to be compared; both default to
Georg Brandl7c3e79f2007-11-02 20:06:17 +0000361 empty strings. The elements of both sequences must be :term:`hashable`.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000362
Terry Reedyd2d2ae92010-11-11 23:22:19 +0000363 The optional argument *autojunk* can be used to disable the automatic junk
364 heuristic.
365
Terry Reedyfc0280f2010-11-12 21:08:56 +0000366 .. versionadded:: 2.7.1
367 The *autojunk* parameter.
368
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000369 :class:`SequenceMatcher` objects have the following methods:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000370
371
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000372 .. method:: set_seqs(a, b)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000373
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000374 Set the two sequences to be compared.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000375
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000376 :class:`SequenceMatcher` computes and caches detailed information about the
377 second sequence, so if you want to compare one sequence against many
378 sequences, use :meth:`set_seq2` to set the commonly used sequence once and
379 call :meth:`set_seq1` repeatedly, once for each of the other sequences.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000380
381
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000382 .. method:: set_seq1(a)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000383
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000384 Set the first sequence to be compared. The second sequence to be compared
385 is not changed.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000386
387
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000388 .. method:: set_seq2(b)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000389
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000390 Set the second sequence to be compared. The first sequence to be compared
391 is not changed.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000392
393
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000394 .. method:: find_longest_match(alo, ahi, blo, bhi)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000395
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000396 Find longest matching block in ``a[alo:ahi]`` and ``b[blo:bhi]``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000397
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000398 If *isjunk* was omitted or ``None``, :meth:`find_longest_match` returns
399 ``(i, j, k)`` such that ``a[i:i+k]`` is equal to ``b[j:j+k]``, where ``alo
400 <= i <= i+k <= ahi`` and ``blo <= j <= j+k <= bhi``. For all ``(i', j',
401 k')`` meeting those conditions, the additional conditions ``k >= k'``, ``i
402 <= i'``, and if ``i == i'``, ``j <= j'`` are also met. In other words, of
403 all maximal matching blocks, return one that starts earliest in *a*, and
404 of all those maximal matching blocks that start earliest in *a*, return
405 the one that starts earliest in *b*.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000406
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000407 >>> s = SequenceMatcher(None, " abcd", "abcd abcd")
408 >>> s.find_longest_match(0, 5, 0, 9)
409 Match(a=0, b=4, size=5)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000410
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000411 If *isjunk* was provided, first the longest matching block is determined
412 as above, but with the additional restriction that no junk element appears
413 in the block. Then that block is extended as far as possible by matching
414 (only) junk elements on both sides. So the resulting block never matches
415 on junk except as identical junk happens to be adjacent to an interesting
416 match.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000417
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000418 Here's the same example as before, but considering blanks to be junk. That
419 prevents ``' abcd'`` from matching the ``' abcd'`` at the tail end of the
420 second sequence directly. Instead only the ``'abcd'`` can match, and
421 matches the leftmost ``'abcd'`` in the second sequence:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000422
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000423 >>> s = SequenceMatcher(lambda x: x==" ", " abcd", "abcd abcd")
424 >>> s.find_longest_match(0, 5, 0, 9)
425 Match(a=1, b=0, size=4)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000426
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000427 If no blocks match, this returns ``(alo, blo, 0)``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000428
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000429 .. versionchanged:: 2.6
430 This method returns a :term:`named tuple` ``Match(a, b, size)``.
Georg Brandle3c3db52008-01-11 09:55:53 +0000431
Georg Brandl8ec7f652007-08-15 14:28:01 +0000432
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000433 .. method:: get_matching_blocks()
Georg Brandl8ec7f652007-08-15 14:28:01 +0000434
Benjamin Petersonc7b05922008-04-25 01:29:10 +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 Brandl8ec7f652007-08-15 14:28:01 +0000438
Benjamin Petersonc7b05922008-04-25 01:29:10 +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 Brandl8ec7f652007-08-15 14:28:01 +0000444
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000445 .. XXX Explain why a dummy is used!
Georg Brandl8ec7f652007-08-15 14:28:01 +0000446
Georg Brandlc62ef8b2009-01-03 20:55:06 +0000447 .. versionchanged:: 2.5
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000448 The guarantee that adjacent triples always describe non-adjacent blocks
449 was implemented.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000450
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000451 .. doctest::
Georg Brandl8ec7f652007-08-15 14:28:01 +0000452
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000453 >>> s = SequenceMatcher(None, "abxcd", "abcd")
454 >>> s.get_matching_blocks()
455 [Match(a=0, b=0, size=2), Match(a=3, b=2, size=2), Match(a=5, b=4, size=0)]
Georg Brandl8ec7f652007-08-15 14:28:01 +0000456
457
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000458 .. method:: get_opcodes()
Georg Brandl8ec7f652007-08-15 14:28:01 +0000459
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000460 Return list of 5-tuples describing how to turn *a* into *b*. Each tuple is
461 of the form ``(tag, i1, i2, j1, j2)``. The first tuple has ``i1 == j1 ==
462 0``, and remaining tuples have *i1* equal to the *i2* from the preceding
463 tuple, and, likewise, *j1* equal to the previous *j2*.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000464
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000465 The *tag* values are strings, with these meanings:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000466
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000467 +---------------+---------------------------------------------+
468 | Value | Meaning |
469 +===============+=============================================+
470 | ``'replace'`` | ``a[i1:i2]`` should be replaced by |
471 | | ``b[j1:j2]``. |
472 +---------------+---------------------------------------------+
473 | ``'delete'`` | ``a[i1:i2]`` should be deleted. Note that |
474 | | ``j1 == j2`` in this case. |
475 +---------------+---------------------------------------------+
476 | ``'insert'`` | ``b[j1:j2]`` should be inserted at |
477 | | ``a[i1:i1]``. Note that ``i1 == i2`` in |
478 | | this case. |
479 +---------------+---------------------------------------------+
480 | ``'equal'`` | ``a[i1:i2] == b[j1:j2]`` (the sub-sequences |
481 | | are equal). |
482 +---------------+---------------------------------------------+
Georg Brandl8ec7f652007-08-15 14:28:01 +0000483
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000484 For example:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000485
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000486 >>> a = "qabxcd"
487 >>> b = "abycdf"
488 >>> s = SequenceMatcher(None, a, b)
489 >>> for tag, i1, i2, j1, j2 in s.get_opcodes():
490 ... print ("%7s a[%d:%d] (%s) b[%d:%d] (%s)" %
491 ... (tag, i1, i2, a[i1:i2], j1, j2, b[j1:j2]))
492 delete a[0:1] (q) b[0:0] ()
493 equal a[1:3] (ab) b[0:2] (ab)
494 replace a[3:4] (x) b[2:3] (y)
495 equal a[4:6] (cd) b[3:5] (cd)
496 insert a[6:6] () b[5:6] (f)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000497
498
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000499 .. method:: get_grouped_opcodes([n])
Georg Brandl8ec7f652007-08-15 14:28:01 +0000500
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000501 Return a :term:`generator` of groups with up to *n* lines of context.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000502
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000503 Starting with the groups returned by :meth:`get_opcodes`, this method
504 splits out smaller change clusters and eliminates intervening ranges which
505 have no changes.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000506
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000507 The groups are returned in the same format as :meth:`get_opcodes`.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000508
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000509 .. versionadded:: 2.3
Georg Brandl8ec7f652007-08-15 14:28:01 +0000510
511
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000512 .. method:: ratio()
Georg Brandl8ec7f652007-08-15 14:28:01 +0000513
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000514 Return a measure of the sequences' similarity as a float in the range [0,
515 1].
Georg Brandl8ec7f652007-08-15 14:28:01 +0000516
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000517 Where T is the total number of elements in both sequences, and M is the
518 number of matches, this is 2.0\*M / T. Note that this is ``1.0`` if the
519 sequences are identical, and ``0.0`` if they have nothing in common.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000520
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000521 This is expensive to compute if :meth:`get_matching_blocks` or
522 :meth:`get_opcodes` hasn't already been called, in which case you may want
523 to try :meth:`quick_ratio` or :meth:`real_quick_ratio` first to get an
524 upper bound.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000525
526
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000527 .. method:: quick_ratio()
Georg Brandl8ec7f652007-08-15 14:28:01 +0000528
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000529 Return an upper bound on :meth:`ratio` relatively quickly.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000530
Georg Brandl8ec7f652007-08-15 14:28:01 +0000531
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000532 .. method:: real_quick_ratio()
Georg Brandl8ec7f652007-08-15 14:28:01 +0000533
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000534 Return an upper bound on :meth:`ratio` very quickly.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000535
Georg Brandl8ec7f652007-08-15 14:28:01 +0000536
537The three methods that return the ratio of matching to total characters can give
538different results due to differing levels of approximation, although
539:meth:`quick_ratio` and :meth:`real_quick_ratio` are always at least as large as
Georg Brandle8f1b002008-03-22 22:04:10 +0000540:meth:`ratio`:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000541
542 >>> s = SequenceMatcher(None, "abcd", "bcde")
543 >>> s.ratio()
544 0.75
545 >>> s.quick_ratio()
546 0.75
547 >>> s.real_quick_ratio()
548 1.0
549
550
551.. _sequencematcher-examples:
552
553SequenceMatcher Examples
554------------------------
555
Georg Brandle8f1b002008-03-22 22:04:10 +0000556This example compares two strings, considering blanks to be "junk:"
Georg Brandl8ec7f652007-08-15 14:28:01 +0000557
558 >>> s = SequenceMatcher(lambda x: x == " ",
559 ... "private Thread currentThread;",
560 ... "private volatile Thread currentThread;")
561
562:meth:`ratio` returns a float in [0, 1], measuring the similarity of the
563sequences. As a rule of thumb, a :meth:`ratio` value over 0.6 means the
Georg Brandle8f1b002008-03-22 22:04:10 +0000564sequences are close matches:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000565
566 >>> print round(s.ratio(), 3)
567 0.866
568
569If you're only interested in where the sequences match,
Georg Brandle8f1b002008-03-22 22:04:10 +0000570:meth:`get_matching_blocks` is handy:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000571
572 >>> for block in s.get_matching_blocks():
573 ... print "a[%d] and b[%d] match for %d elements" % block
574 a[0] and b[0] match for 8 elements
Georg Brandle8f1b002008-03-22 22:04:10 +0000575 a[8] and b[17] match for 21 elements
Georg Brandl8ec7f652007-08-15 14:28:01 +0000576 a[29] and b[38] match for 0 elements
577
578Note that the last tuple returned by :meth:`get_matching_blocks` is always a
579dummy, ``(len(a), len(b), 0)``, and this is the only case in which the last
580tuple element (number of elements matched) is ``0``.
581
582If you want to know how to change the first sequence into the second, use
Georg Brandle8f1b002008-03-22 22:04:10 +0000583:meth:`get_opcodes`:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000584
585 >>> for opcode in s.get_opcodes():
586 ... print "%6s a[%d:%d] b[%d:%d]" % opcode
587 equal a[0:8] b[0:8]
588 insert a[8:8] b[8:17]
Georg Brandle8f1b002008-03-22 22:04:10 +0000589 equal a[8:29] b[17:38]
Georg Brandl8ec7f652007-08-15 14:28:01 +0000590
Raymond Hettinger8b8f8cc2009-04-27 21:12:54 +0000591.. seealso::
592
593 * The :func:`get_close_matches` function in this module which shows how
594 simple code building on :class:`SequenceMatcher` can be used to do useful
595 work.
596
597 * `Simple version control recipe
598 <http://code.activestate.com/recipes/576729/>`_ for a small application
599 built with :class:`SequenceMatcher`.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000600
601
602.. _differ-objects:
603
604Differ Objects
605--------------
606
607Note that :class:`Differ`\ -generated deltas make no claim to be **minimal**
608diffs. To the contrary, minimal diffs are often counter-intuitive, because they
609synch up anywhere possible, sometimes accidental matches 100 pages apart.
610Restricting synch points to contiguous matches preserves some notion of
611locality, at the occasional cost of producing a longer diff.
612
613The :class:`Differ` class has this constructor:
614
615
616.. class:: Differ([linejunk[, charjunk]])
617
618 Optional keyword parameters *linejunk* and *charjunk* are for filter functions
619 (or ``None``):
620
621 *linejunk*: A function that accepts a single string argument, and returns true
622 if the string is junk. The default is ``None``, meaning that no line is
623 considered junk.
624
625 *charjunk*: A function that accepts a single character argument (a string of
626 length 1), and returns true if the character is junk. The default is ``None``,
627 meaning that no character is considered junk.
628
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000629 :class:`Differ` objects are used (deltas generated) via a single method:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000630
631
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000632 .. method:: Differ.compare(a, b)
Georg Brandl8ec7f652007-08-15 14:28:01 +0000633
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000634 Compare two sequences of lines, and generate the delta (a sequence of lines).
Georg Brandl8ec7f652007-08-15 14:28:01 +0000635
Benjamin Petersonc7b05922008-04-25 01:29:10 +0000636 Each sequence must contain individual single-line strings ending with newlines.
637 Such sequences can be obtained from the :meth:`readlines` method of file-like
638 objects. The delta generated also consists of newline-terminated strings, ready
639 to be printed as-is via the :meth:`writelines` method of a file-like object.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000640
641
642.. _differ-examples:
643
644Differ Example
645--------------
646
647This example compares two texts. First we set up the texts, sequences of
648individual single-line strings ending with newlines (such sequences can also be
Georg Brandle8f1b002008-03-22 22:04:10 +0000649obtained from the :meth:`readlines` method of file-like objects):
Georg Brandl8ec7f652007-08-15 14:28:01 +0000650
651 >>> text1 = ''' 1. Beautiful is better than ugly.
652 ... 2. Explicit is better than implicit.
653 ... 3. Simple is better than complex.
654 ... 4. Complex is better than complicated.
655 ... '''.splitlines(1)
656 >>> len(text1)
657 4
658 >>> text1[0][-1]
659 '\n'
660 >>> text2 = ''' 1. Beautiful is better than ugly.
661 ... 3. Simple is better than complex.
662 ... 4. Complicated is better than complex.
663 ... 5. Flat is better than nested.
664 ... '''.splitlines(1)
665
Georg Brandle8f1b002008-03-22 22:04:10 +0000666Next we instantiate a Differ object:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000667
668 >>> d = Differ()
669
670Note that when instantiating a :class:`Differ` object we may pass functions to
671filter out line and character "junk." See the :meth:`Differ` constructor for
672details.
673
Georg Brandle8f1b002008-03-22 22:04:10 +0000674Finally, we compare the two:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000675
676 >>> result = list(d.compare(text1, text2))
677
Georg Brandle8f1b002008-03-22 22:04:10 +0000678``result`` is a list of strings, so let's pretty-print it:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000679
680 >>> from pprint import pprint
681 >>> pprint(result)
682 [' 1. Beautiful is better than ugly.\n',
683 '- 2. Explicit is better than implicit.\n',
684 '- 3. Simple is better than complex.\n',
685 '+ 3. Simple is better than complex.\n',
Georg Brandle8f1b002008-03-22 22:04:10 +0000686 '? ++\n',
Georg Brandl8ec7f652007-08-15 14:28:01 +0000687 '- 4. Complex is better than complicated.\n',
Georg Brandle8f1b002008-03-22 22:04:10 +0000688 '? ^ ---- ^\n',
Georg Brandl8ec7f652007-08-15 14:28:01 +0000689 '+ 4. Complicated is better than complex.\n',
Georg Brandle8f1b002008-03-22 22:04:10 +0000690 '? ++++ ^ ^\n',
Georg Brandl8ec7f652007-08-15 14:28:01 +0000691 '+ 5. Flat is better than nested.\n']
692
Georg Brandle8f1b002008-03-22 22:04:10 +0000693As a single multi-line string it looks like this:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000694
695 >>> import sys
696 >>> sys.stdout.writelines(result)
697 1. Beautiful is better than ugly.
698 - 2. Explicit is better than implicit.
699 - 3. Simple is better than complex.
700 + 3. Simple is better than complex.
701 ? ++
702 - 4. Complex is better than complicated.
703 ? ^ ---- ^
704 + 4. Complicated is better than complex.
705 ? ++++ ^ ^
706 + 5. Flat is better than nested.
707
Georg Brandl080b0942008-02-23 15:19:54 +0000708
709.. _difflib-interface:
710
711A command-line interface to difflib
712-----------------------------------
713
714This example shows how to use difflib to create a ``diff``-like utility.
715It is also contained in the Python source distribution, as
716:file:`Tools/scripts/diff.py`.
717
Georg Brandle8f1b002008-03-22 22:04:10 +0000718.. testcode::
Georg Brandl080b0942008-02-23 15:19:54 +0000719
720 """ Command line interface to difflib.py providing diffs in four formats:
721
722 * ndiff: lists every line and highlights interline changes.
723 * context: highlights clusters of changes in a before/after format.
724 * unified: highlights clusters of changes in an inline format.
725 * html: generates side by side comparison with change highlights.
726
727 """
728
Benjamin Petersona7b55a32009-02-20 03:31:23 +0000729 import sys, os, time, difflib, optparse
Georg Brandl080b0942008-02-23 15:19:54 +0000730
731 def main():
732 # Configure the option parser
733 usage = "usage: %prog [options] fromfile tofile"
734 parser = optparse.OptionParser(usage)
735 parser.add_option("-c", action="store_true", default=False,
736 help='Produce a context format diff (default)')
737 parser.add_option("-u", action="store_true", default=False,
738 help='Produce a unified format diff')
739 hlp = 'Produce HTML side by side diff (can use -c and -l in conjunction)'
740 parser.add_option("-m", action="store_true", default=False, help=hlp)
741 parser.add_option("-n", action="store_true", default=False,
742 help='Produce a ndiff format diff')
743 parser.add_option("-l", "--lines", type="int", default=3,
744 help='Set number of context lines (default 3)')
745 (options, args) = parser.parse_args()
746
747 if len(args) == 0:
748 parser.print_help()
749 sys.exit(1)
750 if len(args) != 2:
751 parser.error("need to specify both a fromfile and tofile")
752
753 n = options.lines
754 fromfile, tofile = args # as specified in the usage string
755
756 # we're passing these as arguments to the diff function
757 fromdate = time.ctime(os.stat(fromfile).st_mtime)
758 todate = time.ctime(os.stat(tofile).st_mtime)
759 fromlines = open(fromfile, 'U').readlines()
760 tolines = open(tofile, 'U').readlines()
761
762 if options.u:
763 diff = difflib.unified_diff(fromlines, tolines, fromfile, tofile,
764 fromdate, todate, n=n)
765 elif options.n:
766 diff = difflib.ndiff(fromlines, tolines)
767 elif options.m:
768 diff = difflib.HtmlDiff().make_file(fromlines, tolines, fromfile,
769 tofile, context=options.c,
770 numlines=n)
771 else:
772 diff = difflib.context_diff(fromlines, tolines, fromfile, tofile,
773 fromdate, todate, n=n)
774
775 # we're using writelines because diff is a generator
776 sys.stdout.writelines(diff)
777
778 if __name__ == '__main__':
779 main()