← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/changeset.py

  • Committer: Robert Collins
  • Date: 2005-10-09 23:12:35 UTC
  • Revision ID: robertc@robertcollins.net-20051009231235-93626e72cac71b78
clean up test dirs on make clean

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/changeset.py
45
45
    return newdict
46
46
 
47
47
       
48
 
class ChangeExecFlag(object):
 
48
class ChangeUnixPermissions(object):
49
49
    """This is two-way change, suitable for file modification, creation,
50
50
    deletion"""
51
 
    def __init__(self, old_exec_flag, new_exec_flag):
52
 
        self.old_exec_flag = old_exec_flag
53
 
        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
54
54
 
55
55
    def apply(self, filename, conflict_handler, reverse=False):
56
56
        if not reverse:
57
 
            from_exec_flag = self.old_exec_flag
58
 
            to_exec_flag = self.new_exec_flag
 
57
            from_mode = self.old_mode
 
58
            to_mode = self.new_mode
59
59
        else:
60
 
            from_exec_flag = self.new_exec_flag
61
 
            to_exec_flag = self.old_exec_flag
 
60
            from_mode = self.new_mode
 
61
            to_mode = self.old_mode
62
62
        try:
63
 
            current_exec_flag = bool(os.stat(filename).st_mode & 0111)
 
63
            current_mode = os.stat(filename).st_mode &0777
64
64
        except OSError, e:
65
65
            if e.errno == errno.ENOENT:
66
 
                if conflict_handler.missing_for_exec_flag(filename) == "skip":
 
66
                if conflict_handler.missing_for_chmod(filename) == "skip":
67
67
                    return
68
68
                else:
69
 
                    current_exec_flag = from_exec_flag
 
69
                    current_mode = from_mode
70
70
 
71
 
        if from_exec_flag is not None and current_exec_flag != from_exec_flag:
72
 
            if conflict_handler.wrong_old_exec_flag(filename,
73
 
                        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":
74
74
                return
75
75
 
76
 
        if to_exec_flag is not None:
77
 
            current_mode = os.stat(filename).st_mode
78
 
            if to_exec_flag:
79
 
                umask = os.umask(0)
80
 
                os.umask(umask)
81
 
                to_mode = current_mode | (0100 & ~umask)
82
 
                # Enable x-bit for others only if they can read it.
83
 
                if current_mode & 0004:
84
 
                    to_mode |= 0001 & ~umask
85
 
                if current_mode & 0040:
86
 
                    to_mode |= 0010 & ~umask
87
 
            else:
88
 
                to_mode = current_mode & ~0111
 
76
        if to_mode is not None:
89
77
            try:
90
78
                os.chmod(filename, to_mode)
91
79
            except IOError, e:
92
80
                if e.errno == errno.ENOENT:
93
 
                    conflict_handler.missing_for_exec_flag(filename)
 
81
                    conflict_handler.missing_for_chmod(filename)
94
82
 
95
83
    def __eq__(self, other):
96
 
        return (isinstance(other, ChangeExecFlag) and
97
 
                self.old_exec_flag == other.old_exec_flag and
98
 
                self.new_exec_flag == other.new_exec_flag)
 
84
        if not isinstance(other, ChangeUnixPermissions):
 
85
            return False
 
86
        elif self.old_mode != other.old_mode:
 
87
            return False
 
88
        elif self.new_mode != other.new_mode:
 
89
            return False
 
90
        else:
 
91
            return True
99
92
 
100
93
    def __ne__(self, other):
101
94
        return not (self == other)
923
916
        Exception.__init__(self, "Conflict applying changes to %s" % this_path)
924
917
        self.this_path = this_path
925
918
 
 
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
 
925
this: %o
 
926
base: %o
 
927
other: %o
 
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)
 
933
 
926
934
class WrongOldContents(Exception):
927
935
    def __init__(self, filename):
928
936
        msg = "Contents mismatch deleting %s" % filename
929
937
        self.filename = filename
930
938
        Exception.__init__(self, msg)
931
939
 
932
 
class WrongOldExecFlag(Exception):
933
 
    def __init__(self, filename, old_exec_flag, new_exec_flag):
934
 
        msg = "Executable flag missmatch on %s:\n" \
935
 
        "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)
936
944
        self.filename = filename
937
945
        Exception.__init__(self, msg)
938
946
 
956
964
        Exception.__init__(self, msg)
957
965
        self.filename = filename
958
966
 
959
 
class MissingForSetExec(Exception):
 
