~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: Martin Pool
  • Date: 2010-07-16 15:20:17 UTC
  • mfrom: (5346.3.1 pathnotchild)
  • mto: This revision was merged to the branch mainline in revision 5351.
  • Revision ID: mbp@canonical.com-20100716152017-t4c73h9y1uoih7fb
PathNotChild should not give a traceback.

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,
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
        #
185
182
               **kwargs):
186
183
        # avoid circular imports
187
184
        from bzrlib import commit
188
 
        if revprops is None:
189
 
            revprops = {}
190
185
        possible_master_transports=[]
191
 
        if not 'branch-nick' in revprops:
192
 
            revprops['branch-nick'] = self.branch._get_nick(
 
186
        revprops = commit.Commit.update_revprops(
 
187
                revprops,
 
188
                self.branch,
 
189
                kwargs.pop('authors', None),
 
190
                kwargs.pop('author', None),
193
191
                kwargs.get('local', False),
194
192
                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
193
        # args for wt.commit start at message from the Commit.commit method,
222
194
        args = (message, ) + args
223
195
        for hook in MutableTree.hooks['start_commit']:
226
198
            revprops=revprops,
227
199
            possible_master_transports=possible_master_transports,
228
200
            *args, **kwargs)
 
201
        post_hook_params = PostCommitHookParams(self)
 
202
        for hook in MutableTree.hooks['post_commit']:
 
203
            hook(post_hook_params)
229
204
        return committed_id
230
205
 
231
206
    def _gather_kinds(self, files, kinds):
232
207
        """Helper function for add - sets the entries of kinds."""
233
208
        raise NotImplementedError(self._gather_kinds)
234
209
 
235
 
    def get_file_with_stat(self, file_id, path=None):
236
 
        """Get a file handle and stat object for file_id.
237
 
 
238
 
        The default implementation returns (self.get_file, None) for backwards
239
 
        compatibility.
240
 
 
241
 
        :param file_id: The file id to read.
242
 
        :param path: The path of the file, if it is known.
243
 
        :return: A tuple (file_handle, stat_value_or_None). If the tree has
244
 
            no stat facility, or need for a stat cache feedback during commit,
245
 
            it may return None for the second element of the tuple.
246
 
        """
247
 
        return (self.get_file(file_id, path), None)
 
210
    @needs_read_lock
 
211
    def has_changes(self, _from_tree=None):
 
212
        """Quickly check that the tree contains at least one commitable change.
 
213
 
 
214
        :param _from_tree: tree to compare against to find changes (default to
 
215
            the basis tree and is intended to be used by tests).
 
216
 
 
217
        :return: True if a change is found. False otherwise
 
218
        """
 
219
        # Check pending merges
 
220
        if len(self.get_parent_ids()) > 1:
 
221
            return True
 
222
        if _from_tree is None:
 
223
            _from_tree = self.basis_tree()
 
224
        changes = self.iter_changes(_from_tree)
 
225
        try:
 
226
            change = changes.next()
 
227
            # Exclude root (talk about black magic... --vila 20090629)
 
228
            if change[4] == (None, None):
 
229
                change = changes.next()
 
230
            return True
 
231
        except StopIteration:
 
232
            # No changes
 
233
            return False
 
234
 
 
235
    @needs_read_lock
 
236
    def check_changed_or_out_of_date(self, strict, opt_name,
 
237
                                     more_error, more_warning):
 
238
        """Check the tree for uncommitted changes and branch synchronization.
 
239
 
 
240
        If strict is None and not set in the config files, a warning is issued.
 
241
        If strict is True, an error is raised.
 
242
        If strict is False, no checks are done and no warning is issued.
 
243
 
 
244
        :param strict: True, False or None, searched in branch config if None.
 
245
 
 
246
        :param opt_name: strict option name to search in config file.
 
247
 
 
248
        :param more_error: Details about how to avoid the check.
 
249
 
 
250
        :param more_warning: Details about what is happening.
 
251
        """
 
252
        if strict is None:
 
253
            strict = self.branch.get_config().get_user_option_as_bool(opt_name)
 
254
        if strict is not False:
 
255
            err_class = None
 
256
            if (self.has_changes()):
 
257
                err_class = errors.UncommittedChanges
 
258
            elif self.last_revision() != self.branch.last_revision():
 
259
                # The tree has lost sync with its branch, there is little
 
260
                # chance that the user is aware of it but he can still force
 
261
                # the action with --no-strict
 
262
                err_class = errors.OutOfDateTree
 
263
            if err_class is not None:
 
264
                if strict is None:
 
265
                    err = err_class(self, more=more_warning)
 
266
                    # We don't want to interrupt the user if he expressed no
 
267
                    # preference about strict.
 
268
                    trace.warning('%s', err._format())
 
269
                else:
 
270
                    err = err_class(self, more=more_error)
 
271
                    raise err
248
272
 
249
273
    @needs_read_lock
250
274
    def last_revision(self):
351
375
        This is designed more towards DWIM for humans than API clarity.
352
376
        For the specific behaviour see the help for cmd_add().
353
377
 
 
378
        :param file_list: List of zero or more paths.  *NB: these are 
 
379
            interpreted relative to the process cwd, not relative to the 
 
380
            tree.*  (Add and most other tree methods use tree-relative
 
381
            paths.)
354
382
        :param action: A reporter to be called with the inventory, parent_ie,
355
383
            path and kind of the path being added. It may return a file_id if
356
384
            a specific one should be used.
369
397
 
370
398
        if not file_list:
371
399
            # no paths supplied: add the entire tree.
 
400
            # FIXME: this assumes we are running in a working tree subdir :-/
 
401
            # -- vila 20100208
372
402
            file_list = [u'.']
373
403
        # mutter("smart add of %r")
374
404
        inv = self.inventory
376
406
        ignored = {}
377
407
        dirs_to_add = []
378
408
        user_dirs = set()
 
409
        conflicts_related = set()
 
410
        # Not all mutable trees can have conflicts
 
411
        if getattr(self, 'conflicts', None) is not None:
 
412
            # Collect all related files without checking whether they exist or
 
413
            # are versioned. It's cheaper to do that once for all conflicts
 
414
            # than trying to find the relevant conflict for each added file.
 
415
            for c in self.conflicts():
 
416
                conflicts_related.update(c.associated_filenames())
379
417
 
380
418
        # validate user file paths and convert all paths to tree
381
419
        # relative : it's cheaper to make a tree relative path an abspath
383
421
        # perform the canonicalization in bulk.
384
422
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
385
423
            rf = _FastPath(filepath)
386
 
            # validate user parameters. Our recursive code avoids adding new files
387
 
            # that need such validation
 
424
            # validate user parameters. Our recursive code avoids adding new
 
425
            # files that need such validation
388
426
            if self.is_control_filename(rf.raw_path):
389
427
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
390
428
 
396
434
            else:
397
435
                if not InventoryEntry.versionable_kind(kind):
398
436
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
399
 
            # ensure the named path is added, so that ignore rules in the later directory
400
 
            # walk dont skip it.
401
 
            # we dont have a parent ie known yet.: use the relatively slower inventory
402
 
            # probing method
 
437
            # ensure the named path is added, so that ignore rules in the later
 
438
            # directory walk dont skip it.
 
439
            # we dont have a parent ie known yet.: use the relatively slower
 
440
            # inventory probing method
403
441
            versioned = inv.has_filename(rf.raw_path)
404
442
            if versioned:
405
443
                continue
421
459
                dirs_to_add.append((path, None))
422
460
            prev_dir = path.raw_path
423
461
 
 
462
        illegalpath_re = re.compile(r'[\r\n]')
424
463
        # dirs_to_add is initialised to a list of directories, but as we scan
425
464
        # directories we append files to it.
426
465
        # XXX: We should determine kind of files when we scan them rather than
435
474
            kind = osutils.file_kind(abspath)
436
475
 
437
476
            if not InventoryEntry.versionable_kind(kind):
438
 
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
 
477
                trace.warning("skipping %s (can't add file of kind '%s')",
 
478
                              abspath, kind)
 
479
                continue
 
480
            if illegalpath_re.search(directory.raw_path):
 
481
                trace.warning("skipping %r (contains \\n or \\r)" % abspath)
 
482
                continue
 
483
            if directory.raw_path in conflicts_related:
 
484
                # If the file looks like one generated for a conflict, don't
 
485
                # add it.
 
486
                trace.warning(
 
487
                    'skipping %s (generated to help resolve conflicts)',
 
488
                    abspath)
439
489
                continue
440
490
 
441
491
            if parent_ie is not None:
464
514
                pass
465
515
                # mutter("%r is already versioned", abspath)
466
516
            elif sub_tree:
467
 
                # XXX: This is wrong; people *might* reasonably be trying to add
468
 
                # subtrees as subtrees.  This should probably only be done in formats
469
 
                # which can represent subtrees, and even then perhaps only when
470
 
                # the user asked to add subtrees.  At the moment you can add them
471
 
                # specially through 'join --reference', which is perhaps
472
 
                # reasonable: adding a new reference is a special operation and
473
 
                # can have a special behaviour.  mbp 20070306
474
 
                mutter("%r is a nested bzr tree", abspath)
 
517
                # XXX: This is wrong; people *might* reasonably be trying to
 
518
                # add subtrees as subtrees.  This should probably only be done
 
519
                # in formats which can represent subtrees, and even then
 
520
                # perhaps only when the user asked to add subtrees.  At the
 
521
                # moment you can add them specially through 'join --reference',
 
522
                # which is perhaps reasonable: adding a new reference is a
 
523
                # special operation and can have a special behaviour.  mbp
 
524
                # 20070306
 
525
                trace.mutter("%r is a nested bzr tree", abspath)
475
526
            else:
476
527
                _add_one(self, inv, parent_ie, directory, kind, action)
477
528
                added.append(directory.raw_path)
484
535
                    # without the parent ie, use the relatively slower inventory
485
536
                    # probing method
486
537
                    this_id = inv.path2id(
487
 
                            self._fix_case_of_inventory_path(directory.raw_path))
 
538
                        self._fix_case_of_inventory_path(directory.raw_path))
