/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/diff.py

  • Committer: Martin Pool
  • Date: 2005-08-04 22:04:40 UTC
  • Revision ID: mbp@sourcefrog.net-20050804220440-99562df8151d1ac5
- add pending merge from aaron

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005, 2006 Canonical Ltd.
2
 
#
 
1
#! /usr/bin/env python
 
2
# -*- coding: UTF-8 -*-
 
3
 
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
5
6
# the Free Software Foundation; either version 2 of the License, or
6
7
# (at your option) any later version.
7
 
#
 
8
 
8
9
# This program is distributed in the hope that it will be useful,
9
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
12
# GNU General Public License for more details.
12
 
#
 
13
 
13
14
# You should have received a copy of the GNU General Public License
14
15
# along with this program; if not, write to the Free Software
15
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
17
 
17
 
import difflib
18
 
import os
19
 
import re
20
 
import sys
21
 
 
22
 
from bzrlib.lazy_import import lazy_import
23
 
lazy_import(globals(), """
24
 
import errno
25
 
import subprocess
26
 
import tempfile
27
 
import time
28
 
 
29
 
from bzrlib import (
30
 
    errors,
31
 
    osutils,
32
 
    patiencediff,
33
 
    textfile,
34
 
    timestamp,
35
 
    )
36
 
""")
37
 
 
38
 
from bzrlib.symbol_versioning import (
39
 
        deprecated_function,
40
 
        )
41
 
from bzrlib.trace import mutter, warning
42
 
 
 
18
from bzrlib.trace import mutter
 
19
from bzrlib.errors import BzrError
 
20
from bzrlib.delta import compare_trees
43
21
 
44
22
# TODO: Rather than building a changeset object, we should probably
45
23
# invoke callbacks on an object.  That object can either accumulate a
46
24
# list, write them out directly, etc etc.
47
25
 
48
 
 
49
 
class _PrematchedMatcher(difflib.SequenceMatcher):
50
 
    """Allow SequenceMatcher operations to use predetermined blocks"""
51
 
 
52
 
    def __init__(self, matching_blocks):
53
 
        difflib.SequenceMatcher(self, None, None)
54
 
        self.matching_blocks = matching_blocks
55
 
        self.opcodes = None
56
 
 
57
 
 
58
 
def internal_diff(old_filename, oldlines, new_filename, newlines, to_file,
59
 
                  allow_binary=False, sequence_matcher=None,
60
 
                  path_encoding='utf8'):
 
26
def internal_diff(old_label, oldlines, new_label, newlines, to_file):
 
27
    import difflib
 
28
    
61
29
    # FIXME: difflib is wrong if there is no trailing newline.
62
30
    # The syntax used by patch seems to be "\ No newline at
63
31
    # end of file" following the last diff line from that
73
41
    # both sequences are empty.
74
42
    if not oldlines and not newlines:
75
43
        return
76
 
    
77
 
    if allow_binary is False:
78
 
        textfile.check_text_lines(oldlines)
79
 
        textfile.check_text_lines(newlines)
80
 
 
81
 
    if sequence_matcher is None:
82
 
        sequence_matcher = patiencediff.PatienceSequenceMatcher
83
 
    ud = patiencediff.unified_diff(oldlines, newlines,
84
 
                      fromfile=old_filename.encode(path_encoding),
85
 
                      tofile=new_filename.encode(path_encoding),
86
 
                      sequencematcher=sequence_matcher)
87
 
 
88
 
    ud = list(ud)
 
44
 
 
45
    ud = difflib.unified_diff(oldlines, newlines,
 
46
                              fromfile=old_label, tofile=new_label)
 
47
 
89
48
    # work-around for difflib being too smart for its own good
90
49
    # if /dev/null is "1,0", patch won't recognize it as /dev/null
91
50
    if not oldlines:
 
51
        ud = list(ud)
92
52
        ud[2] = ud[2].replace('-1,0', '-0,0')
93
53
    elif not newlines:
 
54
        ud = list(ud)
94
55
        ud[2] = ud[2].replace('+1,0', '+0,0')
95
 
    # work around for difflib emitting random spaces after the label
96
 
    ud[0] = ud[0][:-2] + '\n'
97
 
    ud[1] = ud[1][:-2] + '\n'
98
56
 
99
57
    for line in ud:
100
58
        to_file.write(line)
101
59
        if not line.endswith('\n'):
102
60
            to_file.write("\n\\ No newline at end of file\n")
103
 
    to_file.write('\n')
104
 
 
105
 
 
106
 
