33
31
File text can be retrieved from the text store.
36
def __init__(self, branch, inv, revision_id):
37
# for compatability the 'branch' parameter has not been renamed to
38
# repository at this point. However, we should change RevisionTree's
39
# construction to always be via Repository and not via direct
40
# construction - this will mean that we can change the constructor
41
# with much less chance of breaking client code.
42
self._repository = branch
34
def __init__(self, repository, revision_id):
35
self._repository = repository
44
36
self._revision_id = revision_id
45
37
self._rules_searcher = None
64
56
"""Return the revision id associated with this tree."""
65
57
return self._revision_id
59
def get_file_revision(self, file_id, path=None):
60
"""Return the revision id in which a file was last changed."""
61
raise NotImplementedError(self.get_file_revision)
67
63
def get_file_text(self, file_id, path=None):
68
64
_, content = list(self.iter_files_bytes([(file_id, None)]))[0]
69
65
return ''.join(content)
71
67
def get_file(self, file_id, path=None):
72
68
return StringIO(self.get_file_text(file_id))
74
def iter_files_bytes(self, desired_files):
75
"""See Tree.iter_files_bytes.
77
This version is implemented on top of Repository.extract_files_bytes"""
78
repo_desired_files = [(f, self.inventory[f].revision, i)
79
for f, i in desired_files]
71
return self._repository.is_locked()
74
self._repository.lock_read()
78
return '<%s instance at %x, rev_id=%r>' % (
79
self.__class__.__name__, id(self), self._revision_id)
82
self._repository.unlock()
84
def _get_rules_searcher(self, default_searcher):
85
"""See Tree._get_rules_searcher."""
86
if self._rules_searcher is None:
87
self._rules_searcher = super(RevisionTree,
88
self)._get_rules_searcher(default_searcher)
89
return self._rules_searcher
92
class InventoryRevisionTree(RevisionTree,tree.InventoryTree):
94
def __init__(self, repository, inv, revision_id):
95
RevisionTree.__init__(self, repository, revision_id)
98
def get_file_mtime(self, file_id, path=None):
99
ie = self._inventory[file_id]
81
for result in self._repository.iter_files_bytes(repo_desired_files):
83
except errors.RevisionNotPresent, e:
84
raise errors.NoSuchFile(e.revision_id)
86
def annotate_iter(self, file_id,
87
default_revision=revision.CURRENT_REVISION):
88
"""See Tree.annotate_iter"""
89
text_key = (file_id, self.inventory[file_id].revision)
90
annotator = self._repository.texts.get_annotator()
91
annotations = annotator.annotate_flat(text_key)
92
return [(key[-1], line) for key, line in annotations]
101
revision = self._repository.get_revision(ie.revision)
102
except errors.NoSuchRevision:
103
raise errors.FileTimestampUnavailable(self.id2path(file_id))
104
return revision.timestamp
94
106
def get_file_size(self, file_id):
95
"""See Tree.get_file_size"""
96
107
return self._inventory[file_id].text_size
98
109
def get_file_sha1(self, file_id, path=None, stat_value=None):
101
112
return ie.text_sha1
104
def get_file_mtime(self, file_id, path=None):
115
def get_file_revision(self, file_id, path=None):
105
116
ie = self._inventory[file_id]
107
revision = self._repository.get_revision(ie.revision)
108
except errors.NoSuchRevision:
109
raise errors.FileTimestampUnavailable(self.id2path(file_id))
110
return revision.timestamp
112
119
def is_executable(self, file_id, path=None):
113
120
ie = self._inventory[file_id]
114
121
if ie.kind != "file":
116
123
return ie.executable
118
125
def has_filename(self, filename):
135
142
for path, entry in entries:
136
143
yield path, 'V', entry.kind, entry.file_id, entry
138
def get_symlink_target(self, file_id):
145
def get_symlink_target(self, file_id, path=None):
139
146
ie = self._inventory[file_id]
140
147
# Inventories store symlink targets in unicode
141
148
return ie.symlink_target
172
179
def _file_size(self, entry, stat_value):
173
180
return entry.text_size
175
def _get_ancestors(self, default_revision):
176
return set(self._repository.get_ancestry(self._revision_id,
180
self._repository.lock_read()
183
return '<%s instance at %x, rev_id=%r>' % (
184
self.__class__.__name__, id(self), self._revision_id)
187
self._repository.unlock()
189
182
def walkdirs(self, prefix=""):
190
183
_directory = 'directory'
191
184
inv = self.inventory
215
208
if dir[2] == _directory:
216
209
pending.append(dir)
218
def _get_rules_searcher(self, default_searcher):
219
"""See Tree._get_rules_searcher."""
220
if self._rules_searcher is None:
221
self._rules_searcher = super(RevisionTree,
222
self)._get_rules_searcher(default_searcher)
223
return self._rules_searcher
211
def iter_files_bytes(self, desired_files):
212
"""See Tree.iter_files_bytes.
214
This version is implemented on top of Repository.extract_files_bytes"""
215
repo_desired_files = [(f, self.get_file_revision(f), i)
216
for f, i in desired_files]
218
for result in self._repository.iter_files_bytes(repo_desired_files):
220
except errors.RevisionNotPresent, e:
221
raise errors.NoSuchFile(e.revision_id)
223
def annotate_iter(self, file_id,
224
default_revision=revision.CURRENT_REVISION):
225
"""See Tree.annotate_iter"""
226
text_key = (file_id, self.get_file_revision(file_id))
227
annotator = self._repository.texts.get_annotator()
228
annotations = annotator.annotate_flat(text_key)
229
return [(key[-1], line) for key, line in annotations]
226
232
class InterCHKRevisionTree(tree.InterTree):