/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 von Gagern
  • Date: 2010-04-20 08:47:38 UTC
  • mfrom: (5167 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5195.
  • Revision ID: martin.vgagern@gmx.net-20100420084738-ygymnqmdllzrhpfn
merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd.
 
1
# Copyright (C) 2005-2010 Canonical Ltd.
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
from __future__ import absolute_import
18
 
 
19
17
import difflib
20
18
import os
21
19
import re
 
20
import shutil
22
21
import string
23
22
import sys
24
23
 
27
26
import errno
28
27
import subprocess
29
28
import tempfile
 
29
import time
30
30
 
31
31
from bzrlib import (
 
32
    branch as _mod_branch,
 
33
    bzrdir,
 
34
    cmdline,
32
35
    cleanup,
33
 
    cmdline,
34
 
    controldir,
35
36
    errors,
36
37
    osutils,
37
38
    patiencediff,
41
42
    )
42
43
 
43
44
from bzrlib.workingtree import WorkingTree
44
 
from bzrlib.i18n import gettext
45
45
""")
46
46
 
47
47
from bzrlib.registry import (
48
48
    Registry,
49
49
    )
 
50
from bzrlib.symbol_versioning import (
 
51
    deprecated_function,
 
52
    deprecated_in,
 
53
    )
50
54
from bzrlib.trace import mutter, note, warning
51
55
 
52
56
 
96
100
    if sequence_matcher is None:
97
101
        sequence_matcher = patiencediff.PatienceSequenceMatcher
98
102
    ud = patiencediff.unified_diff(oldlines, newlines,
99
 
                      fromfile=old_filename.encode(path_encoding, 'replace'),
100
 
                      tofile=new_filename.encode(path_encoding, 'replace'),
 
103
                      fromfile=old_filename.encode(path_encoding),
 
104
                      tofile=new_filename.encode(path_encoding),
101
105
                      sequencematcher=sequence_matcher)
102
106
 
103
107
    ud = list(ud)
287
291
                        new_abspath, e)
288
292
 
289
293
 
 
294
@deprecated_function(deprecated_in((2, 2, 0)))
 
295
def get_trees_and_branches_to_diff(path_list, revision_specs, old_url, new_url,
 
296
                                   apply_view=True):
 
297
    """Get the trees and specific files to diff given a list of paths.
 
298
 
 
299
    This method works out the trees to be diff'ed and the files of
 
300
    interest within those trees.
 
301
 
 
302
    :param path_list:
 
303
        the list of arguments passed to the diff command
 
304
    :param revision_specs:
 
305
        Zero, one or two RevisionSpecs from the diff command line,
 
306
        saying what revisions to compare.
 
307
    :param old_url:
 
308
        The url of the old branch or tree. If None, the tree to use is
 
309
        taken from the first path, if any, or the current working tree.
 
310
    :param new_url:
 
311
        The url of the new branch or tree. If None, the tree to use is
 
312
        taken from the first path, if any, or the current working tree.
 
313
    :param apply_view:
 
314
        if True and a view is set, apply the view or check that the paths
 
315
        are within it
 
316
    :returns:
 
317
        a tuple of (old_tree, new_tree, old_branch, new_branch,
 
318
        specific_files, extra_trees) where extra_trees is a sequence of
 
319
        additional trees to search in for file-ids.  The trees and branches
 
320
        are not locked.
 
321
    """
 
322
    op = cleanup.OperationWithCleanups(get_trees_and_branches_to_diff_locked)
 
323
    return op.run_simple(path_list, revision_specs, old_url, new_url,
 
324
            op.add_cleanup, apply_view=apply_view)
 
325
    
 
326
 
290
327
def get_trees_and_branches_to_diff_locked(
291
328
    path_list, revision_specs, old_url, new_url, add_cleanup, apply_view=True):
292
329
    """Get the trees and specific files to diff given a list of paths.
358
395
    if old_url is None:
359
396
        old_url = default_location
360
397
    working_tree, branch, relpath = \
361
 
        controldir.ControlDir.open_containing_tree_or_branch(old_url)
 
398
        bzrdir.BzrDir.open_containing_tree_or_branch(old_url)
362
399
    lock_tree_or_branch(working_tree, branch)
363
400
    if consider_relpath and relpath != '':
364
401
        if working_tree is not None and apply_view:
372
409
        new_url = default_location
373
410
    if new_url != old_url:
374
411
        working_tree, branch, relpath = \
375
 
            controldir.ControlDir.open_containing_tree_or_branch(new_url)
 
412
            bzrdir.BzrDir.open_containing_tree_or_branch(new_url)
376
413
        lock_tree_or_branch(working_tree, branch)
377
414
        if consider_relpath and relpath != '':
378
415
            if working_tree is not None and apply_view:
384
421
 
385
422
    # Get the specific files (all files is None, no files is [])
386
423
    if make_paths_wt_relative and working_tree is not None:
387
 
        other_paths = working_tree.safe_relpath_files(
388
 
            other_paths,
 
424
        try:
 
425
            from bzrlib.builtins import safe_relpath_files
 
426
            other_paths = safe_relpath_files(working_tree, other_paths,
389
427
            apply_view=apply_view)
 
428
        except errors.FileInWrongBranch:
 
429
            raise errors.BzrCommandError("Files are in different branches")
390
430
    specific_files.extend(other_paths)
391
431
    if len(specific_files) == 0:
392
432
        specific_files = None
396
436
            if view_files:
397
437
                specific_files = view_files
398
438
                view_str = views.view_display_str(view_files)
399
 
                note(gettext("*** Ignoring files outside view. View is %s") % view_str)
 
439
                note("*** Ignoring files outside view. View is %s" % view_str)
400
440
 
401
441
    # Get extra trees that ought to be searched for file-ids
402
442
    extra_trees = None
403
443
    if working_tree is not None and working_tree not in (old_tree, new_tree):
404
444
        extra_trees = (working_tree,)
405
 
    return (old_tree, new_tree, old_branch, new_branch,
406
 
            specific_files, extra_trees)
 
445
    return old_tree, new_tree, old_branch, new_branch, specific_files, extra_trees
407
446
 
408
447
 
409
448
def _get_tree_to_diff(spec, tree=None, branch=None, basis_is_default=True):
430
469
    """Show in text form the changes from one tree to another.
431
470
 
432
471
    :param to_file: The output stream.
433
 
    :param specific_files: Include only changes to these files - None for all
 
472
    :param specific_files:Include only changes to these files - None for all
434
473
        changes.
435
474
    :param external_diff_options: If set, use an external GNU diff and pass 
436
475
        these options.
668
707
        """
669
708
        def _get_text(tree, file_id, path):
670
709
            if file_id is not None:
671
 
                return tree.get_file_lines(file_id, path)
 
710
                return tree.get_file(file_id, path).readlines()
672
711
            else:
673
712
                return []
674
713
        try:
675
714
            from_text = _get_text(self.old_tree, from_file_id, from_path)
676
715
            to_text = _get_text(self.new_tree, to_file_id, to_path)
677
716
            self.text_differ(from_label, from_text, to_label, to_text,
678
 
                             self.to_file, path_encoding=self.path_encoding)
 
717
                             self.to_file)
679
718
        except errors.BinaryFile:
680
719
            self.to_file.write(
681
720
                  ("Binary files %s and %s differ\n" %
682
 
                  (from_label, to_label)).encode(self.path_encoding,'replace'))
 
721
                  (from_label, to_label)).encode(self.path_encoding))