967
class MissingPermsFile(Exception):
960
968
    def __init__(self, filename):
961
969
        msg = "Attempt to change permissions on  %s, which does not exist" %\
962
970
            filename
1019
1027
        os.unlink(new_file)
1020
1028
        raise MergeConflict(this_path)
1021
1029
 
 
1030
    def permission_conflict(self, this_path, base_path, other_path):
 
1031
        raise MergePermissionConflict(this_path, base_path, other_path)
 
1032
 
1022
1033
    def wrong_old_contents(self, filename, expected_contents):
1023
1034
        raise WrongOldContents(filename)
1024
1035
 
1025
1036
    def rem_contents_conflict(self, filename, this_contents, base_contents):
1026
1037
        raise RemoveContentsConflict(filename)
1027
1038
 
1028
 
    def wrong_old_exec_flag(self, filename, old_exec_flag, new_exec_flag):
1029
 
        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)
1030
1041
 
1031
1042
    def rmdir_non_empty(self, filename):
1032
1043
        raise DeletingNonEmptyDirectory(filename)
1037
1048
    def patch_target_missing(self, filename, contents):
1038
1049
        raise PatchTargetMissing(filename)
1039
1050
 
1040
 
    def missing_for_exec_flag(self, filename):
1041
 
        raise MissingForExecFlag(filename)
 
1051
    def missing_for_chmod(self, filename):
 
1052
        raise MissingPermsFile(filename)
1042
1053
 
1043
1054
    def missing_for_rm(self, filename, change):
1044
1055
        raise MissingForRm(filename)
1254
1265
        return new_meta
1255
1266
    elif new_meta is None:
1256
1267
        return old_meta
1257
 
    elif (isinstance(old_meta, ChangeExecFlag) and
1258
 
          isinstance(new_meta, ChangeExecFlag)):
1259
 
        return ChangeExecFlag(old_meta.old_exec_flag, new_meta.new_exec_flag)
 
1268
    elif isinstance(old_meta, ChangeUnixPermissions) and \
 
1269
        isinstance(new_meta, ChangeUnixPermissions):
 
1270
        return ChangeUnixPermissions(old_meta.old_mode, new_meta.new_mode)
1260
1271
    else:
1261
1272
        return ApplySequence(old_meta, new_meta)
1262
1273
 
1277
1288
def generate_changeset(tree_a, tree_b, interesting_ids=None):
1278
1289
    return ChangesetGenerator(tree_a, tree_b, interesting_ids)()
1279
1290
 
1280
 
 
1281
1291
class ChangesetGenerator(object):
1282
1292
    def __init__(self, tree_a, tree_b, interesting_ids=None):
1283
1293
        object.__init__(self)
1381
1391
        stat_a = self.lstat(full_path_a)
1382
1392
        stat_b = self.lstat(full_path_b)
1383
1393
 
1384
 
        cs_entry.metadata_change = self.make_exec_flag_change(stat_a, stat_b)
 
1394
        cs_entry.metadata_change = self.make_mode_change(stat_a, stat_b)
1385
1395
 
1386
1396
        if id in self.tree_a and id in self.tree_b:
1387
1397
            a_sha1 = self.tree_a.get_file_sha1(id)
1395
1405
                                                             stat_b)
1396
1406
        return cs_entry
1397
1407
 
1398
 
    def make_exec_flag_change(self, stat_a, stat_b):
1399
 
        exec_flag_a = exec_flag_b = None
 
1408
    def make_mode_change(self, stat_a, stat_b):
 
1409
        mode_a = None
1400
1410
        if stat_a is not None and not stat.S_ISLNK(stat_a.st_mode):
1401
 
            exec_flag_a = bool(stat_a.st_mode & 0111)
 
1411
            mode_a = stat_a.st_mode & 0777
 
1412
        mode_b = None
1402
1413
        if stat_b is not None and not stat.S_ISLNK(stat_b.st_mode):
1403
 
            exec_flag_b = bool(stat_b.st_mode & 0111)
1404
 
        if exec_flag_a == exec_flag_b:
 
1414
            mode_b = stat_b.st_mode & 0777
 
1415
        if mode_a == mode_b:
1405
1416
            return None
1406
 
        return ChangeExecFlag(exec_flag_a, exec_flag_b)
 
1417
        return ChangeUnixPermissions(mode_a, mode_b)
1407
1418
 
1408
1419
    def make_contents_change(self, full_path_a, stat_a, full_path_b, stat_b):
1409
1420
        if stat_a is None and stat_b is None: