← 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-08-23 06:52:09 UTC
  • mto: (974.1.50) (1185.1.10) (1092.3.1)
  • mto: This revision was merged to the branch mainline in revision 1139.
  • Revision ID: robertc@robertcollins.net-20050823065209-81cd5962c401751b
move io redirection into each test case from the global runner

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tree.py
17
17
"""Tree classes, representing directory at point in time.
18
18
"""
19
19
 
20
 
from osutils import pumpfile, appendpath, fingerprint_file
 
20
import os
21
21
 
 
22
import bzrlib
22
23
from bzrlib.trace import mutter, note
23
24
from bzrlib.errors import BzrError
 
25
from bzrlib.inventory import Inventory
 
26
from bzrlib.osutils import pumpfile, appendpath, fingerprint_file
24
27
 
25
 
import bzrlib
26
28
 
27
29
exporters = {}
28
30
 
65
67
 
66
68
    def _get_inventory(self):
67
69
        return self._inventory
 
70
    
 
71
    def get_file_by_path(self, path):
 
72
        return self.get_file(self._inventory.path2id(path))
68
73
 
69
74
    inventory = property(_get_inventory,
70
75
                         doc="Inventory of this Tree")
93
98
        pumpfile(self.get_file(fileid), sys.stdout)
94
99
        
95
100
        
96
 
    def export(self, dest, format='dir'):
 
101
    def export(self, dest, format='dir', root=None):
97
102
        """Export this tree."""
98
103
        try:
99
104
            exporter = exporters[format]
100
105
        except KeyError:
 
106
            from bzrlib.errors import BzrCommandError
101
107
            raise BzrCommandError("export format %r not supported" % format)
102
 
        exporter(self, dest)
 
108
        exporter(self, dest, root)
103
109
 
104
110
 
105
111
 
129
135
 
130
136
    def get_file_sha1(self, file_id):
131
137
        ie = self._inventory[file_id]
132
 
        return ie.text_sha1
 
138
        if ie.kind == "file":
 
139
            return ie.text_sha1
133
140
 
134
141
    def has_filename(self, filename):
135
142
        return bool(self.inventory.path2id(filename))
142
149
 
143
150
class EmptyTree(Tree):
144
151
    def __init__(self):
145
 
        from bzrlib.inventory import Inventory
146
152
        self._inventory = Inventory()
147
153
 
148
154
    def has_filename(self, filename):
152
158
        if False:  # just to make it a generator
153
159
            yield None
154
160
    
 
161
    def __contains__(self, file_id):
 
162
        return file_id in self._inventory
 
163
 
 
164
    def get_file_sha1(self, file_id):
 
165
        assert self._inventory[file_id].kind == "root_directory"
 
166
        return None
 
167
 
 
168
 
155
169
 
156
170
 
157
171
######################################################################
223
237
######################################################################
224
238
# export
225
239
 
226
 
def dir_exporter(tree, dest):
 
240
def dir_exporter(tree, dest, root):
227
241
    """Export this tree to a new directory.
228
242
 
229
243
    `dest` should not exist, and will be created holding the
256
270
except ImportError:
257
271
    pass
258
272
else:
259
 
    def tar_exporter(tree, dest, compression=None):
 
273
    def get_root_name(dest):
 
274
        """Get just the root name for a tarball.
 
275
 
 
276
        >>> get_root_name('mytar.tar')
 
277
        'mytar'
 
278
        >>> get_root_name('mytar.tar.bz2')
 
279
        'mytar'
 
280
        >>> get_root_name('tar.tar.tar.tgz')
 
281
        'tar.tar.tar'
 
282
        >>> get_root_name('bzr-0.0.5.tar.gz')
 
283
        'bzr-0.0.5'
 
284
        >>> get_root_name('a/long/path/mytar.tgz')
 
285
        'mytar'
 
286
        >>> get_root_name('../parent/../dir/other.tbz2')
 
287
        'other'
 
288
        """
 
289
        endings = ['.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
 
290
        dest = os.path.basename(dest)
 
291
        for end in endings:
 
292
            if dest.endswith(end):
 
293
                return dest[:-len(end)]
 
294
 
 
295
    def tar_exporter(tree, dest, root, compression=None):
260
296
        """Export this tree to a new tar file.
261
297
 
262
298
        `dest` will be created holding the contents of this tree; if it
265
301
        from time import time
266
302
        now = time()
267
303
        compression = str(compression or '')
 
304
        if root is None:
 
305
            root = get_root_name(dest)
268
306
        try:
269
307
            ball = tarfile.open(dest, 'w:' + compression)
270
308
        except tarfile.CompressionError, e:
273
311
        inv = tree.inventory
274
312
        for dp, ie in inv.iter_entries():
275
313
            mutter("  export {%s} kind %s to %s" % (ie.file_id, ie.kind, dest))
276
 
            item = tarfile.TarInfo(dp)
 
314
            item = tarfile.TarInfo(os.path.join(root, dp))
277
315
            # TODO: would be cool to actually set it to the timestamp of the
278
316
            # revision it was last changed
279
317
            item.mtime = now
296
334
        ball.close()
297
335
    exporters['tar'] = tar_exporter
298
336
 
299
 
    def tgz_exporter(tree, dest):
300
 
        tar_exporter(tree, dest, compression='gz')
 
337
    def tgz_exporter(tree, dest, root):
 
338
        tar_exporter(tree, dest, root, compression='gz')
301
339
    exporters['tgz'] = tgz_exporter
302
340
 
303
 
    def tbz_exporter(tree, dest):
304
 
        tar_exporter(tree, dest, compression='bz2')
 
341
    def tbz_exporter(tree, dest, root):
 
342
        tar_exporter(tree, dest, root, compression='bz2')
305
343
    exporters['tbz2'] = tbz_exporter
306
344
 
307
345