~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: Vincent Ladeuil
  • Date: 2009-06-22 12:52:39 UTC
  • mto: (4471.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4472.
  • Revision ID: v.ladeuil+lp@free.fr-20090622125239-kabo9smxt9c3vnir
Use a consistent scheme for naming pyrex source files.

Show diffs side-by-side

added added

removed removed

Lines of Context:
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""MutableTree object.
18
18
 
28
28
    add,
29
29
    bzrdir,
30
30
    hooks,
 
31
    symbol_versioning,
31
32
    )
32
33
from bzrlib.osutils import dirname
33
34
from bzrlib.revisiontree import RevisionTree
41
42
    )
42
43
from bzrlib.decorators import needs_read_lock, needs_write_lock
43
44
from bzrlib.osutils import splitpath
44
 
from bzrlib.symbol_versioning import DEPRECATED_PARAMETER
45
45
 
46
46
 
47
47
def needs_tree_write_lock(unbound):
69
69
    entirely in memory.
70
70
 
71
71
    For now, we are not treating MutableTree as an interface to provide
72
 
    conformance tests for - rather we are testing MemoryTree specifically, and 
 
72
    conformance tests for - rather we are testing MemoryTree specifically, and
73
73
    interface testing implementations of WorkingTree.
74
74
 
75
75
    A mutable tree always has an associated Branch and BzrDir object - the
76
76
    branch and bzrdir attributes.
77
77
    """
 
78
    def __init__(self, *args, **kw):
 
79
        super(MutableTree, self).__init__(*args, **kw)
 
80
        # Is this tree on a case-insensitive or case-preserving file-system?
 
81
        # Sub-classes may initialize to False if they detect they are being
 
82
        # used on media which doesn't differentiate the case of names.
 
83
        self.case_sensitive = True
78
84
 
79
85
    @needs_tree_write_lock
80
86
    def add(self, files, ids=None, kinds=None):
124
130
            if self.is_control_filename(f):
125
131
                raise errors.ForbiddenControlFileError(filename=f)
126
132
            fp = splitpath(f)
127
 
        # fill out file kinds for all files [not needed when we stop 
 
133
        # fill out file kinds for all files [not needed when we stop
128
134
        # caring about the instantaneous file kind within a uncommmitted tree
129
135
        #
130
136
        self._gather_kinds(files, kinds)
181
187
        from bzrlib import commit
182
188
        if revprops is None:
183
189
            revprops = {}
 
190
        possible_master_transports=[]
184
191
        if not 'branch-nick' in revprops:
185
 
            revprops['branch-nick'] = self.branch.nick
 
192
            revprops['branch-nick'] = self.branch._get_nick(
 
193
                kwargs.get('local', False),
 
194
                possible_master_transports)
 
195
        authors = kwargs.pop('authors', None)
186
196
        author = kwargs.pop('author', None)
 
197
        if authors is not None:
 
198
            if author is not None:
 
199
                raise AssertionError('Specifying both author and authors '
 
200
                        'is not allowed. Specify just authors instead')
 
201
            if 'author' in revprops or 'authors' in revprops:
 
202
                # XXX: maybe we should just accept one of them?
 
203
                raise AssertionError('author property given twice')
 
204
            if authors:
 
205
                for individual in authors:
 
206
                    if '\n' in individual:
 
207
                        raise AssertionError('\\n is not a valid character '
 
208
                                'in an author identity')
 
209
                revprops['authors'] = '\n'.join(authors)
187
210
        if author is not None:
188
 
            if 'author' in revprops:
 
211
            symbol_versioning.warn('The parameter author was deprecated'
 
212
                   ' in version 1.13. Use authors instead',
 
213
                   DeprecationWarning)
 
214
            if 'author' in revprops or 'authors' in revprops:
189
215
                # XXX: maybe we should just accept one of them?
190
216
                raise AssertionError('author property given twice')
191
 
            revprops['author'] = author
 
217
            if '\n' in author:
 
218
                raise AssertionError('\\n is not a valid character '
 
219
                        'in an author identity')
 
220
            revprops['authors'] = author
192
221
        # args for wt.commit start at message from the Commit.commit method,
193
222
        args = (message, ) + args
194
223
        for hook in MutableTree.hooks['start_commit']:
195
224
            hook(self)
196
225
        committed_id = commit.Commit().commit(working_tree=self,
197
 
            revprops=revprops, *args, **kwargs)
 
226
            revprops=revprops,
 
227
            possible_master_transports=possible_master_transports,
 
228
            *args, **kwargs)
198
229
        return committed_id
199
230
 
200
231
    def _gather_kinds(self, files, kinds):
208
239
        In early tree formats the result of last_revision is the same as the
209
240
        branch last_revision, but that is no longer the case for modern tree
210
241
        formats.
211
 
        
 
242
 
212
243
        last_revision returns the left most parent id, or None if there are no
213
244
        parents.
214
245
 
231
262
    def lock_write(self):
232
263
        """Lock the tree and its branch. This allows mutating calls to be made.
