~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: Vincent Ladeuil
  • Date: 2010-02-11 09:27:55 UTC
  • mfrom: (5017.3.46 test-servers)
  • mto: This revision was merged to the branch mainline in revision 5030.
  • Revision ID: v.ladeuil+lp@free.fr-20100211092755-3vvu4vbwiwjjte3s
Move tests servers from bzrlib.transport to bzrlib.tests.test_server

Show diffs side-by-side

added added

removed removed

Lines of Context:
23
23
from bzrlib.lazy_import import lazy_import
24
24
lazy_import(globals(), """
25
25
import os
 
26
import re
26
27
 
27
28
from bzrlib import (
28
29
    add,
29
30
    bzrdir,
 
31
    errors,
30
32
    hooks,
 
33
    osutils,
 
34
    revisiontree,
31
35
    symbol_versioning,
 
36
    trace,
 
37
    tree,
32
38
    )
33
 
from bzrlib.osutils import dirname
34
 
from bzrlib.revisiontree import RevisionTree
35
 
from bzrlib.trace import mutter, warning
36
39
""")
37
40
 
38
 
from bzrlib import (
39
 
    errors,
40
 
    osutils,
41
 
    tree,
42
 
    )
43
41
from bzrlib.decorators import needs_read_lock, needs_write_lock
44
 
from bzrlib.osutils import splitpath
45
42
 
46
43
 
47
44
def needs_tree_write_lock(unbound):
129
126
            # generic constraint checks:
130
127
            if self.is_control_filename(f):
131
128
                raise errors.ForbiddenControlFileError(filename=f)
132
 
            fp = splitpath(f)
 
129
            fp = osutils.splitpath(f)
133
130
        # fill out file kinds for all files [not needed when we stop
134
131
        # caring about the instantaneous file kind within a uncommmitted tree
135
132
        #
226
223
            revprops=revprops,
227
224
            possible_master_transports=possible_master_transports,
228
225
            *args, **kwargs)
 
226
        post_hook_params = PostCommitHookParams(self)
 
227
        for hook in MutableTree.hooks['post_commit']:
 
228
            hook(post_hook_params)
229
229
        return committed_id
230
230
 
231
231
    def _gather_kinds(self, files, kinds):
233
233
        raise NotImplementedError(self._gather_kinds)
234
234
 
235
235
    @needs_read_lock
236
 
    def has_changes(self, from_tree):
237
 
        """Quickly check that the tree contains at least one change.
 
236
    def has_changes(self, _from_tree=None):
 
237
        """Quickly check that the tree contains at least one commitable change.
 
238
 
 
239
        :param _from_tree: tree to compare against to find changes (default to
 
240
            the basis tree and is intended to be used by tests).
238
241
 
239
242
        :return: True if a change is found. False otherwise
