/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: John Arbash Meinel
  • Date: 2009-09-24 19:26:45 UTC
  • mto: (4634.52.3 2.0)
  • mto: This revision was merged to the branch mainline in revision 4716.
  • Revision ID: john@arbash-meinel.com-20090924192645-hyy1ycnnk6u3j5j6
Catch a corner case that we were missing.
The CHKInventory tests were passing, but failed for test_inv because
we were passing None to _getitems(). That only failed for InternalNodes,
but we were using a structure that didn't have internal nodes.
So now the test is parameterized on a small CHKInventory page size
to force those things out into the open.

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
 
    symbol_versioning,
36
 
    trace,
37
41
    tree,
38
42
    )
39
 
""")
40
 
 
41
43
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
44
from bzrlib.osutils import splitpath
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
        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
193
221
        # args for wt.commit start at message from the Commit.commit method,
194
222
        args = (message, ) + args
195
223
        for hook in MutableTree.hooks['start_commit']:
208
236
        raise NotImplementedError(self._gather_kinds)
209
237
 
210
238
    @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).
 
239
    def has_changes(self, from_tree):
 
240
        """Quickly check that the tree contains at least one change.
216
241
 
217
242
        :return: True if a change is found. False otherwise
218
243
        """
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)
 
244
        changes = self.iter_changes(from_tree)
225
245
        try:
226
246
            change = changes.next()
227
247
            # Exclude root (talk about black magic... --vila 20090629)
233
253
            return False
234
254
 
235
255
    @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
272
 
 
273
 
    @needs_read_lock
274
256
    def last_revision(self):
275
257
        """Return the revision id of the last commit performed in this tree.
276
258
 
375
357
        This is designed more towards DWIM for humans than API clarity.
376
358
        For the specific behaviour see the help for cmd_add().
377
359
 
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.)
382
360
        :param action: A reporter to be called with the inventory, parent_ie,
383
361
            path and kind of the path being added. It may return a file_id if
384
362
            a specific one should be used.
397
375
 
398
376
        if not file_list:
399
377
            # no paths supplied: add the entire tree.
400
 
            # FIXME: this assumes we are running in a working tree subdir :-/
401
 
            # -- vila 20100208
402
378
            file_list = [u'.']
403
379
        # mutter("smart add of %r")
404
380
        inv = self.inventory
406
382
        ignored = {}
407
383
        dirs_to_add = []
408
384
        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())
417
385
 
418
386
        # validate user file paths and convert all paths to tree
419
387
        # relative : it's cheaper to make a tree relative path an abspath
421
389
        # perform the canonicalization in bulk.
422
390
        for filepath in osutils.canonical_relpaths(self.basedir, file_list):
423
391
            rf = _FastPath(filepath)
424
 
            # validate user parameters. Our recursive code avoids adding new
425
 
            # files that need such validation
 
392
            # validate user parameters. Our recursive code avoids adding new files
 
393
            # that need such validation
426
394
            if self.is_control_filename(rf.raw_path):
427
395
                raise errors.ForbiddenControlFileError(filename=rf.raw_path)
428
396
 
434
402
            else:
435
403
                if not InventoryEntry.versionable_kind(kind):
436
404
                    raise errors.BadFileKindError(filename=abspath, kind=kind)
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
 
405
            # ensure the named path is added, so that ignore rules in the later directory
 
406
            # walk dont skip it.
 
407
            # we dont have a parent ie known yet.: use the relatively slower inventory
 
408
            # probing method
441
409
            versioned = inv.has_filename(rf.raw_path)
442
410
            if versioned:
443
411
                continue
459
427
                dirs_to_add.append((path, None))
460
428
            prev_dir = path.raw_path
461
429
 
462
 
        illegalpath_re = re.compile(r'[\r\n]')
463
430
        # dirs_to_add is initialised to a list of directories, but as we scan
464
431
        # directories we append files to it.
465
432
        # XXX: We should determine kind of files when we scan them rather than
474
441
            kind = osutils.file_kind(abspath)
475
442
 
476
443
            if not InventoryEntry.versionable_kind(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)
 
444
                warning("skipping %s (can't add file of kind '%s')", abspath, kind)
489
445
                continue
490
446
 
491
447
            if parent_ie is not None:
514
470
                pass
515
471
                # mutter("%r is already versioned", abspath)
516
472
            elif sub_tree:
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)
 
473
                # XXX: This is wrong; people *might* reasonably be trying to add
 
474
                # subtrees as subtrees.  This should probably only be done in formats
 
475
                # which can represent subtrees, and even then perhaps only when
 
476
                # the user asked to add subtrees.  At the moment you can add them
 
477
                # specially through 'join --reference', which is perhaps
 
478
                # reasonable: adding a new reference is a special operation and
 
479
                # can have a special behaviour.  mbp 20070306
 
480
                mutter("%r is a nested bzr tree", abspath)
526
481
            else:
527
482
                _add_one(self, inv, parent_ie, directory, kind, action)
528
483
                added.append(directory.raw_path)
535
490
                    # without the parent ie, use the relatively slower inventory
536
491
                    # probing method
537
492
                    this_id = inv.path2id(
538
 
                        self._fix_case_of_inventory_path(directory.raw_path))
 
493
                            self._fix_case_of_inventory_path(directory.raw_path))
539
494
                    if this_id is None:
540
495
                        this_ie = None
541
496
                    else:
550
505
                    # faster - its impossible for a non root dir to have a
551
506
                    # control file.
552
507
                    if self.is_control_filename(subp):
553
 
                        trace.mutter("skip control directory %r", subp)
 
508
                        mutter("skip control directory %r", subp)
554
509
                    elif subf in this_ie.children:
555
510
                        # recurse into this already versioned subdir.
556
511
                        dirs_to_add.append((_FastPath(subp, subf), this_ie))
612
567
        inventory = basis.inventory._get_mutable_inventory()
613
568
        basis.unlock()
614
569
        inventory.apply_delta(delta)
615
 
        rev_tree = revisiontree.RevisionTree(self.branch.repository,
616
 
                                             inventory, new_revid)
 
570
        rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
617
571
        self.set_parent_trees([(new_revid, rev_tree)])
618
572
 
619
573
 
702
656
        # there are a limited number of dirs we can be nested under, it should
703
657
        # generally find it very fast and not recurse after that.
704
658
        added = _add_one_and_parent(tree, inv, None,
705
 
            _FastPath(osutils.dirname(path.raw_path)), 'directory', action)
706
 
        parent_id = inv.path2id(osutils.dirname(path.raw_path))
 
659
            _FastPath(dirname(path.raw_path)), 'directory', action)
 
660
        parent_id = inv.path2id(dirname(path.raw_path))
707
661
        parent_ie = inv[parent_id]
708
662
    _add_one(tree, inv, parent_ie, path, kind, action)
709
663
    return added + [path.raw_path]