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
23
from .lazy_import import lazy_import
23
from bzrlib.lazy_import import lazy_import
24
24
lazy_import(globals(), """
38
from breezy.workingtree import WorkingTree
39
from breezy.i18n import gettext
41
from bzrlib.workingtree import WorkingTree
45
from .registry import (
44
from bzrlib.registry import (
48
from .trace import mutter, note, warning
49
from .tree import FileTimestampUnavailable
52
DEFAULT_CONTEXT_AMOUNT = 3
47
from bzrlib.symbol_versioning import (
51
from bzrlib.trace import mutter, note, warning
54
class AtTemplate(string.Template):
55
"""Templating class that uses @ instead of $."""
55
60
# TODO: Rather than building a changeset object, we should probably
79
84
# In the meantime we at least make sure the patch isn't
88
# Special workaround for Python2.3, where difflib fails if
89
# both sequences are empty.
90
if not oldlines and not newlines:
82
93
if allow_binary is False:
83
94
textfile.check_text_lines(oldlines)
84
95
textfile.check_text_lines(newlines)
86
97
if sequence_matcher is None:
87
98
sequence_matcher = patiencediff.PatienceSequenceMatcher
88
ud = unified_diff_bytes(
90
fromfile=old_label.encode(path_encoding, 'replace'),
91
tofile=new_label.encode(path_encoding, 'replace'),
92
n=context_lines, sequencematcher=sequence_matcher)
99
ud = patiencediff.unified_diff(oldlines, newlines,
100
fromfile=old_filename.encode(path_encoding, 'replace'),
101
tofile=new_filename.encode(path_encoding, 'replace'),
102
sequencematcher=sequence_matcher)
95
if len(ud) == 0: # Identical contents, nothing to do
105
if len(ud) == 0: # Identical contents, nothing to do
97
107
# work-around for difflib being too smart for its own good
98
108
# if /dev/null is "1,0", patch won't recognize it as /dev/null
100
ud[2] = ud[2].replace(b'-1,0', b'-0,0')
110
ud[2] = ud[2].replace('-1,0', '-0,0')
101
111
elif not newlines:
102
ud[2] = ud[2].replace(b'+1,0', b'+0,0')
112
ud[2] = ud[2].replace('+1,0', '+0,0')
105
115
to_file.write(line)
106
if not line.endswith(b'\n'):
107
to_file.write(b"\n\\ No newline at end of file\n")
111
def unified_diff_bytes(a, b, fromfile=b'', tofile=b'', fromfiledate=b'',
112
tofiledate=b'', n=3, lineterm=b'\n', sequencematcher=None):
114
Compare two sequences of lines; generate the delta as a unified diff.
116
Unified diffs are a compact way of showing line changes and a few
117
lines of context. The number of context lines is set by 'n' which
120
By default, the diff control lines (those with ---, +++, or @@) are
121
created with a trailing newline. This is helpful so that inputs
122
created from file.readlines() result in diffs that are suitable for
123
file.writelines() since both the inputs and outputs have trailing
126
For inputs that do not have trailing newlines, set the lineterm
127
argument to "" so that the output will be uniformly newline free.
129
The unidiff format normally has a header for filenames and modification
130
times. Any or all of these may be specified using strings for
131
'fromfile', 'tofile', 'fromfiledate', and 'tofiledate'. The modification
132
times are normally expressed in the format returned by time.ctime().
136
>>> for line in bytes_unified_diff(b'one two three four'.split(),
137
... b'zero one tree four'.split(), b'Original', b'Current',
138
... b'Sat Jan 26 23:30:50 1991', b'Fri Jun 06 10:20:52 2003',
141
--- Original Sat Jan 26 23:30:50 1991
142
+++ Current Fri Jun 06 10:20:52 2003
151
if sequencematcher is None:
152
sequencematcher = difflib.SequenceMatcher
155
fromfiledate = b'\t' + bytes(fromfiledate)
157
tofiledate = b'\t' + bytes(tofiledate)
160
for group in sequencematcher(None, a, b).get_grouped_opcodes(n):
162
yield b'--- %s%s%s' % (fromfile, fromfiledate, lineterm)
163
yield b'+++ %s%s%s' % (tofile, tofiledate, lineterm)
165
i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4]
166
yield b"@@ -%d,%d +%d,%d @@%s" % (i1 + 1, i2 - i1, j1 + 1, j2 - j1, lineterm)
167
for tag, i1, i2, j1, j2 in group:
169
for line in a[i1:i2]:
172
if tag == 'replace' or tag == 'delete':
173
for line in a[i1:i2]:
175
if tag == 'replace' or tag == 'insert':
176
for line in b[j1:j2]:
116
if not line.endswith('\n'):
117
to_file.write("\n\\ No newline at end of file\n")
180
121
def _spawn_external_diff(diffcmd, capture_errors=True):
181
"""Spawn the external diff process, and return the child handle.
122
"""Spawn the externall diff process, and return the child handle.
183
124
:param diffcmd: The command list to spawn
184
125
:param capture_errors: Capture stderr as well as setting LANG=C
217
# diff style options as of GNU diff v3.2
218
style_option_list = ['-c', '-C', '--context',
220
'-f', '--forward-ed',
224
'-u', '-U', '--unified',
225
'-y', '--side-by-side',
229
def default_style_unified(diff_opts):
230
"""Default to unified diff style if alternative not specified in diff_opts.
232
diff only allows one style to be specified; they don't override.
233
Note that some of these take optargs, and the optargs can be
234
directly appended to the options.
235
This is only an approximate parser; it doesn't properly understand
238
:param diff_opts: List of options for external (GNU) diff.
239
:return: List of options with default style=='unified'.
241
for s in style_option_list:
249
diff_opts.append('-u')
253
def external_diff(old_label, oldlines, new_label, newlines, to_file,
158
def external_diff(old_filename, oldlines, new_filename, newlines, to_file,
255
160
"""Display a diff by calling out to the external diff program."""
256
161
# make sure our own output is properly ordered before the diff
259
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='brz-diff-old-')
260
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='brz-diff-new-')
164
oldtmp_fd, old_abspath = tempfile.mkstemp(prefix='bzr-diff-old-')
165
newtmp_fd, new_abspath = tempfile.mkstemp(prefix='bzr-diff-new-')
261
166
oldtmpf = os.fdopen(oldtmp_fd, 'wb')
262
167
newtmpf = os.fdopen(newtmp_fd, 'wb')
280
185
if sys.platform == 'win32':
281
186
# Popen doesn't do the proper encoding for external commands
282
187
# Since we are dealing with an ANSI api, use mbcs encoding
283
old_label = old_label.encode('mbcs')
284
new_label = new_label.encode('mbcs')
188
old_filename = old_filename.encode('mbcs')
189
new_filename = new_filename.encode('mbcs')
285
190
diffcmd = ['diff',
286
'--label', old_label,
191
'--label', old_filename,
288
'--label', new_label,
193
'--label', new_filename,
293
diff_opts = default_style_unified(diff_opts)
198
# diff only allows one style to be specified; they don't override.
199
# note that some of these take optargs, and the optargs can be
200
# directly appended to the options.
201
# this is only an approximate parser; it doesn't properly understand
203
for s in ['-c', '-u', '-C', '-U',
208
'-y', '--side-by-side',
296
220
diffcmd.extend(diff_opts)
298
222
pipe = _spawn_external_diff(diffcmd, capture_errors=True)
299
out, err = pipe.communicate()
223
out,err = pipe.communicate()
300
224
rc = pipe.returncode
302
226
# internal_diff() adds a trailing newline, add one here for consistency
305
229
# 'diff' gives retcode == 2 for all sorts of errors
306
230
# one of those is 'Binary files differ'.
313
237
out, err = pipe.communicate()
315
239
# Write out the new i18n diff response
316
to_file.write(out + b'\n')
240
to_file.write(out+'\n')
317
241
if pipe.returncode != 2:
318
242
raise errors.BzrError(
319
'external diff failed with exit code 2'
320
' when run with LANG=C and LC_ALL=C,'
321
' but not when run natively: %r' % (diffcmd,))
243
'external diff failed with exit code 2'
244
' when run with LANG=C and LC_ALL=C,'
245
' but not when run natively: %r' % (diffcmd,))
323
first_line = lang_c_out.split(b'\n', 1)[0]
247
first_line = lang_c_out.split('\n', 1)[0]
324
248
# Starting with diffutils 2.8.4 the word "binary" was dropped.
325
m = re.match(b'^(binary )?files.*differ$', first_line, re.I)
249
m = re.match('^(binary )?files.*differ$', first_line, re.I)
327
251
raise errors.BzrError('external diff failed with exit code 2;'
328
252
' command: %r' % (diffcmd,))
341
265
msg = 'exit code %d' % rc
343
267
raise errors.BzrError('external diff failed with %s; command: %r'
347
272
oldtmpf.close() # and delete
351
# Warn in case the file couldn't be deleted (in case windows still
352
# holds the file open, but not if the files have already been
357
if e.errno not in (errno.ENOENT,):
358
warning('Failed to delete temporary file: %s %s', path, e)
274
# Clean up. Warn in case the files couldn't be deleted
275
# (in case windows still holds the file open, but not
276
# if the files have already been deleted)
278
os.remove(old_abspath)
280
if e.errno not in (errno.ENOENT,):
281
warning('Failed to delete temporary file: %s %s',
284
os.remove(new_abspath)
286
if e.errno not in (errno.ENOENT,):
287
warning('Failed to delete temporary file: %s %s',
291
@deprecated_function(deprecated_in((2, 2, 0)))
292
def get_trees_and_branches_to_diff(path_list, revision_specs, old_url, new_url,
294
"""Get the trees and specific files to diff given a list of paths.
296
This method works out the trees to be diff'ed and the files of
297
interest within those trees.
300
the list of arguments passed to the diff command
301
:param revision_specs:
302
Zero, one or two RevisionSpecs from the diff command line,
303
saying what revisions to compare.
305
The url of the old branch or tree. If None, the tree to use is
306
taken from the first path, if any, or the current working tree.
308
The url of the new branch or tree. If None, the tree to use is
309
taken from the first path, if any, or the current working tree.
311
if True and a view is set, apply the view or check that the paths
314
a tuple of (old_tree, new_tree, old_branch, new_branch,
315
specific_files, extra_trees) where extra_trees is a sequence of
316
additional trees to search in for file-ids. The trees and branches
319
op = cleanup.OperationWithCleanups(get_trees_and_branches_to_diff_locked)
320
return op.run_simple(path_list, revision_specs, old_url, new_url,
321
op.add_cleanup, apply_view=apply_view)
364
324
def get_trees_and_branches_to_diff_locked(
365
path_list, revision_specs, old_url, new_url, exit_stack, apply_view=True):
325
path_list, revision_specs, old_url, new_url, add_cleanup, apply_view=True):
366
326
"""Get the trees and specific files to diff given a list of paths.
368
328
This method works out the trees to be diff'ed and the files of
462
424
specific_files.extend(other_paths)
463
425
if len(specific_files) == 0:
464
426
specific_files = None
465
if (working_tree is not None and working_tree.supports_views() and
427
if (working_tree is not None and working_tree.supports_views()
467
429
view_files = working_tree.views.lookup_view()
469
431
specific_files = view_files
470
432
view_str = views.view_display_str(view_files)
471
note(gettext("*** Ignoring files outside view. View is %s") % view_str)
433
note("*** Ignoring files outside view. View is %s" % view_str)
473
435
# Get extra trees that ought to be searched for file-ids
474
436
extra_trees = None
475
437
if working_tree is not None and working_tree not in (old_tree, new_tree):
476
438
extra_trees = (working_tree,)
477
return (old_tree, new_tree, old_branch, new_branch,
478
specific_files, extra_trees)
439
return old_tree, new_tree, old_branch, new_branch, specific_files, extra_trees
481
442
def _get_tree_to_diff(spec, tree=None, branch=None, basis_is_default=True):
498
459
extra_trees=None,
499
460
path_encoding='utf8',
502
context=DEFAULT_CONTEXT_AMOUNT):
503
463
"""Show in text form the changes from one tree to another.
505
465
:param to_file: The output stream.
506
:param specific_files: Include only changes to these files - None for all
466
:param specific_files:Include only changes to these files - None for all
508
:param external_diff_options: If set, use an external GNU diff and pass
468
:param external_diff_options: If set, use an external GNU diff and pass
510
470
:param extra_trees: If set, more Trees to use for looking up file ids
511
:param path_encoding: If set, the path will be encoded as specified,
471
:param path_encoding: If set, the path will be encoded as specified,
512
472
otherwise is supposed to be utf8
513
473
:param format_cls: Formatter class (DiffTree subclass)
516
context = DEFAULT_CONTEXT_AMOUNT
517
475
if format_cls is None:
518
476
format_cls = DiffTree
519
with contextlib.ExitStack() as exit_stack:
520
exit_stack.enter_context(old_tree.lock_read())
521
479
if extra_trees is not None:
522
480
for tree in extra_trees:
523
exit_stack.enter_context(tree.lock_read())
524
exit_stack.enter_context(new_tree.lock_read())
525
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
527
external_diff_options,
528
old_label, new_label, using,
529
context_lines=context)
530
return differ.show_diff(specific_files, extra_trees)
533
def _patch_header_date(tree, path):
484
differ = format_cls.from_trees_options(old_tree, new_tree, to_file,
486
external_diff_options,
487
old_label, new_label, using)
488
return differ.show_diff(specific_files, extra_trees)
491
if extra_trees is not None:
492
for tree in extra_trees:
498
def _patch_header_date(tree, file_id, path):
534
499
"""Returns a timestamp suitable for use in a patch header."""
536
mtime = tree.get_file_mtime(path)
537
except FileTimestampUnavailable:
501
mtime = tree.get_file_mtime(file_id, path)
502
except errors.FileTimestampUnavailable:
539
504
return timestamp.format_patch_date(mtime)
542
507
def get_executable_change(old_is_x, new_is_x):
543
descr = {True: b"+x", False: b"-x", None: b"??"}
508
descr = { True:"+x", False:"-x", None:"??" }
544
509
if old_is_x != new_is_x:
545
return [b"%s to %s" % (descr[old_is_x], descr[new_is_x],)]
510
return ["%s to %s" % (descr[old_is_x], descr[new_is_x],)]
616
582
if None in (old_kind, new_kind):
617
583
return DiffPath.CANNOT_DIFF
618
result = DiffPath._diff_many(
619
self.differs, old_path, new_path, old_kind, None)
584
result = DiffPath._diff_many(self.differs, file_id, old_path,
585
new_path, old_kind, None)
620
586
if result is DiffPath.CANNOT_DIFF:
622
return DiffPath._diff_many(
623
self.differs, old_path, new_path, None, new_kind)
626
class DiffTreeReference(DiffPath):
628
def diff(self, old_path, new_path, old_kind, new_kind):
629
"""Perform comparison between two tree references. (dummy)
632
if 'tree-reference' not in (old_kind, new_kind):
633
return self.CANNOT_DIFF
634
if old_kind not in ('tree-reference', None):
635
return self.CANNOT_DIFF
636
if new_kind not in ('tree-reference', None):
637
return self.CANNOT_DIFF
588
return DiffPath._diff_many(self.differs, file_id, old_path, new_path,
641
592
class DiffDirectory(DiffPath):
643
def diff(self, old_path, new_path, old_kind, new_kind):
594
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
644
595
"""Perform comparison between two directories. (dummy)
700
649
EPOCH_DATE = '1970-01-01 00:00:00 +0000'
702
651
def __init__(self, old_tree, new_tree, to_file, path_encoding='utf-8',
703
old_label='', new_label='', text_differ=internal_diff,
704
context_lines=DEFAULT_CONTEXT_AMOUNT):
652
old_label='', new_label='', text_differ=internal_diff):
705
653
DiffPath.__init__(self, old_tree, new_tree, to_file, path_encoding)
706
654
self.text_differ = text_differ
707
655
self.old_label = old_label
708
656
self.new_label = new_label
709
657
self.path_encoding = path_encoding
710
self.context_lines = context_lines
712
def diff(self, old_path, new_path, old_kind, new_kind):
659
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
713
660
"""Compare two files in unified diff format
662
:param file_id: The file_id of the file to compare
715
663
:param old_path: Path of the file in the old tree
716
664
:param new_path: Path of the file in the new tree
717
665
:param old_kind: Old file-kind of the file
720
668
if 'file' not in (old_kind, new_kind):
721
669
return self.CANNOT_DIFF
670
from_file_id = to_file_id = file_id
722
671
if old_kind == 'file':
723
old_date = _patch_header_date(self.old_tree, old_path)
672
old_date = _patch_header_date(self.old_tree, file_id, old_path)
724
673
elif old_kind is None:
725
674
old_date = self.EPOCH_DATE
727
677
return self.CANNOT_DIFF
728
678
if new_kind == 'file':
729
new_date = _patch_header_date(self.new_tree, new_path)
679
new_date = _patch_header_date(self.new_tree, file_id, new_path)
730
680
elif new_kind is None:
731
681
new_date = self.EPOCH_DATE
733
684
return self.CANNOT_DIFF
734
from_label = '%s%s\t%s' % (
735
self.old_label, old_path or new_path, old_date)
736
to_label = '%s%s\t%s' % (
737
self.new_label, new_path or old_path, new_date)
738
return self.diff_text(old_path, new_path, from_label, to_label)
685
from_label = '%s%s\t%s' % (self.old_label, old_path, old_date)
686
to_label = '%s%s\t%s' % (self.new_label, new_path, new_date)
687
return self.diff_text(from_file_id, to_file_id, from_label, to_label,
740
def diff_text(self, from_path, to_path, from_label, to_label):
690
def diff_text(self, from_file_id, to_file_id, from_label, to_label,
691
from_path=None, to_path=None):
741
692
"""Diff the content of given files in two trees
743
:param from_path: The path in the from tree. If None,
694
:param from_file_id: The id of the file in the from tree. If None,
744
695
the file is not present in the from tree.
745
:param to_path: The path in the to tree. This may refer
746
to a different file from from_path. If None,
696
:param to_file_id: The id of the file in the to tree. This may refer
697
to a different file from from_file_id. If None,
747
698
the file is not present in the to tree.
699
:param from_path: The path in the from tree or None if unknown.
700
:param to_path: The path in the to tree or None if unknown.
749
def _get_text(tree, path):
753
return tree.get_file_lines(path)
754
except errors.NoSuchFile:
702
def _get_text(tree, file_id, path):
703
if file_id is not None:
704
return tree.get_file_lines(file_id, path)
757
from_text = _get_text(self.old_tree, from_path)
758
to_text = _get_text(self.new_tree, to_path)
708
from_text = _get_text(self.old_tree, from_file_id, from_path)
709
to_text = _get_text(self.new_tree, to_file_id, to_path)
759
710
self.text_differ(from_label, from_text, to_label, to_text,
760
self.to_file, path_encoding=self.path_encoding,
761
context_lines=self.context_lines)
711
self.to_file, path_encoding=self.path_encoding)
762
712
except errors.BinaryFile:
763
713
self.to_file.write(
764
("Binary files %s%s and %s%s differ\n" %
765
(self.old_label, from_path or to_path,
766
self.new_label, to_path or from_path)
767
).encode(self.path_encoding, 'replace'))
714
("Binary files %s and %s differ\n" %
715
(from_label, to_label)).encode(self.path_encoding,'replace'))
768
716
return self.CHANGED
795
746
def _get_command(self, old_path, new_path):
796
747
my_map = {'old_path': old_path, 'new_path': new_path}
797
command = [t.format(**my_map) for t in
798
self.command_template]
799
if command == self.command_template:
800
command += [old_path, new_path]
801
if sys.platform == 'win32': # Popen doesn't accept unicode on win32
804
if isinstance(c, str):
805
command_encoded.append(c.encode('mbcs'))
807
command_encoded.append(c)
808
return command_encoded
748
return [AtTemplate(t).substitute(my_map) for t in
749
self.command_template]
812
751
def _execute(self, old_path, new_path):
813
752
command = self._get_command(old_path, new_path)
815
754
proc = subprocess.Popen(command, stdout=subprocess.PIPE,
818
757
if e.errno == errno.ENOENT:
819
758
raise errors.ExecutableMissing(command[0])
822
761
self.to_file.write(proc.stdout.read())
824
762
return proc.wait()
826
764
def _try_symlink_root(self, tree, prefix):
827
if (getattr(tree, 'abspath', None) is None or
828
not osutils.host_os_dereferences_symlinks()):
765
if (getattr(tree, 'abspath', None) is None
766
or not osutils.host_os_dereferences_symlinks()):
831
769
os.symlink(tree.abspath(''), osutils.pathjoin(self._root, prefix))
833
771
if e.errno != errno.EEXIST:
839
"""Returns safe encoding for passing file path to diff tool"""
840
if sys.platform == 'win32':
843
# Don't fallback to 'utf-8' because subprocess may not be able to
844
# handle utf-8 correctly when locale is not utf-8.
845
return sys.getfilesystemencoding() or 'ascii'
847
def _is_safepath(self, path):
848
"""Return true if `path` may be able to pass to subprocess."""
851
return path == path.encode(fenc).decode(fenc)
855
def _safe_filename(self, prefix, relpath):
856
"""Replace unsafe character in `relpath` then join `self._root`,
857
`prefix` and `relpath`."""
859
# encoded_str.replace('?', '_') may break multibyte char.
860
# So we should encode, decode, then replace(u'?', u'_')
861
relpath_tmp = relpath.encode(fenc, 'replace').decode(fenc, 'replace')
862
relpath_tmp = relpath_tmp.replace(u'?', u'_')
863
return osutils.pathjoin(self._root, prefix, relpath_tmp)
865
def _write_file(self, relpath, tree, prefix, force_temp=False,
775
def _write_file(self, file_id, tree, prefix, relpath, force_temp=False,
866
776
allow_write=False):
867
777
if not force_temp and isinstance(tree, WorkingTree):
868
full_path = tree.abspath(relpath)
869
if self._is_safepath(full_path):
872
full_path = self._safe_filename(prefix, relpath)
778
return tree.abspath(tree.id2path(file_id))
780
full_path = osutils.pathjoin(self._root, prefix, relpath)
873
781
if not force_temp and self._try_symlink_root(tree, prefix):
875
783
parent_dir = osutils.dirname(full_path)
877
785
os.makedirs(parent_dir)
879
787
if e.errno != errno.EEXIST:
881
with tree.get_file(relpath) as source, \
882
open(full_path, 'wb') as target:
883
osutils.pumpfile(source, target)
885
mtime = tree.get_file_mtime(relpath)
886
except FileTimestampUnavailable:
789
source = tree.get_file(file_id, relpath)
791
target = open(full_path, 'wb')
793
osutils.pumpfile(source, target)
799
mtime = tree.get_file_mtime(file_id)
800
except errors.FileTimestampUnavailable:
889
803
os.utime(full_path, (mtime, mtime))
891
805
osutils.make_readonly(full_path)
894
def _prepare_files(self, old_path, new_path, force_temp=False,
808
def _prepare_files(self, file_id, old_path, new_path, force_temp=False,
895
809
allow_write_new=False):
896
old_disk_path = self._write_file(
897
old_path, self.old_tree, 'old', force_temp)
898
new_disk_path = self._write_file(
899
new_path, self.new_tree, 'new', force_temp,
900
allow_write=allow_write_new)
810
old_disk_path = self._write_file(file_id, self.old_tree, 'old',
811
old_path, force_temp)
812
new_disk_path = self._write_file(file_id, self.new_tree, 'new',
813
new_path, force_temp,
814
allow_write=allow_write_new)
901
815
return old_disk_path, new_disk_path
903
817
def finish(self):
905
819
osutils.rmtree(self._root)
907
821
if e.errno != errno.ENOENT:
908
822
mutter("The temporary directory \"%s\" was not "
909
"cleanly removed: %s." % (self._root, e))
823
"cleanly removed: %s." % (self._root, e))
911
def diff(self, old_path, new_path, old_kind, new_kind):
825
def diff(self, file_id, old_path, new_path, old_kind, new_kind):
912
826
if (old_kind, new_kind) != ('file', 'file'):
913
827
return DiffPath.CANNOT_DIFF
914
828
(old_disk_path, new_disk_path) = self._prepare_files(
829
file_id, old_path, new_path)
916
830
self._execute(old_disk_path, new_disk_path)
918
def edit_file(self, old_path, new_path):
832
def edit_file(self, file_id):
919
833
"""Use this tool to edit a file.
921
835
A temporary copy will be edited, and the new contents will be
838
:param file_id: The id of the file to edit.
924
839
:return: The new contents of the file.
926
old_abs_path, new_abs_path = self._prepare_files(
927
old_path, new_path, allow_write_new=True, force_temp=True)
928
command = self._get_command(old_abs_path, new_abs_path)
841
old_path = self.old_tree.id2path(file_id)
842
new_path = self.new_tree.id2path(file_id)
843
new_abs_path = self._prepare_files(file_id, old_path, new_path,
844
allow_write_new=True,
846
command = self._get_command(osutils.pathjoin('old', old_path),
847
osutils.pathjoin('new', new_path))
929
848
subprocess.call(command, cwd=self._root)
930
with open(new_abs_path, 'rb') as new_file:
849
new_file = open(new_abs_path, 'r')
931
851
return new_file.read()
934
856
class DiffTree(object):
1028
946
# TODO: Generation of pseudo-diffs for added/deleted files could
1029
947
# be usefully made into a much faster special case.
1030
948
iterator = self.new_tree.iter_changes(self.old_tree,
1031
specific_files=specific_files,
1032
extra_trees=extra_trees,
1033
require_versioned=True)
949
specific_files=specific_files,
950
extra_trees=extra_trees,
951
require_versioned=True)
1036
953
def changes_key(change):
1037
old_path, new_path = change.path
954
old_path, new_path = change[1]
1039
956
if path is None:
1043
959
def get_encoded_path(path):
1044
960
if path is not None:
1045
961
return path.encode(self.path_encoding, "replace")
1046
for change in sorted(iterator, key=changes_key):
962
for (file_id, paths, changed_content, versioned, parent, name, kind,
963
executable) in sorted(iterator, key=changes_key):
1047
964
# The root does not get diffed, and items with no known kind (that
1048
965
# is, missing) in both trees are skipped as well.
1049
if change.parent_id == (None, None) or change.kind == (None, None):
1051
if change.kind[0] == 'symlink' and not self.new_tree.supports_symlinks():
1053
'Ignoring "%s" as symlinks are not '
1054
'supported on this filesystem.' % (change.path[0],))
1056
oldpath, newpath = change.path
1057
oldpath_encoded = get_encoded_path(oldpath)
1058
newpath_encoded = get_encoded_path(newpath)
1059
old_present = (change.kind[0] is not None and change.versioned[0])
1060
new_present = (change.kind[1] is not None and change.versioned[1])
1061
executable = change.executable
1063
renamed = (change.parent_id[0], change.name[0]) != (change.parent_id[1], change.name[1])
966
if parent == (None, None) or kind == (None, None):
968
oldpath, newpath = paths
969
oldpath_encoded = get_encoded_path(paths[0])
970
newpath_encoded = get_encoded_path(paths[1])
971
old_present = (kind[0] is not None and versioned[0])
972
new_present = (kind[1] is not None and versioned[1])
973
renamed = (parent[0], name[0]) != (parent[1], name[1])
1065
975
properties_changed = []
1066
properties_changed.extend(
1067
get_executable_change(executable[0], executable[1]))
976
properties_changed.extend(get_executable_change(executable[0], executable[1]))
1069
978
if properties_changed:
1070
prop_str = b" (properties changed: %s)" % (
1071
b", ".join(properties_changed),)
979
prop_str = " (properties changed: %s)" % (", ".join(properties_changed),)
1075
983
if (old_present, new_present) == (True, False):
1076
self.to_file.write(b"=== removed %s '%s'\n" %
1077
(kind[0].encode('ascii'), oldpath_encoded))
984
self.to_file.write("=== removed %s '%s'\n" %
985
(kind[0], oldpath_encoded))
1078
987
elif (old_present, new_present) == (False, True):
1079
self.to_file.write(b"=== added %s '%s'\n" %
1080
(kind[1].encode('ascii'), newpath_encoded))
988
self.to_file.write("=== added %s '%s'\n" %
989
(kind[1], newpath_encoded))
1082
self.to_file.write(b"=== renamed %s '%s' => '%s'%s\n" %
1083
(kind[0].encode('ascii'), oldpath_encoded, newpath_encoded, prop_str))
992
self.to_file.write("=== renamed %s '%s' => '%s'%s\n" %
993
(kind[0], oldpath_encoded, newpath_encoded, prop_str))
1085
995
# if it was produced by iter_changes, it must be
1086
996
# modified *somehow*, either content or execute bit.
1087
self.to_file.write(b"=== modified %s '%s'%s\n" % (kind[0].encode('ascii'),
1088
newpath_encoded, prop_str))
1089
if change.changed_content:
1090
self._diff(oldpath, newpath, kind[0], kind[1])
997
self.to_file.write("=== modified %s '%s'%s\n" % (kind[0],
998
newpath_encoded, prop_str))
1000
self._diff(file_id, oldpath, newpath, kind[0], kind[1])
1091
1001
has_changes = 1
1093
1003
has_changes = 1
1094
1004
return has_changes
1096
def diff(self, old_path, new_path):
1006
def diff(self, file_id, old_path, new_path):
1097
1007
"""Perform a diff of a single file
1009
:param file_id: file-id of the file
1099
1010
:param old_path: The path of the file in the old tree
1100
1011
:param new_path: The path of the file in the new tree
1102
if old_path is None:
1014
old_kind = self.old_tree.kind(file_id)
1015
except (errors.NoSuchId, errors.NoSuchFile):
1103
1016
old_kind = None
1105
old_kind = self.old_tree.kind(old_path)
1106
if new_path is None:
1018
new_kind = self.new_tree.kind(file_id)
1019
except (errors.NoSuchId, errors.NoSuchFile):
1107
1020
new_kind = None
1109
new_kind = self.new_tree.kind(new_path)
1110
self._diff(old_path, new_path, old_kind, new_kind)
1112
def _diff(self, old_path, new_path, old_kind, new_kind):
1113
result = DiffPath._diff_many(
1114
self.differs, old_path, new_path, old_kind, new_kind)
1021
self._diff(file_id, old_path, new_path, old_kind, new_kind)
1024
def _diff(self, file_id, old_path, new_path, old_kind, new_kind):
1025
result = DiffPath._diff_many(self.differs, file_id, old_path,
1026
new_path, old_kind, new_kind)
1115
1027
if result is DiffPath.CANNOT_DIFF:
1116
1028
error_path = new_path
1117
1029
if error_path is None: