/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/inventory.py

  • Committer: Ian Clatworthy
  • Date: 2009-07-13 06:58:49 UTC
  • mto: (4527.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4529.
  • Revision ID: ian.clatworthy@canonical.com-20090713065849-n7g2rsjyl6dt1mgv
Apply review feedback

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 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
51
51
    )
52
52
from bzrlib.symbol_versioning import deprecated_in, deprecated_method
53
53
from bzrlib.trace import mutter
54
 
from bzrlib.static_tuple import StaticTuple
55
54
 
56
55
 
57
56
class InventoryEntry(object):
261
260
    def versionable_kind(kind):
262
261
        return (kind in ('file', 'directory', 'symlink', 'tree-reference'))
263
262
 
264
 
    def check(self, checker, rev_id, inv):
 
263
    def check(self, checker, rev_id, inv, tree):
265
264
        """Check this inventory entry is intact.
266
265
 
267
266
        This is a template method, override _check for kind specific
273
272
        :param rev_id: Revision id from which this InventoryEntry was loaded.
274
273
             Not necessarily the last-changed revision for this file.
275
274
        :param inv: Inventory from which the entry was loaded.
 
275
        :param tree: RevisionTree for this entry.
276
276
        """
277
277
        if self.parent_id is not None:
278
278
            if not inv.has_id(self.parent_id):
279
279
                raise BzrCheckError('missing parent {%s} in inventory for revision {%s}'
280
280
                        % (self.parent_id, rev_id))
281
 
        checker._add_entry_to_text_key_references(inv, self)
282
 
        self._check(checker, rev_id)
 
281
        self._check(checker, rev_id, tree)
283
282
 
284
 
    def _check(self, checker, rev_id):
 
283
    def _check(self, checker, rev_id, tree):
285
284
        """Check this inventory entry for kind specific errors."""
286
 
        checker._report_items.append(
287
 
            'unknown entry kind %r in revision {%s}' % (self.kind, rev_id))
 
285
        raise BzrCheckError('unknown entry kind %r in revision {%s}' %
 
286
                            (self.kind, rev_id))
288
287
 
289
288
    def copy(self):
290
289
        """Clone this inventory entry."""
403
402
                 'text_id', 'parent_id', 'children', 'executable',
404
403
                 'revision', 'symlink_target', 'reference_revision']
405
404
 
406
 
    def _check(self, checker, rev_id):
 
405
    def _check(self, checker, rev_id, tree):
407
406
        """See InventoryEntry._check"""
408
407
 
409
408
    def __init__(self, file_id):
432
431
                 'text_id', 'parent_id', 'children', 'executable',
433
432
                 'revision', 'symlink_target', 'reference_revision']
434
433
 
435
 
    def _check(self, checker, rev_id):
 
434
    def _check(self, checker, rev_id, tree):
436
435
        """See InventoryEntry._check"""
437
 
        if (self.text_sha1 is not None or self.text_size is not None or
438
 
            self.text_id is not None):
439
 
            checker._report_items.append('directory {%s} has text in revision {%s}'
 
436
        if self.text_sha1 is not None or self.text_size is not None or self.text_id is not None:
 
437
            raise BzrCheckError('directory {%s} has text in revision {%s}'
440
438
                                % (self.file_id, rev_id))
441
 
        # In non rich root repositories we do not expect a file graph for the
442
 
        # root.
443
 
        if self.name == '' and not checker.rich_roots:
444
 
            return
445
 
        # Directories are stored as an empty file, but the file should exist
446
 
        # to provide a per-fileid log. The hash of every directory content is
447
 
        # "da..." below (the sha1sum of '').
448
 
        checker.add_pending_item(rev_id,
449
 
            ('texts', self.file_id, self.revision), 'text',
450
 
             'da39a3ee5e6b4b0d3255bfef95601890afd80709')
451
439
 
452
440
    def copy(self):
453
441
        other = InventoryDirectory(self.file_id, self.name, self.parent_id)
486
474
                 'text_id', 'parent_id', 'children', 'executable',
487
475
                 'revision', 'symlink_target', 'reference_revision']
488
476
 
489
 
    def _check(self, checker, tree_revision_id):
 
477
    def _check(self, checker, tree_revision_id, tree):
490
478
        """See InventoryEntry._check"""
491
 
        # TODO: check size too.
492
 
        checker.add_pending_item(tree_revision_id,
493
 
            ('texts', self.file_id, self.revision), 'text',
494
 
             self.text_sha1)
495
 
        if self.text_size is None:
496
 
            checker._report_items.append(
497
 
                'fileid {%s} in {%s} has None for text_size' % (self.file_id,
498
 
                tree_revision_id))
 
479
        key = (self.file_id, self.revision)
 
480
        if key in checker.checked_texts:
 
481
            prev_sha = checker.checked_texts[key]
 
482
            if prev_sha != self.text_sha1:
 
