/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/revisiontree.py

  • Committer: Robert Collins
  • Date: 2007-10-05 02:41:37 UTC
  • mto: (2592.3.166 repository)
  • mto: This revision was merged to the branch mainline in revision 2896.
  • Revision ID: robertc@robertcollins.net-20071005024137-kn7brcu07nu8cwl1
* The class ``bzrlib.repofmt.knitrepo.KnitRepository3`` has been folded into
  ``KnitRepository`` by parameters to the constructor. (Robert Collins)
* ``bzrlib.xml_serializer.Serializer`` is now responsible for checking that
  mandatory attributes are present on serialisation and deserialisation.
  This fixes some holes in API usage and allows better separation between
  physical storage and object serialisation. (Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/revisiontree.py
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
"""RevisionTree - a Tree implementation backed by repository data for a revision."""
 
18
 
 
19
from cStringIO import StringIO
 
20
 
 
21
from bzrlib import (
 
22
    osutils,
 
23
    revision,
 
24
    symbol_versioning,
 
25
    )
 
26
from bzrlib.tree import Tree
 
27
 
 
28
 
 
29
class RevisionTree(Tree):
 
30
    """Tree viewing a previous revision.
 
31
 
 
32
    File text can be retrieved from the text store.
 
33
    """
 
34
    
 
35
    def __init__(self, branch, inv, revision_id):
 
36
        # for compatability the 'branch' parameter has not been renamed to 
 
37
        # repository at this point. However, we should change RevisionTree's
 
38
        # construction to always be via Repository and not via direct 
 
39
        # construction - this will mean that we can change the constructor
 
40
        # with much less chance of breaking client code.
 
41
        self._repository = branch
 
42
        self._weave_store = branch.weave_store
 
43
        self._inventory = inv
 
44
        self._revision_id = osutils.safe_revision_id(revision_id)
 
45
 
 
46
    def supports_tree_reference(self):
 
47
        return True
 
48
 
 
49
    def get_parent_ids(self):
 
50
        """See Tree.get_parent_ids.
 
51
 
 
52
        A RevisionTree's parents match the revision graph.
 
53
        """
 
54
        if self._revision_id in (None, revision.NULL_REVISION):
 
55
            parent_ids = []
 
56
        else:
 
57
            parent_ids = self._repository.get_revision(
 
58
                self._revision_id).parent_ids
 
59
        return parent_ids
 
60
        
 
61
    def get_revision_id(self):
 
62
        """Return the revision id associated with this tree."""
 
63
        return self._revision_id
 
64
 
 
65
    @symbol_versioning.deprecated_method(symbol_versioning.zero_ninety)
 
66
    def get_weave(self, file_id):
 
67
        return self._get_weave(file_id)
 
68
 
 
69
    def _get_weave(self, file_id):
 
70
        file_id = osutils.safe_file_id(file_id)
 
71
        return self._weave_store.get_weave(file_id,
 
72
                self._repository.get_transaction())
 
73
 
 
74
    def get_file_lines(self, file_id):
 
75
        file_id = osutils.safe_file_id(file_id)
 
76
        ie = self._inventory[file_id]
 
77
        weave = self._get_weave(file_id)
 
78
        return weave.get_lines(ie.revision)
 
79
 
 
80
    def get_file_text(self, file_id):
 
81
        file_id = osutils.safe_file_id(file_id)
 
82
        return ''.join(self.get_file_lines(file_id))
 
83
 
 
84
    def get_file(self, file_id, path=None):
 
85
        file_id = osutils.safe_file_id(file_id)
 
86
        return StringIO(self.get_file_text(file_id))
 
87
 
 
88
    def iter_files_bytes(self, desired_files):
 
89
        """See Tree.extract_files_bytes.
 
90
 
 
91
        This version is implemented on top of Repository.extract_files_bytes"""
 
92
        repo_desired_files = [(f, self.inventory[f].revision, i)
 
93
                              for f, i in desired_files]
 
94
        return self._repository.iter_files_bytes(repo_desired_files)
 
95
 
 
96
    def annotate_iter(self, file_id,
 
97
                      default_revision=revision.CURRENT_REVISION):
 
98
        """See Tree.annotate_iter"""
 
99
        file_id = osutils.safe_file_id(file_id)
 
100
        w = self._get_weave(file_id)
 
101
        return w.annotate_iter(self.inventory[file_id].revision)
 
102
 
 
103
    def get_file_size(self, file_id):
 
104
        file_id = osutils.safe_file_id(file_id)
 
105
        return self._inventory[file_id].text_size
 
106
 
 
107
    def get_file_sha1(self, file_id, path=None, stat_value=None):
 
108
        file_id = osutils.safe_file_id(file_id)
 
109
        ie = self._inventory[file_id]
 
110
        if ie.kind == "file":
 
111
            return ie.text_sha1
 
112
        return None
 
113
 
 
114
    def get_file_mtime(self, file_id, path=None):
 
115
        file_id = osutils.safe_file_id(file_id)
 
116
        ie = self._inventory[file_id]
 
117
        revision = self._repository.get_revision(ie.revision)
 
118
        return revision.timestamp
 
119
 
 
120
    def is_executable(self, file_id, path=None):
 
121
        file_id = osutils.safe_file_id(file_id)
 
122
        ie = self._inventory[file_id]
 
123
        if ie.kind != "file":
 
124
            return None
 
125
        return ie.executable
 
126
 
 
127
    def has_filename(self, filename):
 
128
        return bool(self.inventory.path2id(filename))
 
129
 
 
130
    def list_files(self, include_root=False):
 
131
        # The only files returned by this are those from the version
 
132
        entries = self.inventory.iter_entries()
 
133
        # skip the root for compatability with the current apis.
 
134
        if self.inventory.root is not None and not include_root:
 
135
            # skip the root for compatability with the current apis.
 
136
            entries.next()
 
137
        for path, entry in entries:
 
138
            yield path, 'V', entry.kind, entry.file_id, entry
 
139
 
 
140
    def get_symlink_target(self, file_id):
 
141
        file_id = osutils.safe_file_id(file_id)
 
142
        ie = self._inventory[file_id]
 
143
        return ie.symlink_target;
 
144
 
 
145
    def get_reference_revision(self, file_id, path=None):
 
146
        return self.inventory[file_id].reference_revision
 
147
 
 
148
    def get_root_id(self):
 
149
        if self.inventory.root:
 
150
            return self.inventory.root.file_id
 
151
 
 
152
    def kind(self, file_id):
 
153
        file_id = osutils.safe_file_id(file_id)
 
154
        return self._inventory[file_id].kind
 
155
 
 
156
    def path_content_summary(self, path):
 
157
        """See Tree.path_content_summary."""
 
158
        id = self.inventory.path2id(path)
 
159
        if id is None:
 
160
            return ('missing', None, None, None)
 
161
        entry = self._inventory[id]
 
162
        kind = entry.kind
 
163
        if kind == 'file':
 
164
            return (kind, entry.text_size, entry.executable, entry.text_sha1)
 
165
        elif kind == 'symlink':
 
166
            return (kind, None, None, entry.symlink_target)
 
167
        else:
 
168
            return (kind, None, None, None)
 
169
 
 
170
    def _comparison_data(self, entry, path):
 
171
        if entry is None:
 
172
            return None, False, None
 
173
        return entry.kind, entry.executable, None
 
174
 
 
175
    def _file_size(self, entry, stat_value):
 
176
        assert entry.text_size is not None
 
177
        return entry.text_size
 
178
 
 
179
    def _get_ancestors(self, default_revision):
 
180
        return set(self._repository.get_ancestry(self._revision_id,
 
181
                                                 topo_sorted=False))
 
182
 
 
183
    def lock_read(self):
 
184
        self._repository.lock_read()
 
185
 
 
186
    def __repr__(self):
 
187
        return '<%s instance at %x, rev_id=%r>' % (
 
188
            self.__class__.__name__, id(self), self._revision_id)
 
189
 
 
190
    def unlock(self):
 
191
        self._repository.unlock()
 
192
 
 
193
    def walkdirs(self, prefix=""):
 
194
        _directory = 'directory'
 
195
        inv = self.inventory
 
196
        top_id = inv.path2id(prefix)
 
197
        if top_id is None:
 
198
            pending = []
 
199
        else:
 
200
            pending = [(prefix, '', _directory, None, top_id, None)]
 
201
        while pending:
 
202
            dirblock = []
 
203
            currentdir = pending.pop()
 
204
            # 0 - relpath, 1- basename, 2- kind, 3- stat, id, v-kind
 
205
            if currentdir[0]:
 
206
                relroot = currentdir[0] + '/'
 
207
            else:
 
208
                relroot = ""
 
209
            # FIXME: stash the node in pending
 
210
            entry = inv[currentdir[4]]
 
211
            for name, child in entry.sorted_children():
 
212
                toppath = relroot + name
 
213
                dirblock.append((toppath, name, child.kind, None,
 
214
                    child.file_id, child.kind
 
215
                    ))
 
216
            yield (currentdir[0], entry.file_id), dirblock
 
217
            # push the user specified dirs from dirblock
 
218
            for dir in reversed(dirblock):
 
219
                if dir[2] == _directory:
 
220
                    pending.append(dir)