~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: John Arbash Meinel
  • Date: 2009-02-25 21:13:22 UTC
  • mto: This revision was merged to the branch mainline in revision 4051.
  • Revision ID: john@arbash-meinel.com-20090225211322-qc94czk3s1g7nliq
Some direct tests for _group_keys_for_io

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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""MutableTree object.
18
18
 
23
23
from bzrlib.lazy_import import lazy_import
24
24
lazy_import(globals(), """
25
25
import os
26
 
import re
27
26
 
28
27
from bzrlib import (
29
28
    add,
30
29
    bzrdir,
 
30
    hooks,
 
31
    )
 
32
from bzrlib.osutils import dirname
 
33
from bzrlib.revisiontree import RevisionTree
 
34
from bzrlib.trace import mutter, warning
 
35
""")
 
36
 
 
37
from bzrlib import (
31
38
    errors,
32
 
    hooks,
33
39
    osutils,
34
 
    revisiontree,
35
 
    symbol_versioning,
36
 
    trace,
37
40
    tree,
38
41
    )
39
 
""")
40
 
 
41
42
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
43
from bzrlib.osutils import splitpath
 
44
from bzrlib.symbol_versioning import DEPRECATED_PARAMETER
42
45
 
43
46
 
44
47
def needs_tree_write_lock(unbound):
126
129
            # generic constraint checks:
127
130
            if self.is_control_filename(f):
128
131
                raise errors.ForbiddenControlFileError(filename=f)
129
 
            fp = osutils.splitpath(f)
 
132
            fp = splitpath(f)
130
133
        # fill out file kinds for all files [not needed when we stop
131
134
        # caring about the instantaneous file kind within a uncommmitted tree
132
135
        #
189
192
            revprops['branch-nick'] = self.branch._get_nick(
190
193
                kwargs.get('local', False),
191
194
                possible_master_transports)
192
 
        authors = kwargs.pop('authors', None)
193
195
        author = kwargs.pop('author', None)
194
 
        if authors is not None:
195
 
            if author is not None:
196
 
                raise AssertionError('Specifying both author and authors '
197
 
                        'is not allowed. Specify just authors instead')
198
 
            if 'author' in revprops or 'authors' in revprops:
199
 
                # XXX: maybe we should just accept one of them?
200
 
                raise AssertionError('author property given twice')
201
 
            if authors:
202
 
                for individual in authors:
203
 
                    if '\n' in individual:
204
 
                        raise AssertionError('\\n is not a valid character '
205
 
                                'in an author identity')
206
 
                revprops['authors'] = '\n'.join(authors)
207
196
        if author is not None:
208
 
            symbol_versioning.warn('The parameter author was deprecated'
209
 
                   ' in version 1.13. Use authors instead',
210
 
                   DeprecationWarning)
211
 
            if 'author' in revprops or 'authors' in revprops:
 
197
            if 'author' in revprops:
212
198
                # XXX: maybe we should just accept one of them?
213
199
                raise AssertionError('author property given twice')
214
 
            if '\n' in author:
215
 
                raise AssertionError('\\n is not a valid character '
216
 
                        'in an author identity')
217
 
            revprops['authors'] = author
 
200
            revprops['author'] = author
218
201
        # args for wt.commit start at message from the Commit.commit method,
219
202
        args = (message, ) + args
220
203
        for hook in MutableTree.hooks['start_commit']:
223
206
            revprops=revprops,
224
207
            possible_master_transports=possible_master_transports,
225
208
            *args, **kwargs)
226
 
        post_hook_params = PostCommitHookParams(self)
227
 
        for hook in MutableTree.hooks['post_commit']:
228
 
            hook(post_hook_params)
229
209
        return committed_id
230
210
 
231
211
    def _gather_kinds(self, files, kinds):
232
212
        """Helper function for add - sets the entries of kinds."""
233
213
        raise NotImplementedError(self._gather_kinds)
234
214
 
235
 
    @needs_read_lock
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).
241
 
 
242
 
        :return: True if a change is found. False otherwise
 
215
    def get_file_with_stat(self, file_id, path=None):
 
216
        """Get a file handle and stat object for file_id.
 
217
 
 
218
        The default implementation returns (self.get_file, None) for backwards
 
219
        compatibility.
 
220
 
 
221
        :param file_id: The file id to read.
 
222
        :param path: The path of the file, if it is known.
 
223
        :return: A tuple (file_handle, stat_value_or_None). If the tree has
 
224
            no stat facility, or need for a stat cache feedback during commit,
 
225
            it may return None for the second element of the tuple.