683
722
        return self.CHANGED
684
723
 
685
724
 
701
740
                     path_encoding)
702
741
 
703
742
    @classmethod
704
 
    def make_from_diff_tree(klass, command_string, external_diff_options=None):
 
743
    def make_from_diff_tree(klass, command_string):
705
744
        def from_diff_tree(diff_tree):
706
 
            full_command_string = [command_string]
707
 
            if external_diff_options is not None:
708
 
                full_command_string += ' ' + external_diff_options
709
 
            return klass.from_string(full_command_string, diff_tree.old_tree,
 
745
            return klass.from_string(command_string, diff_tree.old_tree,
710
746
                                     diff_tree.new_tree, diff_tree.to_file)
711
747
        return from_diff_tree
712
748
 
713
749
    def _get_command(self, old_path, new_path):
714
750
        my_map = {'old_path': old_path, 'new_path': new_path}
715
 
        command = [AtTemplate(t).substitute(my_map) for t in
716
 
                   self.command_template]
717
 
        if sys.platform == 'win32': # Popen doesn't accept unicode on win32
718
 
            command_encoded = []
719
 
            for c in command:
720
 
                if isinstance(c, unicode):
721
 
                    command_encoded.append(c.encode('mbcs'))
722
 
                else:
723
 
                    command_encoded.append(c)
724
 
            return command_encoded
725
 
        else:
726
 
            return command
 
751
        return [AtTemplate(t).substitute(my_map) for t in
 
752
                self.command_template]
727
753
 
728
754
    def _execute(self, old_path, new_path):
729
755
        command = self._get_command(old_path, new_path)
749
775
                raise
750
776
        return True
751
777
 
752
 
    @staticmethod
753
 
    def _fenc():
754
 
        """Returns safe encoding for passing file path to diff tool"""
755
 
        if sys.platform == 'win32':
756
 
            return 'mbcs'
757
 
        else:
758
 
            # Don't fallback to 'utf-8' because subprocess may not be able to
759
 
            # handle utf-8 correctly when locale is not utf-8.
760
 
            return sys.getfilesystemencoding() or 'ascii'
761
 
 
762
 
    def _is_safepath(self, path):
763
 
        """Return true if `path` may be able to pass to subprocess."""
764
 
        fenc = self._fenc()
765
 
        try:
766
 
            return path == path.encode(fenc).decode(fenc)
767
 
        except UnicodeError:
768
 
            return False
769
 
 
770
 
    def _safe_filename(self, prefix, relpath):
771
 
        """Replace unsafe character in `relpath` then join `self._root`,
772
 
        `prefix` and `relpath`."""
773
 
        fenc = self._fenc()
774
 
        # encoded_str.replace('?', '_') may break multibyte char.
775
 
        # So we should encode, decode, then replace(u'?', u'_')
776
 
        relpath_tmp = relpath.encode(fenc, 'replace').decode(fenc, 'replace')
777
 
        relpath_tmp = relpath_tmp.replace(u'?', u'_')
778
 
        return osutils.pathjoin(self._root, prefix, relpath_tmp)
779
 
 
780
778
    def _write_file(self, file_id, tree, prefix, relpath, force_temp=False,
781
779
                    allow_write=False):
782
780
        if not force_temp and isinstance(tree, WorkingTree):
783
 
            full_path = tree.abspath(tree.id2path(file_id))
784
 
            if self._is_safepath(full_path):
785
 
                return full_path
786
 
 
787
 
        full_path = self._safe_filename(prefix, relpath)
 
781
            return tree.abspath(tree.id2path(file_id))
 
782
        
 
783
        full_path = osutils.pathjoin(self._root, prefix, relpath)
788
784
        if not force_temp and self._try_symlink_root(tree, prefix):
789
785
            return full_path
790
786
        parent_dir = osutils.dirname(full_path)
847
843
        """
848
844
        old_path = self.old_tree.id2path(file_id)
849
845
        new_path = self.new_tree.id2path(file_id)
850
 
        old_abs_path, new_abs_path = self._prepare_files(
851
 
                                            file_id, old_path, new_path,
852
 
                                            allow_write_new=True,
853
 
                                            force_temp=True)
854
 
        command = self._get_command(old_abs_path, new_abs_path)
 
846
        new_abs_path = self._prepare_files(file_id, old_path, new_path,
 
847
                                           allow_write_new=True,
 
848
                                           force_temp=True)[1]
 
849
        command = self._get_command(osutils.pathjoin('old', old_path),
 
850
                                    osutils.pathjoin('new', new_path))
855
851
        subprocess.call(command, cwd=self._root)
856
 
        new_file = open(new_abs_path, 'rb')
 
852
        new_file = open(new_abs_path, 'r')
857
853
        try:
858
854
            return new_file.read()
859
855
        finally:
909
905
        """Factory for producing a DiffTree.
910
906
 
911
907
        Designed to accept options used by show_diff_trees.
912
 
 
913
908
        :param old_tree: The tree to show as old in the comparison
914
909
        :param new_tree: The tree to show as new in the comparison
915
910
        :param to_file: File to write comparisons to
921
916
        :param using: Commandline to use to invoke an external diff tool
922
917
        """
923
918
        if using is not None:
924
 
            extra_factories = [DiffFromTool.make_from_diff_tree(using, external_diff_options)]
 
919
            extra_factories = [DiffFromTool.make_from_diff_tree(using)]
925
920
        else:
926
921
            extra_factories = []
927
922
        if external_diff_options:
928
923
            opts = external_diff_options.split()
929
 
            def diff_file(olab, olines, nlab, nlines, to_file, path_encoding=None):
930
 
                """:param path_encoding: not used but required
931
 
                        to match the signature of internal_diff.
932
 
                """
 
924
            def diff_file(olab, olines, nlab, nlines, to_file):
933
925
                external_diff(olab, olines, nlab, nlines, to_file, opts)
934
926
        else:
935
927
            diff_file = internal_diff