def _spawn_external_diff(diffcmd, capture_errors=True):
107
 
    """Spawn the externall diff process, and return the child handle.
108
 
 
109
 
    :param diffcmd: The command list to spawn
110
 
    :param capture_errors: Capture stderr as well as setting LANG=C
111
 
        and LC_ALL=C. This lets us read and understand the output of diff,
112
 
        and respond to any errors.
113
 
    :return: A Popen object.
114
 
    """
115
 
    if capture_errors:
116
 
        # construct minimal environment
117
 
        env = {}
118
 
        path = os.environ.get('PATH')
119
 
        if path is not None:
120
 
            env['PATH'] = path
121
 
        env['LANGUAGE'] = 'C'   # on win32 only LANGUAGE has effect
122
 
        env['LANG'] = 'C'
123
 
        env['LC_ALL'] = 'C'
124
 
        stderr = subprocess.PIPE
125
 
    else:
126
 
        env = None
127
 
        stderr = None
128
 
 
129
 
    try:
130
 
        pipe = subprocess.Popen(diffcmd,
131
 
                                stdin=subprocess.PIPE,
132
 
                                stdout=subprocess.PIPE,
133
 
                                stderr=stderr,
134
 
                                env=env)
135
 
    except OSError, e:
136
 
        if e.errno == errno.ENOENT:
137
 
            raise errors.NoDiff(str(e))
138
 
        raise
139
 
 
140
 
    return pipe
141
 
 
142
 
 
143
 
def external_diff(old_filename, oldlines, new_filename, newlines, to_file,
 
61
    print >>to_file
 
62
 
 
63
 
 
64
 
 
65
 
 
66
def external_diff(old_label, oldlines, new_label, newlines, to_file,
144
67
                  diff_opts):
145
68
    """Display a diff by calling out to the external diff program."""
 
69
    import sys
 
70
    
 
71
    if to_file != sys.stdout:
 
72
        raise NotImplementedError("sorry, can't send external diff other than to stdout yet",
 
73
                                  to_file)
 
74
 
146
75
    # make sure our own output is properly ordered before the diff
147
76
    to_file.flush()
148
77
 
149
 
    oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='bzr-diff-old-')
150
 
    newtmp_fd, new_abspath = tempfile.mkstemp(prefix='bzr-diff-new-')
151
 
    oldtmpf = os.fdopen(oldtmp_fd, 'wb')
152
 
    newtmpf = os.fdopen(newtmp_fd, 'wb')
 
78
    from tempfile import NamedTemporaryFile
 
79
    import os
 
80
 
 
81
    oldtmpf = NamedTemporaryFile()
 
82
    newtmpf = NamedTemporaryFile()
153
83
 
154
84
    try:
155
85
        # TODO: perhaps a special case for comparing to or from the empty
162
92
        oldtmpf.writelines(oldlines)
163
93
        newtmpf.writelines(newlines)
164
94
 
165
 
        oldtmpf.close()
166
 
        newtmpf.close()
 
95
        oldtmpf.flush()
 
96
        newtmpf.flush()
167
97
 
168
98
        if not diff_opts:
169
99
            diff_opts = []
170
100
        diffcmd = ['diff',
171
 
                   '--label', old_filename,
172
 
                   old_abspath,
173
 
                   '--label', new_filename,
174
 
                   new_abspath,
175
 
                   '--binary',
176
 
                  ]
 
101
                   '--label', old_label,
 
102
                   oldtmpf.name,
 
103
                   '--label', new_label,
 
104
                   newtmpf.name]
177
105
 
178
106
        # diff only allows one style to be specified; they don't override.
179
107
        # note that some of these take optargs, and the optargs can be
199
127
        if diff_opts:
200
128
            diffcmd.extend(diff_opts)
201
129
 
202
 
        pipe = _spawn_external_diff(diffcmd, capture_errors=True)
203
 
        out,err = pipe.communicate()
204
 
        rc = pipe.returncode
 
130
        rc = os.spawnvp(os.P_WAIT, 'diff', diffcmd)
205
131
        
206
 
        # internal_diff() adds a trailing newline, add one here for consistency
207
 
        out += '\n'
208
 
        if rc == 2:
209
 
            # 'diff' gives retcode == 2 for all sorts of errors
210
 
            # one of those is 'Binary files differ'.
211
 
            # Bad options could also be the problem.
212
 
            # 'Binary files' is not a real error, so we suppress that error.
213
 
            lang_c_out = out
214
 
 
215
 
            # Since we got here, we want to make sure to give an i18n error
216
 
            pipe = _spawn_external_diff(diffcmd, capture_errors=False)
217
 
            out, err = pipe.communicate()
218
 
 
219
 
            # Write out the new i18n diff response