243
226
        """
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)
250
 
        try:
251
 
            change = changes.next()
252
 
            # Exclude root (talk about black magic... --vila 20090629)
253
 
            if change[4] == (None, None):
254
 
                change = changes.next()
255
 
            return True
256
 
        except StopIteration:
257
 
            # No changes
258
 
            return False
 
227
        return (self.get_file(file_id, path), None)
259
228
 
260
229
    @needs_read_lock
261
230
    def last_revision(self):
394
363
        # perform the canonicalization in bulk.
395
364
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
396
365
            rf = _FastPath(filepath)
397
 
            # validate user parameters. Our recursive code avoids adding new
398
 
            # files that need such validation
 
366
            # validate user parameters. Our recursive code avoids adding new files
 
367
            # that need such validation
399
368
            if self.is_control_filename(rf.raw_path):
400
369
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
401
370
 
407
376
            else:
408
377
                if not InventoryEntry.versionable_kind(kind):
409
378
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
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
 
379
            # ensure the named path is added, so that ignore rules in the later directory
 
380
            # walk dont skip it.
 
381
            # we dont have a parent ie known yet.: use the relatively slower inventory
 
382
            # probing method
414
383
            versioned = inv.has_filename(rf.raw_path)
415
384
            if versioned:
416
385
                continue
432
401
                dirs_to_add.append((path, None))
433
402
            prev_dir = path.raw_path
434
403
 
435
 
        illegalpath_re = re.compile(r'[\r\n]')
436
404
        # dirs_to_add is initialised to a list of directories, but as we scan
437
405
        # directories we append files to it.
438
406
        # XXX: We should determine kind of files when we scan them rather than
447
415
            kind = osutils.file_kind(abspath)
448
416
 
449
417
            if not InventoryEntry.versionable_kind(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)
 
418
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
455
419
                continue
456
420
 
457
421
            if parent_ie is not None:
480
444
                pass
481
445
                # mutter("%r is already versioned", abspath)
482
446
            elif sub_tree:
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)
 
447
                # XXX: This is wrong; people *might* reasonably be trying to add
 
448
                # subtrees as subtrees.  This should probably only be done in formats
 
449
                # which can represent subtrees, and even then perhaps only when
 
450
                # the user asked to add subtrees.  At the moment you can add them
 
451
                # specially through 'join --reference', which is perhaps
 
452
                # reasonable: adding a new reference is a special operation and
 
453
                # can have a special behaviour.  mbp 20070306
 
454
                mutter("%r is a nested bzr tree", abspath)
492
455
            else:
493
456
                _add_one(self, inv, parent_ie, directory, kind, action)
494
457
                added.append(directory.raw_path)
501
464
                    # without the parent ie, use the relatively slower inventory
502
465
                    # probing method
503
466
                    this_id = inv.path2id(
504
 
                        self._fix_case_of_inventory_path(directory.raw_path))
 
467
                            self._fix_case_of_inventory_path(directory.raw_path))
505
468
                    if this_id is None:
506
469
                        this_ie = None
507
470
                    else:
516
479
                    # faster - its impossible for a non root dir to have a
517
480
                    # control file.
518
481
                    if self.is_control_filename(subp):
519
 
                        trace.mutter("skip control directory %r", subp)
 
482
                        mutter("skip control directory %r", subp)
520
483
                    elif subf in this_ie.children:
521
484
                        # recurse into this already versioned subdir.
522
485
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
554
517
        for commit which is not required to handle situations that do not arise
555
518
        outside of commit.
556
519
 
557
 
        See the inventory developers documentation for the theory behind
558
 
        inventory deltas.
559
 
 
560
520
        :param new_revid: The new revision id for the trees parent.
561
521
        :param delta: An inventory delta (see apply_inventory_delta) describing
562
522
            the changes from the current left most parent revision to new_revid.
572
532
        # WorkingTree classes for optimised versions for specific format trees.
573
533
        basis = self.basis_tree()
574
534
        basis.lock_read()
575
 
        # TODO: Consider re-evaluating the need for this with CHKInventory
576
 
        # we don't strictly need to mutate an inventory for this
577
 
        # it only makes sense when apply_delta is cheaper than get_inventory()
578
 
        inventory = basis.inventory._get_mutable_inventory()
 
535
        inventory = basis.inventory
579
536
        basis.unlock()
580
537
        inventory.apply_delta(delta)
581
 
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
582
 
                                             inventory, new_revid)
 
538
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
583
539
        self.set_parent_trees([(new_revid, rev_tree)])
584
540
 
585
541
 
593
549
 
594
550
        """
595
551
        hooks.Hooks.__init__(self)
596
 
        self.create_hook(hooks.HookPoint('start_commit',
597
 
            "Called before a commit is performed on a tree. The start commit "
598
 
            "hook is able to change the tree before the commit takes place. "
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))
 
552
        # Invoked before a commit is done in a tree. New in 1.4
 
553
        self['start_commit'] = []
606
554
 
607
555
 
608
556
# install the default hooks into the MutableTree class.
609
557
MutableTree.hooks = MutableTreeHooks()
610
558
 
611
559
 
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
 
 
625
560
class _FastPath(object):
626
561
    """A path object with fast accessors for things like basename."""
627
562
 
668
603
        # there are a limited number of dirs we can be nested under, it should
669
604
        # generally find it very fast and not recurse after that.
670
605
        added = _add_one_and_parent(tree, inv, None,
671
 
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
672
 
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
 
606
            _FastPath(dirname(path.raw_path)), 'directory', action)
 
607
        parent_id = inv.path2id(dirname(path.raw_path))
673
608
        parent_ie = inv[parent_id]
674
609
    _add_one(tree, inv, parent_ie, path, kind, action)
675
610
    return added + [path.raw_path]