233
264
 
234
 
        Some mutating methods will take out implicit write locks, but in 
 
265
        Some mutating methods will take out implicit write locks, but in
235
266
        general you should always obtain a write lock before calling mutating
236
267
        methods on a tree.
237
268
        """
247
278
        """
248
279
        raise NotImplementedError(self.mkdir)
249
280
 
 
281
    def _observed_sha1(self, file_id, path, (sha1, stat_value)):
 
282
        """Tell the tree we have observed a paths sha1.
 
283
 
 
284
        The intent of this function is to allow trees that have a hashcache to
 
285
        update the hashcache during commit. If the observed file is too new
 
286
        (based on the stat_value) to be safely hash-cached the tree will ignore
 
287
        it.
 
288
 
 
289
        The default implementation does nothing.
 
290
 
 
291
        :param file_id: The file id
 
292
        :param path: The file path
 
293
        :param sha1: The sha 1 that was observed.
 
294
        :param stat_value: A stat result for the file the sha1 was read from.
 
295
        :return: None
 
296
        """
 
297
 
 
298
    def _fix_case_of_inventory_path(self, path):
 
299
        """If our tree isn't case sensitive, return the canonical path"""
 
300
        if not self.case_sensitive:
 
301
            path = self.get_canonical_inventory_path(path)
 
302
        return path
 
303
 
250
304
    @needs_write_lock
251
305
    def put_file_bytes_non_atomic(self, file_id, bytes):
252
306
        """Update the content of a file in the tree.
253
 
        
 
307
 
254
308
        Note that the file is written in-place rather than being
255
309
        written to a temporary location and renamed. As a consequence,
256
310
        readers can potentially see the file half-written.
270
324
    def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
271
325
        """Set the parents of the working tree.
272
326
 
273
 
        :param parents_list: A list of (revision_id, tree) tuples. 
 
327
        :param parents_list: A list of (revision_id, tree) tuples.
274
328
            If tree is None, then that element is treated as an unreachable
275
329
            parent tree - i.e. a ghost.