488
539
                    if this_id is None:
489
540
                        this_ie = None
490
541
                    else:
499
550
                    # faster - its impossible for a non root dir to have a
500
551
                    # control file.
501
552
                    if self.is_control_filename(subp):
502
 
                        mutter("skip control directory %r", subp)
 
553
                        trace.mutter("skip control directory %r", subp)
503
554
                    elif subf in this_ie.children:
504
555
                        # recurse into this already versioned subdir.
505
556
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
537
588
        for commit which is not required to handle situations that do not arise
538
589
        outside of commit.
539
590
 
 
591
        See the inventory developers documentation for the theory behind
 
592
        inventory deltas.
 
593
 
540
594
        :param new_revid: The new revision id for the trees parent.
541
595
        :param delta: An inventory delta (see apply_inventory_delta) describing
542
596
            the changes from the current left most parent revision to new_revid.
558
612
        inventory = basis.inventory._get_mutable_inventory()
559
613
        basis.unlock()
560
614
        inventory.apply_delta(delta)
561
 
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
 
615
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
 
616
                                             inventory, new_revid)
562
617
        self.set_parent_trees([(new_revid, rev_tree)])
563
618
 
564
619
 
575
630
        self.create_hook(hooks.HookPoint('start_commit',
576
631
            "Called before a commit is performed on a tree. The start commit "
577
632
            "hook is able to change the tree before the commit takes place. "
578
 
            "start_commit is called with the bzrlib.tree.MutableTree that the "
579
 
            "commit is being performed on.", (1, 4), None))
 
