~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/mutabletree.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-19 17:53:37 UTC
  • mto: This revision was merged to the branch mainline in revision 4466.
  • Revision ID: john@arbash-meinel.com-20090619175337-uozt3bntdd48lh4z
Update time_graph to use X:1 ratios rather than 0.xxx ratios.
It is just easier to track now that the new code is much faster.

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
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
    symbol_versioning,
 
32
    )
 
33
from bzrlib.osutils import dirname
 
34
from bzrlib.revisiontree import RevisionTree
 
35
from bzrlib.trace import mutter, warning
 
36
""")
 
37
 
 
38
from bzrlib import (
31
39
    errors,
32
 
    hooks,
33
40
    osutils,
34
 
    revisiontree,
35
 
    inventory,
36
 
    symbol_versioning,
37
 
    trace,
38
41
    tree,
39
42
    )
40
 
""")
41
 
 
42
43
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
44
from bzrlib.osutils import splitpath
43
45
 
44
46
 
45
47
def needs_tree_write_lock(unbound):
127
129
            # generic constraint checks:
128
130
            if self.is_control_filename(f):
129
131
                raise errors.ForbiddenControlFileError(filename=f)
130
 
            fp = osutils.splitpath(f)
 
132
            fp = splitpath(f)
131
133
        # fill out file kinds for all files [not needed when we stop
132
134
        # caring about the instantaneous file kind within a uncommmitted tree
133
135
        #
183
185
               **kwargs):
184
186
        # avoid circular imports
185
187
        from bzrlib import commit
 
188
        if revprops is None:
 
189
            revprops = {}
186
190
        possible_master_transports=[]
187
 
        revprops = commit.Commit.update_revprops(
188
 
                revprops,
189
 
                self.branch,
190
 
                kwargs.pop('authors', None),
191
 
                kwargs.pop('author', None),
 
191
        if not 'branch-nick' in revprops:
 
192
            revprops['branch-nick'] = self.branch._get_nick(
192
193
                kwargs.get('local', False),
193
194
                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
194
221
        # args for wt.commit start at message from the Commit.commit method,
195
222
        args = (message, ) + args
196
223
        for hook in MutableTree.hooks['start_commit']:
199
226
            revprops=revprops,
200
227
            possible_master_transports=possible_master_transports,
201
228
            *args, **kwargs)
202
 
        post_hook_params = PostCommitHookParams(self)
203
 
        for hook in MutableTree.hooks['post_commit']:
204
 
            hook(post_hook_params)
205
229
        return committed_id
206
230
 
207
231
    def _gather_kinds(self, files, kinds):
209
233
        raise NotImplementedError(self._gather_kinds)
210
234
 
211
235
    @needs_read_lock
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).
217
 
 
218
 
        :return: True if a change is found. False otherwise
219
 
        """
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)
226
 
        try:
227
 
            change = changes.next()
228
 
            # Exclude root (talk about black magic... --vila 20090629)
229
 
            if change[4] == (None, None):
230
 
                change = changes.next()
231
 
            return True
232
 
        except StopIteration:
233
 
            # No changes
234
 
            return False
235
 
 
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
275
236
    def last_revision(self):
276
237
        """Return the revision id of the last commit performed in this tree.
277
238
 
376
337
        This is designed more towards DWIM for humans than API clarity.
377
338
        For the specific behaviour see the help for cmd_add().
378
339
 
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.)
383
340
        :param action: A reporter to be called with the inventory, parent_ie,
384
341
            path and kind of the path being added. It may return a file_id if
385
342
            a specific one should be used.
398
355
 
399
356
        if not file_list:
400
357
            # no paths supplied: add the entire tree.
401
 
            # FIXME: this assumes we are running in a working tree subdir :-/
402
 
            # -- vila 20100208
403
358
            file_list = [u'.']
404
359
        # mutter("smart add of %r")
405
360
        inv = self.inventory
407
362
        ignored = {}
408
363
        dirs_to_add = []
409
364
        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)
422
365
 
423
366
        # validate user file paths and convert all paths to tree
424
367
        # relative : it's cheaper to make a tree relative path an abspath
426
369
        # perform the canonicalization in bulk.
427
370
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
428
371
            rf = _FastPath(filepath)
429
 
            # validate user parameters. Our recursive code avoids adding new
430
 
            # files that need such validation
 
372
            # validate user parameters. Our recursive code avoids adding new files
 
373
            # that need such validation
431
374
            if self.is_control_filename(rf.raw_path):
432
375
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
433
376
 
439
382
            else:
440
383
                if not InventoryEntry.versionable_kind(kind):
441
384
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
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
 
