97
104
"""Print file with id `file_id` to stdout."""
99
106
sys.stdout.write(self.get_file_text(file_id))
102
def export(self, dest, format='dir', root=None):
103
"""Export this tree."""
105
exporter = exporters[format]
107
from bzrlib.errors import BzrCommandError
108
raise BzrCommandError("export format %r not supported" % format)
109
exporter(self, dest, root)
114
def filter_unversioned_files(self, paths):
115
"""Filter out paths that are not versioned.
117
:return: set of paths.
119
# NB: we specifically *don't* call self.has_filename, because for
120
# WorkingTrees that can indicate files that exist on disk but that
122
pred = self.inventory.has_filename
123
return set((p for p in paths if not pred(p)))
113
126
class RevisionTree(Tree):
114
127
"""Tree viewing a previous revision.
120
133
or at least passing a description to the constructor.
123
def __init__(self, weave_store, inv, revision_id):
124
self._weave_store = weave_store
136
def __init__(self, branch, inv, revision_id):
137
self._branch = branch
138
self._weave_store = branch.weave_store
125
139
self._inventory = inv
126
140
self._revision_id = revision_id
142
def get_revision_id(self):
143
"""Return the revision id associated with this tree."""
144
return self._revision_id
128
146
def get_weave(self, file_id):
129
return self._weave_store.get_weave(file_id)
147
return self._weave_store.get_weave(file_id,
148
self._branch.get_transaction())
150
def get_file_lines(self, file_id):
151
ie = self._inventory[file_id]
152
weave = self.get_weave(file_id)
153
return weave.get_lines(ie.revision)
132
155
def get_file_text(self, file_id):
133
ie = self._inventory[file_id]
134
weave = self.get_weave(file_id)
135
idx = weave.lookup(ie.text_version)
136
content = weave.get_text(idx)
137
if len(content) != ie.text_size:
138
raise BzrCheckError('mismatched size on revision %s of file %s: '
140
% (self._revision_id, file_id, len(content),
156
return ''.join(self.get_file_lines(file_id))
144
158
def get_file(self, file_id):
145
159
return StringIO(self.get_file_text(file_id))
147
161
def get_file_size(self, file_id):
148
162
return self._inventory[file_id].text_size
150
def get_file_sha1(self, file_id):
164
def get_file_sha1(self, file_id, path=None):
151
165
ie = self._inventory[file_id]
152
166
if ie.kind == "file":
153
167
return ie.text_sha1
170
def is_executable(self, file_id, path=None):
171
ie = self._inventory[file_id]
172
if ie.kind != "file":
174
return self._inventory[file_id].executable
155
176
def has_filename(self, filename):
156
177
return bool(self.inventory.path2id(filename))
158
179
def list_files(self):
159
180
# The only files returned by this are those from the version
160
181
for path, entry in self.inventory.iter_entries():
161
yield path, 'V', entry.kind, entry.file_id
182
yield path, 'V', entry.kind, entry.file_id, entry
184
def get_symlink_target(self, file_id):
185
ie = self._inventory[file_id]
186
return ie.symlink_target;
188
def kind(self, file_id):
189
return self._inventory[file_id].kind
192
self._branch.lock_read()
195
self._branch.unlock()
164
198
class EmptyTree(Tree):
165
199
def __init__(self):
166
200
self._inventory = Inventory()
202
def get_symlink_target(self, file_id):
168
205
def has_filename(self, filename):
208
def kind(self, file_id):
209
assert self._inventory[file_id].kind == "root_directory"
210
return "root_directory"
171
212
def list_files(self):
172
if False: # just to make it a generator
175
215
def __contains__(self, file_id):
176
216
return file_id in self._inventory
178
def get_file_sha1(self, file_id):
218
def get_file_sha1(self, file_id, path=None):
179
219
assert self._inventory[file_id].kind == "root_directory"
185
223
######################################################################
251
######################################################################
254
def dir_exporter(tree, dest, root):
255
"""Export this tree to a new directory.
257
`dest` should not exist, and will be created holding the
258
contents of this tree.
260
TODO: To handle subdirectories we need to create the
263
:note: If the export fails, the destination directory will be
264
left in a half-assed state.
268
mutter('export version %r' % tree)
270
for dp, ie in inv.iter_entries():
272
fullpath = appendpath(dest, dp)
273
if kind == 'directory':
276
pumpfile(tree.get_file(ie.file_id), file(fullpath, 'wb'))
278
raise BzrError("don't know how to export {%s} of kind %r" % (ie.file_id, kind))
279
mutter(" export {%s} kind %s to %s" % (ie.file_id, kind, fullpath))
280
exporters['dir'] = dir_exporter
287
def get_root_name(dest):
288
"""Get just the root name for a tarball.
290
>>> get_root_name('mytar.tar')
292
>>> get_root_name('mytar.tar.bz2')
294
>>> get_root_name('tar.tar.tar.tgz')
296
>>> get_root_name('bzr-0.0.5.tar.gz')
298
>>> get_root_name('a/long/path/mytar.tgz')
300
>>> get_root_name('../parent/../dir/other.tbz2')
303
endings = ['.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
304
dest = os.path.basename(dest)
306
if dest.endswith(end):
307
return dest[:-len(end)]
309
def tar_exporter(tree, dest, root, compression=None):
310
"""Export this tree to a new tar file.
312
`dest` will be created holding the contents of this tree; if it
313
already exists, it will be clobbered, like with "tar -c".
315
from time import time
317
compression = str(compression or '')
319
root = get_root_name(dest)
321
ball = tarfile.open(dest, 'w:' + compression)
322
except tarfile.CompressionError, e:
323
raise BzrError(str(e))
324
mutter('export version %r' % tree)
326
for dp, ie in inv.iter_entries():
327
mutter(" export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
328
item = tarfile.TarInfo(os.path.join(root, dp))
329
# TODO: would be cool to actually set it to the timestamp of the
330
# revision it was last changed
332
if ie.kind == 'directory':
333
item.type = tarfile.DIRTYPE
338
elif ie.kind == 'file':
339
item.type = tarfile.REGTYPE
340
fileobj = tree.get_file(ie.file_id)
341
item.size = _find_file_size(fileobj)
344
raise BzrError("don't know how to export {%s} of kind %r" %
345
(ie.file_id, ie.kind))
347
ball.addfile(item, fileobj)
349
exporters['tar'] = tar_exporter
351
def tgz_exporter(tree, dest, root):
352
tar_exporter(tree, dest, root, compression='gz')
353
exporters['tgz'] = tgz_exporter
355
def tbz_exporter(tree, dest, root):
356
tar_exporter(tree, dest, root, compression='bz2')
357
exporters['tbz2'] = tbz_exporter
360
def _find_file_size(fileobj):
361
offset = fileobj.tell()
364
size = fileobj.tell()
366
# gzip doesn't accept second argument to seek()
370
nread = len(fileobj.read())
added
removed
bzrlib/tree.py
