/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: Jelmer Vernooij
  • Date: 2009-02-25 02:05:35 UTC
  • mto: This revision was merged to the branch mainline in revision 4053.
  • Revision ID: jelmer@samba.org-20090225020535-qw7udfz9ploqzosn
Add tests for InterBranch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006, 2007 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
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
        #
182
185
               **kwargs):
183
186
        # avoid circular imports
184
187
        from bzrlib import commit
 
188
        if revprops is None:
 
189
            revprops = {}
185
190
        possible_master_transports=[]
186
 
        revprops = commit.Commit.update_revprops(
187
 
                revprops,
188
 
                self.branch,
189
 
                kwargs.pop('authors', None),
190
 
                kwargs.pop('author', None),
 
191
        if not 'branch-nick' in revprops:
 
192
            revprops['branch-nick'] = self.branch._get_nick(
191
193
                kwargs.get('local', False),
192
194
                possible_master_transports)
 
195
        author = kwargs.pop('author', None)
 
196
        if author is not None:
 
197
            if 'author' in revprops:
 
198
                # XXX: maybe we should just accept one of them?
 
199
                raise AssertionError('author property given twice')
 
200
            revprops['author'] = author
193
201
        # args for wt.commit start at message from the Commit.commit method,
194
202
        args = (message, ) + args
195
203
        for hook in MutableTree.hooks['start_commit']:
198
206
            revprops=revprops,
199
207
            possible_master_transports=possible_master_transports,
200
208
            *args, **kwargs)
201
 
        post_hook_params = PostCommitHookParams(self)
202
 
        for hook in MutableTree.hooks['post_commit']:
203
 
            hook(post_hook_params)
204
209
        return committed_id
205
210
 
206
211
    def _gather_kinds(self, files, kinds):
207
212
        """Helper function for add - sets the entries of kinds."""
208
213
        raise NotImplementedError(self._gather_kinds)
209
214
 
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
 
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.
 
