51
class ChangeExecFlag(object):
48
class ChangeUnixPermissions(object):
52
49
"""This is two-way change, suitable for file modification, creation,
54
def __init__(self, old_exec_flag, new_exec_flag):
55
self.old_exec_flag = old_exec_flag
56
self.new_exec_flag = new_exec_flag
51
def __init__(self, old_mode, new_mode):
52
self.old_mode = old_mode
53
self.new_mode = new_mode
58
55
def apply(self, filename, conflict_handler, reverse=False):
60
from_exec_flag = self.old_exec_flag
61
to_exec_flag = self.new_exec_flag
57
from_mode = self.old_mode
58
to_mode = self.new_mode
63
from_exec_flag = self.new_exec_flag
64
to_exec_flag = self.old_exec_flag
60
from_mode = self.new_mode
61
to_mode = self.old_mode
66
current_exec_flag = bool(os.stat(filename).st_mode & 0111)
63
current_mode = os.stat(filename).st_mode &0777
68
65
if e.errno == errno.ENOENT:
69
if conflict_handler.missing_for_exec_flag(filename) == "skip":
66
if conflict_handler.missing_for_chmod(filename) == "skip":
72
current_exec_flag = from_exec_flag
69
current_mode = from_mode
74
if from_exec_flag is not None and current_exec_flag != from_exec_flag:
75
if conflict_handler.wrong_old_exec_flag(filename,
76
from_exec_flag, current_exec_flag) != "continue":
71
if from_mode is not None and current_mode != from_mode:
72
if conflict_handler.wrong_old_perms(filename, from_mode,
73
current_mode) != "continue":
79
if to_exec_flag is not None:
80
current_mode = os.stat(filename).st_mode
84
to_mode = current_mode | (0100 & ~umask)
85
# Enable x-bit for others only if they can read it.
86
if current_mode & 0004:
87
to_mode |= 0001 & ~umask
88
if current_mode & 0040:
89
to_mode |= 0010 & ~umask
91
to_mode = current_mode & ~0111
76
if to_mode is not None:
93
78
os.chmod(filename, to_mode)
95
80
if e.errno == errno.ENOENT:
96
conflict_handler.missing_for_exec_flag(filename)
81
conflict_handler.missing_for_chmod(filename)
98
83
def __eq__(self, other):
99
return (isinstance(other, ChangeExecFlag) and
100
self.old_exec_flag == other.old_exec_flag and
101
self.new_exec_flag == other.new_exec_flag)
84
if not isinstance(other, ChangeUnixPermissions):
86
elif self.old_mode != other.old_mode:
88
elif self.new_mode != other.new_mode:
103
93
def __ne__(self, other):
104
94
return not (self == other)
239
class TreeFileCreate(object):
240
"""Create or delete a file (for use with ReplaceContents)"""
241
def __init__(self, tree, file_id):
244
:param contents: The contents of the file to write
248
self.file_id = file_id
251
return "TreeFileCreate(%s)" % self.file_id
253
def __eq__(self, other):
254
if not isinstance(other, TreeFileCreate):
256
return self.tree.get_file_sha1(self.file_id) == \
257
other.tree.get_file_sha1(other.file_id)
259
def __ne__(self, other):
260
return not (self == other)
262
def write_file(self, filename):
263
outfile = file(filename, "wb")
264
for line in self.tree.get_file(self.file_id):
267
def same_text(self, filename):
268
in_file = file(filename, "rb")
269
return sha_file(in_file) == self.tree.get_file_sha1(self.file_id)
271
def __call__(self, filename, conflict_handler, reverse):
272
"""Create or delete a file
274
:param filename: The name of the file to create
276
:param reverse: Delete the file instead of creating it
281
self.write_file(filename)
283
if e.errno == errno.ENOENT:
284
if conflict_handler.missing_parent(filename)=="continue":
285
self.write_file(filename)
291
if not self.same_text(filename):
292
direction = conflict_handler.wrong_old_contents(filename,
293
self.tree.get_file(self.file_id).read())
294
if direction != "continue":
298
if e.errno != errno.ENOENT:
300
if conflict_handler.missing_for_rm(filename, undo) == "skip":
305
226
def reversed(sequence):
306
227
max = len(sequence) - 1
307
228
for i in range(len(sequence)):
432
340
def __ne__(self, other):
433
341
return not (self == other)
435
def dump_file(self, temp_dir, name, tree):
436
out_path = os.path.join(temp_dir, name)
437
out_file = file(out_path, "wb")
438
in_file = tree.get_file(self.file_id)
443
343
def apply(self, filename, conflict_handler, reverse=False):
444
temp_dir = mkdtemp(prefix="bzr-")
446
new_file = filename+".new"
447
base_file = self.dump_file(temp_dir, "base", self.base)
448
other_file = self.dump_file(temp_dir, "other", self.other)
455
status = patch.diff3(new_file, filename, base, other)
457
os.chmod(new_file, os.stat(filename).st_mode)
458
rename(new_file, filename)
462
def get_lines(filename):
463
my_file = file(filename, "rb")
464
lines = my_file.readlines()
467
base_lines = get_lines(base)
468
other_lines = get_lines(other)
469
conflict_handler.merge_conflict(new_file, filename, base_lines,
344
new_file = filename+".new"
345
base_file = self.base.readonly_path(self.file_id)
346
other_file = self.other.readonly_path(self.file_id)
353
status = patch.diff3(new_file, filename, base, other)
355
os.chmod(new_file, os.stat(filename).st_mode)
356
rename(new_file, filename)
360
def get_lines(filename):
361
my_file = file(filename, "rb")
362
lines = my_file.readlines()
365
base_lines = get_lines(base)
366
other_lines = get_lines(other)
367
conflict_handler.merge_conflict(new_file, filename, base_lines,
1025
916
Exception.__init__(self, "Conflict applying changes to %s" % this_path)
1026
917
self.this_path = this_path
919
class MergePermissionConflict(Exception):
920
def __init__(self, this_path, base_path, other_path):
921
this_perms = os.stat(this_path).st_mode & 0755
922
base_perms = os.stat(base_path).st_mode & 0755
923
other_perms = os.stat(other_path).st_mode & 0755
924
msg = """Conflicting permission for %s
928
""" % (this_path, this_perms, base_perms, other_perms)
929
self.this_path = this_path
930
self.base_path = base_path
931
self.other_path = other_path
932
Exception.__init__(self, msg)
1028
934
class WrongOldContents(Exception):
1029
935
def __init__(self, filename):
1030
936
msg = "Contents mismatch deleting %s" % filename
1031
937
self.filename = filename
1032
938
Exception.__init__(self, msg)
1034
class WrongOldExecFlag(Exception):
1035
def __init__(self, filename, old_exec_flag, new_exec_flag):
1036
msg = "Executable flag missmatch on %s:\n" \
1037
"Expected %s, got %s." % (filename, old_exec_flag, new_exec_flag)
940
class WrongOldPermissions(Exception):
941
def __init__(self, filename, old_perms, new_perms):
942
msg = "Permission missmatch on %s:\n" \
943
"Expected 0%o, got 0%o." % (filename, old_perms, new_perms)
1038
944
self.filename = filename
1039
945
Exception.__init__(self, msg)
1083
989
msg = "Conflicting contents for new file %s" % (filename)
1084
990
Exception.__init__(self, msg)
1086
class WeaveMergeConflict(Exception):
1087
def __init__(self, filename):
1088
msg = "Conflicting contents for file %s" % (filename)
1089
Exception.__init__(self, msg)
1091
class ThreewayContentsConflict(Exception):
1092
def __init__(self, filename):
1093
msg = "Conflicting contents for file %s" % (filename)
1094
Exception.__init__(self, msg)
1097
993
class MissingForMerge(Exception):
1098
994
def __init__(self, filename):
1131
1027
os.unlink(new_file)
1132
1028
raise MergeConflict(this_path)
1030
def permission_conflict(self, this_path, base_path, other_path):
1031
raise MergePermissionConflict(this_path, base_path, other_path)
1134
1033
def wrong_old_contents(self, filename, expected_contents):
1135
1034
raise WrongOldContents(filename)
1137
1036
def rem_contents_conflict(self, filename, this_contents, base_contents):
1138
1037
raise RemoveContentsConflict(filename)
1140
def wrong_old_exec_flag(self, filename, old_exec_flag, new_exec_flag):
1141
raise WrongOldExecFlag(filename, old_exec_flag, new_exec_flag)
1039
def wrong_old_perms(self, filename, old_perms, new_perms):
1040
raise WrongOldPermissions(filename, old_perms, new_perms)
1143
1042
def rmdir_non_empty(self, filename):
1144
1043
raise DeletingNonEmptyDirectory(filename)
1149
1048
def patch_target_missing(self, filename, contents):
1150
1049
raise PatchTargetMissing(filename)
1152
def missing_for_exec_flag(self, filename):
1153
raise MissingForExecFlag(filename)
1051
def missing_for_chmod(self, filename):
1052
raise MissingPermsFile(filename)
1155
1054
def missing_for_rm(self, filename, change):
1156
1055
raise MissingForRm(filename)
1158
def missing_for_rename(self, filename, to_path):
1159
raise MissingForRename(filename, to_path)
1057
def missing_for_rename(self, filename):
1058
raise MissingForRename(filename)
1161
1060
def missing_for_merge(self, file_id, other_path):
1162
1061
raise MissingForMerge(other_path)
1164
1063
def new_contents_conflict(self, filename, other_contents):
1165
1064
raise NewContentsConflict(filename)
1167
def weave_merge_conflict(self, filename, weave, other_i, out_file):
1168
raise WeaveMergeConflict(filename)
1170
def threeway_contents_conflict(self, filename, this_contents,
1171
base_contents, other_contents):
1172
raise ThreewayContentsConflict(filename)
1174
1066
def finalize(self):
1389
class UnsupportedFiletype(Exception):
1390
def __init__(self, kind, full_path):
1391
msg = "The file \"%s\" is a %s, which is not a supported filetype." \
1281
class UnsuppportedFiletype(Exception):
1282
def __init__(self, full_path, stat_result):
1283
msg = "The file \"%s\" is not a supported filetype." % full_path
1393
1284
Exception.__init__(self, msg)
1394
1285
self.full_path = full_path
1286
self.stat_result = stat_result
1397
1288
def generate_changeset(tree_a, tree_b, interesting_ids=None):
1398
1289
return ChangesetGenerator(tree_a, tree_b, interesting_ids)()
1401
1291
class ChangesetGenerator(object):
1402
1292
def __init__(self, tree_a, tree_b, interesting_ids=None):
1403
1293
object.__init__(self)
1504
1399
if None not in (a_sha1, b_sha1) and a_sha1 == b_sha1:
1505
1400
return cs_entry
1507
cs_entry.contents_change = self.make_contents_change(id)
1402
cs_entry.contents_change = self.make_contents_change(full_path_a,
1508
1406
return cs_entry
1510
def make_exec_flag_change(self, file_id):
1511
exec_flag_a = exec_flag_b = None
1512
if file_id in self.tree_a and self.tree_a.kind(file_id) == "file":
1513
exec_flag_a = self.tree_a.is_executable(file_id)
1515
if file_id in self.tree_b and self.tree_b.kind(file_id) == "file":
1516
exec_flag_b = self.tree_b.is_executable(file_id)
1518
if exec_flag_a == exec_flag_b:
1520
return ChangeExecFlag(exec_flag_a, exec_flag_b)
1522
def make_contents_change(self, file_id):
1523
a_contents = get_contents(self.tree_a, file_id)
1524
b_contents = get_contents(self.tree_b, file_id)
1408
def make_mode_change(self, stat_a, stat_b):
1410
if stat_a is not None and not stat.S_ISLNK(stat_a.st_mode):
1411
mode_a = stat_a.st_mode & 0777
1413
if stat_b is not None and not stat.S_ISLNK(stat_b.st_mode):
1414
mode_b = stat_b.st_mode & 0777
1415
if mode_a == mode_b:
1417
return ChangeUnixPermissions(mode_a, mode_b)
1419
def make_contents_change(self, full_path_a, stat_a, full_path_b, stat_b):
1420
if stat_a is None and stat_b is None:
1422
if None not in (stat_a, stat_b) and stat.S_ISDIR(stat_a.st_mode) and\
1423
stat.S_ISDIR(stat_b.st_mode):
1425
if None not in (stat_a, stat_b) and stat.S_ISREG(stat_a.st_mode) and\
1426
stat.S_ISREG(stat_b.st_mode):
1427
if stat_a.st_ino == stat_b.st_ino and \
1428
stat_a.st_dev == stat_b.st_dev:
1431
a_contents = self.get_contents(stat_a, full_path_a)
1432
b_contents = self.get_contents(stat_b, full_path_b)
1525
1433
if a_contents == b_contents:
1527
1435
return ReplaceContents(a_contents, b_contents)
1437
def get_contents(self, stat_result, full_path):
1438
if stat_result is None:
1440
elif stat.S_ISREG(stat_result.st_mode):
1441
return FileCreate(file(full_path, "rb").read())
1442
elif stat.S_ISDIR(stat_result.st_mode):
1444
elif stat.S_ISLNK(stat_result.st_mode):
1445
return SymlinkCreate(os.readlink(full_path))
1447
raise UnsupportedFiletype(full_path, stat_result)
1530
def get_contents(tree, file_id):
1531
"""Return the appropriate contents to create a copy of file_id from tree"""
1532
if file_id not in tree:
1534
kind = tree.kind(file_id)
1536
return TreeFileCreate(tree, file_id)
1537
elif kind in ("directory", "root_directory"):
1539
elif kind == "symlink":
1540
return SymlinkCreate(tree.get_symlink_target(file_id))
1542
raise UnsupportedFiletype(kind, tree.id2path(file_id))
1449
def lstat(self, full_path):
1451
if full_path is not None:
1453
stat_result = os.lstat(full_path)
1455
if e.errno != errno.ENOENT:
1545
1460
def full_path(entry, tree):
1546
return os.path.join(tree.basedir, entry.path)
1461
return os.path.join(tree.root, entry.path)
1548
1463
def new_delete_entry(entry, tree, inventory, delete):
1549
1464
if entry.path == "":