240
243
        """
241
 
        changes = self.iter_changes(from_tree)
 
244
        # Check pending merges
 
245
        if len(self.get_parent_ids()) > 1:
 
246
            return True
 
247
        if _from_tree is None:
 
248
            _from_tree = self.basis_tree()
 
249
        changes = self.iter_changes(_from_tree)
242
250
        try:
243
251
            change = changes.next()
244
252
            # Exclude root (talk about black magic... --vila 20090629)
386
394
        # perform the canonicalization in bulk.
387
395
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
388
396
            rf = _FastPath(filepath)
389
 
            # validate user parameters. Our recursive code avoids adding new files
390
 
            # that need such validation
 
397
            # validate user parameters. Our recursive code avoids adding new
 
398
            # files that need such validation
391
399
            if self.is_control_filename(rf.raw_path):
392
400
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
393
401
 
399
407
            else:
400
408
                if not InventoryEntry.versionable_kind(kind):
401
409
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
402
 
            # ensure the named path is added, so that ignore rules in the later directory
403
 
            # walk dont skip it.
404
 
            # we dont have a parent ie known yet.: use the relatively slower inventory
405
 
            # probing method
 
410
            # ensure the named path is added, so that ignore rules in the later
 
411
            # directory walk dont skip it.
 
412
            # we dont have a parent ie known yet.: use the relatively slower
 
413
            # inventory probing method
406
414
            versioned = inv.has_filename(rf.raw_path)
407
415
            if versioned:
408
416
                continue
424
432
                dirs_to_add.append((path, None))
425
433
            prev_dir = path.raw_path
426
434
 
 
435
        illegalpath_re = re.compile(r'[\r\n]')
427
436
        # dirs_to_add is initialised to a list of directories, but as we scan
428
437
        # directories we append files to it.
429
438
        # XXX: We should determine kind of files when we scan them rather than
438
447
            kind = osutils.file_kind(abspath)
439
448
 
440
449
            if not InventoryEntry.versionable_kind(kind):
441
 
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
 
450
                trace.warning("skipping %s (can't add file of kind '%s')",
 
451
                              abspath, kind)
 
452
                continue
 
453
            if illegalpath_re.search(directory.raw_path):
 
454
                trace.warning("skipping %r (contains \\n or \\r)" % abspath)
442
455
                continue
443
456
 
444
457
            if parent_ie is not None:
467
480
                pass
468
481
                # mutter("%r is already versioned", abspath)
469
482
            elif sub_tree:
470
 
                # XXX: This is wrong; people *might* reasonably be trying to add
471
 
                # subtrees as subtrees.  This should probably only be done in formats
472
 
                # which can represent subtrees, and even then perhaps only when
473
 
                # the user asked to add subtrees.  At the moment you can add them
474
 
                # specially through 'join --reference', which is perhaps
475
 
                # reasonable: adding a new reference is a special operation and
476
 
                # can have a special behaviour.  mbp 20070306
477
 
                mutter("%r is a nested bzr tree", abspath)
 
483
                # XXX: This is wrong; people *might* reasonably be trying to
 
484
                # add subtrees as subtrees.  This should probably only be done
 
485
                # in formats which can represent subtrees, and even then
 
486
                # perhaps only when the user asked to add subtrees.  At the
 
487
                # moment you can add them specially through 'join --reference',
 
488
                # which is perhaps reasonable: adding a new reference is a
 
489
                # special operation and can have a special behaviour.  mbp
 
490
                # 20070306
 
491
                trace.mutter("%r is a nested bzr tree", abspath)
478
492
            else:
479
493
                _add_one(self, inv, parent_ie, directory, kind, action)
480
494
                added.append(directory.raw_path)
487
501
                    # without the parent ie, use the relatively slower inventory
488
502
                    # probing method
489
503
                    this_id = inv.path2id(
490
 
                            self._fix_case_of_inventory_path(directory.raw_path))
 
504
                        self._fix_case_of_inventory_path(directory.raw_path))
491
505
                    if this_id is None:
492
506
                        this_ie = None
493
507
                    else:
502
516
                    # faster - its impossible for a non root dir to have a
503
517
                    # control file.
504
518
                    if self.is_control_filename(subp):
505
 
                        mutter("skip control directory %r", subp)
 
519
                        trace.mutter("skip control directory %r", subp)
506
520
                    elif subf in this_ie.children:
507
521
                        # recurse into this already versioned subdir.
508
522
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
564
578
        inventory = basis.inventory._get_mutable_inventory()
565
579
        basis.unlock()
566
580
        inventory.apply_delta(delta)
567
 
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
 
581
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
 
582
                                             inventory, new_revid)
568
583
        self.set_parent_trees([(new_revid, rev_tree)])
569
584
 
570
585
 
581
596
        self.create_hook(hooks.HookPoint('start_commit',
582
597
            "Called before a commit is performed on a tree. The start commit "
583
598
            "hook is able to change the tree before the commit takes place. "
584
 
            "start_commit is called with the bzrlib.tree.MutableTree that the "
585
 
            "commit is being performed on.", (1, 4), None))
 
599
            "start_commit is called with the bzrlib.mutabletree.MutableTree "
 
600
            "that the commit is being performed on.", (1, 4), None))
 
601
        self.create_hook(hooks.HookPoint('post_commit',
 
602
            "Called after a commit is performed on a tree. The hook is "
 
603
            "called with a bzrlib.mutabletree.PostCommitHookParams object. "
 
604
            "The mutable tree the commit was performed on is available via "
 
605
            "the mutable_tree attribute of that object.", (2, 0), None))
586
606
 
587
607
 
588
608
# install the default hooks into the MutableTree class.
589
609
MutableTree.hooks = MutableTreeHooks()
590
610
 
591
611
 
 
612
class PostCommitHookParams(object):
 
613
    """Parameters for the post_commit hook.
 
614
 
 
615
    To access the parameters, use the following attributes:
 
616
 
 
617
    * mutable_tree - the MutableTree object
 
618
    """
 
619
 
 
620
    def __init__(self, mutable_tree):
 
621
        """Create the parameters for the post_commit hook."""
 
622
        self.mutable_tree = mutable_tree
 
623
 
 
624
 
592
625
class _FastPath(object):
593
626
    """A path object with fast accessors for things like basename."""
594
627
 
635
668
        # there are a limited number of dirs we can be nested under, it should
636
669
        # generally find it very fast and not recurse after that.
637
670
        added = _add_one_and_parent(tree, inv, None,
638
 
            _FastPath(dirname(path.raw_path)), 'directory', action)
639
 
        parent_id = inv.path2id(dirname(path.raw_path))
 
671
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
 
672
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
640
673
        parent_ie = inv[parent_id]
641
674
    _add_one(tree, inv, parent_ie, path, kind, action)
642
675
    return added + [path.raw_path]