← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tree.py

  • Committer: Robert Collins
  • Date: 2005-10-09 23:12:35 UTC
  • Revision ID: robertc@robertcollins.net-20051009231235-93626e72cac71b78
clean up test dirs on make clean

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tree.py
18
18
"""
19
19
 
20
20
import os
 
21
from cStringIO import StringIO
21
22
 
22
23
import bzrlib
23
24
from bzrlib.trace import mutter, note
24
 
from bzrlib.errors import BzrError
 
25
from bzrlib.errors import BzrError, BzrCheckError
25
26
from bzrlib.inventory import Inventory
26
 
from bzrlib.osutils import pumpfile, appendpath, fingerprint_file
 
27
from bzrlib.osutils import appendpath, fingerprint_file
27
28
 
28
29
 
29
30
exporters = {}
75
76
                         doc="Inventory of this Tree")
76
77
 
77
78
    def _check_retrieved(self, ie, f):
 
79
        if not __debug__:
 
80
            return  
78
81
        fp = fingerprint_file(f)
79
82
        f.seek(0)
80
83
        
92
95
                     "store is probably damaged/corrupt"])
93
96
 
94
97
 
95
 
    def print_file(self, fileid):
96
 
        """Print file with id `fileid` to stdout."""
 
98
    def print_file(self, file_id):
 
99
        """Print file with id `file_id` to stdout."""
97
100
        import sys
98
 
        pumpfile(self.get_file(fileid), sys.stdout)
 
101
        sys.stdout.write(self.get_file_text(file_id))
99
102
        
100
103
        
101
104
    def export(self, dest, format='dir', root=None):
119
122
           or at least passing a description to the constructor.
120
123
    """
121
124
    
122
 
    def __init__(self, store, inv):
123
 
        self._store = store
 
125
    def __init__(self, weave_store, inv, revision_id):
 
126
        self._weave_store = weave_store
124
127
        self._inventory = inv
 
128
        self._revision_id = revision_id
 
129
 
 
130
    def get_weave(self, file_id):
 
131
        # FIXME: RevisionTree should be given a branch
 
132
        # not a store, or the store should know the branch.
 
133
        import bzrlib.transactions as transactions
 
134
        return self._weave_store.get_weave(file_id,
 
135
            transactions.PassThroughTransaction())
 
136
 
 
137
 
 
138
    def get_file_lines(self, file_id):
 
139
        ie = self._inventory[file_id]
 
140
        weave = self.get_weave(file_id)
 
141
        return weave.get(ie.revision)
 
142
        
 
143
 
 
144
    def get_file_text(self, file_id):
 
145
        return ''.join(self.get_file_lines(file_id))
 
146
 
125
147
 
126
148
    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)
131
 
        return f
 
149
        return StringIO(self.get_file_text(file_id))
132
150
 
133
151
    def get_file_size(self, file_id):
134
152
        return self._inventory[file_id].text_size
135
153
 
136
154
    def get_file_sha1(self, file_id):
137
155
        ie = self._inventory[file_id]
138
 
        return ie.text_sha1
 
156
        if ie.kind == "file":
 
157
            return ie.text_sha1
 
158
 
 
159
    def is_executable(self, file_id):
 
160
        return self._inventory[file_id].executable
139
161
 
140
162
    def has_filename(self, filename):
141
163
        return bool(self.inventory.path2id(filename))
143
165
    def list_files(self):
144
166
        # The only files returned by this are those from the version
145
167
        for path, entry in self.inventory.iter_entries():
146
 
            yield path, 'V', entry.kind, entry.file_id
 
168
            yield path, 'V', entry.kind, entry.file_id, entry
 
169
 
 
170
    def get_symlink_target(self, file_id):
 
171
        ie = self._inventory[file_id]
 
172
        return ie.symlink_target;
147
173
 
148
174
 
149
175
class EmptyTree(Tree):
150
176
    def __init__(self):
151
177
        self._inventory = Inventory()
152
178
 
 
179
    def get_symlink_target(self, file_id):
 
180
        return None
 
181
 
153
182
    def has_filename(self, filename):
154
183
        return False
155
184
 
156
185
    def list_files(self):
157
 
        if False:  # just to make it a generator
158
 
            yield None
 
186
        return iter([])
159
187
    
 
188
    def __contains__(self, file_id):
 
189
        return file_id in self._inventory
 
190
 
 
191
    def get_file_sha1(self, file_id):
 
192
        assert self._inventory[file_id].kind == "root_directory"
 
193
        return None
160
194
 
161
195
 
162
196
######################################################################
245
279
    mutter('export version %r' % tree)
246
280
    inv = tree.inventory
247
281
    for dp, ie in inv.iter_entries():
248
 
        kind = ie.kind
249
 
        fullpath = appendpath(dest, dp)
250
 
        if kind == 'directory':
251
 
            os.mkdir(fullpath)
252
 
        elif kind == 'file':
253
 
            pumpfile(tree.get_file(ie.file_id), file(fullpath, 'wb'))
254
 
        else:
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))
 
282
        ie.put_on_disk(dest, dp, tree)
 
283
 
257
284
exporters['dir'] = dir_exporter
258
285
 
259
286
try:
302
329
        inv = tree.inventory
303
330
        for dp, ie in inv.iter_entries():
304
331
            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
308
 
            item.mtime = now
309
 
            if ie.kind == 'directory':
310
 
                item.type = tarfile.DIRTYPE
311
 
                fileobj = None
312
 
                item.name += '/'
313
 
                item.size = 0
314
 
                item.mode = 0755
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)
319
 
                item.mode = 0644
320
 
            else:
321
 
                raise BzrError("don't know how to export {%s} of kind %r" %
322
 
                        (ie.file_id, ie.kind))
323
 
 
 
332
            item, fileobj = ie.get_tar_item(root, dp, now, tree)
324
333
            ball.addfile(item, fileobj)
325
334
        ball.close()
 
335
 
326
336
    exporters['tar'] = tar_exporter
327
337
 
328
338
    def tgz_exporter(tree, dest, root):
332
342
    def tbz_exporter(tree, dest, root):
333
343
        tar_exporter(tree, dest, root, compression='bz2')
334
344
    exporters['tbz2'] = tbz_exporter
335
 
 
336
 
 
337
 
def _find_file_size(fileobj):
338
 
    offset = fileobj.tell()
339
 
    try:
340
 
        fileobj.seek(0, 2)
341
 
        size = fileobj.tell()
342
 
    except TypeError:
343
 
        # gzip doesn't accept second argument to seek()
344
 
        fileobj.seek(0)
345
 
        size = 0
346
 
        while True:
347
 
            nread = len(fileobj.read())
348
 
            if nread == 0:
349
 
                break
350
 
            size += nread
351
 
    fileobj.seek(offset)
352
 
    return size