633
            "start_commit is called with the bzrlib.mutabletree.MutableTree "
 
634
            "that the commit is being performed on.", (1, 4), None))
 
635
        self.create_hook(hooks.HookPoint('post_commit',
 
636
            "Called after a commit is performed on a tree. The hook is "
 
637
            "called with a bzrlib.mutabletree.PostCommitHookParams object. "
 
638
            "The mutable tree the commit was performed on is available via "
 
639
            "the mutable_tree attribute of that object.", (2, 0), None))
580
640
 
581
641
 
582
642
# install the default hooks into the MutableTree class.
583
643
MutableTree.hooks = MutableTreeHooks()
584
644
 
585
645
 
 
646
class PostCommitHookParams(object):
 
647
    """Parameters for the post_commit hook.
 
648
 
 
649
    To access the parameters, use the following attributes:
 
650
 
 
651
    * mutable_tree - the MutableTree object
 
652
    """
 
653
 
 
654
    def __init__(self, mutable_tree):
 
655
        """Create the parameters for the post_commit hook."""
 
656
        self.mutable_tree = mutable_tree
 
657
 
 
658
 
586
659
class _FastPath(object):
587
660
    """A path object with fast accessors for things like basename."""
588
661
 
629
702
        # there are a limited number of dirs we can be nested under, it should
630
703
        # generally find it very fast and not recurse after that.
631
704
        added = _add_one_and_parent(tree, inv, None,
632
 
            _FastPath(dirname(path.raw_path)), 'directory', action)
633
 
        parent_id = inv.path2id(dirname(path.raw_path))
 
705
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
 
706
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
634
707
        parent_ie = inv[parent_id]
635
708
    _add_one(tree, inv, parent_ie, path, kind, action)
636
709
    return added + [path.raw_path]