483
                raise BzrCheckError(
 
484
                    'mismatched sha1 on {%s} in {%s} (%s != %s) %r' %
 
485
                    (self.file_id, tree_revision_id, prev_sha, self.text_sha1,
 
486
                     t))
 
487
            else:
 
488
                checker.repeated_text_cnt += 1
 
489
                return
 
490
 
 
491
        checker.checked_text_cnt += 1
 
492
        # We can't check the length, because Weave doesn't store that
 
493
        # information, and the whole point of looking at the weave's
 
494
        # sha1sum is that we don't have to extract the text.
 
495
        if (self.text_sha1 != tree._repository.texts.get_sha1s([key])[key]):
 
496
            raise BzrCheckError('text {%s} version {%s} wrong sha1' % key)
 
497
        checker.checked_texts[key] = self.text_sha1
499
498
 
500
499
    def copy(self):
501
500
        other = InventoryFile(self.file_id, self.name, self.parent_id)
599
598
                 'text_id', 'parent_id', 'children', 'executable',
600
599
                 'revision', 'symlink_target', 'reference_revision']
601
600
 
602
 
    def _check(self, checker, tree_revision_id):
 
601
    def _check(self, checker, rev_id, tree):
603
602
        """See InventoryEntry._check"""
604
603
        if self.text_sha1 is not None or self.text_size is not None or self.text_id is not None:
605
 
            checker._report_items.append(
606
 
               'symlink {%s} has text in revision {%s}'
607
 
                    % (self.file_id, tree_revision_id))
 
604
            raise BzrCheckError('symlink {%s} has text in revision {%s}'
 
605
                    % (self.file_id, rev_id))
608
606
        if self.symlink_target is None:
609
 
            checker._report_items.append(
610
 
                'symlink {%s} has no target in revision {%s}'
611
 
                    % (self.file_id, tree_revision_id))
612
 
        # Symlinks are stored as ''
613
 
        checker.add_pending_item(tree_revision_id,
614
 
            ('texts', self.file_id, self.revision), 'text',
615
 
             'da39a3ee5e6b4b0d3255bfef95601890afd80709')
 
607
            raise BzrCheckError('symlink {%s} has no target in revision {%s}'
 
608
                    % (self.file_id, rev_id))
616
609
 
617
610
    def copy(self):
618
611
        other = InventoryLink(self.file_id, self.name, self.parent_id)
750
743
        """
751
744
        return self.has_id(file_id)
752
745
 
753
 
    def has_filename(self, filename):
754
 
        return bool(self.path2id(filename))
755
 
 
756
746
    def id2path(self, file_id):
757
747
        """Return as a string the path to file_id.
758
748
 
761
751
        >>> e = i.add(InventoryFile('foo-id', 'foo.c', parent_id='src-id'))
762
752
        >>> print i.id2path('foo-id')
763
753
        src/foo.c
764
 
 
765
 
        :raises NoSuchId: If file_id is not present in the inventory.
766
754
        """
767
755
        # get all names, skipping root
768
756
        return '/'.join(reversed(
959
947
        descend(self.root, u'')
960
948
        return accum
961
949
 
962
 
    def path2id(self, relpath):
 
950
    def path2id(self, name):
963
951
        """Walk down through directories to return entry of last component.
964
952
 
965
 
        :param relpath: may be either a list of path components, or a single
966
 
            string, in which case it is automatically split.
 
953
        names may be either a list of path components, or a single
 
954
        string, in which case it is automatically split.
967
955
 
968
956
        This returns the entry of the last component in the path,
969
957
        which may be either a file or a directory.
970
958
 
971
959
        Returns None IFF the path is not found.