220
 
            to_file.write(out+'\n')
221
 
            if pipe.returncode != 2:
222
 
                raise errors.BzrError(
223
 
                               'external diff failed with exit code 2'
224
 
                               ' when run with LANG=C and LC_ALL=C,'
225
 
                               ' but not when run natively: %r' % (diffcmd,))
226
 
 
227
 
            first_line = lang_c_out.split('\n', 1)[0]
228
 
            # Starting with diffutils 2.8.4 the word "binary" was dropped.
229
 
            m = re.match('^(binary )?files.*differ$', first_line, re.I)
230
 
            if m is None:
231
 
                raise errors.BzrError('external diff failed with exit code 2;'
232
 
                                      ' command: %r' % (diffcmd,))
233
 
            else:
234
 
                # Binary files differ, just return
235
 
                return
236
 
 
237
 
        # If we got to here, we haven't written out the output of diff
238
 
        # do so now
239
 
        to_file.write(out)
240
 
        if rc not in (0, 1):
 
132
        if rc != 0 and rc != 1:
241
133
            # returns 1 if files differ; that's OK
242
134
            if rc < 0:
243
135
                msg = 'signal %d' % (-rc)
244
136
            else:
245
137
                msg = 'exit code %d' % rc
246
138
                
247
 
            raise errors.BzrError('external diff failed with %s; command: %r' 
248
 
                                  % (rc, diffcmd))
249
 
 
250
 
 
 
139
            raise BzrError('external diff failed with %s; command: %r' % (rc, diffcmd))
251
140
    finally:
252
141
        oldtmpf.close()                 # and delete
253
142
        newtmpf.close()
254
 
        # Clean up. Warn in case the files couldn't be deleted
255
 
        # (in case windows still holds the file open, but not
256
 
        # if the files have already been deleted)
257
 
        try:
258
 
            os.remove(old_abspath)
259
 
        except OSError, e:
260
 
            if e.errno not in (errno.ENOENT,):
261
 
                warning('Failed to delete temporary file: %s %s',
262
 
                        old_abspath, e)
263
 
        try:
264
 
            os.remove(new_abspath)
265
 
        except OSError:
266
 
            if e.errno not in (errno.ENOENT,):
267
 
                warning('Failed to delete temporary file: %s %s',
268
 
                        new_abspath, e)
269
 
 
270
 
 
271
 
def diff_cmd_helper(tree, specific_files, external_diff_options, 
272
 
                    old_revision_spec=None, new_revision_spec=None,
273
 
                    revision_specs=None,
274
 
                    old_label='a/', new_label='b/'):
275
 
    """Helper for cmd_diff.
276
 
 
277
 
    :param tree:
278
 
        A WorkingTree
279
 
 
280
 
    :param specific_files:
281
 
        The specific files to compare, or None
282
 
 
283
 
    :param external_diff_options:
284
 
        If non-None, run an external diff, and pass it these options
285
 
 
286
 
    :param old_revision_spec:
287
 
        If None, use basis tree as old revision, otherwise use the tree for
288
 
        the specified revision. 
289
 
 
290
 
    :param new_revision_spec:
291
 
        If None, use working tree as new revision, otherwise use the tree for
292
 
        the specified revision.
293
 
    
294
 
    :param revision_specs: 
295
 
        Zero, one or two RevisionSpecs from the command line, saying what revisions 
296
 
        to compare.  This can be passed as an alternative to the old_revision_spec 
297
 
        and new_revision_spec parameters.
298
 
 
 
143
    
 
144
 
 
145
 
 
146
def show_diff(b, revision, specific_files, external_diff_options=None):
 
147
    """Shortcut for showing the diff to the working tree.
 
148
 
 
149
    b
 
150
        Branch.
 
151
 
 
152
    revision
 
153
        None for each, or otherwise the old revision to compare against.
 
154
    
299
155
    The more general form is show_diff_trees(), where the caller
300
156
    supplies any two trees.
