/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: Martin von Gagern
  • Date: 2010-04-20 08:47:38 UTC
  • mfrom: (5167 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5195.
  • Revision ID: martin.vgagern@gmx.net-20100420084738-ygymnqmdllzrhpfn
merge trunk

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
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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
26
27
 
27
28
from bzrlib import (
28
29
    add,
29
30
    bzrdir,
 
31
    errors,
30
32
    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 (
38
 
    errors,
39
33
    osutils,
 
34
    revisiontree,
 
35
    symbol_versioning,
 
36
    trace,
40
37
    tree,
41
38
    )
 
39
""")
 
40
 
42
41
from bzrlib.decorators import needs_read_lock, needs_write_lock
43
 
from bzrlib.osutils import splitpath
44
 
from bzrlib.symbol_versioning import DEPRECATED_PARAMETER
45
42
 
46
43
 
47
44
def needs_tree_write_lock(unbound):
69
66
    entirely in memory.
70
67
 
71
68
    For now, we are not treating MutableTree as an interface to provide
72
 
    conformance tests for - rather we are testing MemoryTree specifically, and 
 
69
    conformance tests for - rather we are testing MemoryTree specifically, and
73
70
    interface testing implementations of WorkingTree.
74
71
 
75
72
    A mutable tree always has an associated Branch and BzrDir object - the
76
73
    branch and bzrdir attributes.
77
74
    """
 
75
    def __init__(self, *args, **kw):
 
76
        super(MutableTree, self).__init__(*args, **kw)
 
77
        # Is this tree on a case-insensitive or case-preserving file-system?
 
78
        # Sub-classes may initialize to False if they detect they are being
 
79
        # used on media which doesn't differentiate the case of names.
 
80
        self.case_sensitive = True
78
81
 
79
82
    @needs_tree_write_lock
80
83
    def add(self, files, ids=None, kinds=None):
123
126
            # generic constraint checks:
124
127
            if self.is_control_filename(f):
125
128
                raise errors.ForbiddenControlFileError(filename=f)
126
 
            fp = splitpath(f)
127
 
        # fill out file kinds for all files [not needed when we stop 
 
129
            fp = osutils.splitpath(f)
 
130
        # fill out file kinds for all files [not needed when we stop
128
131
        # caring about the instantaneous file kind within a uncommmitted tree
129
132
        #
130
133
        self._gather_kinds(files, kinds)
181
184
        from bzrlib import commit
182
185
        if revprops is None:
183
186
            revprops = {}
 
187
        possible_master_transports=[]
184
188
        if not 'branch-nick' in revprops:
185
 
            revprops['branch-nick'] = self.branch.nick
 
189
            revprops['branch-nick'] = self.branch._get_nick(
 
190
                kwargs.get('local', False),
 
191
                possible_master_transports)
 
192
        authors = kwargs.pop('authors', None)
186
193
        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)
187
207
        if author is not None:
188
 
            if 'author' in revprops:
 
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:
189
212
                # XXX: maybe we should just accept one of them?
190
213
                raise AssertionError('author property given twice')
191
 
            revprops['author'] = author
 
214
            if '\n' in author:
 
215
                raise AssertionError('\\n is not a valid character '
 
216
                        'in an author identity')
 
217
            revprops['authors'] = author
192
218
        # args for wt.commit start at message from the Commit.commit method,
193
219
        args = (message, ) + args
194
220
        for hook in MutableTree.hooks['start_commit']:
195
221
            hook(self)
196
222
        committed_id = commit.Commit().commit(working_tree=self,
197
 
            revprops=revprops, *args, **kwargs)
 
223
            revprops=revprops,
 
224
            possible_master_transports=possible_master_transports,
 
225
            *args, **kwargs)
 
226
        post_hook_params = PostCommitHookParams(self)
 
227
        for hook in MutableTree.hooks['post_commit']:
 
228
            hook(post_hook_params)
198
229
        return committed_id
199
230
 
200
231
    def _gather_kinds(self, files, kinds):
