92
118
"store is probably damaged/corrupt"])
95
def print_file(self, fileid):
96
"""Print file with id `fileid` to stdout."""
121
def print_file(self, file_id):
122
"""Print file with id `file_id` to stdout."""
98
pumpfile(self.get_file(fileid), sys.stdout)
101
def export(self, dest, format='dir', root=None):
102
"""Export this tree."""
104
exporter = exporters[format]
106
from bzrlib.errors import BzrCommandError
107
raise BzrCommandError("export format %r not supported" % format)
108
exporter(self, dest, root)
124
sys.stdout.write(self.get_file_text(file_id))
130
"""What files are present in this tree and unknown.
132
:return: an iterator over the unknown files.
139
def filter_unversioned_files(self, paths):
140
"""Filter out paths that are not versioned.
142
:return: set of paths.
144
# NB: we specifically *don't* call self.has_filename, because for
145
# WorkingTrees that can indicate files that exist on disk but that
147
pred = self.inventory.has_filename
148
return set((p for p in paths if not pred(p)))
112
151
class RevisionTree(Tree):
113
152
"""Tree viewing a previous revision.
119
158
or at least passing a description to the constructor.
122
def __init__(self, store, inv):
161
def __init__(self, branch, inv, revision_id):
162
# for compatability the 'branch' parameter has not been renamed to
163
# repository at this point. However, we should change RevisionTree's
164
# construction to always be via Repository and not via direct
165
# construction - this will mean that we can change the constructor
166
# with much less chance of breaking client code.
167
self._repository = branch
168
self._weave_store = branch.weave_store
124
169
self._inventory = inv
170
self._revision_id = revision_id
172
def get_parent_ids(self):
173
"""See Tree.get_parent_ids.
175
A RevisionTree's parents match the revision graph.
177
parent_ids = self._repository.get_revision(self._revision_id).parent_ids
180
def get_revision_id(self):
181
"""Return the revision id associated with this tree."""
182
return self._revision_id
184
def get_weave(self, file_id):
185
return self._weave_store.get_weave(file_id,
186
self._repository.get_transaction())
188
def get_file_lines(self, file_id):
189
ie = self._inventory[file_id]
190
weave = self.get_weave(file_id)
191
return weave.get_lines(ie.revision)
193
def get_file_text(self, file_id):
194
return ''.join(self.get_file_lines(file_id))
126
196
def get_file(self, file_id):
127
ie = self._inventory[file_id]
128
f = self._store[ie.text_id]
129
mutter(" get fileid{%s} from %r" % (file_id, self))
130
self._check_retrieved(ie, f)
197
return StringIO(self.get_file_text(file_id))
133
199
def get_file_size(self, file_id):
134
200
return self._inventory[file_id].text_size
136
def get_file_sha1(self, file_id):
137
ie = self._inventory[file_id]
202
def get_file_sha1(self, file_id, path=None):
203
ie = self._inventory[file_id]
204
if ie.kind == "file":
208
def get_file_mtime(self, file_id, path=None):
209
ie = self._inventory[file_id]
210
revision = self._repository.get_revision(ie.revision)
211
return revision.timestamp
213
def is_executable(self, file_id, path=None):
214
ie = self._inventory[file_id]
215
if ie.kind != "file":
217
return self._inventory[file_id].executable
140
219
def has_filename(self, filename):
141
220
return bool(self.inventory.path2id(filename))
143
222
def list_files(self):
144
223
# The only files returned by this are those from the version
145
for path, entry in self.inventory.iter_entries():
146
yield path, 'V', entry.kind, entry.file_id
224
entries = self.inventory.iter_entries()
225
# skip the root for compatability with the current apis.
227
for path, entry in entries:
228
yield path, 'V', entry.kind, entry.file_id, entry
230
def get_symlink_target(self, file_id):
231
ie = self._inventory[file_id]
232
return ie.symlink_target;
234
def kind(self, file_id):
235
return self._inventory[file_id].kind
238
self._repository.lock_read()
241
self._repository.unlock()
149
244
class EmptyTree(Tree):
150
246
def __init__(self):
151
247
self._inventory = Inventory()
248
warn('EmptyTree is deprecated as of bzr 0.9 please use '
249
'repository.revision_tree instead.',
250
DeprecationWarning, stacklevel=2)
252
def get_parent_ids(self):
255
def get_symlink_target(self, file_id):
153
258
def has_filename(self, filename):
261
def kind(self, file_id):
262
assert self._inventory[file_id].kind == "root_directory"
263
return "root_directory"
156
265
def list_files(self):
157
if False: # just to make it a generator
268
def __contains__(self, file_id):
269
return file_id in self._inventory
271
def get_file_sha1(self, file_id, path=None):
272
assert self._inventory[file_id].kind == "root_directory"
162
276
######################################################################
224
338
yield (old_name, new_name)
228
######################################################################
231
def dir_exporter(tree, dest, root):
232
"""Export this tree to a new directory.
234
`dest` should not exist, and will be created holding the
235
contents of this tree.
237
TODO: To handle subdirectories we need to create the
240
:note: If the export fails, the destination directory will be
241
left in a half-assed state.
245
mutter('export version %r' % tree)
247
for dp, ie in inv.iter_entries():
249
fullpath = appendpath(dest, dp)
250
if kind == 'directory':
253
pumpfile(tree.get_file(ie.file_id), file(fullpath, 'wb'))
255
raise BzrError("don't know how to export {%s} of kind %r" % (ie.file_id, kind))
256
mutter(" export {%s} kind %s to %s" % (ie.file_id, kind, fullpath))
257
exporters['dir'] = dir_exporter
264
def get_root_name(dest):
265
"""Get just the root name for a tarball.
267
>>> get_root_name('mytar.tar')
269
>>> get_root_name('mytar.tar.bz2')
271
>>> get_root_name('tar.tar.tar.tgz')
273
>>> get_root_name('bzr-0.0.5.tar.gz')
275
>>> get_root_name('a/long/path/mytar.tgz')
277
>>> get_root_name('../parent/../dir/other.tbz2')
280
endings = ['.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
281
dest = os.path.basename(dest)
283
if dest.endswith(end):
284
return dest[:-len(end)]
286
def tar_exporter(tree, dest, root, compression=None):
287
"""Export this tree to a new tar file.
289
`dest` will be created holding the contents of this tree; if it
290
already exists, it will be clobbered, like with "tar -c".
292
from time import time
294
compression = str(compression or '')
296
root = get_root_name(dest)
298
ball = tarfile.open(dest, 'w:' + compression)
299
except tarfile.CompressionError, e:
300
raise BzrError(str(e))
301
mutter('export version %r' % tree)
303
for dp, ie in inv.iter_entries():
304
mutter(" export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
305
item = tarfile.TarInfo(os.path.join(root, dp))
306
# TODO: would be cool to actually set it to the timestamp of the
307
# revision it was last changed
309
if ie.kind == 'directory':
310
item.type = tarfile.DIRTYPE
315
elif ie.kind == 'file':
316
item.type = tarfile.REGTYPE
317
fileobj = tree.get_file(ie.file_id)
318
item.size = _find_file_size(fileobj)
321
raise BzrError("don't know how to export {%s} of kind %r" %
322
(ie.file_id, ie.kind))
324
ball.addfile(item, fileobj)
326
exporters['tar'] = tar_exporter
328
def tgz_exporter(tree, dest, root):
329
tar_exporter(tree, dest, root, compression='gz')
330
exporters['tgz'] = tgz_exporter
332
def tbz_exporter(tree, dest, root):
333
tar_exporter(tree, dest, root, compression='bz2')
334
exporters['tbz2'] = tbz_exporter
337
def _find_file_size(fileobj):
338
offset = fileobj.tell()
341
size = fileobj.tell()
343
# gzip doesn't accept second argument to seek()
347
nread = len(fileobj.read())
341
def find_ids_across_trees(filenames, trees, require_versioned=True):
342
"""Find the ids corresponding to specified filenames.
344
All matches in all trees will be used, and all children of matched
345
directories will be used.
347
:param filenames: The filenames to find file_ids for
348
:param trees: The trees to find file_ids within
349
:param require_versioned: if true, all specified filenames must occur in
351
:return: a set of file ids for the specified filenames and their children.
355
specified_ids = _find_filename_ids_across_trees(filenames, trees,
357
return _find_children_across_trees(specified_ids, trees)
360
def _find_filename_ids_across_trees(filenames, trees, require_versioned):
361
"""Find the ids corresponding to specified filenames.
363
All matches in all trees will be used.
365
:param filenames: The filenames to find file_ids for
366
:param trees: The trees to find file_ids within
367
:param require_versioned: if true, all specified filenames must occur in
369
:return: a set of file ids for the specified filenames
372
interesting_ids = set()
373
for tree_path in filenames:
376
file_id = tree.inventory.path2id(tree_path)
377
if file_id is not None:
378
interesting_ids.add(file_id)
381
not_versioned.append(tree_path)
382
if len(not_versioned) > 0 and require_versioned:
383
raise errors.PathsNotVersionedError(not_versioned)
384
return interesting_ids
387
def _find_children_across_trees(specified_ids, trees):
388
"""Return a set including specified ids and their children
390
All matches in all trees will be used.
392
:param trees: The trees to find file_ids within
393
:return: a set containing all specified ids and their children
395
interesting_ids = set(specified_ids)
396
pending = interesting_ids
397
# now handle children of interesting ids
398
# we loop so that we handle all children of each id in both trees
399
while len(pending) > 0:
401
for file_id in pending:
403
if file_id not in tree:
405
entry = tree.inventory[file_id]
406
for child in getattr(entry, 'children', {}).itervalues():
407
if child.file_id not in interesting_ids:
408
new_pending.add(child.file_id)
409
interesting_ids.update(new_pending)
410
pending = new_pending
411
return interesting_ids
added
removed
bzrlib/tree.py