301
157
    """
302
 
 
303
 
    # TODO: perhaps remove the old parameters old_revision_spec and
304
 
    # new_revision_spec, since this is only really for use from cmd_diff and
305
 
    # it now always passes through a sequence of revision_specs -- mbp
306
 
    # 20061221
307
 
 
308
 
    def spec_tree(spec):
309
 
        if tree:
310
 
            revision = spec.in_store(tree.branch)
311
 
        else:
312
 
            revision = spec.in_store(None)
313
 
        revision_id = revision.rev_id
314
 
        branch = revision.branch
315
 
        return branch.repository.revision_tree(revision_id)
316
 
 
317
 
    if revision_specs is not None:
318
 
        assert (old_revision_spec is None
319
 
                and new_revision_spec is None)
320
 
        if len(revision_specs) > 0:
321
 
            old_revision_spec = revision_specs[0]
322
 
        if len(revision_specs) > 1:
323
 
            new_revision_spec = revision_specs[1]
324
 
 
325
 
    if old_revision_spec is None:
326
 
        old_tree = tree.basis_tree()
327
 
    else:
328
 
        old_tree = spec_tree(old_revision_spec)
329
 
 
330
 
    if (new_revision_spec is None
331
 
        or new_revision_spec.spec is None):
332
 
        new_tree = tree
333
 
    else:
334
 
        new_tree = spec_tree(new_revision_spec)
335
 
 
336
 
    if new_tree is not tree:
337
 
        extra_trees = (tree,)
338
 
    else:
339
 
        extra_trees = None
340
 
 
341
 
    return show_diff_trees(old_tree, new_tree, sys.stdout, specific_files,
342
 
                           external_diff_options,
343
 
                           old_label=old_label, new_label=new_label,
344
 
                           extra_trees=extra_trees)
 
158
    import sys
 
159
 
 
160
    if revision == None:
 
161
        old_tree = b.basis_tree()
 
162
    else:
 
163
        old_tree = b.revision_tree(b.lookup_revision(revision))
 
164
        
 
165
    new_tree = b.working_tree()
 
166
 
 
167
    show_diff_trees(old_tree, new_tree, sys.stdout, specific_files,
 
168
                    external_diff_options)
 
169
 
345
170
 
346
171
 
347
172
def show_diff_trees(old_tree, new_tree, to_file, specific_files=None,
348
 
                    external_diff_options=None,
349
 
                    old_label='a/', new_label='b/',
350
 
                    extra_trees=None,
351
 
                    path_encoding='utf8'):
 
173
                    external_diff_options=None):
352
174
    """Show in text form the changes from one tree to another.
353
175
 
354
176
    to_files
356
178
 
357
179
    external_diff_options
358
180
        If set, use an external GNU diff and pass these options.
359
 
 
360
 
    extra_trees
361
 
        If set, more Trees to use for looking up file ids
362
 
 
363
 
    path_encoding
364
 
        If set, the path will be encoded as specified, otherwise is supposed
365
 
        to be utf8
366
181
    """
367
 
    old_tree.lock_read()
368
 
    try:
369
 
        if extra_trees is not None:
370
 
            for tree in extra_trees:
371
 
                tree.lock_read()
372
 
        new_tree.lock_read()
373
 
        try:
374
 
            differ = DiffTree.from_trees_options(old_tree, new_tree, to_file,
375
 
                                                   path_encoding,
376
 
                                                   external_diff_options,
377
 
                                                   old_label, new_label)
378
 
            return differ.show_diff(specific_files, extra_trees)
379
 
        finally:
380
 
            new_tree.unlock()
381
 
            if extra_trees is not None:
382
 
                for tree in extra_trees:
383
 
                    tree.unlock()
384
 
    finally:
385
 
        old_tree.unlock()
386
 
 
387
 
 
388
 
def _patch_header_date(tree, file_id, path):
389
 
    """Returns a timestamp suitable for use in a patch header."""
390
 
    mtime = tree.get_file_mtime(file_id, path)
391
 
    assert mtime is not None, \
392
 
        "got an mtime of None for file-id %s, path %s in tree %s" % (
393
 
                file_id, path, tree)
394
 
    return timestamp.format_patch_date(mtime)
395
 
 
396
 
 
397
 
def _raise_if_nonexistent(paths, old_tree, new_tree):
398
 
    """Complain if paths are not in either inventory or tree.
399
 
 
400
 
    It's OK with the files exist in either tree's inventory, or 
401
 
    if they exist in the tree but are not versioned.
 
182
 
 
183
    # TODO: Options to control putting on a prefix or suffix, perhaps as a format string
 
184
    old_label = ''
 
185
    new_label = ''
 
186
 
 
187
    DEVNULL = '/dev/null'
 
188
    # Windows users, don't panic about this filename -- it is a
 
189
    # special signal to GNU patch that the file should be created or
 
190
    # deleted respectively.
 
191
 
 
192
    # TODO: Generation of pseudo-diffs for added/deleted files could
 
193
    # be usefully made into a much faster special case.
 
194
 
 
195
    if external_diff_options:
 
196
        assert isinstance(external_diff_options, basestring)
 
197
        opts = external_diff_options.split()
 
198
        def diff_file(olab, olines, nlab, nlines, to_file):
 