226
        """
 
227
        return (self.get_file(file_id, path), None)
272
228
 
273
229
    @needs_read_lock
274
230
    def last_revision(self):
393
349
 
394
350
        if not file_list:
395
351
            # no paths supplied: add the entire tree.
396
 
            # FIXME: this assumes we are running in a working tree subdir :-/
397
 
            # -- vila 20100208
398
352
            file_list = [u'.']
399
353
        # mutter("smart add of %r")
400
354
        inv = self.inventory
402
356
        ignored = {}
403
357
        dirs_to_add = []
404
358
        user_dirs = set()
405
 
        conflicts_related = set()
406
 
        # Not all mutable trees can have conflicts
407
 
        if getattr(self, 'conflicts', None) is not None:
408
 
            # Collect all related files without checking whether they exist or
409
 
            # are versioned. It's cheaper to do that once for all conflicts
410
 
            # than trying to find the relevant conflict for each added file.
411
 
            for c in self.conflicts():
412
 
                conflicts_related.update(c.associated_filenames())
413
359
 
414
360
        # validate user file paths and convert all paths to tree
415
361
        # relative : it's cheaper to make a tree relative path an abspath
417
363
        # perform the canonicalization in bulk.
418
364
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
419
365
            rf = _FastPath(filepath)
420
 
            # validate user parameters. Our recursive code avoids adding new
421
 
            # files that need such validation
 
366
            # validate user parameters. Our recursive code avoids adding new files
 
367
            # that need such validation
422
368
            if self.is_control_filename(rf.raw_path):
423
369
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
424
370
 
430
376
            else:
431
377
                if not InventoryEntry.versionable_kind(kind):
432
378
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
433
 
            # ensure the named path is added, so that ignore rules in the later
434
 
            # directory walk dont skip it.
435
 
            # we dont have a parent ie known yet.: use the relatively slower
436
 
            # 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
437
383
            versioned = inv.has_filename(rf.raw_path)
438
384
            if versioned:
439
385
                continue
455
401
                dirs_to_add.append((path, None))
456
402
            prev_dir = path.raw_path
457
403
 
458
 
        illegalpath_re = re.compile(r'[\r\n]')
459
404
        # dirs_to_add is initialised to a list of directories, but as we scan
460
405
        # directories we append files to it.
461
406
        # XXX: We should determine kind of files when we scan them rather than
470
415
            kind = osutils.file_kind(abspath)
471
416
 
472
417
            if not InventoryEntry.versionable_kind(kind):
473
 
                trace.warning("skipping %s (can't add file of kind '%s')",
474
 
                              abspath, kind)
475
 
                continue
476
 
            if illegalpath_re.search(directory.raw_path):
477
 
                trace.warning("skipping %r (contains \\n or \\r)" % abspath)
478
 
                continue
479
 
            if directory.raw_path in conflicts_related:
480
 
                # If the file looks like one generated for a conflict, don't
481
 
                # add it.
482
 
                trace.warning(
483
 
                    'skipping %s (generated to help resolve conflicts)',
484
 
                    abspath)
 
418
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
485
419
                continue
486
420
 
487
421
            if parent_ie is not None:
510
444
                pass
511
445
                # mutter("%r is already versioned", abspath)
512
446
            elif sub_tree:
513
 
                # XXX: This is wrong; people *might* reasonably be trying to
514
 
                # add subtrees as subtrees.  This should probably only be done
515
 
                # in formats which can represent subtrees, and even then
516
 
                # perhaps only when the user asked to add subtrees.  At the
517
 
                # moment you can add them specially through 'join --reference',
518
 
                # which is perhaps reasonable: adding a new reference is a
519
 
                # special operation and can have a special behaviour.  mbp
520
 
                # 20070306
521
 
                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)
522
455
            else:
523
456
                _add_one(self, inv, parent_ie, directory, kind, action)
524
457
                added.append(directory.raw_path)
531
464
                    # without the parent ie, use the relatively slower inventory
532
465
                    # probing method
533
466
                    this_id = inv.path2id(
534
 
                        self._fix_case_of_inventory_path(directory.raw_path))
 
467
                            self._fix_case_of_inventory_path(directory.raw_path))
535
468
                    if this_id is None:
536
469
                        this_ie = None
537
470
                    else:
546
479
                    # faster - its impossible for a non root dir to have a
547
480
                    # control file.
548
481
                    if self.is_control_filename(subp):
549
 
                        trace.mutter("skip control directory %r", subp)
 
482
                        mutter("skip control directory %r", subp)
550
483
                    elif subf in this_ie.children:
551
484
                        # recurse into this already versioned subdir.
552
485
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
584
517
        for commit which is not required to handle situations that do not arise
585
518
        outside of commit.
586
519
 
587
 
        See the inventory developers documentation for the theory behind
588
 
        inventory deltas.
589
 
 
590
520
        :param new_revid: The new revision id for the trees parent.
591
521
        :param delta: An inventory delta (see apply_inventory_delta) describing
592
522
            the changes from the current left most parent revision to new_revid.
602
532
        # WorkingTree classes for optimised versions for specific format trees.
603
533
        basis = self.basis_tree()
604
534
        basis.lock_read()
605
 
        # TODO: Consider re-evaluating the need for this with CHKInventory
606
 
        # we don't strictly need to mutate an inventory for this
607
 
        # it only makes sense when apply_delta is cheaper than get_inventory()
608
 
        inventory = basis.inventory._get_mutable_inventory()
 
535
        inventory = basis.inventory
609
536
        basis.unlock()
610
537
        inventory.apply_delta(delta)
611
 
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
612
 
                                             inventory, new_revid)
 
538
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
613
539
        self.set_parent_trees([(new_revid, rev_tree)])
614
540
 
615
541
 
623
549
 
624
550
        """
625
551
        hooks.Hooks.__init__(self)
626
 
        self.create_hook(hooks.HookPoint('start_commit',
627
 
            "Called before a commit is performed on a tree. The start commit "
628
 
            "hook is able to change the tree before the commit takes place. "
629
 
            "start_commit is called with the bzrlib.mutabletree.MutableTree "
630
 
            "that the commit is being performed on.", (1, 4), None))
631
 
        self.create_hook(hooks.HookPoint('post_commit',
632
 
            "Called after a commit is performed on a tree. The hook is "
633
 
            "called with a bzrlib.mutabletree.PostCommitHookParams object. "
634
 
            "The mutable tree the commit was performed on is available via "
635
 
            "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'] = []
636
554
 
637
555
 
638
556
# install the default hooks into the MutableTree class.
639
557
MutableTree.hooks = MutableTreeHooks()
640
558
 
641
559
 
642
 
class PostCommitHookParams(object):
643
 
    """Parameters for the post_commit hook.
644
 
 
645
 
    To access the parameters, use the following attributes:
646
 
 
647
 
    * mutable_tree - the MutableTree object
648
 
    """
649
 
 
650
 
    def __init__(self, mutable_tree):
651
 
        """Create the parameters for the post_commit hook."""
652
 
        self.mutable_tree = mutable_tree
653
 
 
654
 
 
655
560
class _FastPath(object):
656
561
    """A path object with fast accessors for things like basename."""
657
562
 
698
603
        # there are a limited number of dirs we can be nested under, it should
699
604
        # generally find it very fast and not recurse after that.
700
605
        added = _add_one_and_parent(tree, inv, None,
701
 
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
702
 
        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))
703
608
        parent_ie = inv[parent_id]
704
609
    _add_one(tree, inv, parent_ie, path, kind, action)
705
610
    return added + [path.raw_path]