972
960
        """
973
 
        if isinstance(relpath, basestring):
974
 
            names = osutils.splitpath(relpath)
975
 
        else:
976
 
            names = relpath
 
961
        if isinstance(name, basestring):
 
962
            name = osutils.splitpath(name)
 
963
 
 
964
        # mutter("lookup path %r" % name)
977
965
 
978
966
        try:
979
967
            parent = self.root
982
970
            return None
983
971
        if parent is None:
984
972
            return None
985
 
        for f in names:
 
973
        for f in name:
986
974
            try:
987
975
                children = getattr(parent, 'children', None)
988
976
                if children is None:
1101
1089
        See the inventory developers documentation for the theory behind
1102
1090
        inventory deltas.
1103
1091
 
1104
 
        If delta application fails the inventory is left in an indeterminate
1105
 
        state and must not be used.
1106
 
 
1107
1092
        :param delta: A list of changes to apply. After all the changes are
1108
1093
            applied the final inventory must be internally consistent, but it
1109
1094
            is ok to supply changes which, if only half-applied would have an
1144
1129
        # facility.
1145
1130
        list(_check_delta_unique_ids(_check_delta_unique_new_paths(
1146
1131
            _check_delta_unique_old_paths(_check_delta_ids_match_entry(
1147
 
            _check_delta_ids_are_valid(
1148
 
            _check_delta_new_path_entry_both_or_None(
1149
 
            delta)))))))
 
1132
            delta)))))
1150
1133
 
1151
1134
        children = {}
1152
1135
        # Remove all affected items which were in the original inventory,
1155
1138
        # modified children remaining by the time we examine it.
1156
1139
        for old_path, file_id in sorted(((op, f) for op, np, f, e in delta
1157
1140
                                        if op is not None), reverse=True):
 
1141
            if file_id not in self:
 
1142
                # adds come later
 
1143
                continue
1158
1144
            # Preserve unaltered children of file_id for later reinsertion.
1159
1145
            file_id_children = getattr(self[file_id], 'children', {})
1160
1146
            if len(file_id_children):
1161
1147
                children[file_id] = file_id_children
1162
 
            if self.id2path(file_id) != old_path:
1163
 
                raise errors.InconsistentDelta(old_path, file_id,
1164
 
                    "Entry was at wrong other path %r." % self.id2path(file_id))
1165
1148
            # Remove file_id and the unaltered children. If file_id is not
1166
1149
            # being deleted it will be reinserted back later.
1167
1150
            self.remove_recursive_id(file_id)
1170
1153
        # longest, ensuring that items which were modified and whose parents in
1171
1154
        # the resulting inventory were also modified, are inserted after their
1172
1155
        # parents.
1173
 
        for new_path, f, new_entry in sorted((np, f, e) for op, np, f, e in
 
1156
        for new_path, new_entry in sorted((np, e) for op, np, f, e in
1174
1157
                                          delta if np is not None):
1175
1158
            if new_entry.kind == 'directory':
1176
1159
                # Pop the child which to allow detection of children whose
1183
1166
                new_entry = replacement
1184
1167
            try:
1185
1168
                self.add(new_entry)
1186
 
            except errors.DuplicateFileId:
1187
 
                raise errors.InconsistentDelta(new_path, new_entry.file_id,
1188
 
                    "New id is already present in target.")
1189
1169
            except AttributeError:
1190
1170
                raise errors.InconsistentDelta(new_path, new_entry.file_id,
1191
1171
                    "Parent is not a directory.")
1192
 
            if self.id2path(new_entry.file_id) != new_path:
1193
 
                raise errors.InconsistentDelta(new_path, new_entry.file_id,
1194
 
                    "New path is not consistent with parent path.")
1195
1172
        if len(children):
1196
1173
            # Get the parent id that was deleted
1197
1174
            parent_id, children = children.popitem()
1198
1175
            raise errors.InconsistentDelta("<deleted>", parent_id,
1199
1176
                "The file id was deleted but its children were not deleted.")
1200
1177
 
1201
 
    def create_by_apply_delta(self, inventory_delta, new_revision_id,
1202
 
                              propagate_caches=False):
1203
 
        """See CHKInventory.create_by_apply_delta()"""
1204
 
        new_inv = self.copy()
1205
 
        new_inv.apply_delta(inventory_delta)
1206
 
        new_inv.revision_id = new_revision_id
1207
 
        return new_inv
1208
 
 
1209
1178
    def _set_root(self, ie):
1210
1179
        self.root = ie
1211
1180
        self._byid = {self.root.file_id: self.root}
1285
1254
        To add  a file to a branch ready to be committed, use Branch.add,
1286
1255
        which calls this.
1287
1256
 
1288
 
        :return: entry
 
1257
        Returns the new entry object.
1289
1258
        """
1290
1259
        if entry.file_id in self._byid:
1291
1260
            raise errors.DuplicateFileId(entry.file_id,
1292
1261
                                         self._byid[entry.file_id])
 
1262
 
1293
1263
        if entry.parent_id is None:
1294
1264
            self.root = entry
1295
1265
        else:
1383
1353
            yield ie
1384
1354
            file_id = ie.parent_id
1385
1355
 
 
1356
    def has_filename(self, filename):
 
1357
        return bool(self.path2id(filename))
 
1358
 
1386
1359
    def has_id(self, file_id):
1387
1360
        return (file_id in self._byid)
1388
1361
 
1498
1471
        self._fileid_to_entry_cache = {}
1499
1472
        self._path_to_fileid_cache = {}
1500
1473
        self._search_key_name = search_key_name
1501
 
        self.root_id = None
1502
1474
 
1503
1475
    def __eq__(self, other):
1504
1476
        """Compare two sets by comparing their contents."""
1555
1527
        else:
1556
1528
            raise ValueError("unknown kind %r" % entry.kind)
1557
1529
 
1558
 
    def _expand_fileids_to_parents_and_children(self, file_ids):
1559
 
        """Give a more wholistic view starting with the given file_ids.
1560
 
 
1561
 
        For any file_id which maps to a directory, we will include all children
1562
 
        of that directory. We will also include all directories which are
1563
 
        parents of the given file_ids, but we will not include their children.
1564
 
 
1565
 
        eg:
1566
 
          /     # TREE_ROOT
1567
 
          foo/  # foo-id
1568
 
            baz # baz-id
1569
 
            frob/ # frob-id
1570
 
              fringle # fringle-id
1571
 
          bar/  # bar-id
1572
 
            bing # bing-id
1573
 
 
1574
 
        if given [foo-id] we will include
1575
 
            TREE_ROOT as interesting parents
1576
 
        and 
1577
 
            foo-id, baz-id, frob-id, fringle-id
1578
 
        As interesting ids.
1579
 
        """