199
            external_diff(olab, olines, nlab, nlines, to_file, opts)
 
200
    else:
 
201
        diff_file = internal_diff
402
202
    
403
 
    This can be used by operations such as bzr status that can accept
404
 
    unknown or ignored files.
405
 
    """
406
 
    mutter("check paths: %r", paths)
407
 
    if not paths:
408
 
        return
409
 
    s = old_tree.filter_unversioned_files(paths)
410
 
    s = new_tree.filter_unversioned_files(s)
411
 
    s = [path for path in s if not new_tree.has_filename(path)]
412
 
    if s:
413
 
        raise errors.PathsDoNotExist(sorted(s))
414
 
 
415
 
 
416
 
def get_prop_change(meta_modified):
417
 
    if meta_modified:
418
 
        return " (properties changed)"
419
 
    else:
420
 
        return  ""
421
 
 
422
 
 
423
 
class DiffPath(object):
424
 
    """Base type for command object that compare files"""
425
 
 
426
 
    # The type or contents of the file were unsuitable for diffing
427
 
    CANNOT_DIFF = 'CANNOT_DIFF'
428
 
    # The file has changed in a semantic way
429
 
    CHANGED = 'CHANGED'
430
 
    # The file content may have changed, but there is no semantic change
431
 
    UNCHANGED = 'UNCHANGED'
432
 
 
433
 
    def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8'):
434
 
        """Constructor.
435
 
 
436
 
        :param old_tree: The tree to show as the old tree in the comparison
437
 
        :param new_tree: The tree to show as new in the comparison
438
 
        :param to_file: The file to write comparison data to
439
 
        :param path_encoding: The character encoding to write paths in
440
 
        """
441
 
        self.old_tree = old_tree
442
 
        self.new_tree = new_tree
443
 
        self.to_file = to_file
444
 
        self.path_encoding = path_encoding
445
 
 
446
 
    @classmethod
447
 
    def from_diff_tree(klass, diff_tree):
448
 
        return klass(diff_tree.old_tree, diff_tree.new_tree,
449
 
                     diff_tree.to_file, diff_tree.path_encoding)
450
 
 
451
 
    @staticmethod
452
 
    def _diff_many(differs, file_id, old_path, new_path, old_kind, new_kind):
453
 
        for file_differ in differs:
454
 
            result = file_differ.diff(file_id, old_path, new_path, old_kind,
455
 
                                      new_kind)
456
 
            if result is not DiffPath.CANNOT_DIFF:
457
 
                return result
458
 
        else:
459
 
            return DiffPath.CANNOT_DIFF
460
 
 
461
 
 
462
 
class DiffKindChange(object):
463
 
    """Special differ for file kind changes.
464
 
 
465
 
    Represents kind change as deletion + creation.  Uses the other differs
466
 
    to do this.
467
 
    """
468
 
    def __init__(self, differs):
469
 
        self.differs = differs
470
 
 
471
 
    @classmethod
472
 
    def from_diff_tree(klass, diff_tree):
473
 
        return klass(diff_tree.differs)
474
 
 
475
 
    def diff(self, file_id, old_path, new_path, old_kind, new_kind):
476
 
        """Perform comparison
477
 
 
478
 
        :param file_id: The file_id of the file to compare
479
 
        :param old_path: Path of the file in the old tree
480
 
        :param new_path: Path of the file in the new tree
481
 
        :param old_kind: Old file-kind of the file
482
 
        :param new_kind: New file-kind of the file
483
 
        """
484
 
        if None in (old_kind, new_kind):
485
 
            return DiffPath.CANNOT_DIFF
486
 
        result = DiffPath._diff_many(self.differs, file_id, old_path,
487
 
                                       new_path, old_kind, None)
488
 
        if result is DiffPath.CANNOT_DIFF:
489
 
            return result
490
 
        return DiffPath._diff_many(self.differs, file_id, old_path, new_path,
491
 
                                     None, new_kind)
492
 
 
493
 
 
494
 
class DiffDirectory(DiffPath):
495
 
 
496
 
    def diff(self, file_id, old_path, new_path, old_kind, new_kind):
497
 
        """Perform comparison between two directories.  (dummy)
498
 
 
499
 
        """
500
 
        if 'directory' not in (old_kind, new_kind):
501
 
            return self.CANNOT_DIFF
502
 
        if old_kind not in ('directory', None):
503
 
            return self.CANNOT_DIFF
504
 
        if new_kind not in ('directory', None):
505
 
            return self.CANNOT_DIFF
506
 
        return self.CHANGED
507
 
 
508
 
 
509
 
class DiffSymlink(DiffPath):
510
 
 
511
 
    def diff(self, file_id, old_path, new_path, old_kind, new_kind):
512
 
        """Perform comparison between two symlinks
