← 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-09-28 04:58:18 UTC
  • mto: (1092.2.19)
  • mto: This revision was merged to the branch mainline in revision 1391.
  • Revision ID: robertc@robertcollins.net-20050928045818-c5ce6c7cc796f6fc
patch from Rob Weir to correct bzr-man.py

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 = {}
58
57
    def has_id(self, file_id):
59
58
        return self.inventory.has_id(file_id)
60
59
 
61
 
    def has_or_had_id(self, file_id):
62
 
        if file_id == self.inventory.root.file_id:
63
 
            return True
64
 
        return self.inventory.has_id(file_id)
65
 
 
66
60
    __contains__ = has_id
67
61
 
68
62
    def __iter__(self):
71
65
    def id2path(self, file_id):
72
66
        return self.inventory.id2path(file_id)
73
67
 
74
 
    def kind(self, file_id):
75
 
        raise NotImplementedError("subclasses must implement kind")
76
 
 
77
68
    def _get_inventory(self):
78
69
        return self._inventory
79
70
    
84
75
                         doc="Inventory of this Tree")
85
76
 
86
77
    def _check_retrieved(self, ie, f):
87
 
        if not __debug__:
88
 
            return  
89
78
        fp = fingerprint_file(f)
90
79
        f.seek(0)
91
80
        
103
92
                     "store is probably damaged/corrupt"])
104
93
 
105
94
 
106
 
    def print_file(self, file_id):
107
 
        """Print file with id `file_id` to stdout."""
 
95
    def print_file(self, fileid):
 
96
        """Print file with id `fileid` to stdout."""
108
97
        import sys
109
 
        sys.stdout.write(self.get_file_text(file_id))
 
98
        pumpfile(self.get_file(fileid), sys.stdout)
110
99
        
111
100
        
112
101
    def export(self, dest, format='dir', root=None):
130
119
           or at least passing a description to the constructor.
131
120
    """
132
121
    
133
 
    def __init__(self, weave_store, inv, revision_id):
134
 
        self._weave_store = weave_store
 
122
    def __init__(self, store, inv):
 
123
        self._store = store
135
124
        self._inventory = inv
136
 
        self._revision_id = revision_id
137
 
 
138
 
    def get_weave(self, file_id):
139
 
        # FIXME: RevisionTree should be given a branch
140
 
        # not a store, or the store should know the branch.
141
 
        import bzrlib.transactions as transactions
142
 
        return self._weave_store.get_weave(file_id,
143
 
            transactions.PassThroughTransaction())
144
 
 
145
 
 
146
 
    def get_file_lines(self, file_id):
147
 
        ie = self._inventory[file_id]
148
 
        weave = self.get_weave(file_id)
149
 
        return weave.get(ie.revision)
150
 
        
151
 
 
152
 
    def get_file_text(self, file_id):
153
 
        return ''.join(self.get_file_lines(file_id))
154
 
 
155
125
 
156
126
    def get_file(self, file_id):
157
 
        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
158
132
 
159
133
    def get_file_size(self, file_id):
160
134
        return self._inventory[file_id].text_size
164
138
        if ie.kind == "file":
165
139
            return ie.text_sha1
166
140
 
167
 
    def is_executable(self, file_id):
168
 
        ie = self._inventory[file_id]
169
 
        if ie.kind != "file":
170
 
            return None 
171
 
        return self._inventory[file_id].executable
172
 
 
173
141
    def has_filename(self, filename):
174
142
        return bool(self.inventory.path2id(filename))
175
143
 
176
144
    def list_files(self):
177
145
        # The only files returned by this are those from the version
178
146
        for path, entry in self.inventory.iter_entries():
179
 
            yield path, 'V', entry.kind, entry.file_id, entry
180
 
 
181
 
    def get_symlink_target(self, file_id):
182
 
        ie = self._inventory[file_id]
183
 
        return ie.symlink_target;
184
 
 
185
 
    def kind(self, file_id):
186
 
        return self._inventory[file_id].kind
 
147
            yield path, 'V', entry.kind, entry.file_id
187
148
 
188
149
 
189
150
class EmptyTree(Tree):
190
151
    def __init__(self):
191
152
        self._inventory = Inventory()
192
153
 
193
 
    def get_symlink_target(self, file_id):
194
 
        return None
195
 
 
196
154
    def has_filename(self, filename):
197
155
        return False
198
156
 
199
 
    def kind(self, file_id):
200
 
        assert self._inventory[file_id].kind == "root_directory"
201
 
        return "root_directory"
202
 
 
203
157
    def list_files(self):
204
 
        return iter([])
 
158
        if False:  # just to make it a generator
 
159
            yield None
205
160
    
206
161
    def __contains__(self, file_id):
207
162
        return file_id in self._inventory
211
166
        return None
212
167
 
213
168
 
 
169
 
 
170
 
214
171
######################################################################
215
172
# diff
216
173
 
297
254
    mutter('export version %r' % tree)
298
255
    inv = tree.inventory
299
256
    for dp, ie in inv.iter_entries():
300
 
        ie.put_on_disk(dest, dp, tree)
301
 
 
 
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))
302
266
exporters['dir'] = dir_exporter
303
267
 
304
268
try:
347
311
        inv = tree.inventory
348
312
        for dp, ie in inv.iter_entries():
349
313
            mutter("  export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
350
 
            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
 
351
333
            ball.addfile(item, fileobj)
352
334
        ball.close()
353
 
 
354
335
    exporters['tar'] = tar_exporter
355
336
 
356
337
    def tgz_exporter(tree, dest, root):
360
341
    def tbz_exporter(tree, dest, root):
361
342
        tar_exporter(tree, dest, root, compression='bz2')
362
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