← 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-01 11:53:02 UTC
  • Revision ID: mbp@sourcefrog.net-20050901115302-2fcc6c750f0abe34
- make external commands work again

  code is now much simpler; no translation to objects and back again

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
 
        # 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
 
 
147
125
 
148
126
    def get_file(self, file_id):
149
 
        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
150
132
 
151
133
    def get_file_size(self, file_id):
152
134
        return self._inventory[file_id].text_size
156
138
        if ie.kind == "file":
157
139
            return ie.text_sha1
158
140
 
159
 
    def is_executable(self, file_id):
160
 
        return self._inventory[file_id].executable
161
 
 
162
141
    def has_filename(self, filename):
163
142
        return bool(self.inventory.path2id(filename))
164
143
 
165
144
    def list_files(self):
166
145
        # The only files returned by this are those from the version
167
146
        for path, entry in self.inventory.iter_entries():
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
            yield path, 'V', entry.kind, entry.file_id
173
148
 
174
149
 
175
150
class EmptyTree(Tree):
176
151
    def __init__(self):
177
152
        self._inventory = Inventory()
178
153
 
179
 
    def get_symlink_target(self, file_id):
180
 
        return None
181
 
 
182
154
    def has_filename(self, filename):
183
155
        return False
184
156
 
185
157
    def list_files(self):
186
 
        return iter([])
 
158
        if False:  # just to make it a generator
 
159
            yield None
187
160
    
188
161
    def __contains__(self, file_id):
189
162
        return file_id in self._inventory
193
166
        return None
194
167
 
195
168
 
 
169
 
 
170
 
196
171
######################################################################
197
172
# diff
198
173
 
279
254
    mutter('export version %r' % tree)
280
255
    inv = tree.inventory
281
256
    for dp, ie in inv.iter_entries():
282
 
        ie.put_on_disk(dest, dp, tree)
283
 
 
 
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))
284
266
exporters['dir'] = dir_exporter
285
267
 
286
268
try:
329
311
        inv = tree.inventory
330
312
        for dp, ie in inv.iter_entries():
331
313
            mutter("  export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
332
 
            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
 
333
333
            ball.addfile(item, fileobj)
334
334
        ball.close()
335
 
 
336
335
    exporters['tar'] = tar_exporter
337
336
 
338
337
    def tgz_exporter(tree, dest, root):
342
341
    def tbz_exporter(tree, dest, root):
343
342
        tar_exporter(tree, dest, root, compression='bz2')
344
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