23
23
from bzrlib.lazy_import import lazy_import
24
24
lazy_import(globals(), """
27
28
from bzrlib import (
33
from bzrlib.osutils import dirname
34
from bzrlib.revisiontree import RevisionTree
35
from bzrlib.trace import mutter, warning
43
41
from bzrlib.decorators import needs_read_lock, needs_write_lock
44
from bzrlib.osutils import splitpath
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)
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
226
223
revprops=revprops,
227
224
possible_master_transports=possible_master_transports,
226
post_hook_params = PostCommitHookParams(self)
227
for hook in MutableTree.hooks['post_commit']:
228
hook(post_hook_params)
229
229
return committed_id
231
231
def _gather_kinds(self, files, kinds):
233
233
raise NotImplementedError(self._gather_kinds)
236
def has_changes(self, from_tree):
237
"""Quickly check that the tree contains at least one change.
236
def has_changes(self, _from_tree=None):
237
"""Quickly check that the tree contains at least one commitable change.
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).
239
242
:return: True if a change is found. False otherwise
241
changes = self.iter_changes(from_tree)
244
# Check pending merges
245
if len(self.get_parent_ids()) > 1:
247
if _from_tree is None:
248
_from_tree = self.basis_tree()
249
changes = self.iter_changes(_from_tree)
243
251
change = changes.next()
244
252
# Exclude root (talk about black magic... --vila 20090629)
381
391
user_dirs = set()
392
conflicts_related = set()
393
# Not all mutable trees can have conflicts
394
if getattr(self, 'conflicts', None) is not None:
395
# Collect all related files without checking whether they exist or
396
# are versioned. It's cheaper to do that once for all conflicts
397
# than trying to find the relevant conflict for each added file.
398
for c in self.conflicts():
399
conflicts_related.update(c.associated_filenames())
383
401
# validate user file paths and convert all paths to tree
384
402
# relative : it's cheaper to make a tree relative path an abspath
386
404
# perform the canonicalization in bulk.
387
405
for filepath in osutils.canonical_relpaths(self.basedir, file_list):
388
406
rf = _FastPath(filepath)
389
# validate user parameters. Our recursive code avoids adding new files
390
# that need such validation
407
# validate user parameters. Our recursive code avoids adding new
408
# files that need such validation
391
409
if self.is_control_filename(rf.raw_path):
392
410
raise errors.ForbiddenControlFileError(filename=rf.raw_path)
400
418
if not InventoryEntry.versionable_kind(kind):
401
419
raise errors.BadFileKindError(filename=abspath, kind=kind)
402
# ensure the named path is added, so that ignore rules in the later directory
404
# we dont have a parent ie known yet.: use the relatively slower inventory
420
# ensure the named path is added, so that ignore rules in the later
421
# directory walk dont skip it.
422
# we dont have a parent ie known yet.: use the relatively slower
423
# inventory probing method
406
424
versioned = inv.has_filename(rf.raw_path)
424
442
dirs_to_add.append((path, None))
425
443
prev_dir = path.raw_path
445
illegalpath_re = re.compile(r'[\r\n]')
427
446
# dirs_to_add is initialised to a list of directories, but as we scan
428
447
# directories we append files to it.
429
448
# XXX: We should determine kind of files when we scan them rather than
438
457
kind = osutils.file_kind(abspath)
440
459
if not InventoryEntry.versionable_kind(kind):
441
warning("skipping %s (can't add file of kind '%s')", abspath, kind)
460
trace.warning("skipping %s (can't add file of kind '%s')",
463
if illegalpath_re.search(directory.raw_path):
464
trace.warning("skipping %r (contains \\n or \\r)" % abspath)
466
if directory.raw_path in conflicts_related:
467
# If the file looks like one generated for a conflict, don't
470
'skipping %s (generated to help resolve conflicts)',
444
474
if parent_ie is not None:
468
498
# mutter("%r is already versioned", abspath)
470
# XXX: This is wrong; people *might* reasonably be trying to add
471
# subtrees as subtrees. This should probably only be done in formats
472
# which can represent subtrees, and even then perhaps only when
473
# the user asked to add subtrees. At the moment you can add them
474
# specially through 'join --reference', which is perhaps
475
# reasonable: adding a new reference is a special operation and
476
# can have a special behaviour. mbp 20070306
477
mutter("%r is a nested bzr tree", abspath)
500
# XXX: This is wrong; people *might* reasonably be trying to
501
# add subtrees as subtrees. This should probably only be done
502
# in formats which can represent subtrees, and even then
503
# perhaps only when the user asked to add subtrees. At the
504
# moment you can add them specially through 'join --reference',
505
# which is perhaps reasonable: adding a new reference is a
506
# special operation and can have a special behaviour. mbp
508
trace.mutter("%r is a nested bzr tree", abspath)
479
510
_add_one(self, inv, parent_ie, directory, kind, action)
480
511
added.append(directory.raw_path)
502
533
# faster - its impossible for a non root dir to have a
504
535
if self.is_control_filename(subp):
505
mutter("skip control directory %r", subp)
536
trace.mutter("skip control directory %r", subp)
506
537
elif subf in this_ie.children:
507
538
# recurse into this already versioned subdir.
508
539
dirs_to_add.append((_FastPath(subp, subf), this_ie))
564
595
inventory = basis.inventory._get_mutable_inventory()
566
597
inventory.apply_delta(delta)
567
rev_tree = RevisionTree(self.branch.repository, inventory, new_revid)
598
rev_tree = revisiontree.RevisionTree(self.branch.repository,
599
inventory, new_revid)
568
600
self.set_parent_trees([(new_revid, rev_tree)])
581
613
self.create_hook(hooks.HookPoint('start_commit',
582
614
"Called before a commit is performed on a tree. The start commit "
583
615
"hook is able to change the tree before the commit takes place. "
584
"start_commit is called with the bzrlib.tree.MutableTree that the "
585
"commit is being performed on.", (1, 4), None))
616
"start_commit is called with the bzrlib.mutabletree.MutableTree "
617
"that the commit is being performed on.", (1, 4), None))
618
self.create_hook(hooks.HookPoint('post_commit',
619
"Called after a commit is performed on a tree. The hook is "
620
"called with a bzrlib.mutabletree.PostCommitHookParams object. "
621
"The mutable tree the commit was performed on is available via "
622
"the mutable_tree attribute of that object.", (2, 0), None))
588
625
# install the default hooks into the MutableTree class.
589
626
MutableTree.hooks = MutableTreeHooks()
629
class PostCommitHookParams(object):
630
"""Parameters for the post_commit hook.
632
To access the parameters, use the following attributes:
634
* mutable_tree - the MutableTree object
637
def __init__(self, mutable_tree):
638
"""Create the parameters for the post_commit hook."""
639
self.mutable_tree = mutable_tree
592
642
class _FastPath(object):
593
643
"""A path object with fast accessors for things like basename."""
635
685
# there are a limited number of dirs we can be nested under, it should
636
686
# generally find it very fast and not recurse after that.
637
687
added = _add_one_and_parent(tree, inv, None,
638
_FastPath(dirname(path.raw_path)), 'directory', action)
639
parent_id = inv.path2id(dirname(path.raw_path))
688
_FastPath(osutils.dirname(path.raw_path)), 'directory', action)
689
parent_id = inv.path2id(osutils.dirname(path.raw_path))
640
690
parent_ie = inv[parent_id]
641
691
_add_one(tree, inv, parent_ie, path, kind, action)
642
692
return added + [path.raw_path]
added
removed
bzrlib/mutabletree.py
