← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tree.py

  • Committer: Martin Pool
  • Date: 2005-07-22 23:43:26 UTC
  • Revision ID: mbp@sourcefrog.net-20050722234326-13104af9d837ed2b
todo

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tree.py
17
17
"""Tree classes, representing directory at point in time.
18
18
"""
19
19
 
20
 
from sets import Set
21
 
import os.path, os, fnmatch
22
 
 
23
 
from osutils import pumpfile, filesize, quotefn, sha_file, \
24
 
     joinpath, splitpath, appendpath, isdir, isfile, file_kind, fingerprint_file
25
 
import errno
26
 
from stat import S_ISREG, S_ISDIR, ST_MODE, ST_SIZE
27
 
 
28
 
from inventory import Inventory
29
 
from trace import mutter, note
30
 
from errors import bailout
31
 
import branch
 
20
from osutils import pumpfile, appendpath, fingerprint_file
 
21
import os
 
22
 
 
23
from bzrlib.trace import mutter, note
 
24
from bzrlib.errors import BzrError
32
25
 
33
26
import bzrlib
34
27
 
 
28
exporters = {}
 
29
 
35
30
class Tree(object):
36
31
    """Abstract file tree.
37
32
 
81
76
        
82
77
        if ie.text_size != None:
83
78
            if ie.text_size != fp['size']:
84
 
                bailout("mismatched size for file %r in %r" % (ie.file_id, self._store),
 
79
                raise BzrError("mismatched size for file %r in %r" % (ie.file_id, self._store),
85
80
                        ["inventory expects %d bytes" % ie.text_size,
86
81
                         "file is actually %d bytes" % fp['size'],
87
82
                         "store is probably damaged/corrupt"])
88
83
 
89
84
        if ie.text_sha1 != fp['sha1']:
90
 
            bailout("wrong SHA-1 for file %r in %r" % (ie.file_id, self._store),
 
85
            raise BzrError("wrong SHA-1 for file %r in %r" % (ie.file_id, self._store),
91
86
                    ["inventory expects %s" % ie.text_sha1,
92
87
                     "file is actually %s" % fp['sha1'],
93
88
                     "store is probably damaged/corrupt"])
99
94
        pumpfile(self.get_file(fileid), sys.stdout)
100
95
        
101
96
        
102
 
    def export(self, dest):        
103
 
        """Export this tree to a new directory.
104
 
 
105
 
        `dest` should not exist, and will be created holding the
106
 
        contents of this tree.
107
 
 
108
 
        TODO: To handle subdirectories we need to create the
109
 
               directories first.
110
 
 
111
 
        :note: If the export fails, the destination directory will be
112
 
               left in a half-assed state.
113
 
        """
114
 
        os.mkdir(dest)
115
 
        mutter('export version %r' % self)
116
 
        inv = self.inventory
117
 
        for dp, ie in inv.iter_entries():
118
 
            kind = ie.kind
119
 
            fullpath = appendpath(dest, dp)
120
 
            if kind == 'directory':
121
 
                os.mkdir(fullpath)
122
 
            elif kind == 'file':
123
 
                pumpfile(self.get_file(ie.file_id), file(fullpath, 'wb'))
124
 
            else:
125
 
                bailout("don't know how to export {%s} of kind %r" % (ie.file_id, kind))
126
 
            mutter("  export {%s} kind %s to %s" % (ie.file_id, kind, fullpath))
 
97
    def export(self, dest, format='dir', root=None):
 
98
        """Export this tree."""
 
99
        try:
 
100
            exporter = exporters[format]
 
101
        except KeyError:
 
102
            from bzrlib.errors import BzrCommandError
 
103
            raise BzrCommandError("export format %r not supported" % format)
 
104
        exporter(self, dest, root)
127
105
 
128
106
 
129
107
 
165
143
 
166
144
 
167
145
class EmptyTree(Tree):
168
 
    def __init__(self):
169
 
        self._inventory = Inventory()
 
146
    def __init__(self, root_id):
 
147
        from bzrlib.inventory import Inventory
 
148
        self._inventory = Inventory(root_id)
170
149
 
171
150
    def has_filename(self, filename):
172
151
        return False
241
220
        if old_name != new_name:
242
221
            yield (old_name, new_name)
243
222
            
 
223
 
 
224
 
 
225
######################################################################
 
226
# export
 
227
 
 
228
def dir_exporter(tree, dest, root):
 
229
    """Export this tree to a new directory.
 
230
 
 
231
    `dest` should not exist, and will be created holding the
 
232
    contents of this tree.
 
233
 
 
234
    TODO: To handle subdirectories we need to create the
 
235
           directories first.
 
236
 
 
237
    :note: If the export fails, the destination directory will be
 
238
           left in a half-assed state.
 
239
    """
 
240
    import os
 
241
    os.mkdir(dest)
 
242
    mutter('export version %r' % tree)
 
243
    inv = tree.inventory
 
244
    for dp, ie in inv.iter_entries():
 
245
        kind = ie.kind
 
246
        fullpath = appendpath(dest, dp)
 
247
        if kind == 'directory':
 
248
            os.mkdir(fullpath)
 
249
        elif kind == 'file':
 
250
            pumpfile(tree.get_file(ie.file_id), file(fullpath, 'wb'))
 
251
        else:
 
252
            raise BzrError("don't know how to export {%s} of kind %r" % (ie.file_id, kind))
 
253
        mutter("  export {%s} kind %s to %s" % (ie.file_id, kind, fullpath))
 
254
exporters['dir'] = dir_exporter
 
255
 
 
256
try:
 
257
    import tarfile
 
258
except ImportError:
 
259
    pass
 
260
else:
 
261
    def get_root_name(dest):
 
262
        """Get just the root name for a tarball.
 
263
 
 
264
        >>> get_root_name('mytar.tar')
 
265
        'mytar'
 
266
        >>> get_root_name('mytar.tar.bz2')
 
267
        'mytar'
 
268
        >>> get_root_name('tar.tar.tar.tgz')
 
269
        'tar.tar.tar'
 
270
        >>> get_root_name('bzr-0.0.5.tar.gz')
 
271
        'bzr-0.0.5'
 
272
        >>> get_root_name('a/long/path/mytar.tgz')
 
273
        'mytar'
 
274
        >>> get_root_name('../parent/../dir/other.tbz2')
 
275
        'other'
 
276
        """
 
277
        endings = ['.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
 
278
        dest = os.path.basename(dest)
 
279
        for end in endings:
 
280
            if dest.endswith(end):
 
281
                return dest[:-len(end)]
 
282
 
 
283
    def tar_exporter(tree, dest, root, compression=None):
 
284
        """Export this tree to a new tar file.
 
285
 
 
286
        `dest` will be created holding the contents of this tree; if it
 
287
        already exists, it will be clobbered, like with "tar -c".
 
288
        """
 
289
        from time import time
 
290
        now = time()
 
291
        compression = str(compression or '')
 
292
        if root is None:
 
293
            root = get_root_name(dest)
 
294
        try:
 
295
            ball = tarfile.open(dest, 'w:' + compression)
 
296
        except tarfile.CompressionError, e:
 
297
            raise BzrError(str(e))
 
298
        mutter('export version %r' % tree)
 
299
        inv = tree.inventory
 
300
        for dp, ie in inv.iter_entries():
 
301
            mutter("  export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
 
302
            item = tarfile.TarInfo(os.path.join(root, dp))
 
303
            # TODO: would be cool to actually set it to the timestamp of the
 
304
            # revision it was last changed
 
305
            item.mtime = now
 
306
            if ie.kind == 'directory':
 
307
                item.type = tarfile.DIRTYPE
 
308
                fileobj = None
 
309
                item.name += '/'
 
310
                item.size = 0
 
311
                item.mode = 0755
 
312
            elif ie.kind == 'file':
 
313
                item.type = tarfile.REGTYPE
 
314
                fileobj = tree.get_file(ie.file_id)
 
315
                item.size = _find_file_size(fileobj)
 
316
                item.mode = 0644
 
317
            else:
 
318
                raise BzrError("don't know how to export {%s} of kind %r" %
 
319
                        (ie.file_id, ie.kind))
 
320
 
 
321
            ball.addfile(item, fileobj)
 
322
        ball.close()
 
323
    exporters['tar'] = tar_exporter
 
324
 
 
325
    def tgz_exporter(tree, dest, root):
 
326
        tar_exporter(tree, dest, root, compression='gz')
 
327
    exporters['tgz'] = tgz_exporter
 
328
 
 
329
    def tbz_exporter(tree, dest, root):
 
330
        tar_exporter(tree, dest, root, compression='bz2')
 
331
    exporters['tbz2'] = tbz_exporter
 
332
 
 
333
 
 
334
def _find_file_size(fileobj):
 
335
    offset = fileobj.tell()
 
336
    try:
 
337
        fileobj.seek(0, 2)
 
338
        size = fileobj.tell()
 
339
    except TypeError:
 
340
        # gzip doesn't accept second argument to seek()
 
341
        fileobj.seek(0)
 
342
        size = 0
 
343
        while True:
 
344
            nread = len(fileobj.read())
 
345
            if nread == 0:
 
346
                break
 
347
            size += nread
 
348
    fileobj.seek(offset)
 
349
    return size