~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
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,
 
35
    inventory,
31
36
    symbol_versioning,
 
37
    trace,
 
38
    tree,
32
39
    )
33
 
from bzrlib.osutils import dirname
34
 
from bzrlib.revisiontree import RevisionTree
35
 
from bzrlib.trace import mutter, warning
36
40
""")
37
41
 
38
 
from bzrlib import (
39
 
    errors,
40
 
    osutils,
41
 
    tree,
42
 
    )
43
42
from bzrlib.decorators import needs_read_lock, needs_write_lock
44
 
from bzrlib.osutils import splitpath
45
43
 
46
44
 
47
45
def needs_tree_write_lock(unbound):
129
127
            # generic constraint checks:
130
128
            if self.is_control_filename(f):
131
129
                raise errors.ForbiddenControlFileError(filename=f)
132
 
            fp = splitpath(f)
 
130
            fp = osutils.splitpath(f)
133
131
        # fill out file kinds for all files [not needed when we stop
134
132
        # caring about the instantaneous file kind within a uncommmitted tree
135
133
        #
185
183
               **kwargs):
186
184
        # avoid circular imports
187
185
        from bzrlib import commit
188
 
        if revprops is None:
189
 
            revprops = {}
190
186
        possible_master_transports=[]
191
 
        if not 'branch-nick' in revprops:
192
 
            revprops['branch-nick'] = self.branch._get_nick(
 
187
        revprops = commit.Commit.update_revprops(
 
188
                revprops,
 
189
                self.branch,
 
190
                kwargs.pop('authors', None),
 
191
                kwargs.pop('author', None),
193
192
                kwargs.get('local', False),
194
193
                possible_master_transports)
195
 
        authors = kwargs.pop('authors', None)
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)
210
 
        if author is not None:
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:
215
 
                # XXX: maybe we should just accept one of them?
216
 
                raise AssertionError('author property given twice')
217
 
            if '\n' in author:
218
 
                raise AssertionError('\\n is not a valid character '
219
 
                        'in an author identity')
220
 
            revprops['authors'] = author
221
194
        # args for wt.commit start at message from the Commit.commit method,
222
195
        args = (message, ) + args
223
196
        for hook in MutableTree.hooks['start_commit']:
226
199
            revprops=revprops,
227
200
            possible_master_transports=possible_master_transports,
228
201
            *args, **kwargs)
 
202
        post_hook_params = PostCommitHookParams(self)
 
203
        for hook in MutableTree.hooks['post_commit']:
 
204
            hook(post_hook_params)
229
205
        return committed_id
230
206
 
231
207
    def _gather_kinds(self, files, kinds):
233
209
        raise NotImplementedError(self._gather_kinds)
234
210
 
235
211
    @needs_read_lock
236
 
    def has_changes(self, from_tree):
237
 
        """Quickly check that the tree contains at least one change.
 
212
    def has_changes(self, _from_tree=None):
 
213
        """Quickly check that the tree contains at least one commitable change.
 
214
 
 
215
        :param _from_tree: tree to compare against to find changes (default to
 
216
            the basis tree and is intended to be used by tests).
238
217
 
239
218
        :return: True if a change is found. False otherwise
240
219
        """
241
 
        changes = self.iter_changes(from_tree)
 
220
        # Check pending merges
 
221
        if len(self.get_parent_ids()) > 1:
 
222
            return True
 
223
        if _from_tree is None:
 
224
            _from_tree = self.basis_tree()
 
225
        changes = self.iter_changes(_from_tree)
242
226
        try:
243
227
            change = changes.next()
244
228
            # Exclude root (talk about black magic... --vila 20090629)
250
234
            return False
251
235
 
252
236
    @needs_read_lock
 
237
    def check_changed_or_out_of_date(self, strict, opt_name,
 
238
                                     more_error, more_warning):
 
239
        """Check the tree for uncommitted changes and branch synchronization.
 
240
 
 
241
        If strict is None and not set in the config files, a warning is issued.
 
242
        If strict is True, an error is raised.
 
243
        If strict is False, no checks are done and no warning is issued.
 
244
 
 
245
        :param strict: True, False or None, searched in branch config if None.
 
246
 
 
247
        :param opt_name: strict option name to search in config file.
 
248
 
 
249
        :param more_error: Details about how to avoid the check.
 
250
 
 
251
        :param more_warning: Details about what is happening.
 
252
        """
 
253
        if strict is None:
 
254
            strict = self.branch.get_config().get_user_option_as_bool(opt_name)
 
255
        if strict is not False:
 
256
            err_class = None
 
