2
# -*- coding: UTF-8 -*-
 
 
4
# This program is free software; you can redistribute it and/or modify
 
 
5
# it under the terms of the GNU General Public License as published by
 
 
6
# the Free Software Foundation; either version 2 of the License, or
 
 
7
# (at your option) any later version.
 
 
9
# This program is distributed in the hope that it will be useful,
 
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
 
12
# GNU General Public License for more details.
 
 
14
# You should have received a copy of the GNU General Public License
 
 
15
# along with this program; if not, write to the Free Software
 
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 
18
from bzrlib.trace import mutter
 
 
20
class TreeDelta(object):
 
 
21
    """Describes changes from one tree to another.
 
 
30
        (oldpath, newpath, id, kind, text_modified)
 
 
36
    Each id is listed only once.
 
 
38
    Files that are both modified and renamed are listed only in
 
 
39
    renamed, with the text_modified flag true.
 
 
41
    Files are only considered renamed if their name has changed or
 
 
42
    their parent directory has changed.  Renaming a directory
 
 
43
    does not count as renaming all its contents.
 
 
45
    The lists are normally sorted when the delta is created.
 
 
54
    def __eq__(self, other):
 
 
55
        if not isinstance(other, TreeDelta):
 
 
57
        return self.added == other.added \
 
 
58
               and self.removed == other.removed \
 
 
59
               and self.renamed == other.renamed \
 
 
60
               and self.modified == other.modified \
 
 
61
               and self.unchanged == other.unchanged
 
 
63
    def __ne__(self, other):
 
 
64
        return not (self == other)
 
 
67
        return "TreeDelta(added=%r, removed=%r, renamed=%r, modified=%r," \
 
 
68
            " unchanged=%r)" % (self.added, self.removed, self.renamed,
 
 
69
            self.modified, self.unchanged)
 
 
71
    def has_changed(self):
 
 
72
        changes = len(self.added) + len(self.removed) + len(self.renamed)
 
 
73
        changes += len(self.modified) 
 
 
76
    def touches_file_id(self, file_id):
 
 
77
        """Return True if file_id is modified by this delta."""
 
 
78
        for l in self.added, self.removed, self.modified:
 
 
82
        for v in self.renamed:
 
 
88
    def show(self, to_file, show_ids=False, show_unchanged=False):
 
 
90
            for path, fid, kind in files:
 
 
91
                if kind == 'directory':
 
 
93
                elif kind == 'symlink':
 
 
97
                    print >>to_file, '  %-30s %s' % (path, fid)
 
 
99
                    print >>to_file, ' ', path
 
 
102
            print >>to_file, 'removed:'
 
 
103
            show_list(self.removed)
 
 
106
            print >>to_file, 'added:'
 
 
107
            show_list(self.added)
 
 
110
            print >>to_file, 'renamed:'
 
 
111
            for oldpath, newpath, fid, kind, text_modified in self.renamed:
 
 
113
                    print >>to_file, '  %s => %s %s' % (oldpath, newpath, fid)
 
 
115
                    print >>to_file, '  %s => %s' % (oldpath, newpath)
 
 
118
            print >>to_file, 'modified:'
 
 
119
            show_list(self.modified)
 
 
121
        if show_unchanged and self.unchanged:
 
 
122
            print >>to_file, 'unchanged:'
 
 
123
            show_list(self.unchanged)
 
 
127
def compare_trees(old_tree, new_tree, want_unchanged=False, specific_files=None):
 
 
128
    """Describe changes from one tree to another.
 
 
130
    Returns a TreeDelta with details of added, modified, renamed, and
 
 
133
    The root entry is specifically exempt.
 
 
135
    This only considers versioned files.
 
 
138
        If true, also list files unchanged from one version to
 
 
142
        If true, only check for changes to specified names or
 
 
146
    from osutils import is_inside_any
 
 
148
    old_inv = old_tree.inventory
 
 
149
    new_inv = new_tree.inventory
 
 
151
    mutter('start compare_trees')
 
 
153
    # TODO: match for specific files can be rather smarter by finding
 
 
154
    # the IDs of those files up front and then considering only that.
 
 
156
    for file_id in old_tree:
 
 
157
        if file_id in new_tree:
 
 
158
            old_ie = old_inv[file_id]
 
 
159
            new_ie = new_inv[file_id]
 
 
162
            assert kind == new_ie.kind
 
 
164
            assert kind in ('file', 'directory', 'symlink', 'root_directory'), \
 
 
165
                   'invalid file kind %r' % kind
 
 
167
            if kind == 'root_directory':
 
 
171
                if (not is_inside_any(specific_files, old_inv.id2path(file_id)) 
 
 
172
                    and not is_inside_any(specific_files, new_inv.id2path(file_id))):
 
 
176
                old_sha1 = old_tree.get_file_sha1(file_id)
 
 
177
                new_sha1 = new_tree.get_file_sha1(file_id)
 
 
178
                text_modified = (old_sha1 != new_sha1)
 
 
180
                ## mutter("no text to check for %r %r" % (file_id, kind))
 
 
181
                text_modified = False
 
 
183
            # TODO: Can possibly avoid calculating path strings if the
 
 
184
            # two files are unchanged and their names and parents are
 
 
185
            # the same and the parents are unchanged all the way up.
 
 
186
            # May not be worthwhile.
 
 
188
            if (old_ie.name != new_ie.name
 
 
189
                or old_ie.parent_id != new_ie.parent_id):
 
 
190
                delta.renamed.append((old_inv.id2path(file_id),
 
 
191
                                      new_inv.id2path(file_id),
 
 
195
                delta.modified.append((new_inv.id2path(file_id), file_id, kind))
 
 
197
                delta.unchanged.append((new_inv.id2path(file_id), file_id, kind))
 
 
199
            kind = old_inv.get_file_kind(file_id)
 
 
200
            if kind == 'root_directory':
 
 
202
            old_path = old_inv.id2path(file_id)
 
 
204
                if not is_inside_any(specific_files, old_path):
 
 
206
            delta.removed.append((old_path, file_id, kind))
 
 
208
    mutter('start looking for new files')
 
 
209
    for file_id in new_inv:
 
 
210
        if file_id in old_inv:
 
 
212
        kind = new_inv.get_file_kind(file_id)
 
 
213
        if kind == 'root_directory':
 
 
215
        new_path = new_inv.id2path(file_id)
 
 
217
            if not is_inside_any(specific_files, new_path):
 
 
219
        delta.added.append((new_path, file_id, kind))
 
 
224
    delta.modified.sort()
 
 
225
    delta.unchanged.sort()