201
232
        """Helper function for add - sets the entries of kinds."""
202
233
        raise NotImplementedError(self._gather_kinds)
203
234
 
204
 
    def get_file_with_stat(self, file_id, path=None):
205
 
        """Get a file handle and stat object for file_id.
206
 
 
207
 
        The default implementation returns (self.get_file, None) for backwards
208
 
        compatibility.
209
 
 
210
 
        :param file_id: The file id to read.
211
 
        :param path: The path of the file, if it is known.
212
 
        :return: A tuple (file_handle, stat_value_or_None). If the tree has
213
 
            no stat facility, or need for a stat cache feedback during commit,
214
 
            it may return None for the second element of the tuple.
215
 
        """
216
 
        return (self.get_file(file_id, path), None)
 
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
 
243
        """
 
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
 
259
 
 
260
    @needs_read_lock
 
261
    def warn_if_changed_or_out_of_date(self, strict, opt_name, more_msg):
 
262
        """Check the tree for uncommitted changes and branch synchronization.
 
263
 
 
264
        If strict is None and not set in the config files, a warning is issued.
 
265
        If strict is True, an error is raised.
 
266
        If strict is False, no checks are done and no warning is issued.
 
267
 
 
268
        :param strict: True, False or None, searched in branch config if None.
 
269
 
 
270
        :param opt_name: strict option name to search in config file.
 
271
 
 
272
        :param more_msg: Details about how to avoid the warnings.
 
273
        """
 
274
        if strict is None:
 
275
            strict = self.branch.get_config().get_user_option_as_bool(opt_name)
 
276
        if strict is not False:
 
277
            err = None
 
278
            if (self.has_changes()):
 
279
                err = errors.UncommittedChanges(self, more=more_msg)
 
280
            elif self.last_revision() != self.branch.last_revision():
 
281
                # The tree has lost sync with its branch, there is little
 
282
                # chance that the user is aware of it but he can still force
 
283
                # the action with --no-strict
 
284
                err = errors.OutOfDateTree(self, more=more_msg)
 
285
            if err is not None:
 
286
                if strict is None:
 
287
                    # We don't want to interrupt the user if he expressed no
 
288
                    # preference about strict.
 
289
                    trace.warning('%s', err._format())
 
290
                else:
 
291
                    raise err
217
292
 
218
293
    @needs_read_lock
219
294
    def last_revision(self):
222
297
        In early tree formats the result of last_revision is the same as the
223
298
        branch last_revision, but that is no longer the case for modern tree
224
299
        formats.
225
 
        
 
300
 
226
301
        last_revision returns the left most parent id, or None if there are no
227
302
        parents.
228
303
 
245
320
    def lock_write(self):
246
321
        """Lock the tree and its branch. This allows mutating calls to be made.
247
322
 
248
 
        Some mutating methods will take out implicit write locks, but in 
 
323
        Some mutating methods will take out implicit write locks, but in
249
324
        general you should always obtain a write lock before calling mutating
250
325
        methods on a tree.
251
326
        """
267
342
        The intent of this function is to allow trees that have a hashcache to
268
343
        update the hashcache during commit. If the observed file is too new
269
344
        (based on the stat_value) to be safely hash-cached the tree will ignore
270
 
        it. 
 
345
        it.
271
346
 
272
347
        The default implementation does nothing.
273
348
 
278
353
        :return: None
279
354
        """
280
355
 
 
356
    def _fix_case_of_inventory_path(self, path):
 
357
        """If our tree isn't case sensitive, return the canonical path"""
 
358
        if not self.case_sensitive:
 
359
            path = self.get_canonical_inventory_path(path)
 
360
        return path
 
361
 
281
362
    @needs_write_lock
282
363
    def put_file_bytes_non_atomic(self, file_id, bytes):
283
364
        """Update the content of a file in the tree.
284
 
        
 
365
 
285
366
        Note that the file is written in-place rather than being
286
367
        written to a temporary location and renamed. As a consequence,
287
368
        readers can potentially see the file half-written.
301
382
    def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
302
383
        """Set the parents of the working tree.