513
 
 
514
 
        :param file_id: The file_id of the file to compare
515
 
        :param old_path: Path of the file in the old tree
516
 
        :param new_path: Path of the file in the new tree
517
 
        :param old_kind: Old file-kind of the file
518
 
        :param new_kind: New file-kind of the file
519
 
        """
520
 
        if 'symlink' not in (old_kind, new_kind):
521
 
            return self.CANNOT_DIFF
522
 
        if old_kind == 'symlink':
523
 
            old_target = self.old_tree.get_symlink_target(file_id)
524
 
        elif old_kind is None:
525
 
            old_target = None
526
 
        else:
527
 
            return self.CANNOT_DIFF
528
 
        if new_kind == 'symlink':
529
 
            new_target = self.new_tree.get_symlink_target(file_id)
530
 
        elif new_kind is None:
531
 
            new_target = None
532
 
        else:
533
 
            return self.CANNOT_DIFF
534
 
        return self.diff_symlink(old_target, new_target)
535
 
 
536
 
    def diff_symlink(self, old_target, new_target):
537
 
        if old_target is None:
538
 
            self.to_file.write('=== target is %r\n' % new_target)
539
 
        elif new_target is None:
540
 
            self.to_file.write('=== target was %r\n' % old_target)
541
 
        else:
542
 
            self.to_file.write('=== target changed %r => %r\n' %
543
 
                              (old_target, new_target))
544
 
        return self.CHANGED
545
 
 
546
 
 
547
 
class DiffText(DiffPath):
548
 
 
549
 
    # GNU Patch uses the epoch date to detect files that are being added
550
 
    # or removed in a diff.
551
 
    EPOCH_DATE = '1970-01-01 00:00:00 +0000'
552
 
 
553
 
    def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8',
554
 
                 old_label='', new_label='', text_differ=internal_diff):
555
 
        DiffPath.__init__(self, old_tree, new_tree, to_file, path_encoding)
556
 
        self.text_differ = text_differ
557
 
        self.old_label = old_label
558
 
        self.new_label = new_label
559
 
        self.path_encoding = path_encoding
560
 
 
561
 
    def diff(self, file_id, old_path, new_path, old_kind, new_kind):
562
 
        """Compare two files in unified diff format
563
 
 
564
 
        :param file_id: The file_id of the file to compare
565
 
        :param old_path: Path of the file in the old tree
566
 
        :param new_path: Path of the file in the new tree
567
 
        :param old_kind: Old file-kind of the file
568
 
        :param new_kind: New file-kind of the file
569
 
        """
570
 
        if 'file' not in (old_kind, new_kind):
571
 
            return self.CANNOT_DIFF
572
 
        from_file_id = to_file_id = file_id
573
 
        if old_kind == 'file':
574
 
            old_date = _patch_header_date(self.old_tree, file_id, old_path)
575
 
        elif old_kind is None:
576
 
            old_date = self.EPOCH_DATE
577
 
            from_file_id = None
578
 
        else:
579
 
            return self.CANNOT_DIFF
580
 
        if new_kind == 'file':
581
 
            new_date = _patch_header_date(self.new_tree, file_id, new_path)
582
 
        elif new_kind is None:
583
 
            new_date = self.EPOCH_DATE
584
 
            to_file_id = None
585
 
        else:
586
 
            return self.CANNOT_DIFF
587
 
        from_label = '%s%s\t%s' % (self.old_label, old_path, old_date)
588
 
        to_label = '%s%s\t%s' % (self.new_label, new_path, new_date)
589
 
        return self.diff_text(from_file_id, to_file_id, from_label, to_label)
590
 
 
591
 
    def diff_text(self, from_file_id, to_file_id, from_label, to_label):
592
 
        """Diff the content of given files in two trees
593
 
 
594
 
        :param from_file_id: The id of the file in the from tree.  If None,
595
 
            the file is not present in the from tree.
596
 
        :param to_file_id: The id of the file in the to tree.  This may refer
597
 
            to a different file from from_file_id.  If None,
598
 
            the file is not present in the to tree.
599
 
        """
600
 
        def _get_text(tree, file_id):
601
 
            if file_id is not None:
602
 
                return tree.get_file(file_id).readlines()
603
 
            else:
604
 
                return []
605
 
        try:
606
 
            from_text = _get_text(self.old_tree, from_file_id)
607
 
            to_text = _get_text(self.new_tree, to_file_id)
608
 
            self.text_differ(from_label, from_text, to_label, to_text,
609
 
                             self.to_file)
610
 
        except errors.BinaryFile:
611
 
            self.to_file.write(
612
 
                  ("Binary files %s and %s differ\n" %
613
 
                  (from_label, to_label)).encode(self.path_encoding))
614
 
        return self.CHANGED
615
 
 
616
 
 
617
 
class DiffTree(object):
618
 
    """Provides textual representations of the difference between two trees.
