← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tree.py

add a clean target

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