257
            if (self.has_changes()):
 
258
                err_class = errors.UncommittedChanges
 
259
            elif self.last_revision() != self.branch.last_revision():
 
260
                # The tree has lost sync with its branch, there is little
 
261
                # chance that the user is aware of it but he can still force
 
262
                # the action with --no-strict
 
263
                err_class = errors.OutOfDateTree
 
264
            if err_class is not None:
 
265
                if strict is None:
 
266
                    err = err_class(self, more=more_warning)
 
267
                    # We don't want to interrupt the user if he expressed no
 
268
                    # preference about strict.
 
269
                    trace.warning('%s', err._format())
 
270
                else:
 
271
                    err = err_class(self, more=more_error)
 
272
                    raise err
 
273
 
 
274
    @needs_read_lock
253
275
    def last_revision(self):
254
276
        """Return the revision id of the last commit performed in this tree.
255
277
 
354
376
        This is designed more towards DWIM for humans than API clarity.
355
377
        For the specific behaviour see the help for cmd_add().
356
378
 
 
379
        :param file_list: List of zero or more paths.  *NB: these are 
 
380
            interpreted relative to the process cwd, not relative to the 
 
381
            tree.*  (Add and most other tree methods use tree-relative
 
382
            paths.)
357
383
        :param action: A reporter to be called with the inventory, parent_ie,
358
384
            path and kind of the path being added. It may return a file_id if
359
385
            a specific one should be used.
372
398
 
373
399
        if not file_list:
374
400
            # no paths supplied: add the entire tree.
 
401
            # FIXME: this assumes we are running in a working tree subdir :-/
 
402
            # -- vila 20100208
375
403
            file_list = [u'.']
376
404
        # mutter("smart add of %r")
377
405
        inv = self.inventory
379
407
        ignored = {}
380
408
        dirs_to_add = []
381
409
        user_dirs = set()
 
410
        conflicts_related = set()
 
411
        # Not all mutable trees can have conflicts
 
412
        if getattr(self, 'conflicts', None) is not None:
 
413
            # Collect all related files without checking whether they exist or
 
414
            # are versioned. It's cheaper to do that once for all conflicts
 
415
            # than trying to find the relevant conflict for each added file.
 
416
            for c in self.conflicts():
 
417
                conflicts_related.update(c.associated_filenames())
 
418
 
 
419
        # expand any symlinks in the directory part, while leaving the
 
420
        # filename alone
 
421
        file_list = map(osutils.normalizepath, file_list)
382
422
 
383
423
        # validate user file paths and convert all paths to tree
384
424
        # relative : it's cheaper to make a tree relative path an abspath
386
426
        # perform the canonicalization in bulk.
387
427
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
388
428
            rf = _FastPath(filepath)
389
 
            # validate user parameters. Our recursive code avoids adding new files
390
 
            # that need such validation
 
429
            # validate user parameters. Our recursive code avoids adding new
 
430
            # files that need such validation
391
431
            if self.is_control_filename(rf.raw_path):
392
432
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
393
433
 
399
439
            else:
400
440
                if not InventoryEntry.versionable_kind(kind):
401
441
                    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
 
442
            # ensure the named path is added, so that ignore rules in the later
 
443
            # directory walk dont skip it.
 
444
            # we dont have a parent ie known yet.: use the relatively slower
 
445
            # inventory probing method
406
446
            versioned = inv.has_filename(rf.raw_path)
407
447
            if versioned:
408
448
                continue
424
464
                dirs_to_add.append((path, None))
425
465
            prev_dir = path.raw_path
426
466
 
 
467
        illegalpath_re = re.compile(r'[\r\n]')
427
468
        # dirs_to_add is initialised to a list of directories, but as we scan
428
469
        # directories we append files to it.
429
470
        # XXX: We should determine kind of files when we scan them rather than
438
479
            kind = osutils.file_kind(abspath)
439
480
 
440
481
            if not InventoryEntry.versionable_kind(kind):
441
 
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
 
482
                trace.warning("skipping %s (can't add file of kind '%s')",
 
483
                              abspath, kind)
 
484
                continue
 
485
            if illegalpath_re.search(directory.raw_path):
 
486
                trace.warning("skipping %r (contains \\n or \\r)" % abspath)
 
487
                continue
 
488
            if directory.raw_path in conflicts_related:
 
489
                # If the file looks like one generated for a conflict, don't
 
490
                # add it.
 
491
                trace.warning(
 
492
                    'skipping %s (generated to help resolve conflicts)',
 
493
                    abspath)
442
494
                continue
443
495
 
444
496
            if parent_ie is not None:
467
519
                pass
468
520
                # mutter("%r is already versioned", abspath)
469
521
            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)
 
522
                # XXX: This is wrong; people *might* reasonably be trying to
 
523
                # add subtrees as subtrees.  This should probably only be done
 
524
                # in formats which can represent subtrees, and even then
 
525
                # perhaps only when the user asked to add subtrees.  At the
 
526
                # moment you can add them specially through 'join --reference',
 
527
                # which is perhaps reasonable: adding a new reference is a
 
528
                # special operation and can have a special behaviour.  mbp
 
529
                # 20070306
 
530
                trace.mutter("%r is a nested bzr tree", abspath)
478
531
            else:
479
532
                _add_one(self, inv, parent_ie, directory, kind, action)
480
533
                added.append(directory.raw_path)
487
540
                    # without the parent ie, use the relatively slower inventory
488
541
                    # probing method
489
542
                    this_id = inv.path2id(
490
 
                            self._fix_case_of_inventory_path(directory.raw_path))
 
543
                        self._fix_case_of_inventory_path(directory.raw_path))
491
544
                    if this_id is None:
492
545
                        this_ie = None
493
546
                    else:
502
555
                    # faster - its impossible for a non root dir to have a
503
556
                    # control file.
504
557
                    if self.is_control_filename(subp):
505
 
                        mutter("skip control directory %r", subp)
 
558
                        trace.mutter("skip control directory %r", subp)
506
559
                    elif subf in this_ie.children:
507
560
                        # recurse into this already versioned subdir.
508
561
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
564
617
        inventory = basis.inventory._get_mutable_inventory()
565
618
        basis.unlock()
566
619
        inventory.apply_delta(delta)
567
 
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
 
620
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
 
621
                                             inventory, new_revid)
568
622
        self.set_parent_trees([(new_revid, rev_tree)])
569
623
 
570
624
 
581
635
        self.create_hook(hooks.HookPoint('start_commit',
582
636
            "Called before a commit is performed on a tree. The start commit "
583
637
            "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))
 
638
            "start_commit is called with the bzrlib.mutabletree.MutableTree "
 
639
            "that the commit is being performed on.", (1, 4), None))
 
640
        self.create_hook(hooks.HookPoint('post_commit',
 
641
            "Called after a commit is performed on a tree. The hook is "
 
642
            "called with a bzrlib.mutabletree.PostCommitHookParams object. "
 
643
            "The mutable tree the commit was performed on is available via "
 
644
            "the mutable_tree attribute of that object.", (2, 0), None))
586
645
 
587
646
 
588
647
# install the default hooks into the MutableTree class.
589
648
MutableTree.hooks = MutableTreeHooks()
590
649
 
591
650
 
 
651
class PostCommitHookParams(object):
 
652
    """Parameters for the post_commit hook.
 
