~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transform.py

(jameinel) Transform.rename test shouldn't assume the str form of the
 exception because of locale issues. (Martin [gz])

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2006-2011 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
19
19
from stat import S_ISREG, S_IEXEC
20
20
import time
21
21
 
22
 
from bzrlib.lazy_import import lazy_import
23
 
lazy_import(globals(), """
 
22
import bzrlib
 
23
from bzrlib import (
 
24
    errors,
 
25
    lazy_import,
 
26
    registry,
 
27
    tree,
 
28
    )
 
29
lazy_import.lazy_import(globals(), """
24
30
from bzrlib import (
25
31
    annotate,
26
32
    bencode,
27
33
    bzrdir,
 
34
    commit,
28
35
    delta,
29
36
    errors,
30
37
    inventory,
31
38
    multiparent,
32
39
    osutils,
33
40
    revision as _mod_revision,
 
41
    trace,
34
42
    ui,
35
43
    )
36
44
""")
44
52
    delete_any,
45
53
    file_kind,
46
54
    has_symlinks,
47
 
    lexists,
48
55
    pathjoin,
49
56
    sha_file,
50
57
    splitpath,
51
58
    supports_executable,
52
 
)
 
59
    )
53
60
from bzrlib.progress import ProgressPhase
54
61
from bzrlib.symbol_versioning import (
55
 
        deprecated_function,
56
 
        deprecated_in,
57
 
        )
58
 
from bzrlib.trace import mutter, warning
59
 
from bzrlib import tree
 
62
    deprecated_function,
 
63
    deprecated_in,
 
64
    deprecated_method,
 
65
    )
 
66
from bzrlib.trace import warning
60
67
import bzrlib.ui
61
68
import bzrlib.urlutils as urlutils
62
69
 
63
70
 
64
71
ROOT_PARENT = "root-parent"
65
72
 
66
 
 
67
73
def unique_add(map, key, value):
68
74
    if key in map:
69
75
        raise DuplicateKey(key=key)
70
76
    map[key] = value
71
77
 
72
78
 
 
79
 
73
80
class _TransformResults(object):
74
81
    def __init__(self, modified_paths, rename_count):
75
82
        object.__init__(self)
123
130
            self._new_root = self.trans_id_tree_file_id(root_id)
124
131
        else:
125
132
            self._new_root = None
126
 
        # Indictor of whether the transform has been applied
 
133
        # Indicator of whether the transform has been applied
127
134
        self._done = False
128
135
        # A progress bar
129
136
        self._pb = pb
314
321
 
315
322
    def delete_contents(self, trans_id):
316
323
        """Schedule the contents of a path entry for deletion"""
317
 
        # Ensure that the object exists in the WorkingTree, this will raise an
318
 
        # exception if there is a problem
319
 
        self.tree_kind(trans_id)
320
 
        self._removed_contents.add(trans_id)
 
324
        kind = self.tree_kind(trans_id)
 
325
        if kind is not None:
 
326
            self._removed_contents.add(trans_id)
321
327
 
322
328
    def cancel_deletion(self, trans_id):
323
329
        """Cancel a scheduled deletion"""
388
394
        changed_kind = set(self._removed_contents)
389
395
        changed_kind.intersection_update(self._new_contents)
390
396
        changed_kind.difference_update(new_ids)
391
 
        changed_kind = (t for t in changed_kind if self.tree_kind(t) !=
392
 
                        self.final_kind(t))
 
397
        changed_kind = (t for t in changed_kind
 
398
                        if self.tree_kind(t) != self.final_kind(t))
393
399
        new_ids.update(changed_kind)
394
400
        return sorted(FinalPaths(self).get_paths(new_ids))
395
401
 
396
402
    def final_kind(self, trans_id):
397
403
        """Determine the final file kind, after any changes applied.
398
404
 
399
 
        Raises NoSuchFile if the file does not exist/has no contents.
400
 
        (It is conceivable that a path would be created without the
401
 
        corresponding contents insertion command)
 
405
        :return: None if the file does not exist/has no contents.  (It is
 
406
            conceivable that a path would be created without the corresponding
 
407
            contents insertion command)
402
408
        """
403
409
        if trans_id in self._new_contents:
404
410
            return self._new_contents[trans_id]
405
411
        elif trans_id in self._removed_contents:
406
 
            raise NoSuchFile(None)
 
412
            return None
407
413
        else:
408
414
            return self.tree_kind(trans_id)
409
415
 
522
528
        for trans_id in self._removed_id:
523
529
            file_id = self.tree_file_id(trans_id)
524
530
            if file_id is not None:
 
531
                # XXX: This seems like something that should go via a different
 
532
                #      indirection.
525
533
                if self._tree.inventory[file_id].kind == 'directory':
526
534
                    parents.append(trans_id)
527
535
            elif self.tree_kind(trans_id) == 'directory':
531
539
            # ensure that all children are registered with the transaction
532
540
            list(self.iter_tree_children(parent_id))
533
541
 
 
542
    @deprecated_method(deprecated_in((2, 3, 0)))
534
543
    def has_named_child(self, by_parent, parent_id, name):
535
 
        try:
536
 
            children = by_parent[parent_id]
537
 
        except KeyError:
538
 
            children = []
539
 
        for child in children:
 
544
        return self._has_named_child(
 
545
            name, parent_id, known_children=by_parent.get(parent_id, []))
 
546
 
 
547
    def _has_named_child(self, name, parent_id, known_children):
 
548
        """Does a parent already have a name child.
 
549
 
 
550
        :param name: The searched for name.
 
551
 
 
552
        :param parent_id: The parent for which the check is made.
 
553
 
 
554
        :param known_children: The already known children. This should have
 
555
            been recently obtained from `self.by_parent.get(parent_id)`
 
556
            (or will be if None is passed).
 
557
        """
 
558
        if known_children is None:
 
559
            known_children = self.by_parent().get(parent_id, [])
 
560
        for child in known_children:
540
561
            if self.final_name(child) == name:
541
562
                return True
542
 
        try:
543
 
            path = self._tree_id_paths[parent_id]
544
 
        except KeyError:
 
563
        parent_path = self._tree_id_paths.get(parent_id, None)
 
564
        if parent_path is None:
 
565
            # No parent... no children
545
566
            return False
546
 
        childpath = joinpath(path, name)
547
 
        child_id = self._tree_path_ids.get(childpath)
 
567
        child_path = joinpath(parent_path, name)
 
568
        child_id = self._tree_path_ids.get(child_path, None)
548
569
        if child_id is None:
549
 
            return lexists(self._tree.abspath(childpath))
 
570
            # Not known by the tree transform yet, check the filesystem
 
571
            return osutils.lexists(self._tree.abspath(child_path))
550
572
        else:
551
 
            if self.final_parent(child_id) != parent_id:
552
 
                return False
553
 
            if child_id in self._removed_contents:
554
 
                # XXX What about dangling file-ids?
555
 
                return False
556
 
            else:
557
 
                return True
 
573
            raise AssertionError('child_id is missing: %s, %s, %s'
 
574
                                 % (name, parent_id, child_id))
 
575
 
 
576
    def _available_backup_name(self, name, target_id):
 
577
        """Find an available backup name.
 
578
 
 
579
        :param name: The basename of the file.
 
580
 
 
581
        :param target_id: The directory trans_id where the backup should 
 
582
            be placed.
 
583
        """
 
584
        known_children = self.by_parent().get(target_id, [])
 
585
        return osutils.available_backup_name(
 
586
            name,
 
587
            lambda base: self._has_named_child(
 
588
                base, target_id, known_children))
558
589
 
559
590
    def _parent_loops(self):
560
591
        """No entry should be its own ancestor"""
595
626
        """
596
627
        conflicts = []
597
628
        for trans_id in self._new_id.iterkeys():
598
 
            try:
599
 
                kind = self.final_kind(trans_id)
600
 
            except NoSuchFile:
 
629
            kind = self.final_kind(trans_id)
 
630
            if kind is None:
601
631
                conflicts.append(('versioning no contents', trans_id))
602
632
                continue
603
633
            if not InventoryEntry.versionable_kind(kind):
617
647
            if self.final_file_id(trans_id) is None:
618
648
                conflicts.append(('unversioned executability', trans_id))
619
649
            else:
620
 
                try:
621
 
                    non_file = self.final_kind(trans_id) != "file"
622
 
                except NoSuchFile:
623
 
                    non_file = True
624
 
                if non_file is True:
 
650
                if self.final_kind(trans_id) != "file":
625
651
                    conflicts.append(('non-file executability', trans_id))
626
652
        return conflicts
627
653
 
629
655
        """Check for overwrites (not permitted on Win32)"""
630
656
        conflicts = []
631
657
        for trans_id in self._new_contents:
632
 
            try:
633
 
                self.tree_kind(trans_id)
634
 
            except NoSuchFile:
 
658
            if self.tree_kind(trans_id) is None:
635
659
                continue
636
660
            if trans_id not in self._removed_contents:
637
661
                conflicts.append(('overwrite', trans_id,
644
668
        if (self._new_name, self._new_parent) == ({}, {}):
645
669
            return conflicts
646
670
        for children in by_parent.itervalues():
647
 
            name_ids = [(self.final_name(t), t) for t in children]
648
 
            if not self._case_sensitive_target:
649
 
                name_ids = [(n.lower(), t) for n, t in name_ids]
 
671
            name_ids = []
 
672
            for child_tid in children:
 
673
                name = self.final_name(child_tid)
 
674
                if name is not None:
 
675
                    # Keep children only if they still exist in the end
 
676
                    if not self._case_sensitive_target:
 
677
                        name = name.lower()
 
678
                    name_ids.append((name, child_tid))
650
679
            name_ids.sort()
651
680
            last_name = None
652
681
            last_trans_id = None
653
682
            for name, trans_id in name_ids:
654
 
                try:
655
 
                    kind = self.final_kind(trans_id)
656
 
                except NoSuchFile:
657
 
                    kind = None
 
683
                kind = self.final_kind(trans_id)
658
684
                file_id = self.final_file_id(trans_id)
659
685
                if kind is None and file_id is None:
660
686
                    continue
679
705
        return conflicts
680
706
 
681
707
    def _parent_type_conflicts(self, by_parent):
682
 
        """parents must have directory 'contents'."""
 
708
        """Children must have a directory parent"""
683
709
        conflicts = []
684
710
        for parent_id, children in by_parent.iteritems():
685
711
            if parent_id is ROOT_PARENT:
686
712
                continue
687
 
            if not self._any_contents(children):
 
713
            no_children = True
 
714
            for child_id in children:
 
715
                if self.final_kind(child_id) is not None:
 
716
                    no_children = False
 
717
                    break
 
718
            if no_children:
688
719
                continue
689
 
            for child in children:
690
 
                try:
691
 
                    self.final_kind(child)
692
 
                except NoSuchFile:
693
 
                    continue
694
 
            try:
695
 
                kind = self.final_kind(parent_id)
696
 
            except NoSuchFile:
697
 
                kind = None
 
720
            # There is at least a child, so we need an existing directory to
 
721
            # contain it.
 
722
            kind = self.final_kind(parent_id)
698
723
            if kind is None:
 
724
                # The directory will be deleted
699
725
                conflicts.append(('missing parent', parent_id))
700
726
            elif kind != "directory":
 
727
                # Meh, we need a *directory* to put something in it
701
728
                conflicts.append(('non-directory parent', parent_id))
702
729
        return conflicts
703
730
 
704
 
    def _any_contents(self, trans_ids):
705
 
        """Return true if any of the trans_ids, will have contents."""
706
 
        for trans_id in trans_ids:
707
 
            try:
708
 
                kind = self.final_kind(trans_id)
709
 
            except NoSuchFile:
710
 
                continue
711
 
            return True
712
 
        return False
713
 
 
714
731
    def _set_executability(self, path, trans_id):
715
732
        """Set the executability of versioned files """
716
733
        if supports_executable():
780
797
        self.create_symlink(target, trans_id)
781
798
        return trans_id
782
799
 
 
800
    def new_orphan(self, trans_id, parent_id):
 
801
        """Schedule an item to be orphaned.
 
802
 
 
803
        When a directory is about to be removed, its children, if they are not
 
804
        versioned are moved out of the way: they don't have a parent anymore.
 
805
 
 
806
        :param trans_id: The trans_id of the existing item.
 
807
        :param parent_id: The parent trans_id of the item.
 
808
        """
 
809
        raise NotImplementedError(self.new_orphan)
 
810
 
 
811
    def _get_potential_orphans(self, dir_id):
 
812
        """Find the potential orphans in a directory.
 
813
 
 
814
        A directory can't be safely deleted if there are versioned files in it.
 
815
        If all the contained files are unversioned then they can be orphaned.
 
816
 
 
817
        The 'None' return value means that the directory contains at least one
 
818
        versioned file and should not be deleted.
 
819
 
 
820
        :param dir_id: The directory trans id.
 
821
 
 
822
        :return: A list of the orphan trans ids or None if at least one
 
823
             versioned file is present.
 
824
        """
 
825
        orphans = []
 
826
        # Find the potential orphans, stop if one item should be kept
 
827
        for child_tid in self.by_parent()[dir_id]:
 
828
            if child_tid in self._removed_contents:
 
829
                # The child is removed as part of the transform. Since it was
 
830
                # versioned before, it's not an orphan
 
831
                continue
 
832
            elif self.final_file_id(child_tid) is None:
 
833
                # The child is not versioned
 
834
                orphans.append(child_tid)
 
835
            else:
 
836
                # We have a versioned file here, searching for orphans is
 
837
                # meaningless.
 
838
                orphans = None
 
839
                break
 
840
        return orphans
 
841
 
783
842
    def _affected_ids(self):
784
843
        """Return the set of transform ids affected by the transform"""
785
844
        trans_ids = set(self._removed_id)
844
903
        Return a (name, parent, kind, executable) tuple
845
904
        """
846
905
        to_name = self.final_name(to_trans_id)
847
 
        try:
848
 
            to_kind = self.final_kind(to_trans_id)
849
 
        except NoSuchFile:
850
 
            to_kind = None
 
906
        to_kind = self.final_kind(to_trans_id)
851
907
        to_parent = self.final_file_id(self.final_parent(to_trans_id))
852
908
        if to_trans_id in self._new_executability:
853
909
            to_executable = self._new_executability[to_trans_id]
927
983
        """
928
984
        return _PreviewTree(self)
929
985
 
930
 
    def commit(self, branch, message, merge_parents=None, strict=False):
 
986
    def commit(self, branch, message, merge_parents=None, strict=False,
 
987
               timestamp=None, timezone=None, committer=None, authors=None,
 
988
               revprops=None, revision_id=None):
931
989
        """Commit the result of this TreeTransform to a branch.
932
990
 
933
991
        :param branch: The branch to commit to.
934
992
        :param message: The message to attach to the commit.
935
 
        :param merge_parents: Additional parents specified by pending merges.
 
993
        :param merge_parents: Additional parent revision-ids specified by
 
994
            pending merges.
 
995
        :param strict: If True, abort the commit if there are unversioned
 
996
            files.
 
997
        :param timestamp: if not None, seconds-since-epoch for the time and
 
998
            date.  (May be a float.)
 
999
        :param timezone: Optional timezone for timestamp, as an offset in
 
1000
            seconds.
 
1001
        :param committer: Optional committer in email-id format.
 
1002
            (e.g. "J Random Hacker <jrandom@example.com>")
 
1003
        :param authors: Optional list of authors in email-id format.
 
1004
        :param revprops: Optional dictionary of revision properties.
 
1005
        :param revision_id: Optional revision id.  (Specifying a revision-id
 
1006
            may reduce performance for some non-native formats.)
936
1007
        :return: The revision_id of the revision committed.
937
1008
        """
938
1009
        self._check_malformed()
955
1026
        if self._tree.get_revision_id() != last_rev_id:
956
1027
            raise ValueError('TreeTransform not based on branch basis: %s' %
957
1028
                             self._tree.get_revision_id())
958
 
        builder = branch.get_commit_builder(parent_ids)
 
1029
        revprops = commit.Commit.update_revprops(revprops, branch, authors)
 
1030
        builder = branch.get_commit_builder(parent_ids,
 
1031
                                            timestamp=timestamp,
 
1032
                                            timezone=timezone,
 
1033
                                            committer=committer,
 
1034
                                            revprops=revprops,
 
1035
                                            revision_id=revision_id)
959
1036
        preview = self.get_preview_tree()
960
1037
        list(builder.record_iter_changes(preview, last_rev_id,
961
1038
                                         self.iter_changes()))
1160
1237
            if trans_id not in self._new_contents:
1161
1238
                continue
1162
1239
            new_path = self._limbo_name(trans_id)
1163
 
            osutils.rename(old_path, new_path)
 
1240
            os.rename(old_path, new_path)
1164
1241
            for descendant in self._limbo_descendants(trans_id):
1165
1242
                desc_path = self._limbo_files[descendant]
1166
1243
                desc_path = new_path + desc_path[len(old_path):]
1274
1351
            del self._limbo_children_names[trans_id]
1275
1352
        delete_any(self._limbo_name(trans_id))
1276
1353
 
 
1354
    def new_orphan(self, trans_id, parent_id):
 
1355
        # FIXME: There is no tree config, so we use the branch one (it's weird
 
1356
        # to define it this way as orphaning can only occur in a working tree,
 
1357
        # but that's all we have (for now). It will find the option in
 
1358
        # locations.conf or bazaar.conf though) -- vila 20100916
 
1359
        conf = self._tree.branch.get_config()
 
1360
        conf_var_name = 'bzr.transform.orphan_policy'
 
1361
        orphan_policy = conf.get_user_option(conf_var_name)
 
1362
        default_policy = orphaning_registry.default_key
 
1363
        if orphan_policy is None:
 
1364
            orphan_policy = default_policy
 
1365
        if orphan_policy not in orphaning_registry:
 
1366
            trace.warning('%s (from %s) is not a known policy, defaulting to %s'
 
1367
                          % (orphan_policy, conf_var_name, default_policy))
 
1368
            orphan_policy = default_policy
 
1369
        handle_orphan = orphaning_registry.get(orphan_policy)
 
1370
        handle_orphan(self, trans_id, parent_id)
 
1371
 
 
1372
 
 
1373
class OrphaningError(errors.BzrError):
 
1374
 
 
1375
    # Only bugs could lead to such exception being seen by the user
 
1376
    internal_error = True
 
1377
    _fmt = "Error while orphaning %s in %s directory"
 
1378
 
 
1379
    def __init__(self, orphan, parent):
 
1380
        errors.BzrError.__init__(self)
 
1381
        self.orphan = orphan
 
1382
        self.parent = parent
 
1383
 
 
1384
 
 
1385
class OrphaningForbidden(OrphaningError):
 
1386
 
 
1387
    _fmt = "Policy: %s doesn't allow creating orphans."
 
1388
 
 
1389
    def __init__(self, policy):
 
1390
        errors.BzrError.__init__(self)
 
1391
        self.policy = policy
 
1392
 
 
1393
 
 
1394
def move_orphan(tt, orphan_id, parent_id):
 
1395
    """See TreeTransformBase.new_orphan.
 
1396
 
 
1397
    This creates a new orphan in the `bzr-orphans` dir at the root of the
 
1398
    `TreeTransform`.
 
1399
 
 
1400
    :param tt: The TreeTransform orphaning `trans_id`.
 
1401
 
 
1402
    :param orphan_id: The trans id that should be orphaned.
 
1403
 
 
1404
    :param parent_id: The orphan parent trans id.
 
1405
    """
 
1406
    # Add the orphan dir if it doesn't exist
 
1407
    orphan_dir_basename = 'bzr-orphans'
 
1408
    od_id = tt.trans_id_tree_path(orphan_dir_basename)
 
1409
    if tt.final_kind(od_id) is None:
 
1410
        tt.create_directory(od_id)
 
1411
    parent_path = tt._tree_id_paths[parent_id]
 
1412
    # Find a name that doesn't exist yet in the orphan dir
 
1413
    actual_name = tt.final_name(orphan_id)
 
1414
    new_name = tt._available_backup_name(actual_name, od_id)
 
1415
    tt.adjust_path(new_name, od_id, orphan_id)
 
1416
    trace.warning('%s has been orphaned in %s'
 
1417
                  % (joinpath(parent_path, actual_name), orphan_dir_basename))
 
1418
 
 
1419
 
 
1420
def refuse_orphan(tt, orphan_id, parent_id):
 
1421
    """See TreeTransformBase.new_orphan.
 
1422
 
 
1423
    This refuses to create orphan, letting the caller handle the conflict.
 
1424
    """
 
1425
    raise OrphaningForbidden('never')
 
1426
 
 
1427
 
 
1428
orphaning_registry = registry.Registry()
 
1429
orphaning_registry.register(
 
1430
    'conflict', refuse_orphan,
 
1431
    'Leave orphans in place and create a conflict on the directory.')
 
1432
orphaning_registry.register(
 
1433
    'move', move_orphan,
 
1434
    'Move orphans into the bzr-orphans directory.')
 
1435
orphaning_registry._set_default_key('conflict')
 
1436
 
1277
1437
 
1278
1438
class TreeTransform(DiskTreeTransform):
1279
1439
    """Represent a tree transformation.
1397
1557
    def tree_kind(self, trans_id):
1398
1558
        """Determine the file kind in the working tree.
1399
1559
 
1400
 
        Raises NoSuchFile if the file does not exist
 
1560
        :returns: The file kind or None if the file does not exist
1401
1561
        """
1402
1562
        path = self._tree_id_paths.get(trans_id)
1403
1563
        if path is None:
1404
 
            raise NoSuchFile(None)
 
1564
            return None
1405
1565
        try:
1406
1566
            return file_kind(self._tree.abspath(path))
1407
 
        except OSError, e:
1408
 
            if e.errno != errno.ENOENT:
1409
 
                raise
1410
 
            else:
1411
 
                raise NoSuchFile(path)
 
1567
        except errors.NoSuchFile:
 
1568
            return None
1412
1569
 
1413
1570
    def _set_mode(self, trans_id, mode_id, typefunc):
1414
1571
        """Set the mode of new file contents.
1583
1740
                if file_id is None:
1584
1741
                    continue
1585
1742
                needs_entry = False
1586
 
                try:
1587
 
                    kind = self.final_kind(trans_id)
1588
 
                except NoSuchFile:
 
1743
                kind = self.final_kind(trans_id)
 
1744
                if kind is None:
1589
1745
                    kind = self._tree.stored_kind(file_id)
1590
1746
                parent_trans_id = self.final_parent(trans_id)
1591
1747
                parent_file_id = new_path_file_ids.get(parent_trans_id)
1635
1791
                      or trans_id in self._new_parent):
1636
1792
                    try:
1637
1793
                        mover.rename(full_path, self._limbo_name(trans_id))
1638
 
                    except OSError, e:
 
1794
                    except errors.TransformRenameFailed, e:
1639
1795
                        if e.errno != errno.ENOENT:
1640
1796
                            raise
1641
1797
                    else:
1666
1822
                if trans_id in self._needs_rename:
1667
1823
                    try:
1668
1824
                        mover.rename(self._limbo_name(trans_id), full_path)
1669
 
                    except OSError, e:
 
1825
                    except errors.TransformRenameFailed, e:
1670
1826
                        # We may be renaming a dangling inventory id
1671
1827
                        if e.errno != errno.ENOENT:
1672
1828
                            raise
1703
1859
    def tree_kind(self, trans_id):
1704
1860
        path = self._tree_id_paths.get(trans_id)
1705
1861
        if path is None:
1706
 
            raise NoSuchFile(None)
 
1862
            return None
1707
1863
        file_id = self._tree.path2id(path)
1708
 
        return self._tree.kind(file_id)
 
1864
        try:
 
1865
            return self._tree.kind(file_id)
 
1866
        except errors.NoSuchFile:
 
1867
            return None
1709
1868
 
1710
1869
    def _set_mode(self, trans_id, mode_id, typefunc):
1711
1870
        """Set the mode of new file contents.
1731
1890
            childpath = joinpath(path, child)
1732
1891
            yield self.trans_id_tree_path(childpath)
1733
1892
 
 
1893
    def new_orphan(self, trans_id, parent_id):
 
1894
        raise NotImplementedError(self.new_orphan)
 
1895
 
1734
1896
 
1735
1897
class _PreviewTree(tree.Tree):
1736
1898
    """Partial implementation of Tree to support show_diff_trees"""
1770
1932
        parent_keys = [(file_id, self._file_revision(t, file_id)) for t in
1771
1933
                       self._iter_parent_trees()]
1772
1934
        vf.add_lines((file_id, tree_revision), parent_keys,
1773
 
                     self.get_file(file_id).readlines())
 
1935
                     self.get_file_lines(file_id))
1774
1936
        repo = self._get_repository()
1775
1937
        base_vf = repo.texts
1776
1938
        if base_vf not in vf.fallback_versionedfiles:
1907
2069
            if (specific_file_ids is not None
1908
2070
                and file_id not in specific_file_ids):
1909
2071
                continue
1910
 
            try:
1911
 
                kind = self._transform.final_kind(trans_id)
1912
 
            except NoSuchFile:
 
2072
            kind = self._transform.final_kind(trans_id)
 
2073
            if kind is None:
1913
2074
                kind = self._transform._tree.stored_kind(file_id)
1914
2075
            new_entry = inventory.make_entry(
1915
2076
                kind,
2147
2308
                path_from_root = self._final_paths.get_path(child_id)
2148
2309
                basename = self._transform.final_name(child_id)
2149
2310
                file_id = self._transform.final_file_id(child_id)
2150
 
                try:
2151
 
                    kind = self._transform.final_kind(child_id)
 
2311
                kind  = self._transform.final_kind(child_id)
 
2312
                if kind is not None:
2152
2313
                    versioned_kind = kind
2153
 
                except NoSuchFile:
 
2314
                else:
2154
2315
                    kind = 'unknown'
2155
2316
                    versioned_kind = self._transform._tree.stored_kind(file_id)
2156
2317
                if versioned_kind == 'directory':
2269
2430
    for num, _unused in enumerate(wt.all_file_ids()):
2270
2431
        if num > 0:  # more than just a root
2271
2432
            raise errors.WorkingTreeAlreadyPopulated(base=wt.basedir)
2272
 
    existing_files = set()
2273
 
    for dir, files in wt.walkdirs():
2274
 
        existing_files.update(f[0] for f in files)
2275
2433
    file_trans_id = {}
2276
2434
    top_pb = bzrlib.ui.ui_factory.nested_progress_bar()
2277
2435
    pp = ProgressPhase("Build phase", 2, top_pb)
2301
2459
                precomputed_delta = []
2302
2460
            else:
2303
2461
                precomputed_delta = None
 
2462
            # Check if tree inventory has content. If so, we populate
 
2463
            # existing_files with the directory content. If there are no
 
2464
            # entries we skip populating existing_files as its not used.
 
2465
            # This improves performance and unncessary work on large
 
2466
            # directory trees. (#501307)
 
2467
            if total > 0:
 
2468
                existing_files = set()
 
2469
                for dir, files in wt.walkdirs():
 
2470
                    existing_files.update(f[0] for f in files)
2304
2471
            for num, (tree_path, entry) in \
2305
2472
                enumerate(tree.inventory.iter_entries_by_dir()):
2306
2473
                pb.update("Building tree", num - len(deferred_contents), total)
2426
2593
    for child in tt.iter_tree_children(old_parent):
2427
2594
        tt.adjust_path(tt.final_name(child), new_parent, child)
2428
2595
 
 
2596
 
2429
2597
def _reparent_transform_children(tt, old_parent, new_parent):
2430
2598
    by_parent = tt.by_parent()
2431
2599
    for child in by_parent[old_parent]:
2432
2600
        tt.adjust_path(tt.final_name(child), new_parent, child)
2433
2601
    return by_parent[old_parent]
2434
2602
 
 
2603
 
2435
2604
def _content_match(tree, entry, file_id, kind, target_path):
2436
2605
    if entry.kind != kind:
2437
2606
        return False
2438
2607
    if entry.kind == "directory":
2439
2608
        return True
2440
2609
    if entry.kind == "file":
2441
 
        if tree.get_file(file_id).read() == file(target_path, 'rb').read():
2442
 
            return True
 
2610
        f = file(target_path, 'rb')
 
2611
        try:
 
2612
            if tree.get_file_text(file_id) == f.read():
 
2613
                return True
 
2614
        finally:
 
2615
            f.close()
2443
2616
    elif entry.kind == "symlink":
2444
2617
        if tree.get_symlink_target(file_id) == os.readlink(target_path):
2445
2618
            return True
2497
2670
        raise errors.BadFileKindError(name, kind)
2498
2671
 
2499
2672
 
2500
 
@deprecated_function(deprecated_in((1, 9, 0)))
2501
 
def create_by_entry(tt, entry, tree, trans_id, lines=None, mode_id=None):
2502
 
    """Create new file contents according to an inventory entry.
2503
 
 
2504
 
    DEPRECATED.  Use create_from_tree instead.
2505
 
    """
2506
 
    if entry.kind == "file":
2507
 
        if lines is None:
2508
 
            lines = tree.get_file(entry.file_id).readlines()
2509
 
        tt.create_file(lines, trans_id, mode_id=mode_id)
2510
 
    elif entry.kind == "symlink":
2511
 
        tt.create_symlink(tree.get_symlink_target(entry.file_id), trans_id)
2512
 
    elif entry.kind == "directory":
2513
 
        tt.create_directory(trans_id)
2514
 
 
2515
 
 
2516
2673
def create_from_tree(tt, trans_id, tree, file_id, bytes=None,
2517
2674
    filter_tree_path=None):
2518
2675
    """Create new file contents according to tree contents.
2549
2706
        tt.set_executability(entry.executable, trans_id)
2550
2707
 
2551
2708
 
 
2709
@deprecated_function(deprecated_in((2, 3, 0)))
2552
2710
def get_backup_name(entry, by_parent, parent_trans_id, tt):
2553
2711
    return _get_backup_name(entry.name, by_parent, parent_trans_id, tt)
2554
2712
 
2555
2713
 
 
2714
@deprecated_function(deprecated_in((2, 3, 0)))
2556
2715
def _get_backup_name(name, by_parent, parent_trans_id, tt):
2557
2716
    """Produce a backup-style name that appears to be available"""
2558
2717
    def name_gen():
2679
2838
                        tt.delete_contents(trans_id)
2680
2839
                    elif kind[1] is not None:
2681
2840
                        parent_trans_id = tt.trans_id_file_id(parent[0])
2682
 
                        by_parent = tt.by_parent()
2683
 
                        backup_name = _get_backup_name(name[0], by_parent,
2684
 
                                                       parent_trans_id, tt)
 
2841
                        backup_name = tt._available_backup_name(
 
2842
                            name[0], parent_trans_id)
2685
2843
                        tt.adjust_path(backup_name, parent_trans_id, trans_id)
2686
2844
                        new_trans_id = tt.create_path(name[0], parent_trans_id)
2687
2845
                        if versioned == (True, True):
2810
2968
 
2811
2969
        elif c_type == 'missing parent':
2812
2970
            trans_id = conflict[1]
2813
 
            try:
2814
 
                tt.cancel_deletion(trans_id)
2815
 
                new_conflicts.add(('deleting parent', 'Not deleting',
2816
 
                                   trans_id))
2817
 
            except KeyError:
 
2971
            if trans_id in tt._removed_contents:
 
2972
                cancel_deletion = True
 
2973
                orphans = tt._get_potential_orphans(trans_id)
 
2974
                if orphans:
 
2975
                    cancel_deletion = False
 
2976
                    # All children are orphans
 
2977
                    for o in orphans:
 
2978
                        try:
 
2979
                            tt.new_orphan(o, trans_id)
 
2980
                        except OrphaningError:
 
2981
                            # Something bad happened so we cancel the directory
 
2982
                            # deletion which will leave it in place with a
 
2983
                            # conflict. The user can deal with it from there.
 
2984
                            # Note that this also catch the case where we don't
 
2985
                            # want to create orphans and leave the directory in
 
2986
                            # place.
 
2987
                            cancel_deletion = True
 
2988
                            break
 
2989
                if cancel_deletion:
 
2990
                    # Cancel the directory deletion
 
2991
                    tt.cancel_deletion(trans_id)
 
2992
                    new_conflicts.add(('deleting parent', 'Not deleting',
 
2993
                                       trans_id))
 
2994
            else:
2818
2995
                create = True
2819
2996
                try:
2820
2997
                    tt.final_name(trans_id)
2845
3022
            file_id = tt.inactive_file_id(conflict[1])
2846
3023
            # special-case the other tree root (move its children instead)
2847
3024
            if path_tree and file_id in path_tree:
2848
 
                if path_tree.inventory[file_id].parent_id is None:
 
3025
                if path_tree.path2id('') == file_id:
 
3026
                    # This is the root entry, skip it
2849
3027
                    continue
2850
3028
            tt.version_file(file_id, conflict[1])
2851
3029
            new_conflicts.add((c_type, 'Versioned directory', conflict[1]))
2903
3081
    def rename(self, from_, to):
2904
3082
        """Rename a file from one path to another."""
2905
3083
        try:
2906
 
            osutils.rename(from_, to)
 
3084
            os.rename(from_, to)
2907
3085
        except OSError, e:
2908
3086
            if e.errno in (errno.EEXIST, errno.ENOTEMPTY):
2909
3087
                raise errors.FileExists(to, str(e))
2910
 
            raise
 
3088
            # normal OSError doesn't include filenames so it's hard to see where
 
3089
            # the problem is, see https://bugs.launchpad.net/bzr/+bug/491763
 
3090
            raise errors.TransformRenameFailed(from_, to, str(e), e.errno)
2911
3091
        self.past_renames.append((from_, to))
2912
3092
 
2913
3093
    def pre_delete(self, from_, to):
2923
3103
    def rollback(self):
2924
3104
        """Reverse all renames that have been performed"""
2925
3105
        for from_, to in reversed(self.past_renames):
2926
 
            osutils.rename(to, from_)
 
3106
            try:
 
3107
                os.rename(to, from_)
 
3108
            except OSError, e:
 
3109
                raise errors.TransformRenameFailed(to, from_, str(e), e.errno)
2927
3110
        # after rollback, don't reuse _FileMover
2928
3111
        past_renames = None
2929
3112
        pending_deletions = None