1580
 
        interesting = set()
1581
 
        # TODO: Pre-pass over the list of fileids to see if anything is already
1582
 
        #       deserialized in self._fileid_to_entry_cache
1583
 
 
1584
 
        directories_to_expand = set()
1585
 
        children_of_parent_id = {}
1586
 
        # It is okay if some of the fileids are missing
1587
 
        for entry in self._getitems(file_ids):
1588
 
            if entry.kind == 'directory':
1589
 
                directories_to_expand.add(entry.file_id)
1590
 
            interesting.add(entry.parent_id)
1591
 
            children_of_parent_id.setdefault(entry.parent_id, []
1592
 
                                             ).append(entry.file_id)
1593
 
 
1594
 
        # Now, interesting has all of the direct parents, but not the
1595
 
        # parents of those parents. It also may have some duplicates with
1596
 
        # specific_fileids
1597
 
        remaining_parents = interesting.difference(file_ids)
1598
 
        # When we hit the TREE_ROOT, we'll get an interesting parent of None,
1599
 
        # but we don't actually want to recurse into that
1600
 
        interesting.add(None) # this will auto-filter it in the loop
1601
 
        remaining_parents.discard(None) 
1602
 
        while remaining_parents:
1603
 
            next_parents = set()
1604
 
            for entry in self._getitems(remaining_parents):
1605
 
                next_parents.add(entry.parent_id)
1606
 
                children_of_parent_id.setdefault(entry.parent_id, []
1607
 
                                                 ).append(entry.file_id)
1608
 
            # Remove any search tips we've already processed
1609
 
            remaining_parents = next_parents.difference(interesting)
1610
 
            interesting.update(remaining_parents)
1611
 
            # We should probably also .difference(directories_to_expand)
1612
 
        interesting.update(file_ids)
1613
 
        interesting.discard(None)
1614
 
        while directories_to_expand:
1615
 
            # Expand directories by looking in the
1616
 
            # parent_id_basename_to_file_id map
1617
 
            keys = [StaticTuple(f,).intern() for f in directories_to_expand]
1618
 
            directories_to_expand = set()
1619
 
            items = self.parent_id_basename_to_file_id.iteritems(keys)
1620
 
            next_file_ids = set([item[1] for item in items])
1621
 
            next_file_ids = next_file_ids.difference(interesting)
1622
 
            interesting.update(next_file_ids)
1623
 
            for entry in self._getitems(next_file_ids):
1624
 
                if entry.kind == 'directory':
1625
 
                    directories_to_expand.add(entry.file_id)
1626
 
                children_of_parent_id.setdefault(entry.parent_id, []
1627
 
                                                 ).append(entry.file_id)
1628
 
        return interesting, children_of_parent_id
1629
 
 
1630
 
    def filter(self, specific_fileids):
1631
 
        """Get an inventory view filtered against a set of file-ids.
1632
 
 
1633
 
        Children of directories and parents are included.
1634
 
 
1635
 
        The result may or may not reference the underlying inventory
1636
 
        so it should be treated as immutable.
1637
 
        """
1638
 
        (interesting,
1639
 
         parent_to_children) = self._expand_fileids_to_parents_and_children(
1640
 
                                specific_fileids)
1641
 
        # There is some overlap here, but we assume that all interesting items
1642
 
        # are in the _fileid_to_entry_cache because we had to read them to
1643
 
        # determine if they were a dir we wanted to recurse, or just a file
1644
 
        # This should give us all the entries we'll want to add, so start
1645
 
        # adding
1646
 
        other = Inventory(self.root_id)
1647
 
        other.root.revision = self.root.revision
1648
 
        other.revision_id = self.revision_id
1649
 
        if not interesting or not parent_to_children:
1650
 
            # empty filter, or filtering entrys that don't exist
1651
 
            # (if even 1 existed, then we would have populated
1652
 
            # parent_to_children with at least the tree root.)
1653
 
            return other
1654
 
        cache = self._fileid_to_entry_cache
1655
 
        remaining_children = collections.deque(parent_to_children[self.root_id])
1656
 
        while remaining_children:
1657
 
            file_id = remaining_children.popleft()
1658
 
            ie = cache[file_id]
1659
 
            if ie.kind == 'directory':
1660
 
                ie = ie.copy() # We create a copy to depopulate the .children attribute
1661
 
            # TODO: depending on the uses of 'other' we should probably alwyas
1662
 
            #       '.copy()' to prevent someone from mutating other and
1663
 
            #       invaliding our internal cache
1664
 
            other.add(ie)
1665
 
            if file_id in parent_to_children:
1666
 
                remaining_children.extend(parent_to_children[file_id])
1667
 
        return other
1668
 
 
1669
1530
    @staticmethod
1670
1531
    def _bytes_to_utf8name_key(bytes):
1671
1532
        """Get the file_id, revision_id key out of bytes."""
1673
1534
        # to filter out empty names because of non rich-root...
1674
1535
        sections = bytes.split('\n')
1675
1536
        kind, file_id = sections[0].split(': ')
1676
 
        return (sections[2], intern(file_id), intern(sections[3]))
 
1537
        return (sections[2], file_id, sections[3])
1677
1538
 
1678
1539
    def _bytes_to_entry(self, bytes):
1679
1540
        """Deserialise a serialised entry."""
1701
1562
            result.reference_revision = sections[4]
1702
1563
        else:
1703
1564
            raise ValueError("Not a serialised entry %r" % bytes)
1704
 
        result.file_id = intern(result.file_id)
1705
 
        result.revision = intern(sections[3])
 
1565
        result.revision = sections[3]
1706
1566
        if result.parent_id == '':
1707
1567
            result.parent_id = None
1708
1568
        self._fileid_to_entry_cache[result.file_id] = result
1730
1590
          copied to and updated for the result.
1731
1591
        :return: The new CHKInventory.