653
 
 
654
    To access the parameters, use the following attributes:
 
655
 
 
656
    * mutable_tree - the MutableTree object
 
657
    """
 
658
 
 
659
    def __init__(self, mutable_tree):
 
660
        """Create the parameters for the post_commit hook."""
 
661
        self.mutable_tree = mutable_tree
 
662
 
 
663
 
592
664
class _FastPath(object):
593
665
    """A path object with fast accessors for things like basename."""
594
666
 
635
707
        # there are a limited number of dirs we can be nested under, it should
636
708
        # generally find it very fast and not recurse after that.
637
709
        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))
 
710
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
 
711
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
640
712
        parent_ie = inv[parent_id]
641
713
    _add_one(tree, inv, parent_ie, path, kind, action)
642
714
    return added + [path.raw_path]
652
724
        file_id or None to generate a new file id
653
725
    :returns: None
654
726
    """
 
727
    # if the parent exists, but isn't a directory, we have to do the
 
728
    # kind change now -- really the inventory shouldn't pretend to know
 
729
    # the kind of wt files, but it does.
 
730
    if parent_ie.kind != 'directory':
 
731
        # nb: this relies on someone else checking that the path we're using
 
732
        # doesn't contain symlinks.
 
733
        new_parent_ie = inventory.make_entry('directory', parent_ie.name,
 
734
            parent_ie.parent_id, parent_ie.file_id)
 
735
        del inv[parent_ie.file_id]
 
736
        inv.add(new_parent_ie)
 
737
        parent_ie = new_parent_ie
655
738
    file_id = file_id_callback(inv, parent_ie, path, kind)
656
739
    entry = inv.make_entry(kind, path.base_path, parent_ie.file_id,
657
740
        file_id=file_id)