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
17
from __future__ import absolute_import
24
from .lazy_import import lazy_import
23
from bzrlib.lazy_import import lazy_import
25
24
lazy_import(globals(), """
31
branch as _mod_branch,
40
from breezy.workingtree import WorkingTree
41
from breezy.i18n import gettext
43
from bzrlib.workingtree import WorkingTree
47
from .registry import (
46
from bzrlib.registry import (
50
from .sixish import text_type
51
from .trace import mutter, note, warning
52
from .tree import FileTimestampUnavailable
55
DEFAULT_CONTEXT_AMOUNT = 3
49
from bzrlib.symbol_versioning import (
53
from bzrlib.trace import mutter, note, warning
56
class AtTemplate(string.Template):
57
"""Templating class that uses @ instead of $."""
58
62
# TODO: Rather than building a changeset object, we should probably
82
86
# In the meantime we at least make sure the patch isn't
90
# Special workaround for Python2.3, where difflib fails if
91
# both sequences are empty.
92
if not oldlines and not newlines:
85
95
if allow_binary is False:
86
96
textfile.check_text_lines(oldlines)
87
97
textfile.check_text_lines(newlines)
89
99
if sequence_matcher is None:
90
100
sequence_matcher = patiencediff.PatienceSequenceMatcher
91
ud = unified_diff_bytes(
93
fromfile=old_label.encode(path_encoding, 'replace'),
94
tofile=new_label.encode(path_encoding, 'replace'),
95
n=context_lines, sequencematcher=sequence_matcher)
101
ud = patiencediff.unified_diff(oldlines, newlines,
102
fromfile=old_filename.encode(path_encoding, 'replace'),
103
tofile=new_filename.encode(path_encoding, 'replace'),
104
sequencematcher=sequence_matcher)
98
if len(ud) == 0: # Identical contents, nothing to do
107
if len(ud) == 0: # Identical contents, nothing to do
100
109
# work-around for difflib being too smart for its own good
101
110
# if /dev/null is "1,0", patch won't recognize it as /dev/null
103
ud[2] = ud[2].replace(b'-1,0', b'-0,0')
112
ud[2] = ud[2].replace('-1,0', '-0,0')
104
113
elif not newlines:
105
ud[2] = ud[2].replace(b'+1,0', b'+0,0')
114
ud[2] = ud[2].replace('+1,0', '+0,0')
108
117
to_file.write(line)
109
if not line.endswith(b'\n'):
110
to_file.write(b"\n\\ No newline at end of file\n")
114
def unified_diff_bytes(a, b, fromfile=b'', tofile=b'', fromfiledate=b'',
115
tofiledate=b'', n=3, lineterm=b'\n', sequencematcher=None):
117
Compare two sequences of lines; generate the delta as a unified diff.
119
Unified diffs are a compact way of showing line changes and a few
120
lines of context. The number of context lines is set by 'n' which
123
By default, the diff control lines (those with ---, +++, or @@) are
124
created with a trailing newline. This is helpful so that inputs
125
created from file.readlines() result in diffs that are suitable for
126
file.writelines() since both the inputs and outputs have trailing
129
For inputs that do not have trailing newlines, set the lineterm
130
argument to "" so that the output will be uniformly newline free.
132
The unidiff format normally has a header for filenames and modification
133
times. Any or all of these may be specified using strings for
134
'fromfile', 'tofile', 'fromfiledate', and 'tofiledate'. The modification
135
times are normally expressed in the format returned by time.ctime().
139
>>> for line in bytes_unified_diff(b'one two three four'.split(),
140
... b'zero one tree four'.split(), b'Original', b'Current',
141
... b'Sat Jan 26 23:30:50 1991', b'Fri Jun 06 10:20:52 2003',
144
--- Original Sat Jan 26 23:30:50 1991
145
+++ Current Fri Jun 06 10:20:52 2003
154
if sequencematcher is None:
155
sequencematcher = difflib.SequenceMatcher
158
fromfiledate = b'\t' + bytes(fromfiledate)
160
tofiledate = b'\t' + bytes(tofiledate)
163
for group in sequencematcher(None, a, b).get_grouped_opcodes(n):
165
yield b'--- %s%s%s' % (fromfile, fromfiledate, lineterm)
166
yield b'+++ %s%s%s' % (tofile, tofiledate, lineterm)
168
i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
169
yield b"@@ -%d,%d +%d,%d @@%s" % (i1 + 1, i2 - i1, j1 + 1, j2 - j1, lineterm)
170
for tag, i1, i2, j1, j2 in group:
172
for line in a[i1:i2]:
175
if tag == 'replace' or tag == 'delete':
176
for line in a[i1:i2]:
178
if tag == 'replace' or tag == 'insert':
179
for line in b[j1:j2]:
118
if not line.endswith('\n'):
119
to_file.write("\n\\ No newline at end of file\n")
183
123
def _spawn_external_diff(diffcmd, capture_errors=True):
184
"""Spawn the external diff process, and return the child handle.
124
"""Spawn the externall diff process, and return the child handle.
186
126
:param diffcmd: The command list to spawn
187
127
:param capture_errors: Capture stderr as well as setting LANG=C
220
# diff style options as of GNU diff v3.2
221
style_option_list = ['-c', '-C', '--context',
223
'-f', '--forward-ed',
227
'-u', '-U', '--unified',
228
'-y', '--side-by-side',
232
def default_style_unified(diff_opts):
233
"""Default to unified diff style if alternative not specified in diff_opts.
235
diff only allows one style to be specified; they don't override.
236
Note that some of these take optargs, and the optargs can be
237
directly appended to the options.
238
This is only an approximate parser; it doesn't properly understand
241
:param diff_opts: List of options for external (GNU) diff.
242
:return: List of options with default style=='unified'.
244
for s in style_option_list:
252
diff_opts.append('-u')
256
def external_diff(old_label, oldlines, new_label, newlines, to_file,
160
def external_diff(old_filename, oldlines, new_filename, newlines, to_file,
258
162
"""Display a diff by calling out to the external diff program."""
259
163
# make sure our own output is properly ordered before the diff
262
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='brz-diff-old-')
263
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='brz-diff-new-')
166
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='bzr-diff-old-')
167
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='bzr-diff-new-')
264
168
oldtmpf = os.fdopen(oldtmp_fd, 'wb')
265
169
newtmpf = os.fdopen(newtmp_fd, 'wb')
283
187
if sys.platform == 'win32':
284
188
# Popen doesn't do the proper encoding for external commands
285
189
# Since we are dealing with an ANSI api, use mbcs encoding
286
old_label = old_label.encode('mbcs')
287
new_label = new_label.encode('mbcs')
190
old_filename = old_filename.encode('mbcs')
191
new_filename = new_filename.encode('mbcs')
288
192
diffcmd = ['diff',
289
'--label', old_label,
193
'--label', old_filename,
291
'--label', new_label,
195
'--label', new_filename,
296
diff_opts = default_style_unified(diff_opts)
200
# diff only allows one style to be specified; they don't override.
201
# note that some of these take optargs, and the optargs can be
202
# directly appended to the options.
203
# this is only an approximate parser; it doesn't properly understand
205
for s in ['-c', '-u', '-C', '-U',
210
'-y', '--side-by-side',
299
222
diffcmd.extend(diff_opts)
301
224
pipe = _spawn_external_diff(diffcmd, capture_errors=True)
302
out, err = pipe.communicate()
225
out,err = pipe.communicate()
303
226
rc = pipe.returncode
305
228
# internal_diff() adds a trailing newline, add one here for consistency
308
231
# 'diff' gives retcode == 2 for all sorts of errors
309
232
# one of those is 'Binary files differ'.
316
239
out, err = pipe.communicate()
318
241
# Write out the new i18n diff response
319
to_file.write(out + b'\n')
242
to_file.write(out+'\n')
320
243
if pipe.returncode != 2:
321
244
raise errors.BzrError(
322
'external diff failed with exit code 2'
323
' when run with LANG=C and LC_ALL=C,'
324
' but not when run natively: %r' % (diffcmd,))
245
'external diff failed with exit code 2'
246
' when run with LANG=C and LC_ALL=C,'
247
' but not when run natively: %r' % (diffcmd,))
326
first_line = lang_c_out.split(b'\n', 1)[0]
249
first_line = lang_c_out.split('\n', 1)[0]
327
250
# Starting with diffutils 2.8.4 the word "binary" was dropped.
328
m = re.match(b'^(binary )?files.*differ$', first_line, re.I)
251
m = re.match('^(binary )?files.*differ$', first_line, re.I)
330
253
raise errors.BzrError('external diff failed with exit code 2;'
331
254
' command: %r' % (diffcmd,))
344
267
msg = 'exit code %d' % rc
346
269
raise errors.BzrError('external diff failed with %s; command: %r'
350
274
oldtmpf.close() # and delete
354
# Warn in case the file couldn't be deleted (in case windows still
355
# holds the file open, but not if the files have already been
360
if e.errno not in (errno.ENOENT,):
361
warning('Failed to delete temporary file: %s %s', path, e)
276
# Clean up. Warn in case the files couldn't be deleted
277
# (in case windows still holds the file open, but not
278
# if the files have already been deleted)
280
os.remove(old_abspath)
282
if e.errno not in (errno.ENOENT,):
283
warning('Failed to delete temporary file: %s %s',
286
os.remove(new_abspath)
288
if e.errno not in (errno.ENOENT,):
289
warning('Failed to delete temporary file: %s %s',
293
@deprecated_function(deprecated_in((2, 2, 0)))
294
def get_trees_and_branches_to_diff(path_list, revision_specs, old_url, new_url,
296
"""Get the trees and specific files to diff given a list of paths.
298
This method works out the trees to be diff'ed and the files of
299
interest within those trees.
302
the list of arguments passed to the diff command
303
:param revision_specs:
304
Zero, one or two RevisionSpecs from the diff command line,
305
saying what revisions to compare.
307
The url of the old branch or tree. If None, the tree to use is
308
taken from the first path, if any, or the current working tree.
310
The url of the new branch or tree. If None, the tree to use is
311
taken from the first path, if any, or the current working tree.
313
if True and a view is set, apply the view or check that the paths
316
a tuple of (old_tree, new_tree, old_branch, new_branch,
317
specific_files, extra_trees) where extra_trees is a sequence of
318
additional trees to search in for file-ids. The trees and branches
321
op = cleanup.OperationWithCleanups(get_trees_and_branches_to_diff_locked)
322
return op.run_simple(path_list, revision_specs, old_url, new_url,
323
op.add_cleanup, apply_view=apply_view)
367
326
def get_trees_and_branches_to_diff_locked(
368
path_list, revision_specs, old_url, new_url, exit_stack, apply_view=True):
327
path_list, revision_specs, old_url, new_url, add_cleanup, apply_view=True):
369
328
"""Get the trees and specific files to diff given a list of paths.
371
330
This method works out the trees to be diff'ed and the files of
447
408
new_url = default_location
448
409
if new_url != old_url:
449
410
working_tree, branch, relpath = \
450
controldir.ControlDir.open_containing_tree_or_branch(new_url)
411
bzrdir.BzrDir.open_containing_tree_or_branch(new_url)
451
412
lock_tree_or_branch(working_tree, branch)
452
413
if consider_relpath and relpath != '':
453
414
if working_tree is not None and apply_view:
454
415
views.check_path_in_view(working_tree, relpath)
455
416
specific_files.append(relpath)
456
417
new_tree = _get_tree_to_diff(new_revision_spec, working_tree, branch,
457
basis_is_default=working_tree is None)
418
basis_is_default=working_tree is None)
458
419
new_branch = branch
460
421
# Get the specific files (all files is None, no files is [])
461
422
if make_paths_wt_relative and working_tree is not None:
462
other_paths = working_tree.safe_relpath_files(
424
from bzrlib.builtins import safe_relpath_files
425
other_paths = safe_relpath_files(working_tree, other_paths,
464
426
apply_view=apply_view)
427
except errors.FileInWrongBranch:
428
raise errors.BzrCommandError("Files are in different branches")
465
429
specific_files.extend(other_paths)
466
430
if len(specific_files) == 0:
467
431
specific_files = None
468
if (working_tree is not None and working_tree.supports_views() and
432
if (working_tree is not None and working_tree.supports_views()
470
434
view_files = working_tree.views.lookup_view()
472
436
specific_files = view_files
473
437
view_str = views.view_display_str(view_files)
474
note(gettext("*** Ignoring files outside view. View is %s") % view_str)
438
note("*** Ignoring files outside view. View is %s" % view_str)
476
440
# Get extra trees that ought to be searched for file-ids
477
441
extra_trees = None
478
442
if working_tree is not None and working_tree not in (old_tree, new_tree):
479
443
extra_trees = (working_tree,)
480
return (old_tree, new_tree, old_branch, new_branch,
481
specific_files, extra_trees)
444
return old_tree, new_tree, old_branch, new_branch, specific_files, extra_trees
484
447
def _get_tree_to_diff(spec, tree=None, branch=None, basis_is_default=True):
501
464
extra_trees=None,
502
465
path_encoding='utf8',
505
context=DEFAULT_CONTEXT_AMOUNT):
506
468
"""Show in text form the changes from one tree to another.
508
470
:param to_file: The output stream.
509
:param specific_files: Include only changes to these files - None for all
471
:param specific_files:Include only changes to these files - None for all
511
:param external_diff_options: If set, use an external GNU diff and pass
473
:param external_diff_options: If set, use an external GNU diff and pass
513
475
:param extra_trees: If set, more Trees to use for looking up file ids
514
:param path_encoding: If set, the path will be encoded as specified,
476
:param path_encoding: If set, the path will be encoded as specified,
515
477
otherwise is supposed to be utf8
516
478
:param format_cls: Formatter class (DiffTree subclass)
519
context = DEFAULT_CONTEXT_AMOUNT
520
480
if format_cls is None:
521
481
format_cls = DiffTree
522
with cleanup.ExitStack() as exit_stack:
523
exit_stack.enter_context(old_tree.lock_read())
524
484
if extra_trees is not None:
525
485
for tree in extra_trees:
526
exit_stack.enter_context(tree.lock_read())
527
exit_stack.enter_context(new_tree.lock_read())
528
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
530
external_diff_options,
531
old_label, new_label, using,
532
context_lines=context)
533
return differ.show_diff(specific_files, extra_trees)
536
def _patch_header_date(tree, path):
489
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
491
external_diff_options,
492
old_label, new_label, using)
493
return differ.show_diff(specific_files, extra_trees)
496
if extra_trees is not None:
497
for tree in extra_trees:
503
def _patch_header_date(tree, file_id, path):
537
504
"""Returns a timestamp suitable for use in a patch header."""
539
mtime = tree.get_file_mtime(path)
540
except FileTimestampUnavailable:
506
mtime = tree.get_file_mtime(file_id, path)
507
except errors.FileTimestampUnavailable:
542
509
return timestamp.format_patch_date(mtime)
545
512
def get_executable_change(old_is_x, new_is_x):
546
descr = {True: b"+x", False: b"-x", None: b"??"}
513
descr = { True:"+x", False:"-x", None:"??" }
547
514
if old_is_x != new_is_x:
548
return [b"%s to %s" % (descr[old_is_x], descr[new_is_x],)]
515
return ["%s to %s" % (descr[old_is_x], descr[new_is_x],)]
619
587
if None in (old_kind, new_kind):
620
588
return DiffPath.CANNOT_DIFF
621
result = DiffPath._diff_many(
622
self.differs, old_path, new_path, old_kind, None)
589
result = DiffPath._diff_many(self.differs, file_id, old_path,
590
new_path, old_kind, None)
623
591
if result is DiffPath.CANNOT_DIFF:
625
return DiffPath._diff_many(
626
self.differs, old_path, new_path, None, new_kind)
629
class DiffTreeReference(DiffPath):
631
def diff(self, old_path, new_path, old_kind, new_kind):
632
"""Perform comparison between two tree references. (dummy)
635
if 'tree-reference' not in (old_kind, new_kind):
636
return self.CANNOT_DIFF
637
if old_kind not in ('tree-reference', None):
638
return self.CANNOT_DIFF
639
if new_kind not in ('tree-reference', None):
640
return self.CANNOT_DIFF
593
return DiffPath._diff_many(self.differs, file_id, old_path, new_path,
644
597
class DiffDirectory(DiffPath):
646
def diff(self, old_path, new_path, old_kind, new_kind):
599
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
647
600
"""Perform comparison between two directories. (dummy)
703
654
EPOCH_DATE = '1970-01-01 00:00:00 +0000'
705
656
def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8',
706
old_label='', new_label='', text_differ=internal_diff,
707
context_lines=DEFAULT_CONTEXT_AMOUNT):
657
old_label='', new_label='', text_differ=internal_diff):
708
658
DiffPath.__init__(self, old_tree, new_tree, to_file, path_encoding)
709
659
self.text_differ = text_differ
710
660
self.old_label = old_label
711
661
self.new_label = new_label
712
662
self.path_encoding = path_encoding
713
self.context_lines = context_lines
715
def diff(self, old_path, new_path, old_kind, new_kind):
664
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
716
665
"""Compare two files in unified diff format
667
:param file_id: The file_id of the file to compare
718
668
:param old_path: Path of the file in the old tree
719
669
:param new_path: Path of the file in the new tree
720
670
:param old_kind: Old file-kind of the file
723
673
if 'file' not in (old_kind, new_kind):
724
674
return self.CANNOT_DIFF
675
from_file_id = to_file_id = file_id
725
676
if old_kind == 'file':
726
old_date = _patch_header_date(self.old_tree, old_path)
677
old_date = _patch_header_date(self.old_tree, file_id, old_path)
727
678
elif old_kind is None:
728
679
old_date = self.EPOCH_DATE
730
682
return self.CANNOT_DIFF
731
683
if new_kind == 'file':
732
new_date = _patch_header_date(self.new_tree, new_path)
684
new_date = _patch_header_date(self.new_tree, file_id, new_path)
733
685
elif new_kind is None:
734
686
new_date = self.EPOCH_DATE
736
689
return self.CANNOT_DIFF
737
from_label = '%s%s\t%s' % (self.old_label, old_path,
739
to_label = '%s%s\t%s' % (self.new_label, new_path,
741
return self.diff_text(old_path, new_path, from_label, to_label)
690
from_label = '%s%s\t%s' % (self.old_label, old_path, old_date)
691
to_label = '%s%s\t%s' % (self.new_label, new_path, new_date)
692
return self.diff_text(from_file_id, to_file_id, from_label, to_label,
743
def diff_text(self, from_path, to_path, from_label, to_label):
695
def diff_text(self, from_file_id, to_file_id, from_label, to_label,
696
from_path=None, to_path=None):
744
697
"""Diff the content of given files in two trees
746
:param from_path: The path in the from tree. If None,
699
:param from_file_id: The id of the file in the from tree. If None,
747
700
the file is not present in the from tree.
748
:param to_path: The path in the to tree. This may refer
749
to a different file from from_path. If None,
701
:param to_file_id: The id of the file in the to tree. This may refer
702
to a different file from from_file_id. If None,
750
703
the file is not present in the to tree.
704
:param from_path: The path in the from tree or None if unknown.
705
:param to_path: The path in the to tree or None if unknown.
752
def _get_text(tree, path):
756
return tree.get_file_lines(path)
757
except errors.NoSuchFile:
707
def _get_text(tree, file_id, path):
708
if file_id is not None:
709
return tree.get_file_lines(file_id, path)
760
from_text = _get_text(self.old_tree, from_path)
761
to_text = _get_text(self.new_tree, to_path)
713
from_text = _get_text(self.old_tree, from_file_id, from_path)
714
to_text = _get_text(self.new_tree, to_file_id, to_path)
762
715
self.text_differ(from_label, from_text, to_label, to_text,
763
self.to_file, path_encoding=self.path_encoding,
764
context_lines=self.context_lines)
716
self.to_file, path_encoding=self.path_encoding)
765
717
except errors.BinaryFile:
766
718
self.to_file.write(
767
("Binary files %s%s and %s%s differ\n" %
768
(self.old_label, from_path, self.new_label, to_path)).encode(self.path_encoding, 'replace'))
719
("Binary files %s and %s differ\n" %
720
(from_label, to_label)).encode(self.path_encoding,'replace'))
769
721
return self.CHANGED
775
727
path_encoding='utf-8'):
776
728
DiffPath.__init__(self, old_tree, new_tree, to_file, path_encoding)
777
729
self.command_template = command_template
778
self._root = osutils.mkdtemp(prefix='brz-diff-')
730
self._root = osutils.mkdtemp(prefix='bzr-diff-')
781
def from_string(klass, command_template, old_tree, new_tree, to_file,
733
def from_string(klass, command_string, old_tree, new_tree, to_file,
782
734
path_encoding='utf-8'):
735
command_template = cmdline.split(command_string)
736
if '@' not in command_string:
737
command_template.extend(['@old_path', '@new_path'])
783
738
return klass(command_template, old_tree, new_tree, to_file,
787
def make_from_diff_tree(klass, command_string, external_diff_options=None):
742
def make_from_diff_tree(klass, command_string):
788
743
def from_diff_tree(diff_tree):
789
full_command_string = [command_string]
790
if external_diff_options is not None:
791
full_command_string += ' ' + external_diff_options
792
return klass.from_string(full_command_string, diff_tree.old_tree,
744
return klass.from_string(command_string, diff_tree.old_tree,
793
745
diff_tree.new_tree, diff_tree.to_file)
794
746
return from_diff_tree
796
748
def _get_command(self, old_path, new_path):
797
749
my_map = {'old_path': old_path, 'new_path': new_path}
798
command = [t.format(**my_map) for t in
799
self.command_template]
800
if command == self.command_template:
801
command += [old_path, new_path]
802
if sys.platform == 'win32': # Popen doesn't accept unicode on win32
805
if isinstance(c, text_type):
806
command_encoded.append(c.encode('mbcs'))
808
command_encoded.append(c)
809
return command_encoded
750
return [AtTemplate(t).substitute(my_map) for t in
751
self.command_template]
813
753
def _execute(self, old_path, new_path):
814
754
command = self._get_command(old_path, new_path)
816
756
proc = subprocess.Popen(command, stdout=subprocess.PIPE,
819
759
if e.errno == errno.ENOENT:
820
760
raise errors.ExecutableMissing(command[0])
823
763
self.to_file.write(proc.stdout.read())
825
764
return proc.wait()
827
766
def _try_symlink_root(self, tree, prefix):
828
if (getattr(tree, 'abspath', None) is None or
829
not osutils.host_os_dereferences_symlinks()):
767
if (getattr(tree, 'abspath', None) is None
768
or not osutils.host_os_dereferences_symlinks()):
832
771
os.symlink(tree.abspath(''), osutils.pathjoin(self._root, prefix))
834
773
if e.errno != errno.EEXIST:
840
"""Returns safe encoding for passing file path to diff tool"""
841
if sys.platform == 'win32':
844
# Don't fallback to 'utf-8' because subprocess may not be able to
845
# handle utf-8 correctly when locale is not utf-8.
846
return sys.getfilesystemencoding() or 'ascii'
848
def _is_safepath(self, path):
849
"""Return true if `path` may be able to pass to subprocess."""
852
return path == path.encode(fenc).decode(fenc)
856
def _safe_filename(self, prefix, relpath):
857
"""Replace unsafe character in `relpath` then join `self._root`,
858
`prefix` and `relpath`."""
860
# encoded_str.replace('?', '_') may break multibyte char.
861
# So we should encode, decode, then replace(u'?', u'_')
862
relpath_tmp = relpath.encode(fenc, 'replace').decode(fenc, 'replace')
863
relpath_tmp = relpath_tmp.replace(u'?', u'_')
864
return osutils.pathjoin(self._root, prefix, relpath_tmp)
866
def _write_file(self, relpath, tree, prefix, force_temp=False,
777
def _write_file(self, file_id, tree, prefix, relpath, force_temp=False,
867
778
allow_write=False):
868
779
if not force_temp and isinstance(tree, WorkingTree):
869
full_path = tree.abspath(relpath)
870
if self._is_safepath(full_path):
873
full_path = self._safe_filename(prefix, relpath)
780
return tree.abspath(tree.id2path(file_id))
782
full_path = osutils.pathjoin(self._root, prefix, relpath)
874
783
if not force_temp and self._try_symlink_root(tree, prefix):
876
785
parent_dir = osutils.dirname(full_path)
878
787
os.makedirs(parent_dir)
880
789
if e.errno != errno.EEXIST:
882
with tree.get_file(relpath) as source, \
883
open(full_path, 'wb') as target:
884
osutils.pumpfile(source, target)
886
mtime = tree.get_file_mtime(relpath)
887
except FileTimestampUnavailable:
791
source = tree.get_file(file_id, relpath)
793
target = open(full_path, 'wb')
795
osutils.pumpfile(source, target)
801
mtime = tree.get_file_mtime(file_id)
802
except errors.FileTimestampUnavailable:
890
805
os.utime(full_path, (mtime, mtime))
892
807
osutils.make_readonly(full_path)
895
def _prepare_files(self, old_path, new_path, force_temp=False,
810
def _prepare_files(self, file_id, old_path, new_path, force_temp=False,
896
811
allow_write_new=False):
897
old_disk_path = self._write_file(
898
old_path, self.old_tree, 'old', force_temp)
899
new_disk_path = self._write_file(
900
new_path, self.new_tree, 'new', force_temp,
901
allow_write=allow_write_new)
812
old_disk_path = self._write_file(file_id, self.old_tree, 'old',
813
old_path, force_temp)
814
new_disk_path = self._write_file(file_id, self.new_tree, 'new',
815
new_path, force_temp,
816
allow_write=allow_write_new)
902
817
return old_disk_path, new_disk_path
904
819
def finish(self):
906
821
osutils.rmtree(self._root)
908
823
if e.errno != errno.ENOENT:
909
824
mutter("The temporary directory \"%s\" was not "
910
"cleanly removed: %s." % (self._root, e))
825
"cleanly removed: %s." % (self._root, e))
912
def diff(self, old_path, new_path, old_kind, new_kind):
827
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
913
828
if (old_kind, new_kind) != ('file', 'file'):
914
829
return DiffPath.CANNOT_DIFF
915
830
(old_disk_path, new_disk_path) = self._prepare_files(
831
file_id, old_path, new_path)
917
832
self._execute(old_disk_path, new_disk_path)
919
def edit_file(self, old_path, new_path):
834
def edit_file(self, file_id):
920
835
"""Use this tool to edit a file.
922
837
A temporary copy will be edited, and the new contents will be
840
:param file_id: The id of the file to edit.
925
841
:return: The new contents of the file.
927
old_abs_path, new_abs_path = self._prepare_files(
928
old_path, new_path, allow_write_new=True, force_temp=True)
929
command = self._get_command(old_abs_path, new_abs_path)
843
old_path = self.old_tree.id2path(file_id)
844
new_path = self.new_tree.id2path(file_id)
845
new_abs_path = self._prepare_files(file_id, old_path, new_path,
846
allow_write_new=True,
848
command = self._get_command(osutils.pathjoin('old', old_path),
849
osutils.pathjoin('new', new_path))
930
850
subprocess.call(command, cwd=self._root)
931
with open(new_abs_path, 'rb') as new_file:
851
new_file = open(new_abs_path, 'r')
932
853
return new_file.read()
935
858
class DiffTree(object):
1029
948
# TODO: Generation of pseudo-diffs for added/deleted files could
1030
949
# be usefully made into a much faster special case.
1031
950
iterator = self.new_tree.iter_changes(self.old_tree,
1032
specific_files=specific_files,
1033
extra_trees=extra_trees,
1034
require_versioned=True)
951
specific_files=specific_files,
952
extra_trees=extra_trees,
953
require_versioned=True)
1037
955
def changes_key(change):
1038
old_path, new_path = change.path
956
old_path, new_path = change[1]
1040
958
if path is None:
1044
961
def get_encoded_path(path):
1045
962
if path is not None:
1046
963
return path.encode(self.path_encoding, "replace")
1047
for change in sorted(iterator, key=changes_key):
964
for (file_id, paths, changed_content, versioned, parent, name, kind,
965
executable) in sorted(iterator, key=changes_key):
1048
966
# The root does not get diffed, and items with no known kind (that
1049
967
# is, missing) in both trees are skipped as well.
1050
if change.parent_id == (None, None) or change.kind == (None, None):
1052
if change.kind[0] == 'symlink' and not self.new_tree.supports_symlinks():
1054
'Ignoring "%s" as symlinks are not '
1055
'supported on this filesystem.' % (change.path[0],))
1057
oldpath, newpath = change.path
1058
oldpath_encoded = get_encoded_path(change.path[0])
1059
newpath_encoded = get_encoded_path(change.path[1])
1060
old_present = (change.kind[0] is not None and change.versioned[0])
1061
new_present = (change.kind[1] is not None and change.versioned[1])
1062
executable = change.executable
1064
renamed = (change.parent_id[0], change.name[0]) != (change.parent_id[1], change.name[1])
968
if parent == (None, None) or kind == (None, None):
970
oldpath, newpath = paths
971
oldpath_encoded = get_encoded_path(paths[0])
972
newpath_encoded = get_encoded_path(paths[1])
973
old_present = (kind[0] is not None and versioned[0])
974
new_present = (kind[1] is not None and versioned[1])
975
renamed = (parent[0], name[0]) != (parent[1], name[1])
1066
977
properties_changed = []
1067
properties_changed.extend(
1068
get_executable_change(executable[0], executable[1]))
978
properties_changed.extend(get_executable_change(executable[0], executable[1]))
1070
980
if properties_changed:
1071
prop_str = b" (properties changed: %s)" % (
1072
b", ".join(properties_changed),)
981
prop_str = " (properties changed: %s)" % (", ".join(properties_changed),)
1076
985
if (old_present, new_present) == (True, False):
1077
self.to_file.write(b"=== removed %s '%s'\n" %
1078
(kind[0].encode('ascii'), oldpath_encoded))
986
self.to_file.write("=== removed %s '%s'\n" %
987
(kind[0], oldpath_encoded))
1079
988
newpath = oldpath
1080
989
elif (old_present, new_present) == (False, True):
1081
self.to_file.write(b"=== added %s '%s'\n" %
1082
(kind[1].encode('ascii'), newpath_encoded))
990
self.to_file.write("=== added %s '%s'\n" %
991
(kind[1], newpath_encoded))
1083
992
oldpath = newpath
1085
self.to_file.write(b"=== renamed %s '%s' => '%s'%s\n" %
1086
(kind[0].encode('ascii'), oldpath_encoded, newpath_encoded, prop_str))
994
self.to_file.write("=== renamed %s '%s' => '%s'%s\n" %
995
(kind[0], oldpath_encoded, newpath_encoded, prop_str))
1088
997
# if it was produced by iter_changes, it must be
1089
998
# modified *somehow*, either content or execute bit.
1090
self.to_file.write(b"=== modified %s '%s'%s\n" % (kind[0].encode('ascii'),
1091
newpath_encoded, prop_str))
1092
if change.changed_content:
1093
self._diff(oldpath, newpath, kind[0], kind[1])
999
self.to_file.write("=== modified %s '%s'%s\n" % (kind[0],
1000
newpath_encoded, prop_str))
1002
self._diff(file_id, oldpath, newpath, kind[0], kind[1])
1094
1003
has_changes = 1
1096
1005
has_changes = 1
1097
1006
return has_changes
1099
def diff(self, old_path, new_path):
1008
def diff(self, file_id, old_path, new_path):
1100
1009
"""Perform a diff of a single file
1011
:param file_id: file-id of the file
1102
1012
:param old_path: The path of the file in the old tree
1103
1013
:param new_path: The path of the file in the new tree
1105
if old_path is None:
1016
old_kind = self.old_tree.kind(file_id)
1017
except (errors.NoSuchId, errors.NoSuchFile):
1106
1018
old_kind = None
1108
old_kind = self.old_tree.kind(old_path)
1109
if new_path is None:
1020
new_kind = self.new_tree.kind(file_id)
1021
except (errors.NoSuchId, errors.NoSuchFile):
1110
1022
new_kind = None
1112
new_kind = self.new_tree.kind(new_path)
1113
self._diff(old_path, new_path, old_kind, new_kind)
1115
def _diff(self, old_path, new_path, old_kind, new_kind):
1116
result = DiffPath._diff_many(
1117
self.differs, old_path, new_path, old_kind, new_kind)
1023
self._diff(file_id, old_path, new_path, old_kind, new_kind)
1026
def _diff(self, file_id, old_path, new_path, old_kind, new_kind):
1027
result = DiffPath._diff_many(self.differs, file_id, old_path,
1028
new_path, old_kind, new_kind)
1118
1029
if result is DiffPath.CANNOT_DIFF:
1119
1030
error_path = new_path
1120
1031
if error_path is None: