← 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-09-16 09:56:24 UTC
  • Revision ID: mbp@sourcefrog.net-20050916095623-ca0dff452934f21f
- make progress bar more tolerant of out-of-range values

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tree.py
24
24
from bzrlib.trace import mutter, note
25
25
from bzrlib.errors import BzrError, BzrCheckError
26
26
from bzrlib.inventory import Inventory
27
 
from bzrlib.osutils import appendpath, fingerprint_file
 
27
from bzrlib.osutils import pumpfile, appendpath, fingerprint_file
 
28
 
 
29
 
 
30
exporters = {}
28
31
 
29
32
class Tree(object):
30
33
    """Abstract file tree.
55
58
    def has_id(self, file_id):
56
59
        return self.inventory.has_id(file_id)
57
60
 
58
 
    def has_or_had_id(self, file_id):
59
 
        if file_id == self.inventory.root.file_id:
60
 
            return True
61
 
        return self.inventory.has_id(file_id)
62
 
 
63
61
    __contains__ = has_id
64
62
 
65
63
    def __iter__(self):
68
66
    def id2path(self, file_id):
69
67
        return self.inventory.id2path(file_id)
70
68
 
71
 
    def kind(self, file_id):
72
 
        raise NotImplementedError("subclasses must implement kind")
73
 
 
74
69
    def _get_inventory(self):
75
70
        return self._inventory
76
71
    
81
76
                         doc="Inventory of this Tree")
82
77
 
83
78
    def _check_retrieved(self, ie, f):
84
 
        if not __debug__:
85
 
            return  
86
79
        fp = fingerprint_file(f)
87
80
        f.seek(0)
88
81
        
104
97
        """Print file with id `file_id` to stdout."""
105
98
        import sys
106
99
        sys.stdout.write(self.get_file_text(file_id))
107
 
 
108
 
    def lock_read(self):
109
 
        pass
110
 
 
111
 
    def unlock(self):
112
 
        pass
113
 
        
114
 
        
 
100
        
 
101
        
 
102
    def export(self, dest, format='dir', root=None):
 
103
        """Export this tree."""
 
104
        try:
 
105
            exporter = exporters[format]
 
106
        except KeyError:
 
107
            from bzrlib.errors import BzrCommandError
 
108
            raise BzrCommandError("export format %r not supported" % format)
 
109
        exporter(self, dest, root)
 
110
 
 
111
 
 
112
 
115
113
class RevisionTree(Tree):
116
114
    """Tree viewing a previous revision.
117
115
 
122
120
           or at least passing a description to the constructor.
123
121
    """
124
122
    
125
 
    def __init__(self, branch, inv, revision_id):
126
 
        self._branch = branch
127
 
        self._weave_store = branch.weave_store
 
123
    def __init__(self, weave_store, inv, revision_id):
 
124
        self._weave_store = weave_store
128
125
        self._inventory = inv
129
126
        self._revision_id = revision_id
130
127
 
131
128
    def get_weave(self, file_id):
132
 
        import bzrlib.transactions as transactions
133
 
        return self._weave_store.get_weave(file_id,
134
 
                self._branch.get_transaction())
135
 
 
136
 
    def get_weave_prelude(self, file_id):
137
 
        import bzrlib.transactions as transactions
138
 
        return self._weave_store.get_weave_prelude(file_id,
139
 
                self._branch.get_transaction())
140
 
 
141
 
    def get_file_lines(self, file_id):
 
129
        return self._weave_store.get_weave(file_id)
 
130
        
 
131
 
 
132
    def get_file_text(self, file_id):
142
133
        ie = self._inventory[file_id]
143
134
        weave = self.get_weave(file_id)
144
 
        return weave.get(ie.revision)
145
 
 
146
 
    def get_file_text(self, file_id):
147
 
        return ''.join(self.get_file_lines(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: '
 
139
                                '%d vs %d bytes'
 
140
                                % (self._revision_id, file_id, len(content),
 
141
                                   ie.text_size))
 
142
        return content
148
143
 
149
144
    def get_file(self, file_id):
150
145
        return StringIO(self.get_file_text(file_id))
157
152
        if ie.kind == "file":
158
153
            return ie.text_sha1
159
154
 
160
 
    def is_executable(self, file_id):
161
 
        ie = self._inventory[file_id]
162
 
        if ie.kind != "file":
163
 
            return None 
164
 
        return self._inventory[file_id].executable
165
 
 
166
155
    def has_filename(self, filename):
167
156
        return bool(self.inventory.path2id(filename))
168
157
 
169
158
    def list_files(self):
170
159
        # The only files returned by this are those from the version
171
160
        for path, entry in self.inventory.iter_entries():
172
 
            yield path, 'V', entry.kind, entry.file_id, entry
173
 
 
174
 
    def get_symlink_target(self, file_id):
175
 
        ie = self._inventory[file_id]
176
 
        return ie.symlink_target;
177
 
 
178
 
    def kind(self, file_id):
179
 
        return self._inventory[file_id].kind
180
 
 
181
 
    def lock_read(self):
182
 
        self._branch.lock_read()
183
 
 
184
 
    def unlock(self):
185
 
        self._branch.unlock()
 
161
            yield path, 'V', entry.kind, entry.file_id
186
162
 
187
163
 
188
164
class EmptyTree(Tree):
189
165
    def __init__(self):
190
166
        self._inventory = Inventory()
191
167
 
192
 
    def get_symlink_target(self, file_id):
193
 
        return None
194
 
 
195
168
    def has_filename(self, filename):
196
169
        return False
197
170
 
198
 
    def kind(self, file_id):
199
 
        assert self._inventory[file_id].kind == "root_directory"
200
 
        return "root_directory"
201
 
 
202
171
    def list_files(self):
203
 
        return iter([])
 
172
        if False:  # just to make it a generator
 
173
            yield None
204
174
    
205
175
    def __contains__(self, file_id):
206
176
        return file_id in self._inventory
210
180
        return None
211
181
 
212
182
 
 
183
 
 
184
 
213
185
######################################################################
214
186
# diff
215
187
 
276
248
            
277
249
 
278
250
 
 
251
######################################################################
 
252
# export
 
253
 
 
254
def dir_exporter(tree, dest, root):
 
255
    """Export this tree to a new directory.
 
256
 
 
257
    `dest` should not exist, and will be created holding the
 
258
    contents of this tree.
 
259
 
 
260
    TODO: To handle subdirectories we need to create the
 
261
           directories first.
 
262
 
 
263
    :note: If the export fails, the destination directory will be
 
264
           left in a half-assed state.
 
265
    """
 
266
    import os
 
267
    os.mkdir(dest)
 
268
    mutter('export version %r' % tree)
 
269
    inv = tree.inventory
 
270
    for dp, ie in inv.iter_entries():
 
271
        kind = ie.kind
 
272
        fullpath = appendpath(dest, dp)
 
273
        if kind == 'directory':
 
274
            os.mkdir(fullpath)
 
275
        elif kind == 'file':
 
276
            pumpfile(tree.get_file(ie.file_id), file(fullpath, 'wb'))
 
277
        else:
 
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
 
281
 
 
282
try:
 
283
    import tarfile
 
284
except ImportError:
 
285
    pass
 
286
else:
 
287
    def get_root_name(dest):
 
288
        """Get just the root name for a tarball.
 
289
 
 
290
        >>> get_root_name('mytar.tar')
 
291
        'mytar'
 
292
        >>> get_root_name('mytar.tar.bz2')
 
293
        'mytar'
 
294
        >>> get_root_name('tar.tar.tar.tgz')
 
295
        'tar.tar.tar'
 
296
        >>> get_root_name('bzr-0.0.5.tar.gz')
 
297
        'bzr-0.0.5'
 
298
        >>> get_root_name('a/long/path/mytar.tgz')
 
299
        'mytar'
 
300
        >>> get_root_name('../parent/../dir/other.tbz2')
 
301
        'other'
 
302
        """
 
303
        endings = ['.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
 
304
        dest = os.path.basename(dest)
 
305
        for end in endings:
 
306
            if dest.endswith(end):
 
307
                return dest[:-len(end)]
 
308
 
 
309
    def tar_exporter(tree, dest, root, compression=None):
 
310
        """Export this tree to a new tar file.
 
311
 
 
312
        `dest` will be created holding the contents of this tree; if it
 
313
        already exists, it will be clobbered, like with "tar -c".
 
314
        """
 
315
        from time import time
 
316
        now = time()
 
317
        compression = str(compression or '')
 
318
        if root is None:
 
319
            root = get_root_name(dest)
 
320
        try:
 
321
            ball = tarfile.open(dest, 'w:' + compression)
 
322
        except tarfile.CompressionError, e:
 
323
            raise BzrError(str(e))
 
324
        mutter('export version %r' % tree)
 
325
        inv = tree.inventory
 
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
 
331
            item.mtime = now
 
332
            if ie.kind == 'directory':
 
333
                item.type = tarfile.DIRTYPE
 
334
                fileobj = None
 
335
                item.name += '/'
 
336
                item.size = 0
 
337
                item.mode = 0755
 
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)
 
342
                item.mode = 0644
 
343
            else:
 
344
                raise BzrError("don't know how to export {%s} of kind %r" %
 
345
                        (ie.file_id, ie.kind))
 
346
 
 
347
            ball.addfile(item, fileobj)
 
348
        ball.close()
 
349
    exporters['tar'] = tar_exporter
 
350
 
 
351
    def tgz_exporter(tree, dest, root):
 
352
        tar_exporter(tree, dest, root, compression='gz')
 
353
    exporters['tgz'] = tgz_exporter
 
354
 
 
355
    def tbz_exporter(tree, dest, root):
 
356
        tar_exporter(tree, dest, root, compression='bz2')
 
357
    exporters['tbz2'] = tbz_exporter
 
358
 
 
359
 
 
360
def _find_file_size(fileobj):
 
361
    offset = fileobj.tell()
 
362
    try:
 
363
        fileobj.seek(0, 2)
 
364
        size = fileobj.tell()
 
365
    except TypeError:
 
366
        # gzip doesn't accept second argument to seek()
 
367
        fileobj.seek(0)
 
368
        size = 0
 
369
        while True:
 
370
            nread = len(fileobj.read())
 
371
            if nread == 0:
 
372
                break
 
373
            size += nread
 
374
    fileobj.seek(offset)
 
375
    return size