303
384
 
304
 
        :param parents_list: A list of (revision_id, tree) tuples. 
 
385
        :param parents_list: A list of (revision_id, tree) tuples.
305
386
            If tree is None, then that element is treated as an unreachable
306
387
            parent tree - i.e. a ghost.
307
388
        """
315
396
        For the specific behaviour see the help for cmd_add().
316
397
 
317
398
        :param action: A reporter to be called with the inventory, parent_ie,
318
 
            path and kind of the path being added. It may return a file_id if 
 
399
            path and kind of the path being added. It may return a file_id if
319
400
            a specific one should be used.
320
401
        :param save: Save the inventory after completing the adds. If False
321
402
            this provides dry-run functionality by doing the add and not saving
329
410
        from bzrlib.inventory import InventoryEntry
330
411
        if action is None:
331
412
            action = add.AddAction()
332
 
        
 
413
 
333
414
        if not file_list:
334
415
            # no paths supplied: add the entire tree.
 
416
            # FIXME: this assumes we are running in a working tree subdir :-/
 
417
            # -- vila 20100208
335
418
            file_list = [u'.']
336
419
        # mutter("smart add of %r")
337
420
        inv = self.inventory
339
422
        ignored = {}
340
423
        dirs_to_add = []
341
424
        user_dirs = set()
 
425
        conflicts_related = set()
 
426
        # Not all mutable trees can have conflicts
 
427
        if getattr(self, 'conflicts', None) is not None:
 
428
            # Collect all related files without checking whether they exist or
 
429
            # are versioned. It's cheaper to do that once for all conflicts
 
430
            # than trying to find the relevant conflict for each added file.
 
431
            for c in self.conflicts():
 
432
                conflicts_related.update(c.associated_filenames())
342
433
 
343
 
        # validate user file paths and convert all paths to tree 
 
434
        # validate user file paths and convert all paths to tree
344
435
        # relative : it's cheaper to make a tree relative path an abspath
345
 
        # than to convert an abspath to tree relative.
346
 
        for filepath in file_list:
347
 
            rf = _FastPath(self.relpath(filepath))
348
 
            # validate user parameters. Our recursive code avoids adding new files
349
 
            # that need such validation 
 
436
        # than to convert an abspath to tree relative, and it's cheaper to
 
437
        # perform the canonicalization in bulk.
 
438
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
 
439
            rf = _FastPath(filepath)
 
440
            # validate user parameters. Our recursive code avoids adding new
 
441
            # files that need such validation
350
442
            if self.is_control_filename(rf.raw_path):
351
443
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
352
 
            
 
444
 
353
445
            abspath = self.abspath(rf.raw_path)
354
446
            kind = osutils.file_kind(abspath)
355
447
            if kind == 'directory':
358
450
            else:
359
451
                if not InventoryEntry.versionable_kind(kind):
360
452
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
361
 
            # ensure the named path is added, so that ignore rules in the later directory
362
 
            # walk dont skip it.
363
 
            # we dont have a parent ie known yet.: use the relatively slower inventory 
364
 
            # probing method
 
453
            # ensure the named path is added, so that ignore rules in the later
 
454
            # directory walk dont skip it.
 
455
            # we dont have a parent ie known yet.: use the relatively slower
 
456
            # inventory probing method
365
457
            versioned = inv.has_filename(rf.raw_path)
366
458
            if versioned:
367
459
                continue
383
475
                dirs_to_add.append((path, None))
384
476
            prev_dir = path.raw_path
385
477
 
 
478
        illegalpath_re = re.compile(r'[\r\n]')
386
479
        # dirs_to_add is initialised to a list of directories, but as we scan
387
480
        # directories we append files to it.
388
481
        # XXX: We should determine kind of files when we scan them rather than
397
490
            kind = osutils.file_kind(abspath)
398
491
 
399
492
            if not InventoryEntry.versionable_kind(kind):
400
 
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
 
493
                trace.warning("skipping %s (can't add file of kind '%s')",
 
494
                              abspath, kind)
 
495
                continue
 
496
            if illegalpath_re.search(directory.raw_path):
 
497
                trace.warning("skipping %r (contains \\n or \\r)" % abspath)
 
498
                continue
 
499
            if directory.raw_path in conflicts_related:
 
500
                # If the file looks like one generated for a conflict, don't
 
501
                # add it.
 
502
                trace.warning(
 
503
                    'skipping %s (generated to help resolve conflicts)',
 
504
                    abspath)
401
505
                continue
402
506
 
403
507
            if parent_ie is not None:
404
508
                versioned = directory.base_path in parent_ie.children
405
509
            else:
406
 
                # without the parent ie, use the relatively slower inventory 
 
510
                # without the parent ie, use the relatively slower inventory
407
511
                # probing method
408
 
                versioned = inv.has_filename(directory.raw_path)
 
512
                versioned = inv.has_filename(
 
513
                        self._fix_case_of_inventory_path(directory.raw_path))
409
514
 
410
515
            if kind == 'directory':
411
516
                try:
425
530
                pass
426
531
                # mutter("%r is already versioned", abspath)
427
532
            elif sub_tree:
428
 
                # XXX: This is wrong; people *might* reasonably be trying to add
429
 
                # subtrees as subtrees.  This should probably only be done in formats 
430
 
                # which can represent subtrees, and even then perhaps only when
431
 
                # the user asked to add subtrees.  At the moment you can add them
432
 
                # specially through 'join --reference', which is perhaps
433
 
                # reasonable: adding a new reference is a special operation and
434
 
                # can have a special behaviour.  mbp 20070306
435
 
                mutter("%r is a nested bzr tree", abspath)
 
533
                # XXX: This is wrong; people *might* reasonably be trying to
 
534
                # add subtrees as subtrees.  This should probably only be done
 
535
                # in formats which can represent subtrees, and even then
 
536
                # perhaps only when the user asked to add subtrees.  At the
 
537
                # moment you can add them specially through 'join --reference',
 
538
                # which is perhaps reasonable: adding a new reference is a
 
539
                # special operation and can have a special behaviour.  mbp
 
540
                # 20070306
 
541
                trace.mutter("%r is a nested bzr tree", abspath)
436
542
            else:
437
543
                _add_one(self, inv, parent_ie, directory, kind, action)
438
544
                added.append(directory.raw_path)
442
548
                    # must be present:
443
549
                    this_ie = parent_ie.children[directory.base_path]
444
550
                else:
445
 
                    # without the parent ie, use the relatively slower inventory 
 
551
                    # without the parent ie, use the relatively slower inventory
446
552
                    # probing method
447
 
                    this_id = inv.path2id(directory.raw_path)
 
553
                    this_id = inv.path2id(
 
554
                        self._fix_case_of_inventory_path(directory.raw_path))
448
555
                    if this_id is None:
449
556
                        this_ie = None
450
557
                    else:
451
558
                        this_ie = inv[this_id]
452
559
 
453
560
                for subf in sorted(os.listdir(abspath)):
454
 
                    # here we could use TreeDirectory rather than 
 
561
                    # here we could use TreeDirectory rather than
455
562
                    # string concatenation.
456
563
                    subp = osutils.pathjoin(directory.raw_path, subf)
457
 
                    # TODO: is_control_filename is very slow. Make it faster. 
458
 
                    # TreeDirectory.is_control_filename could also make this 
459
 
                    # faster - its impossible for a non root dir to have a 
 
564
                    # TODO: is_control_filename is very slow. Make it faster.
 
565
                    # TreeDirectory.is_control_filename could also make this
 
566
                    # faster - its impossible for a non root dir to have a
460
567
                    # control file.
461
568
                    if self.is_control_filename(subp):
462
 
                        mutter("skip control directory %r", subp)
 
569
                        trace.mutter("skip control directory %r", subp)
463
570
                    elif subf in this_ie.children:
464
571
                        # recurse into this already versioned subdir.
465
572
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
497
604
        for commit which is not required to handle situations that do not arise
498
605
        outside of commit.
499
606
 
 
607
        See the inventory developers documentation for the theory behind
 
608
        inventory deltas.
 
609
 
500
610
        :param new_revid: The new revision id for the trees parent.
501
611
        :param delta: An inventory delta (see apply_inventory_delta) describing
502
612
            the changes from the current left most parent revision to new_revid.
512
622
        # WorkingTree classes for optimised versions for specific format trees.
513
623
        basis = self.basis_tree()
514
624
        basis.lock_read()
515
 
        inventory = basis.inventory
 
625
        # TODO: Consider re-evaluating the need for this with CHKInventory
 
626
        # we don't strictly need to mutate an inventory for this
 
627
        # it only makes sense when apply_delta is cheaper than get_inventory()
 
628
        inventory = basis.inventory._get_mutable_inventory()
516
629
        basis.unlock()
517
630
        inventory.apply_delta(delta)
518
 
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
 
631
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
 
632
                                             inventory, new_revid)
519
633
        self.set_parent_trees([(new_revid, rev_tree)])
520
634
 
521
635
 
522
636
class MutableTreeHooks(hooks.Hooks):
523
 
    """A dictionary mapping a hook name to a list of callables for mutabletree 
 