385
            # ensure the named path is added, so that ignore rules in the later directory
 
386
            # walk dont skip it.
 
387
            # we dont have a parent ie known yet.: use the relatively slower inventory
 
388
            # probing method
446
389
            versioned = inv.has_filename(rf.raw_path)
447
390
            if versioned:
448
391
                continue
464
407
                dirs_to_add.append((path, None))
465
408
            prev_dir = path.raw_path
466
409
 
467
 
        illegalpath_re = re.compile(r'[\r\n]')
468
410
        # dirs_to_add is initialised to a list of directories, but as we scan
469
411
        # directories we append files to it.
470
412
        # XXX: We should determine kind of files when we scan them rather than
479
421
            kind = osutils.file_kind(abspath)
480
422
 
481
423
            if not InventoryEntry.versionable_kind(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)
 
424
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
494
425
                continue
495
426
 
496
427
            if parent_ie is not None:
519
450
                pass
520
451
                # mutter("%r is already versioned", abspath)
521
452
            elif sub_tree:
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)
 
453
                # XXX: This is wrong; people *might* reasonably be trying to add
 
454
                # subtrees as subtrees.  This should probably only be done in formats
 
455
                # which can represent subtrees, and even then perhaps only when
 
456
                # the user asked to add subtrees.  At the moment you can add them
 
457
                # specially through 'join --reference', which is perhaps
 
458
                # reasonable: adding a new reference is a special operation and
 
459
                # can have a special behaviour.  mbp 20070306
 
460
                mutter("%r is a nested bzr tree", abspath)
531
461
            else:
532
462
                _add_one(self, inv, parent_ie, directory, kind, action)
533
463
                added.append(directory.raw_path)
540
470
                    # without the parent ie, use the relatively slower inventory
541
471
                    # probing method
542
472
                    this_id = inv.path2id(
543
 
                        self._fix_case_of_inventory_path(directory.raw_path))
 
473
                            self._fix_case_of_inventory_path(directory.raw_path))
544
474
                    if this_id is None:
545
475
                        this_ie = None
546
476
                    else:
555
485
                    # faster - its impossible for a non root dir to have a
556
486
                    # control file.
557
487
                    if self.is_control_filename(subp):
558
 
                        trace.mutter("skip control directory %r", subp)
 
488
                        mutter("skip control directory %r", subp)
559
489
                    elif subf in this_ie.children:
560
490
                        # recurse into this already versioned subdir.
561
491
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
593
523
        for commit which is not required to handle situations that do not arise
594
524
        outside of commit.
595
525
 
596
 
        See the inventory developers documentation for the theory behind
597
 
        inventory deltas.
598
 
 
599
526
        :param new_revid: The new revision id for the trees parent.
600
527
        :param delta: An inventory delta (see apply_inventory_delta) describing
601
528
            the changes from the current left most parent revision to new_revid.
617
544
        inventory = basis.inventory._get_mutable_inventory()
618
545
        basis.unlock()
619
546
        inventory.apply_delta(delta)
620
 
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
621
 
                                             inventory, new_revid)
 
547
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
622
548
        self.set_parent_trees([(new_revid, rev_tree)])
623
549
 
624
550
 
635
561
        self.create_hook(hooks.HookPoint('start_commit',
636
562
            "Called before a commit is performed on a tree. The start commit "
637
563
            "hook is able to change the tree before the commit takes place. "
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))
 
564
            "start_commit is called with the bzrlib.tree.MutableTree that the "
 
565
            "commit is being performed on.", (1, 4), None))
645
566
 
646
567
 
647
568
# install the default hooks into the MutableTree class.
648
569
MutableTree.hooks = MutableTreeHooks()
649
570
 
650
571
 
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
 
 
664
572
class _FastPath(object):
665
573
    """A path object with fast accessors for things like basename."""
666
574
 
707
615
        # there are a limited number of dirs we can be nested under, it should
708
616
        # generally find it very fast and not recurse after that.
709
617
        added = _add_one_and_parent(tree, inv, None,
710
 
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
711
 
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
 
618
            _FastPath(dirname(path.raw_path)), 'directory', action)
 
619
        parent_id = inv.path2id(dirname(path.raw_path))
712
620
        parent_ie = inv[parent_id]
713
621
    _add_one(tree, inv, parent_ie, path, kind, action)
714
622
    return added + [path.raw_path]
724
632
        file_id or None to generate a new file id
725
633
    :returns: None
726
634
    """
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
738
635
    file_id = file_id_callback(inv, parent_ie, path, kind)
739
636
    entry = inv.make_entry(kind, path.base_path, parent_ie.file_id,
740
637
        file_id=file_id)