1732
1592
        """
1733
 
        split = osutils.split
1734
1593
        result = CHKInventory(self._search_key_name)
1735
1594
        if propagate_caches:
1736
1595
            # Just propagate the path-to-fileid cache for now
1745
1604
            search_key_func=search_key_func)
1746
1605
        result.id_to_entry._ensure_root()
1747
1606
        result.id_to_entry._root_node.set_maximum_size(maximum_size)
1748
 
        # Change to apply to the parent_id_basename delta. The dict maps
1749
 
        # (parent_id, basename) -> (old_key, new_value). We use a dict because
1750
 
        # when a path has its id replaced (e.g. the root is changed, or someone
1751
 
        # does bzr mv a b, bzr mv c a, we should output a single change to this
1752
 
        # map rather than two.
1753
 
        parent_id_basename_delta = {}
 
1607
        parent_id_basename_delta = []
1754
1608
        if self.parent_id_basename_to_file_id is not None:
1755
1609
            result.parent_id_basename_to_file_id = chk_map.CHKMap(
1756
1610
                self.parent_id_basename_to_file_id._store,
1776
1630
        inventory_delta = _check_delta_unique_new_paths(inventory_delta)
1777
1631
        # Check for entries that don't match the fileid
1778
1632
        inventory_delta = _check_delta_ids_match_entry(inventory_delta)
1779
 
        # Check for nonsense fileids
1780
 
        inventory_delta = _check_delta_ids_are_valid(inventory_delta)
1781
 
        # Check for new_path <-> entry consistency
1782
 
        inventory_delta = _check_delta_new_path_entry_both_or_None(
1783
 
            inventory_delta)
1784
 
        # All changed entries need to have their parents be directories and be
1785
 
        # at the right path. This set contains (path, id) tuples.
 
1633
        # All changed entries need to have their parents be directories.
1786
1634
        parents = set()
1787
 
        # When we delete an item, all the children of it must be either deleted
1788
 
        # or altered in their own right. As we batch process the change via
1789
 
        # CHKMap.apply_delta, we build a set of things to use to validate the
1790
 
        # delta.
1791
 
        deletes = set()
1792
 
        altered = set()
1793
1635
        for old_path, new_path, file_id, entry in inventory_delta:
1794
1636
            # file id changes
1795
1637
            if new_path == '':
1804
1646
                        del result._path_to_fileid_cache[old_path]
1805
1647
                    except KeyError:
1806
1648
                        pass
1807
 
                deletes.add(file_id)
1808
1649
            else:
1809
 
                new_key = StaticTuple(file_id,)
 
1650
                new_key = (file_id,)
1810
1651
                new_value = result._entry_to_bytes(entry)
1811
1652
                # Update caches. It's worth doing this whether
1812
1653
                # we're propagating the old caches or not.
1813
1654
                result._path_to_fileid_cache[new_path] = file_id
1814
 
                parents.add((split(new_path)[0], entry.parent_id))
 
1655
                parents.add(entry.parent_id)
1815
1656
            if old_path is None:
1816
1657
                old_key = None
1817
1658
            else:
1818
 
                old_key = StaticTuple(file_id,)
1819
 
                if self.id2path(file_id) != old_path:
1820
 
                    raise errors.InconsistentDelta(old_path, file_id,
1821
 
                        "Entry was at wrong other path %r." %
1822
 
                        self.id2path(file_id))
1823
 
                altered.add(file_id)
1824
 
            id_to_entry_delta.append(StaticTuple(old_key, new_key, new_value))
 
1659
                old_key = (file_id,)
 
1660
            id_to_entry_delta.append((old_key, new_key, new_value))
1825
1661
            if result.parent_id_basename_to_file_id is not None:
1826
1662
                # parent_id, basename changes
1827
1663
                if old_path is None:
1835
1671
                else:
1836
1672
                    new_key = self._parent_id_basename_key(entry)
1837
1673
                    new_value = file_id
1838
 
                # If the two keys are the same, the value will be unchanged
1839
 
                # as its always the file id for this entry.
1840
1674
                if old_key != new_key:
1841
 
                    # Transform a change into explicit delete/add preserving
1842
 
                    # a possible match on the key from a different file id.
1843
 
                    if old_key is not None:
1844
 
                        parent_id_basename_delta.setdefault(
1845
 
                            old_key, [None, None])[0] = old_key
1846
 
                    if new_key is not None:
1847
 
                        parent_id_basename_delta.setdefault(
1848
 
                            new_key, [None, None])[1] = new_value
1849
 
        # validate that deletes are complete.
1850
 
        for file_id in deletes:
1851
 
            entry = self[file_id]
1852
 
            if entry.kind != 'directory':
1853
 
                continue
1854
 
            # This loop could potentially be better by using the id_basename
1855
 
            # map to just get the child file ids.
1856
 
            for child in entry.children.values():
1857
 
                if child.file_id not in altered:
1858
 
                    raise errors.InconsistentDelta(self.id2path(child.file_id),
1859
 
                        child.file_id, "Child not deleted or reparented when "
1860
 
                        "parent deleted.")
 
1675
                    # If the two keys are the same, the value will be unchanged
 
1676
                    # as its always the file id.
 
1677
                    parent_id_basename_delta.append((old_key, new_key, new_value))
1861
1678
        result.id_to_entry.apply_delta(id_to_entry_delta)
1862
1679
        if parent_id_basename_delta:
1863
 
            # Transform the parent_id_basename delta data into a linear delta
1864
 
            # with only one record for a given key. Optimally this would allow
1865
 
            # re-keying, but its simpler to just output that as a delete+add
1866
 
            # to spend less time calculating the delta.
1867
 
            delta_list = []
1868
 
            for key, (old_key, value) in parent_id_basename_delta.iteritems():
1869
 
                if value is not None:
1870
 
                    delta_list.append((old_key, key, value))
1871
 
                else:
1872
 
                    delta_list.append((old_key, None, None))
1873
 
            result.parent_id_basename_to_file_id.apply_delta(delta_list)
1874
 
        parents.discard(('', None))
1875
 
        for parent_path, parent in parents:
 
1680
            result.parent_id_basename_to_file_id.apply_delta(parent_id_basename_delta)
 
1681
        parents.discard(None)
 
1682
        for parent in parents:
1876
1683
            try:
1877
1684
                if result[parent].kind != 'directory':
1878
1685
                    raise errors.InconsistentDelta(result.id2path(parent), parent,
1880
1687
            except errors.NoSuchId:
1881
1688
                raise errors.InconsistentDelta("<unknown>", parent,
1882
1689
                    "Parent is not present in resulting inventory.")
1883
 
            if result.path2id(parent_path) != parent:
1884
 
                raise errors.InconsistentDelta(parent_path, parent,
1885
 
                    "Parent has wrong path %r." % result.path2id(parent_path))
1886
1690
        return result
1887
1691
 
1888
1692
    @classmethod
1914
1718
                raise errors.BzrError('Duplicate key in inventory: %r\n%r'
1915
1719
                                      % (key, bytes))
1916
1720
            info[key] = value
1917
 
        revision_id = intern(info['revision_id'])
1918
 
        root_id = intern(info['root_id'])
1919
 
        search_key_name = intern(info.get('search_key_name', 'plain'))
1920
 
        parent_id_basename_to_file_id = intern(info.get(
1921
 
            'parent_id_basename_to_file_id', None))
1922
 
        if not parent_id_basename_to_file_id.startswith('sha1:'):
1923
 
            raise ValueError('parent_id_basename_to_file_id should be a sha1'
1924
 
                             ' key not %r' % (parent_id_basename_to_file_id,))
 
1721
        revision_id = info['revision_id']
 
1722
        root_id = info['root_id']
 
1723
        search_key_name = info.get('search_key_name', 'plain')
 
1724
        parent_id_basename_to_file_id = info.get(
 
1725
            'parent_id_basename_to_file_id', None)
1925
1726
        id_to_entry = info['id_to_entry']
1926
 
        if not id_to_entry.startswith('sha1:'):
1927
 
            raise ValueError('id_to_entry should be a sha1'
1928
 
                             ' key not %r' % (id_to_entry,))
1929
1727
 
1930
1728
        result = CHKInventory(search_key_name)
1931
1729
        result.revision_id = revision_id
1934
1732
                            result._search_key_name)
1935
1733
        if parent_id_basename_to_file_id is not None:
1936
1734
            result.parent_id_basename_to_file_id = chk_map.CHKMap(
1937
 
                chk_store, StaticTuple(parent_id_basename_to_file_id,),
 
1735
                chk_store, (parent_id_basename_to_file_id,),
1938
1736
                search_key_func=search_key_func)
1939
1737
        else:
1940
1738
            result.parent_id_basename_to_file_id = None
1941
1739
 
1942
 
        result.id_to_entry = chk_map.CHKMap(chk_store,
1943
 
                                            StaticTuple(id_to_entry,),
 
1740
        result.id_to_entry = chk_map.CHKMap(chk_store, (id_to_entry,),
1944
1741
                                            search_key_func=search_key_func)
1945
1742
        if (result.revision_id,) != expected_revision_id:
1946
1743
            raise ValueError("Mismatched revision id and expected: %r, %r" %
1968
1765
        id_to_entry_dict = {}
1969
1766
        parent_id_basename_dict = {}
1970
1767
        for path, entry in inventory.iter_entries():
1971
 
            key = StaticTuple(entry.file_id,).intern()
1972
 
            id_to_entry_dict[key] = entry_to_bytes(entry)
 
1768
            id_to_entry_dict[(entry.file_id,)] = entry_to_bytes(entry)
1973
1769
            p_id_key = parent_id_basename_key(entry)
1974
1770
            parent_id_basename_dict[p_id_key] = entry.file_id
1975
1771
 
1998
1794
            parent_id = entry.parent_id
1999
1795
        else:
2000
1796
            parent_id = ''
2001
 
        return StaticTuple(parent_id, entry.name.encode('utf8')).intern()
 
1797
        return parent_id, entry.name.encode('utf8')
2002
1798
 
2003
1799
    def __getitem__(self, file_id):
2004
1800
        """map a single file_id -> InventoryEntry."""
2009
1805
            return result
2010
1806
        try:
2011
1807
            return self._bytes_to_entry(
2012
 
                self.id_to_entry.iteritems([StaticTuple(file_id,)]).next()[1])
 
1808
                self.id_to_entry.iteritems([(file_id,)]).next()[1])
2013
1809
        except StopIteration:
2014
1810
            # really we're passing an inventory, not a tree...
2015
1811
            raise errors.NoSuchId(self, file_id)
2016
1812
 
2017
 
    def _getitems(self, file_ids):
2018
 
        """Similar to __getitem__, but lets you query for multiple.
