← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/patiencediff.py

  • Committer: Martin Pool
  • Date: 2005-05-09 04:38:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050509043831-d45f7832b7d4d5b0
- better message when refusing to add symlinks (from mpe)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/patiencediff.py
1
 
#!/usr/bin/env python
2
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
 
 
18
 
 
19
 
from bzrlib.lazy_import import lazy_import
20
 
lazy_import(globals(), """
21
 
import os
22
 
import sys
23
 
import time
24
 
import difflib
25
 
""")
26
 
 
27
 
 
28
 
__all__ = ['PatienceSequenceMatcher', 'unified_diff', 'unified_diff_files']
29
 
 
30
 
 
31
 
# This is a version of unified_diff which only adds a factory parameter
32
 
# so that you can override the default SequenceMatcher
33
 
# this has been submitted as a patch to python
34
 
def unified_diff(a, b, fromfile='', tofile='', fromfiledate='',
35
 
                 tofiledate='', n=3, lineterm='\n',
36
 
                 sequencematcher=None):
37
 
    r"""
38
 
    Compare two sequences of lines; generate the delta as a unified diff.
39
 
 
40
 
    Unified diffs are a compact way of showing line changes and a few
41
 
    lines of context.  The number of context lines is set by 'n' which
42
 
    defaults to three.
43
 
 
44
 
    By default, the diff control lines (those with ---, +++, or @@) are
45
 
    created with a trailing newline.  This is helpful so that inputs
46
 
    created from file.readlines() result in diffs that are suitable for
47
 
    file.writelines() since both the inputs and outputs have trailing
48
 
    newlines.
49
 
 
50
 
    For inputs that do not have trailing newlines, set the lineterm
51
 
    argument to "" so that the output will be uniformly newline free.
52
 
 
53
 
    The unidiff format normally has a header for filenames and modification
54
 
    times.  Any or all of these may be specified using strings for
55
 
    'fromfile', 'tofile', 'fromfiledate', and 'tofiledate'.  The modification
56
 
    times are normally expressed in the format returned by time.ctime().
57
 
 
58
 
    Example:
59
 
 
60
 
    >>> for line in unified_diff('one two three four'.split(),
61
 
    ...             'zero one tree four'.split(), 'Original', 'Current',
62
 
    ...             'Sat Jan 26 23:30:50 1991', 'Fri Jun 06 10:20:52 2003',
63
 
    ...             lineterm=''):
64
 
    ...     print line
65
 
    --- Original Sat Jan 26 23:30:50 1991
66
 
    +++ Current Fri Jun 06 10:20:52 2003
67
 
    @@ -1,4 +1,4 @@
68
 
    +zero
69
 
     one
70
 
    -two
71
 
    -three
72
 
    +tree
73
 
     four
74
 
    """
75
 
    if sequencematcher is None:
76
 
        import difflib
77
 
        sequencematcher = difflib.SequenceMatcher
78
 
 
79
 
    started = False
80
 
    for group in sequencematcher(None,a,b).get_grouped_opcodes(n):
81
 
        if not started:
82
 
            yield '--- %s %s%s' % (fromfile, fromfiledate, lineterm)
83
 
            yield '+++ %s %s%s' % (tofile, tofiledate, lineterm)
84
 
            started = True
85
 
        i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
86
 
        yield "@@ -%d,%d +%d,%d @@%s" % (i1+1, i2-i1, j1+1, j2-j1, lineterm)
87
 
        for tag, i1, i2, j1, j2 in group:
88
 
            if tag == 'equal':
89
 
                for line in a[i1:i2]:
90
 
                    yield ' ' + line
91
 
                continue
92
 
            if tag == 'replace' or tag == 'delete':
93
 
                for line in a[i1:i2]:
94
 
                    yield '-' + line
95
 
            if tag == 'replace' or tag == 'insert':
96
 
                for line in b[j1:j2]:
97
 
                    yield '+' + line
98
 
 
99
 
 
100
 
def unified_diff_files(a, b, sequencematcher=None):
101
 
    """Generate the diff for two files.
102
 
    """
103
 
    # Should this actually be an error?
104
 
    if a == b:
105
 
        return []
106
 
    if a == '-':
107
 
        file_a = sys.stdin
108
 
        time_a = time.time()
109
 
    else:
110
 
        file_a = open(a, 'rb')
111
 
        time_a = os.stat(a).st_mtime
112
 
 
113
 
    if b == '-':
114
 
        file_b = sys.stdin
115
 
        time_b = time.time()
116
 
    else:
117
 
        file_b = open(b, 'rb')
118
 
        time_b = os.stat(b).st_mtime
119
 
 
120
 
    # TODO: Include fromfiledate and tofiledate
121
 
    return unified_diff(file_a.readlines(), file_b.readlines(),
122
 
                        fromfile=a, tofile=b,
123
 
                        sequencematcher=sequencematcher)
124
 
 
125
 
 
126
 
try:
127
 
    from bzrlib._patiencediff_c import (
128
 
        unique_lcs_c as unique_lcs,
129
 
        recurse_matches_c as recurse_matches,
130
 
        PatienceSequenceMatcher_c as PatienceSequenceMatcher
131
 
        )
132
 
except ImportError:
133
 
    from bzrlib._patiencediff_py import (
134
 
        unique_lcs_py as unique_lcs,
135
 
        recurse_matches_py as recurse_matches,
136
 
        PatienceSequenceMatcher_py as PatienceSequenceMatcher
137
 
        )
138
 
 
139
 
 
140
 
def main(args):
141
 
    import optparse
142
 
    p = optparse.OptionParser(usage='%prog [options] file_a file_b'
143
 
                                    '\nFiles can be "-" to read from stdin')
144
 
    p.add_option('--patience', dest='matcher', action='store_const', const='patience',
145
 
                 default='patience', help='Use the patience difference algorithm')
146
 
    p.add_option('--difflib', dest='matcher', action='store_const', const='difflib',
147
 
                 default='patience', help='Use python\'s difflib algorithm')
148
 
 
149
 
    algorithms = {'patience':PatienceSequenceMatcher, 'difflib':difflib.SequenceMatcher}
150
 
 
151
 
    (opts, args) = p.parse_args(args)
152
 
    matcher = algorithms[opts.matcher]
153
 
 
154
 
    if len(args) != 2:
155
 
        print 'You must supply 2 filenames to diff'
156
 
        return -1
157
 
 
158
 
    for line in unified_diff_files(args[0], args[1], sequencematcher=matcher):
159
 
        sys.stdout.write(line)
160
 
 
161
 
 
162
 
if __name__ == '__main__':
163
 
    sys.exit(main(sys.argv[1:]))