276
330
        """
284
338
        For the specific behaviour see the help for cmd_add().
285
339
 
286
340
        :param action: A reporter to be called with the inventory, parent_ie,
287
 
            path and kind of the path being added. It may return a file_id if 
 
341
            path and kind of the path being added. It may return a file_id if
288
342
            a specific one should be used.
289
343
        :param save: Save the inventory after completing the adds. If False
290
344
            this provides dry-run functionality by doing the add and not saving
298
352
        from bzrlib.inventory import InventoryEntry
299
353
        if action is None:
300
354
            action = add.AddAction()
301
 
        
 
355
 
302
356
        if not file_list:
303
357
            # no paths supplied: add the entire tree.
304
358
            file_list = [u'.']
309
363
        dirs_to_add = []
310
364
        user_dirs = set()
311
365
 
312
 
        # validate user file paths and convert all paths to tree 
 
366
        # validate user file paths and convert all paths to tree
313
367
        # relative : it's cheaper to make a tree relative path an abspath
314
 
        # than to convert an abspath to tree relative.
315
 
        for filepath in file_list:
316
 
            rf = _FastPath(self.relpath(filepath))
 
368
        # than to convert an abspath to tree relative, and it's cheaper to
 
369
        # perform the canonicalization in bulk.
 
370
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
 
371
            rf = _FastPath(filepath)
317
372
            # validate user parameters. Our recursive code avoids adding new files
318
 
            # that need such validation 
 
373
            # that need such validation
319
374
            if self.is_control_filename(rf.raw_path):
320
375
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
321
 
            
 
376
 
322
377
            abspath = self.abspath(rf.raw_path)
323
378
            kind = osutils.file_kind(abspath)
324
379
            if kind == 'directory':
329
384
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
330
385
            # ensure the named path is added, so that ignore rules in the later directory
331
386
            # walk dont skip it.
332
 
            # we dont have a parent ie known yet.: use the relatively slower inventory 
 
387
            # we dont have a parent ie known yet.: use the relatively slower inventory
333
388
            # probing method
334
389
            versioned = inv.has_filename(rf.raw_path)
335
390
            if versioned:
372
427
            if parent_ie is not None:
373
428
                versioned = directory.base_path in parent_ie.children
374
429
            else:
375
 
                # without the parent ie, use the relatively slower inventory 
 
430
                # without the parent ie, use the relatively slower inventory
376
431
                # probing method
377
 
                versioned = inv.has_filename(directory.raw_path)
 
432
                versioned = inv.has_filename(
 
433
                        self._fix_case_of_inventory_path(directory.raw_path))
378
434
 
379
435
            if kind == 'directory':
380
436
                try:
395
451
                # mutter("%r is already versioned", abspath)
396
452
            elif sub_tree:
397
453
                # XXX: This is wrong; people *might* reasonably be trying to add
398
 
                # subtrees as subtrees.  This should probably only be done in formats 
 
454
                # subtrees as subtrees.  This should probably only be done in formats
399
455
                # which can represent subtrees, and even then perhaps only when
400
456
                # the user asked to add subtrees.  At the moment you can add them
401
457
                # specially through 'join --reference', which is perhaps
411
467
                    # must be present:
412
468
                    this_ie = parent_ie.children[directory.base_path]
413
469
                else:
414
 
                    # without the parent ie, use the relatively slower inventory 
 
470
                    # without the parent ie, use the relatively slower inventory
415
471
                    # probing method
416
 
                    this_id = inv.path2id(directory.raw_path)
 
472
                    this_id = inv.path2id(
 
473
                            self._fix_case_of_inventory_path(directory.raw_path))
417
474
                    if this_id is None:
418
475
                        this_ie = None
419
476
                    else:
420
477
                        this_ie = inv[this_id]
421
478
 
422
479
                for subf in sorted(os.listdir(abspath)):
423
 
                    # here we could use TreeDirectory rather than 
 
480
                    # here we could use TreeDirectory rather than
424
481
                    # string concatenation.
425
482
                    subp = osutils.pathjoin(directory.raw_path, subf)
426
 
                    # TODO: is_control_filename is very slow. Make it faster. 
427
 
                    # TreeDirectory.is_control_filename could also make this 
428
 
                    # faster - its impossible for a non root dir to have a 
 
483
                    # TODO: is_control_filename is very slow. Make it faster.
 
484
                    # TreeDirectory.is_control_filename could also make this
 
485
                    # faster - its impossible for a non root dir to have a
429
486
                    # control file.
430
487
                    if self.is_control_filename(subp):
431
488
                        mutter("skip control directory %r", subp)
481
538
        # WorkingTree classes for optimised versions for specific format trees.
482
539
        basis = self.basis_tree()
483
540
        basis.lock_read()
484
 
        inventory = basis.inventory
 
541
        # TODO: Consider re-evaluating the need for this with CHKInventory
 
542
        # we don't strictly need to mutate an inventory for this
 
543
        # it only makes sense when apply_delta is cheaper than get_inventory()
 
544
        inventory = basis.inventory._get_mutable_inventory()
485
545
        basis.unlock()
486
546
        inventory.apply_delta(delta)
487
547
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
489
549
 
490
550
 
491
551
class MutableTreeHooks(hooks.Hooks):
492
 
    """A dictionary mapping a hook name to a list of callables for mutabletree 
 
552
    """A dictionary mapping a hook name to a list of callables for mutabletree
493
553
    hooks.
494
554
    """
495
555
 
498
558
 
499
559
        """
500
560
        hooks.Hooks.__init__(self)
501
 
        # Invoked before a commit is done in a tree. New in 1.4
502
 
        self['start_commit'] = []
 
561
        self.create_hook(hooks.HookPoint('start_commit',
 
562
            "Called before a commit is performed on a tree. The start commit "
 
563
            "hook is able to change the tree before the commit takes place. "
 
564
            "start_commit is called with the bzrlib.tree.MutableTree that the "
 
565
            "commit is being performed on.", (1, 4), None))
503
566
 
504
567
 
505
568
# install the default hooks into the MutableTree class.
545
608
        added = []
546
609
    else:
547
610
        # slower but does not need parent_ie
548
 
        if inv.has_filename(path.raw_path):
 
611
        if inv.has_filename(tree._fix_case_of_inventory_path(path.raw_path)):
549
612
            return []
550
613
        # its really not there : add the parent
551
614
        # note that the dirname use leads to some extra str copying etc but as