637
    """A dictionary mapping a hook name to a list of callables for mutabletree
524
638
    hooks.
525
639
    """
526
640
 
529
643
 
530
644
        """
531
645
        hooks.Hooks.__init__(self)
532
 
        # Invoked before a commit is done in a tree. New in 1.4
533
 
        self['start_commit'] = []
 
646
        self.create_hook(hooks.HookPoint('start_commit',
 
647
            "Called before a commit is performed on a tree. The start commit "
 
648
            "hook is able to change the tree before the commit takes place. "
 
649
            "start_commit is called with the bzrlib.mutabletree.MutableTree "
 
650
            "that the commit is being performed on.", (1, 4), None))
 
651
        self.create_hook(hooks.HookPoint('post_commit',
 
652
            "Called after a commit is performed on a tree. The hook is "
 
653
            "called with a bzrlib.mutabletree.PostCommitHookParams object. "
 
654
            "The mutable tree the commit was performed on is available via "
 
655
            "the mutable_tree attribute of that object.", (2, 0), None))
534
656
 
535
657
 
536
658
# install the default hooks into the MutableTree class.
537
659
MutableTree.hooks = MutableTreeHooks()
538
660
 
539
661
 
 
662
class PostCommitHookParams(object):
 
663
    """Parameters for the post_commit hook.
 