619
 
 
620
 
    A DiffTree examines two trees and where a file-id has altered
621
 
    between them, generates a textual representation of the difference.
622
 
    DiffTree uses a sequence of DiffPath objects which are each
623
 
    given the opportunity to handle a given altered fileid. The list
624
 
    of DiffPath objects can be extended globally by appending to
625
 
    DiffTree.diff_factories, or for a specific diff operation by
626
 
    supplying the extra_factories option to the appropriate method.
627
 
    """
628
 
 
629
 
    # list of factories that can provide instances of DiffPath objects
630
 
    # may be extended by plugins.
631
 
    diff_factories = [DiffSymlink.from_diff_tree,
632
 
                      DiffDirectory.from_diff_tree]
633
 
 
634
 
    def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8',
635
 
                 diff_text=None, extra_factories=None):
636
 
        """Constructor
637
 
 
638
 
        :param old_tree: Tree to show as old in the comparison
639
 
        :param new_tree: Tree to show as new in the comparison
640
 
        :param to_file: File to write comparision to
641
 
        :param path_encoding: Character encoding to write paths in
642
 
        :param diff_text: DiffPath-type object to use as a last resort for
643
 
            diffing text files.
644
 
        :param extra_factories: Factories of DiffPaths to try before any other
645
 
            DiffPaths"""
646
 
        if diff_text is None:
647
 
            diff_text = DiffText(old_tree, new_tree, to_file, path_encoding,
648
 
                                 '', '',  internal_diff)
649
 
        self.old_tree = old_tree
650
 
        self.new_tree = new_tree
651
 
        self.to_file = to_file
652
 
        self.path_encoding = path_encoding
653
 
        self.differs = []
654
 
        if extra_factories is not None:
655
 
            self.differs.extend(f(self) for f in extra_factories)
656
 
        self.differs.extend(f(self) for f in self.diff_factories)
657
 
        self.differs.extend([diff_text, DiffKindChange.from_diff_tree(self)])
658
 
 
659
 
    @classmethod
660
 
    def from_trees_options(klass, old_tree, new_tree, to_file,
661
 
                           path_encoding, external_diff_options, old_label,
662
 
                           new_label):
663
 
        """Factory for producing a DiffTree.
664
 
 
665
 
        Designed to accept options used by show_diff_trees.
666
 
        :param old_tree: The tree to show as old in the comparison
667
 
        :param new_tree: The tree to show as new in the comparison
668
 
        :param to_file: File to write comparisons to
669
 
        :param path_encoding: Character encoding to use for writing paths
670
 
        :param external_diff_options: If supplied, use the installed diff
671
 
            binary to perform file comparison, using supplied options.
672
 
        :param old_label: Prefix to use for old file labels
673
 
        :param new_label: Prefix to use for new file labels
674
 
        """
675
 
        if external_diff_options:
676
 
            assert isinstance(external_diff_options, basestring)
677
 
            opts = external_diff_options.split()
678
 
            def diff_file(olab, olines, nlab, nlines, to_file):
679
 
                external_diff(olab, olines, nlab, nlines, to_file, opts)
680
 
        else:
681
 
            diff_file = internal_diff
682
 
        diff_text = DiffText(old_tree, new_tree, to_file, path_encoding,
683
 
                             old_label, new_label, diff_file)
684
 
        return klass(old_tree, new_tree, to_file, path_encoding, diff_text)
685
 
 
686
 
    def show_diff(self, specific_files, extra_trees=None):
687
 
        """Write tree diff to self.to_file
688
 
 
689
 
        :param sepecific_files: the specific files to compare (recursive)
690
 
        :param extra_trees: extra trees to use for mapping paths to file_ids