2019
 
        
2020
 
        The returned order is undefined. And currently if an item doesn't
2021
 
        exist, it isn't included in the output.
2022
 
        """
2023
 
        result = []
2024
 
        remaining = []
2025
 
        for file_id in file_ids:
2026
 
            entry = self._fileid_to_entry_cache.get(file_id, None)
2027
 
            if entry is None:
2028
 
                remaining.append(file_id)
2029
 
            else:
2030
 
                result.append(entry)
2031
 
        file_keys = [StaticTuple(f,).intern() for f in remaining]
2032
 
        for file_key, value in self.id_to_entry.iteritems(file_keys):
2033
 
            entry = self._bytes_to_entry(value)
2034
 
            result.append(entry)
2035
 
            self._fileid_to_entry_cache[entry.file_id] = entry
2036
 
        return result
2037
 
 
2038
1813
    def has_id(self, file_id):
2039
1814
        # Perhaps have an explicit 'contains' method on CHKMap ?
2040
1815
        if self._fileid_to_entry_cache.get(file_id, None) is not None:
2041
1816
            return True
2042
 
        return len(list(
2043
 
            self.id_to_entry.iteritems([StaticTuple(file_id,)]))) == 1
 
1817
        return len(list(self.id_to_entry.iteritems([(file_id,)]))) == 1
2044
1818
 
2045
1819
    def is_root(self, file_id):
2046
1820
        return file_id == self.root_id
2175
1949
            delta.append((old_path, new_path, file_id, entry))
2176
1950
        return delta
2177
1951
 
2178
 
    def path2id(self, relpath):
 
1952
    def path2id(self, name):
2179
1953
        """See CommonInventory.path2id()."""
2180
 
        # TODO: perhaps support negative hits?
2181
 
        result = self._path_to_fileid_cache.get(relpath, None)
2182
 
        if result is not None:
2183
 
            return result
2184
 
        if isinstance(relpath, basestring):
2185
 
            names = osutils.splitpath(relpath)
2186
 
        else:
2187
 
            names = relpath
2188
 
        current_id = self.root_id
2189
 
        if current_id is None:
2190
 
            return None
2191
 
        parent_id_index = self.parent_id_basename_to_file_id
2192
 
        cur_path = None
2193
 
        for basename in names:
2194
 
            if cur_path is None:
2195
 
                cur_path = basename
2196
 
            else:
2197
 
                cur_path = cur_path + '/' + basename
2198
 
            basename_utf8 = basename.encode('utf8')
2199
 
            file_id = self._path_to_fileid_cache.get(cur_path, None)
2200
 
            if file_id is None:
2201
 
                key_filter = [StaticTuple(current_id, basename_utf8)]
2202
 
                items = parent_id_index.iteritems(key_filter)
2203
 
                for (parent_id, name_utf8), file_id in items:
2204
 
                    if parent_id != current_id or name_utf8 != basename_utf8:
2205
 
                        raise errors.BzrError("corrupt inventory lookup! "
2206
 
                            "%r %r %r %r" % (parent_id, current_id, name_utf8,
2207
 
                            basename_utf8))
2208
 
                if file_id is None:
2209
 
                    return None
2210
 
                else:
2211
 
                    self._path_to_fileid_cache[cur_path] = file_id
2212
 
            current_id = file_id
2213
 
        return current_id
 
1954
        result = self._path_to_fileid_cache.get(name, None)
 
1955
        if result is None:
 
1956
            result = CommonInventory.path2id(self, name)
 
1957
            self._path_to_fileid_cache[name] = result
 
1958
        return result
2214
1959
 
2215
1960
    def to_lines(self):
2216
1961
        """Serialise the inventory to lines."""
2220
1965
            lines.append('search_key_name: %s\n' % (self._search_key_name,))
2221
1966
            lines.append("root_id: %s\n" % self.root_id)
2222
1967
            lines.append('parent_id_basename_to_file_id: %s\n' %
2223
 
                (self.parent_id_basename_to_file_id.key()[0],))
 
1968
                self.parent_id_basename_to_file_id.key())
2224
1969
            lines.append("revision_id: %s\n" % self.revision_id)
2225
 
            lines.append("id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
 
1970
            lines.append("id_to_entry: %s\n" % self.id_to_entry.key())
2226
1971
        else:
2227
1972
            lines.append("revision_id: %s\n" % self.revision_id)
2228
1973
            lines.append("root_id: %s\n" % self.root_id)
2229
1974
            if self.parent_id_basename_to_file_id is not None:
2230
1975
                lines.append('parent_id_basename_to_file_id: %s\n' %
2231
 
                    (self.parent_id_basename_to_file_id.key()[0],))
2232
 
            lines.append("id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
 
1976
                    self.parent_id_basename_to_file_id.key())
 
1977
            lines.append("id_to_entry: %s\n" % self.id_to_entry.key())
2233
1978
        return lines
2234
1979
 
2235
1980
    @property
2276
2021
        parent_id_index = self._chk_inventory.parent_id_basename_to_file_id
2277
2022
        child_keys = set()
2278
2023
        for (parent_id, name_utf8), file_id in parent_id_index.iteritems(
2279
 
            key_filter=[StaticTuple(self.file_id,)]):
2280
 
            child_keys.add(StaticTuple(file_id,))
 
2024
            key_filter=[(self.file_id,)]):
 
2025
            child_keys.add((file_id,))
2281
2026
        cached = set()
2282
2027
        for file_id_key in child_keys:
2283
2028
            entry = self._chk_inventory._fileid_to_entry_cache.get(
2316
2061
    try:
2317
2062
        factory = entry_factory[kind]
2318
2063
    except KeyError:
2319
 
        raise errors.BadFileKindError(name, kind)
 
2064
        raise BzrError("unknown kind %r" % kind)
2320
2065
    return factory(file_id, name, parent_id)
2321
2066
 
2322
2067
 
2399
2144
        yield item
2400
2145
 
2401
2146
 
2402
 
def _check_delta_ids_are_valid(delta):
2403
 
    """Decorate a delta and check that the ids in it are valid.