664
 
 
665
    To access the parameters, use the following attributes:
 
666
 
 
667
    * mutable_tree - the MutableTree object
 
668
    """
 
669
 
 
670
    def __init__(self, mutable_tree):
 
671
        """Create the parameters for the post_commit hook."""
 
672
        self.mutable_tree = mutable_tree
 
673
 
 
674
 
540
675
class _FastPath(object):
541
676
    """A path object with fast accessors for things like basename."""
542
677
 
576
711
        added = []
577
712
    else:
578
713
        # slower but does not need parent_ie
579
 
        if inv.has_filename(path.raw_path):
 
714
        if inv.has_filename(tree._fix_case_of_inventory_path(path.raw_path)):
580
715
            return []
581
716
        # its really not there : add the parent
582
717
        # note that the dirname use leads to some extra str copying etc but as
583
718
        # there are a limited number of dirs we can be nested under, it should
584
719
        # generally find it very fast and not recurse after that.
585
720
        added = _add_one_and_parent(tree, inv, None,
586
 
            _FastPath(dirname(path.raw_path)), 'directory', action)
587
 
        parent_id = inv.path2id(dirname(path.raw_path))
 
721
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
 
722
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
588
723
        parent_ie = inv[parent_id]
589
724
    _add_one(tree, inv, parent_ie, path, kind, action)
590
725
    return added + [path.raw_path]