691
 
        """
692
 
        # TODO: Generation of pseudo-diffs for added/deleted files could
693
 
        # be usefully made into a much faster special case.
694
 
 
695
 
        delta = self.new_tree.changes_from(self.old_tree,
696
 
            specific_files=specific_files,
697
 
            extra_trees=extra_trees, require_versioned=True)
698
 
 
699
 
        has_changes = 0
700
 
        for path, file_id, kind in delta.removed:
701
 
            has_changes = 1
702
 
            path_encoded = path.encode(self.path_encoding, "replace")
703
 
            self.to_file.write("=== removed %s '%s'\n" % (kind, path_encoded))
704
 
            self.diff(file_id, path, path)
705
 
 
706
 
        for path, file_id, kind in delta.added:
707
 
            has_changes = 1
708
 
            path_encoded = path.encode(self.path_encoding, "replace")
709
 
            self.to_file.write("=== added %s '%s'\n" % (kind, path_encoded))
710
 
            self.diff(file_id, path, path)
711
 
        for (old_path, new_path, file_id, kind,
712
 
             text_modified, meta_modified) in delta.renamed:
713
 
            has_changes = 1
714
 
            prop_str = get_prop_change(meta_modified)
715
 
            oldpath_encoded = old_path.encode(self.path_encoding, "replace")
716
 
            newpath_encoded = new_path.encode(self.path_encoding, "replace")
717
 
            self.to_file.write("=== renamed %s '%s' => '%s'%s\n" % (kind,
718
 
                                oldpath_encoded, newpath_encoded, prop_str))
719
 
            if text_modified:
720
 
                self.diff(file_id, old_path, new_path)
721
 
        for path, file_id, kind, text_modified, meta_modified in\
722
 
            delta.modified:
723
 
            has_changes = 1
724
 
            prop_str = get_prop_change(meta_modified)
725
 
            path_encoded = path.encode(self.path_encoding, "replace")
726
 
            self.to_file.write("=== modified %s '%s'%s\n" % (kind,
727
 
                                path_encoded, prop_str))
728
 
            # The file may be in a different location in the old tree (because
729
 
            # the containing dir was renamed, but the file itself was not)
730
 
            if text_modified:
731
 
                old_path = self.old_tree.id2path(file_id)
732
 
                self.diff(file_id, old_path, path)
733
 
        return has_changes
734
 
 
735
 
    def diff(self, file_id, old_path, new_path):
736
 
        """Perform a diff of a single file
737
 
 
738
 
        :param file_id: file-id of the file
739
 
        :param old_path: The path of the file in the old tree
740
 
        :param new_path: The path of the file in the new tree
741
 
        """
742
 
        try:
743
 
            old_kind = self.old_tree.kind(file_id)
744
 
        except errors.NoSuchId:
745
 
            old_kind = None
746
 
        try:
747
 
            new_kind = self.new_tree.kind(file_id)
748
 
        except errors.NoSuchId:
749
 
            new_kind = None
750
 
 
751
 
        result = DiffPath._diff_many(self.differs, file_id, old_path,
752
 
                                       new_path, old_kind, new_kind)
753
 
        if result is DiffPath.CANNOT_DIFF:
754
 
            error_path = new_path
755
 
            if error_path is None:
756
 
                error_path = old_path
757
 
            raise errors.NoDiffFound(error_path)
 
203
 
 
204
    delta = compare_trees(old_tree, new_tree, want_unchanged=False,
 
205
                          specific_files=specific_files)
 
206
 
 
207
    for path, file_id, kind in delta.removed:
 
208
        print >>to_file, '*** removed %s %r' % (kind, path)
 
209
        if kind == 'file':
 
210
            diff_file(old_label + path,
 
211
                      old_tree.get_file(file_id).readlines(),
 
212
                      DEVNULL, 
 
213
                      [],
 
214
                      to_file)
 
215
 
 
216
    for path, file_id, kind in delta.added:
 
217
        print >>to_file, '*** added %s %r' % (kind, path)
 
218
        if kind == 'file':
 
219
            diff_file(DEVNULL,
 
220
                      [],
 
221
                      new_label + path,
 
222
                      new_tree.get_file(file_id).readlines(),
 
223
                      to_file)
 
224
 
 
225
    for old_path, new_path, file_id, kind, text_modified in delta.renamed:
 
226
        print >>to_file, '*** renamed %s %r => %r' % (kind, old_path, new_path)
 
227
        if text_modified:
 
228
            diff_file(old_label + old_path,
 
229
                      old_tree.get_file(file_id).readlines(),
 
230
                      new_label + new_path,
 
231
                      new_tree.get_file(file_id).readlines(),
 
232
                      to_file)
 
233
 
 
234
    for path, file_id, kind in delta.modified:
 
235
        print >>to_file, '*** modified %s %r' % (kind, path)
 
236
        if kind == 'file':
 
237
            diff_file(old_label + path,
 
238
                      old_tree.get_file(file_id).readlines(),
 
239
                      new_label + path,
 
240
                      new_tree.get_file(file_id).readlines(),
 
241
                      to_file)
 
242
 
 
243
 
 
244
 
 
245
 
 
246