2404
 
 
2405
 
    :return: A generator over delta.
2406
 
    """
2407
 
    for item in delta:
2408
 
        entry = item[3]
2409
 
        if item[2] is None:
2410
 
            raise errors.InconsistentDelta(item[0] or item[1], item[2],
2411
 
                "entry with file_id None %r" % entry)
2412
 
        if type(item[2]) != str:
2413
 
            raise errors.InconsistentDelta(item[0] or item[1], item[2],
2414
 
                "entry with non bytes file_id %r" % entry)
2415
 
        yield item
2416
 
 
2417
 
 
2418
2147
def _check_delta_ids_match_entry(delta):
2419
2148
    """Decorate a delta and check that the ids in it match the entry.file_id.
2420
2149
 
2427
2156
                raise errors.InconsistentDelta(item[0] or item[1], item[2],
2428
2157
                    "mismatched id with %r" % entry)
2429
2158
        yield item
2430
 
 
2431
 
 
2432
 
def _check_delta_new_path_entry_both_or_None(delta):
2433
 
    """Decorate a delta and check that the new_path and entry are paired.
2434
 
 
2435
 
    :return: A generator over delta.
2436
 
    """
2437
 
    for item in delta:
2438
 
        new_path = item[1]
2439
 
        entry = item[3]
2440
 
        if new_path is None and entry is not None:
2441
 
            raise errors.InconsistentDelta(item[0], item[1],
2442
 
                "Entry with no new_path")
2443
 
        if new_path is not None and entry is None:
2444
 
            raise errors.InconsistentDelta(new_path, item[1],
2445
 
                "new_path with no entry")
2446
 
        yield item