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

  • Committer: Martin
  • Date: 2017-06-11 14:25:56 UTC
  • mto: This revision was merged to the branch mainline in revision 6688.
  • Revision ID: gzlist@googlemail.com-20170611142556-zbxdnifdfsk9t0c8
Update rio tests to pass on Python 3

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-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
23
23
# But those depend on its position within a particular inventory, and
24
24
# it would be nice not to need to hold the backpointer here.
25
25
 
 
26
from __future__ import absolute_import
 
27
 
26
28
# This should really be an id randomly assigned when the tree is
27
29
# created, but it's not for now.
28
 
ROOT_ID = "TREE_ROOT"
 
30
ROOT_ID = b"TREE_ROOT"
29
31
 
30
 
from bzrlib.lazy_import import lazy_import
 
32
from .lazy_import import lazy_import
31
33
lazy_import(globals(), """
32
34
import collections
33
35
import copy
34
 
import os
35
36
import re
36
37
import tarfile
37
38
 
38
 
import bzrlib
39
 
from bzrlib import (
 
39
from breezy import (
40
40
    chk_map,
41
41
    errors,
42
42
    generate_ids,
43
43
    osutils,
44
 
    symbol_versioning,
45
44
    )
46
45
""")
47
46
 
48
 
from bzrlib.errors import (
49
 
    BzrCheckError,
50
 
    BzrError,
51
 
    )
52
 
from bzrlib.symbol_versioning import deprecated_in, deprecated_method
53
 
from bzrlib.trace import mutter
54
 
from bzrlib.static_tuple import StaticTuple
 
47
from . import (
 
48
    lazy_regex,
 
49
    trace,
 
50
    )
 
51
from .sixish import (
 
52
    bytesintern,
 
53
    PY3,
 
54
    viewitems,
 
55
    viewvalues,
 
56
    )
 
57
from .static_tuple import StaticTuple
55
58
 
56
59
 
57
60
class InventoryEntry(object):
104
107
    InventoryDirectory('2325', 'wibble', parent_id='123', revision=None)
105
108
    >>> i.path2id('src/wibble')
106
109
    '2325'
107
 
    >>> '2325' in i
108
 
    True
109
110
    >>> i.add(InventoryFile('2326', 'wibble.c', '2325'))
110
111
    InventoryFile('2326', 'wibble.c', parent_id='2325', sha1=None, len=None, revision=None)
111
112
    >>> i['2326']
131
132
    RENAMED = 'renamed'
132
133
    MODIFIED_AND_RENAMED = 'modified and renamed'
133
134
 
134
 
    __slots__ = []
 
135
    __slots__ = ['file_id', 'revision', 'parent_id', 'name']
 
136
 
 
137
    # Attributes that all InventoryEntry instances are expected to have, but
 
138
    # that don't vary for all kinds of entry.  (e.g. symlink_target is only
 
139
    # relevant to InventoryLink, so there's no reason to make every
 
140
    # InventoryFile instance allocate space to hold a value for it.)
 
141
    # Attributes that only vary for files: executable, text_sha1, text_size,
 
142
    # text_id
 
143
    executable = False
 
144
    text_sha1 = None
 
145
    text_size = None
 
146
    text_id = None
 
147
    # Attributes that only vary for symlinks: symlink_target
 
148
    symlink_target = None
 
149
    # Attributes that only vary for tree-references: reference_revision
 
150
    reference_revision = None
 
151
 
135
152
 
136
153
    def detect_changes(self, old_entry):
137
154
        """Return a (text_modified, meta_modified) from this to old_entry.
158
175
        candidates = {}
159
176
        # identify candidate head revision ids.
160
177
        for inv in previous_inventories:
161
 
            if self.file_id in inv:
 
178
            if inv.has_id(self.file_id):
162
179
                ie = inv[self.file_id]
163
180
                if ie.revision in candidates:
164
181
                    # same revision value in two different inventories:
176
193
                    candidates[ie.revision] = ie
177
194
        return candidates
178
195
 
179
 
    @deprecated_method(deprecated_in((1, 6, 0)))
180
 
    def get_tar_item(self, root, dp, now, tree):
181
 
        """Get a tarfile item and a file stream for its content."""
182
 
        item = tarfile.TarInfo(osutils.pathjoin(root, dp).encode('utf8'))
183
 
        # TODO: would be cool to actually set it to the timestamp of the
184
 
        # revision it was last changed
185
 
        item.mtime = now
186
 
        fileobj = self._put_in_tar(item, tree)
187
 
        return item, fileobj
188
 
 
189
196
    def has_text(self):
190
197
        """Return true if the object this entry represents has textual data.
191
198
 
197
204
        """
198
205
        return False
199
206
 
200
 
    def __init__(self, file_id, name, parent_id, text_id=None):
 
207
    def __init__(self, file_id, name, parent_id):
201
208
        """Create an InventoryEntry
202
209
 
203
210
        The filename must be a single component, relative to the
212
219
        Traceback (most recent call last):
213
220
        InvalidEntryName: Invalid entry name: src/hello.c
214
221
        """
215
 
        if '/' in name or '\\' in name:
 
222
        if u'/' in name or u'\\' in name:
216
223
            raise errors.InvalidEntryName(name=name)
217
 
        self.executable = False
 
224
        self.file_id = file_id
218
225
        self.revision = None
219
 
        self.text_sha1 = None
220
 
        self.text_size = None
221
 
        self.file_id = file_id
222
226
        self.name = name
223
 
        self.text_id = text_id
224
227
        self.parent_id = parent_id
225
 
        self.symlink_target = None
226
 
        self.reference_revision = None
227
228
 
228
229
    def kind_character(self):
229
230
        """Return a short kind indicator useful for appending to names."""
230
 
        raise BzrError('unknown kind %r' % self.kind)
 
231
        raise errors.BzrError('unknown kind %r' % self.kind)
231
232
 
232
233
    known_kinds = ('file', 'directory', 'symlink')
233
234
 
234
 
    def _put_in_tar(self, item, tree):
235
 
        """populate item for stashing in a tar, and return the content stream.
236
 
 
237
 
        If no content is available, return None.
238
 
        """
239
 
        raise BzrError("don't know how to export {%s} of kind %r" %
240
 
                       (self.file_id, self.kind))
241
 
 
242
 
    @deprecated_method(deprecated_in((1, 6, 0)))
243
 
    def put_on_disk(self, dest, dp, tree):
244
 
        """Create a representation of self on disk in the prefix dest.
245
 
 
246
 
        This is a template method - implement _put_on_disk in subclasses.
247
 
        """
248
 
        fullpath = osutils.pathjoin(dest, dp)
249
 
        self._put_on_disk(fullpath, tree)
250
 
        # mutter("  export {%s} kind %s to %s", self.file_id,
251
 
        #         self.kind, fullpath)
252
 
 
253
 
    def _put_on_disk(self, fullpath, tree):
254
 
        """Put this entry onto disk at fullpath, from tree tree."""
255
 
        raise BzrError("don't know how to export {%s} of kind %r" % (self.file_id, self.kind))
256
 
 
257
 
    def sorted_children(self):
258
 
        return sorted(self.children.items())
259
 
 
260
235
    @staticmethod
261
236
    def versionable_kind(kind):
262
237
        return (kind in ('file', 'directory', 'symlink', 'tree-reference'))
276
251
        """
277
252
        if self.parent_id is not None:
278
253
            if not inv.has_id(self.parent_id):
279
 
                raise BzrCheckError('missing parent {%s} in inventory for revision {%s}'
280
 
                        % (self.parent_id, rev_id))
 
254
                raise errors.BzrCheckError(
 
255
                    'missing parent {%s} in inventory for revision {%s}' % (
 
256
                        self.parent_id, rev_id))
281
257
        checker._add_entry_to_text_key_references(inv, self)
282
258
        self._check(checker, rev_id)
283
259
 
397
373
        pass
398
374
 
399
375
 
400
 
class RootEntry(InventoryEntry):
401
 
 
402
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
403
 
                 'text_id', 'parent_id', 'children', 'executable',
404
 
                 'revision', 'symlink_target', 'reference_revision']
405
 
 
406
 
    def _check(self, checker, rev_id):
407
 
        """See InventoryEntry._check"""
408
 
 
409
 
    def __init__(self, file_id):
410
 
        self.file_id = file_id
411
 
        self.children = {}
412
 
        self.kind = 'directory'
413
 
        self.parent_id = None
414
 
        self.name = u''
415
 
        self.revision = None
416
 
        symbol_versioning.warn('RootEntry is deprecated as of bzr 0.10.'
417
 
                               '  Please use InventoryDirectory instead.',
418
 
                               DeprecationWarning, stacklevel=2)
419
 
 
420
 
    def __eq__(self, other):
421
 
        if not isinstance(other, RootEntry):
422
 
            return NotImplemented
423
 
 
424
 
        return (self.file_id == other.file_id) \
425
 
               and (self.children == other.children)
426
 
 
427
 
 
428
376
class InventoryDirectory(InventoryEntry):
429
377
    """A directory in an inventory."""
430
378
 
431
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
432
 
                 'text_id', 'parent_id', 'children', 'executable',
433
 
                 'revision', 'symlink_target', 'reference_revision']
 
379
    __slots__ = ['children']
 
380
 
 
381
    kind = 'directory'
434
382
 
435
383
    def _check(self, checker, rev_id):
436
384
        """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}'
440
 
                                % (self.file_id, rev_id))
441
385
        # In non rich root repositories we do not expect a file graph for the
442
386
        # root.
443
387
        if self.name == '' and not checker.rich_roots:
446
390
        # to provide a per-fileid log. The hash of every directory content is
447
391
        # "da..." below (the sha1sum of '').
448
392
        checker.add_pending_item(rev_id,
449
 
            ('texts', self.file_id, self.revision), 'text',
450
 
             'da39a3ee5e6b4b0d3255bfef95601890afd80709')
 
393
            (b'texts', self.file_id, self.revision), b'text',
 
394
             b'da39a3ee5e6b4b0d3255bfef95601890afd80709')
451
395
 
452
396
    def copy(self):
453
397
        other = InventoryDirectory(self.file_id, self.name, self.parent_id)
459
403
    def __init__(self, file_id, name, parent_id):
460
404
        super(InventoryDirectory, self).__init__(file_id, name, parent_id)
461
405
        self.children = {}
462
 
        self.kind = 'directory'
 
406
 
 
407
    def sorted_children(self):
 
408
        return sorted(viewitems(self.children))
463
409
 
464
410
    def kind_character(self):
465
411
        """See InventoryEntry.kind_character."""
466
412
        return '/'
467
413
 
468
 
    def _put_in_tar(self, item, tree):
469
 
        """See InventoryEntry._put_in_tar."""
470
 
        item.type = tarfile.DIRTYPE
471
 
        fileobj = None
472
 
        item.name += '/'
473
 
        item.size = 0
474
 
        item.mode = 0755
475
 
        return fileobj
476
 
 
477
 
    def _put_on_disk(self, fullpath, tree):
478
 
        """See InventoryEntry._put_on_disk."""
479
 
        os.mkdir(fullpath)
480
 
 
481
414
 
482
415
class InventoryFile(InventoryEntry):
483
416
    """A file in an inventory."""
484
417
 
485
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
486
 
                 'text_id', 'parent_id', 'children', 'executable',
487
 
                 'revision', 'symlink_target', 'reference_revision']
 
418
    __slots__ = ['text_sha1', 'text_size', 'text_id', 'executable']
 
419
 
 
420
    kind = 'file'
 
421
 
 
422
    def __init__(self, file_id, name, parent_id):
 
423
        super(InventoryFile, self).__init__(file_id, name, parent_id)
 
424
        self.text_sha1 = None
 
425
        self.text_size = None
 
426
        self.text_id = None
 
427
        self.executable = False
488
428
 
489
429
    def _check(self, checker, tree_revision_id):
490
430
        """See InventoryEntry._check"""
491
431
        # TODO: check size too.
492
432
        checker.add_pending_item(tree_revision_id,
493
 
            ('texts', self.file_id, self.revision), 'text',
 
433
            (b'texts', self.file_id, self.revision), b'text',
494
434
             self.text_sha1)
495
435
        if self.text_size is None:
496
436
            checker._report_items.append(
515
455
    def _diff(self, text_diff, from_label, tree, to_label, to_entry, to_tree,
516
456
             output_to, reverse=False):
517
457
        """See InventoryEntry._diff."""
518
 
        from bzrlib.diff import DiffText
 
458
        from breezy.diff import DiffText
519
459
        from_file_id = self.file_id
520
460
        if to_entry:
521
461
            to_file_id = to_entry.file_id
533
473
        """See InventoryEntry.has_text."""
534
474
        return True
535
475
 
536
 
    def __init__(self, file_id, name, parent_id):
537
 
        super(InventoryFile, self).__init__(file_id, name, parent_id)
538
 
        self.kind = 'file'
539
 
 
540
476
    def kind_character(self):
541
477
        """See InventoryEntry.kind_character."""
542
478
        return ''
543
479
 
544
 
    def _put_in_tar(self, item, tree):
545
 
        """See InventoryEntry._put_in_tar."""
546
 
        item.type = tarfile.REGTYPE
547
 
        fileobj = tree.get_file(self.file_id)
548
 
        item.size = self.text_size
549
 
        if tree.is_executable(self.file_id):
550
 
            item.mode = 0755
551
 
        else:
552
 
            item.mode = 0644
553
 
        return fileobj
554
 
 
555
 
    def _put_on_disk(self, fullpath, tree):
556
 
        """See InventoryEntry._put_on_disk."""
557
 
        osutils.pumpfile(tree.get_file(self.file_id), file(fullpath, 'wb'))
558
 
        if tree.is_executable(self.file_id):
559
 
            os.chmod(fullpath, 0755)
560
 
 
561
480
    def _read_tree_state(self, path, work_tree):
562
481
        """See InventoryEntry._read_tree_state."""
563
482
        self.text_sha1 = work_tree.get_file_sha1(self.file_id, path=path)
595
514
class InventoryLink(InventoryEntry):
596
515
    """A file in an inventory."""
597
516
 
598
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
599
 
                 'text_id', 'parent_id', 'children', 'executable',
600
 
                 'revision', 'symlink_target', 'reference_revision']
 
517
    __slots__ = ['symlink_target']
 
518
 
 
519
    kind = 'symlink'
 
520
 
 
521
    def __init__(self, file_id, name, parent_id):
 
522
        super(InventoryLink, self).__init__(file_id, name, parent_id)
 
523
        self.symlink_target = None
601
524
 
602
525
    def _check(self, checker, tree_revision_id):
603
526
        """See InventoryEntry._check"""
604
 
        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))
608
527
        if self.symlink_target is None:
609
528
            checker._report_items.append(
610
529
                'symlink {%s} has no target in revision {%s}'
611
530
                    % (self.file_id, tree_revision_id))
612
531
        # Symlinks are stored as ''
613
532
        checker.add_pending_item(tree_revision_id,
614
 
            ('texts', self.file_id, self.revision), 'text',
615
 
             'da39a3ee5e6b4b0d3255bfef95601890afd80709')
 
533
            (b'texts', self.file_id, self.revision), b'text',
 
534
             b'da39a3ee5e6b4b0d3255bfef95601890afd80709')
616
535
 
617
536
    def copy(self):
618
537
        other = InventoryLink(self.file_id, self.name, self.parent_id)
625
544
        # FIXME: which _modified field should we use ? RBC 20051003
626
545
        text_modified = (self.symlink_target != old_entry.symlink_target)
627
546
        if text_modified:
628
 
            mutter("    symlink target changed")
 
547
            trace.mutter("    symlink target changed")
629
548
        meta_modified = False
630
549
        return text_modified, meta_modified
631
550
 
632
551
    def _diff(self, text_diff, from_label, tree, to_label, to_entry, to_tree,
633
552
             output_to, reverse=False):
634
553
        """See InventoryEntry._diff."""
635
 
        from bzrlib.diff import DiffSymlink
 
554
        from breezy.diff import DiffSymlink
636
555
        old_target = self.symlink_target
637
556
        if to_entry is not None:
638
557
            new_target = to_entry.symlink_target
648
567
        differ = DiffSymlink(old_tree, new_tree, output_to)
649
568
        return differ.diff_symlink(old_target, new_target)
650
569
 
651
 
    def __init__(self, file_id, name, parent_id):
652
 
        super(InventoryLink, self).__init__(file_id, name, parent_id)
653
 
        self.kind = 'symlink'
654
 
 
655
570
    def kind_character(self):
656
571
        """See InventoryEntry.kind_character."""
657
572
        return ''
658
573
 
659
 
    def _put_in_tar(self, item, tree):
660
 
        """See InventoryEntry._put_in_tar."""
661
 
        item.type = tarfile.SYMTYPE
662
 
        fileobj = None
663
 
        item.size = 0
664
 
        item.mode = 0755
665
 
        item.linkname = self.symlink_target
666
 
        return fileobj
667
 
 
668
 
    def _put_on_disk(self, fullpath, tree):
669
 
        """See InventoryEntry._put_on_disk."""
670
 
        try:
671
 
            os.symlink(self.symlink_target, fullpath)
672
 
        except OSError,e:
673
 
            raise BzrError("Failed to create symlink %r -> %r, error: %s" % (fullpath, self.symlink_target, e))
674
 
 
675
574
    def _read_tree_state(self, path, work_tree):
676
575
        """See InventoryEntry._read_tree_state."""
677
576
        self.symlink_target = work_tree.get_symlink_target(self.file_id)
689
588
 
690
589
class TreeReference(InventoryEntry):
691
590
 
 
591
    __slots__ = ['reference_revision']
 
592
 
692
593
    kind = 'tree-reference'
693
594
 
694
595
    def __init__(self, file_id, name, parent_id, revision=None,
733
634
    inserted, other than through the Inventory API.
734
635
    """
735
636
 
736
 
    def __contains__(self, file_id):
737
 
        """True if this entry contains a file with given id.
738
 
 
739
 
        >>> inv = Inventory()
740
 
        >>> inv.add(InventoryFile('123', 'foo.c', ROOT_ID))
741
 
        InventoryFile('123', 'foo.c', parent_id='TREE_ROOT', sha1=None, len=None, revision=None)
742
 
        >>> '123' in inv
743
 
        True
744
 
        >>> '456' in inv
745
 
        False
746
 
 
747
 
        Note that this method along with __iter__ are not encouraged for use as
748
 
        they are less clear than specific query methods - they may be rmeoved
749
 
        in the future.
750
 
        """
751
 
        return self.has_id(file_id)
752
 
 
753
637
    def has_filename(self, filename):
754
638
        return bool(self.path2id(filename))
755
639
 
786
670
 
787
671
        # unrolling the recursive called changed the time from
788
672
        # 440ms/663ms (inline/total) to 116ms/116ms
789
 
        children = from_dir.children.items()
790
 
        children.sort()
 
673
        children = sorted(viewitems(from_dir.children))
791
674
        if not recursive:
792
675
            for name, ie in children:
793
676
                yield name, ie
812
695
                    continue
813
696
 
814
697
                # But do this child first
815
 
                new_children = ie.children.items()
816
 
                new_children.sort()
 
698
                new_children = sorted(viewitems(ie.children))
817
699
                new_children = collections.deque(new_children)
818
700
                stack.append((path, new_children))
819
701
                # Break out of inner loop, so that we start outer loop with child
822
704
                # if we finished all children, pop it off the stack
823
705
                stack.pop()
824
706
 
 
707
    def _preload_cache(self):
 
708
        """Populate any caches, we are about to access all items.
 
709
        
 
710
        The default implementation does nothing, because CommonInventory doesn't
 
711
        have a cache.
 
712
        """
 
713
        pass
 
714
    
825
715
    def iter_entries_by_dir(self, from_dir=None, specific_file_ids=None,
826
716
        yield_parents=False):
827
717
        """Iterate over the entries in a directory first order.
840
730
            specific_file_ids = set(specific_file_ids)
841
731
        # TODO? Perhaps this should return the from_dir so that the root is
842
732
        # yielded? or maybe an option?
 
733
        if from_dir is None and specific_file_ids is None:
 
734
            # They are iterating from the root, and have not specified any
 
735
            # specific entries to look at. All current callers fully consume the
 
736
            # iterator, so we can safely assume we are accessing all entries
 
737
            self._preload_cache()
843
738
        if from_dir is None:
844
739
            if self.root is None:
845
740
                return
847
742
            if (not yield_parents and specific_file_ids is not None and
848
743
                len(specific_file_ids) == 1):
849
744
                file_id = list(specific_file_ids)[0]
850
 
                if file_id in self:
 
745
                if self.has_id(file_id):
851
746
                    yield self.id2path(file_id), self[file_id]
852
747
                return
853
748
            from_dir = self.root
863
758
            parents = set()
864
759
            byid = self
865
760
            def add_ancestors(file_id):
866
 
                if file_id not in byid:
 
761
                if not byid.has_id(file_id):
867
762
                    return
868
763
                parent_id = byid[file_id].parent_id
869
764
                if parent_id is None:
881
776
            cur_relpath, cur_dir = stack.pop()
882
777
 
883
778
            child_dirs = []
884
 
            for child_name, child_ie in sorted(cur_dir.children.iteritems()):
 
779
            for child_name, child_ie in sorted(viewitems(cur_dir.children)):
885
780
 
886
781
                child_relpath = cur_relpath + child_name
887
782
 
913
808
                    file_id, self[file_id]))
914
809
        return delta
915
810
 
916
 
    def _get_mutable_inventory(self):
917
 
        """Returns a mutable copy of the object.
918
 
 
919
 
        Some inventories are immutable, yet working trees, for example, needs
920
 
        to mutate exisiting inventories instead of creating a new one.
921
 
        """
922
 
        raise NotImplementedError(self._get_mutable_inventory)
923
 
 
924
811
    def make_entry(self, kind, name, parent_id, file_id=None):
925
 
        """Simple thunk to bzrlib.inventory.make_entry."""
 
812
        """Simple thunk to breezy.inventory.make_entry."""
926
813
        return make_entry(kind, name, parent_id, file_id)
927
814
 
928
815
    def entries(self):
932
819
        """
933
820
        accum = []
934
821
        def descend(dir_ie, dir_path):
935
 
            kids = dir_ie.children.items()
936
 
            kids.sort()
 
822
            kids = sorted(viewitems(dir_ie.children))
937
823
            for name, ie in kids:
938
824
                child_path = osutils.pathjoin(dir_path, name)
939
825
                accum.append((child_path, ie))
940
826
                if ie.kind == 'directory':
941
827
                    descend(ie, child_path)
942
828
 
943
 
        descend(self.root, u'')
944
 
        return accum
945
 
 
946
 
    def directories(self):
947
 
        """Return (path, entry) pairs for all directories, including the root.
948
 
        """
949
 
        accum = []
950
 
        def descend(parent_ie, parent_path):
951
 
            accum.append((parent_path, parent_ie))
952
 
 
953
 
            kids = [(ie.name, ie) for ie in parent_ie.children.itervalues() if ie.kind == 'directory']
954
 
            kids.sort()
955
 
 
956
 
            for name, child_ie in kids:
957
 
                child_path = osutils.pathjoin(parent_path, name)
958
 
                descend(child_ie, child_path)
959
 
        descend(self.root, u'')
 
829
        if self.root is not None:
 
830
            descend(self.root, u'')
960
831
        return accum
961
832
 
962
833
    def path2id(self, relpath):
1060
931
 
1061
932
    >>> inv.path2id('hello.c')
1062
933
    '123-123'
1063
 
    >>> '123-123' in inv
 
934
    >>> inv.has_id('123-123')
1064
935
    True
1065
936
 
1066
937
    There are iterators over the contents:
1223
1094
            other.add(entry.copy())
1224
1095
        return other
1225
1096
 
1226
 
    def _get_mutable_inventory(self):
1227
 
        """See CommonInventory._get_mutable_inventory."""
1228
 
        return copy.deepcopy(self)
1229
 
 
1230
1097
    def __iter__(self):
1231
1098
        """Iterate over all file-ids."""
1232
1099
        return iter(self._byid)
1240
1107
        XXX: We may not want to merge this into bzr.dev.
1241
1108
        """
1242
1109
        if self.root is None:
1243
 
            return
1244
 
        for _, ie in self._byid.iteritems():
1245
 
            yield ie
 
1110
            return ()
 
1111
        return iter(viewvalues(self._byid))
1246
1112
 
1247
1113
    def __len__(self):
1248
1114
        """Returns number of entries."""
1272
1138
    def _add_child(self, entry):
1273
1139
        """Add an entry to the inventory, without adding it to its parent"""
1274
1140
        if entry.file_id in self._byid:
1275
 
            raise BzrError("inventory already contains entry with id {%s}" %
1276
 
                           entry.file_id)
 
1141
            raise errors.BzrError(
 
1142
                "inventory already contains entry with id {%s}" %
 
1143
                entry.file_id)
1277
1144
        self._byid[entry.file_id] = entry
1278
 
        for child in getattr(entry, 'children', {}).itervalues():
1279
 
            self._add_child(child)
 
1145
        children = getattr(entry, 'children', {})
 
1146
        if children is not None:
 
1147
            for child in viewvalues(children):
 
1148
                self._add_child(child)
1280
1149
        return entry
1281
1150
 
1282
1151
    def add(self, entry):
1283
1152
        """Add entry to inventory.
1284
1153
 
1285
 
        To add  a file to a branch ready to be committed, use Branch.add,
1286
 
        which calls this.
1287
 
 
1288
1154
        :return: entry
1289
1155
        """
1290
1156
        if entry.file_id in self._byid:
1335
1201
        >>> inv = Inventory()
1336
1202
        >>> inv.add(InventoryFile('123', 'foo.c', ROOT_ID))
1337
1203
        InventoryFile('123', 'foo.c', parent_id='TREE_ROOT', sha1=None, len=None, revision=None)
1338
 
        >>> '123' in inv
 
1204
        >>> inv.has_id('123')
1339
1205
        True
1340
1206
        >>> del inv['123']
1341
 
        >>> '123' in inv
 
1207
        >>> inv.has_id('123')
1342
1208
        False
1343
1209
        """
1344
1210
        ie = self[file_id]
1428
1294
            ie = to_find_delete.pop()
1429
1295
            to_delete.append(ie.file_id)
1430
1296
            if ie.kind == 'directory':
1431
 
                to_find_delete.extend(ie.children.values())
 
1297
                to_find_delete.extend(viewvalues(ie.children))
1432
1298
        for file_id in reversed(to_delete):
1433
1299
            ie = self[file_id]
1434
1300
            del self._byid[file_id]
1446
1312
        """
1447
1313
        new_name = ensure_normalized_name(new_name)
1448
1314
        if not is_valid_name(new_name):
1449
 
            raise BzrError("not an acceptable filename: %r" % new_name)
 
1315
            raise errors.BzrError("not an acceptable filename: %r" % new_name)
1450
1316
 
1451
1317
        new_parent = self._byid[new_parent_id]
1452
1318
        if new_name in new_parent.children:
1453
 
            raise BzrError("%r already exists in %r" % (new_name, self.id2path(new_parent_id)))
 
1319
            raise errors.BzrError("%r already exists in %r" %
 
1320
                (new_name, self.id2path(new_parent_id)))
1454
1321
 
1455
1322
        new_parent_idpath = self.get_idpath(new_parent_id)
1456
1323
        if file_id in new_parent_idpath:
1457
 
            raise BzrError("cannot move directory %r into a subdirectory of itself, %r"
 
1324
            raise errors.BzrError(
 
1325
                "cannot move directory %r into a subdirectory of itself, %r"
1458
1326
                    % (self.id2path(file_id), self.id2path(new_parent_id)))
1459
1327
 
1460
1328
        file_ie = self._byid[file_id]
1496
1364
    def __init__(self, search_key_name):
1497
1365
        CommonInventory.__init__(self)
1498
1366
        self._fileid_to_entry_cache = {}
 
1367
        self._fully_cached = False
1499
1368
        self._path_to_fileid_cache = {}
1500
1369
        self._search_key_name = search_key_name
1501
1370
        self.root_id = None
1531
1400
        if entry.parent_id is not None:
1532
1401
            parent_str = entry.parent_id
1533
1402
        else:
1534
 
            parent_str = ''
 
1403
            parent_str = b''
1535
1404
        name_str = entry.name.encode("utf8")
1536
1405
        if entry.kind == 'file':
1537
1406
            if entry.executable:
1538
 
                exec_str = "Y"
 
1407
                exec_str = b"Y"
1539
1408
            else:
1540
 
                exec_str = "N"
1541
 
            return "file: %s\n%s\n%s\n%s\n%s\n%d\n%s" % (
 
1409
                exec_str = b"N"
 
1410
            return b"file: %s\n%s\n%s\n%s\n%s\n%d\n%s" % (
1542
1411
                entry.file_id, parent_str, name_str, entry.revision,
1543
1412
                entry.text_sha1, entry.text_size, exec_str)
1544
1413
        elif entry.kind == 'directory':
1545
 
            return "dir: %s\n%s\n%s\n%s" % (
 
1414
            return b"dir: %s\n%s\n%s\n%s" % (
1546
1415
                entry.file_id, parent_str, name_str, entry.revision)
1547
1416
        elif entry.kind == 'symlink':
1548
 
            return "symlink: %s\n%s\n%s\n%s\n%s" % (
 
1417
            return b"symlink: %s\n%s\n%s\n%s\n%s" % (
1549
1418
                entry.file_id, parent_str, name_str, entry.revision,
1550
1419
                entry.symlink_target.encode("utf8"))
1551
1420
        elif entry.kind == 'tree-reference':
1552
 
            return "tree: %s\n%s\n%s\n%s\n%s" % (
 
1421
            return b"tree: %s\n%s\n%s\n%s\n%s" % (
1553
1422
                entry.file_id, parent_str, name_str, entry.revision,
1554
1423
                entry.reference_revision)
1555
1424
        else:
1588
1457
            if entry.kind == 'directory':
1589
1458
                directories_to_expand.add(entry.file_id)
1590
1459
            interesting.add(entry.parent_id)
1591
 
            children_of_parent_id.setdefault(entry.parent_id, []
1592
 
                                             ).append(entry.file_id)
 
1460
            children_of_parent_id.setdefault(entry.parent_id, set()
 
1461
                                             ).add(entry.file_id)
1593
1462
 
1594
1463
        # Now, interesting has all of the direct parents, but not the
1595
1464
        # parents of those parents. It also may have some duplicates with
1603
1472
            next_parents = set()
1604
1473
            for entry in self._getitems(remaining_parents):
1605
1474
                next_parents.add(entry.parent_id)
1606
 
                children_of_parent_id.setdefault(entry.parent_id, []
1607
 
                                                 ).append(entry.file_id)
 
1475
                children_of_parent_id.setdefault(entry.parent_id, set()
 
1476
                                                 ).add(entry.file_id)
1608
1477
            # Remove any search tips we've already processed
1609
1478
            remaining_parents = next_parents.difference(interesting)
1610
1479
            interesting.update(remaining_parents)
1617
1486
            keys = [StaticTuple(f,).intern() for f in directories_to_expand]
1618
1487
            directories_to_expand = set()
1619
1488
            items = self.parent_id_basename_to_file_id.iteritems(keys)
1620
 
            next_file_ids = set([item[1] for item in items])
 
1489
            next_file_ids = {item[1] for item in items}
1621
1490
            next_file_ids = next_file_ids.difference(interesting)
1622
1491
            interesting.update(next_file_ids)
1623
1492
            for entry in self._getitems(next_file_ids):
1624
1493
                if entry.kind == 'directory':
1625
1494
                    directories_to_expand.add(entry.file_id)
1626
 
                children_of_parent_id.setdefault(entry.parent_id, []
1627
 
                                                 ).append(entry.file_id)
 
1495
                children_of_parent_id.setdefault(entry.parent_id, set()
 
1496
                                                 ).add(entry.file_id)
1628
1497
        return interesting, children_of_parent_id
1629
1498
 
1630
1499
    def filter(self, specific_fileids):
1652
1521
            # parent_to_children with at least the tree root.)
1653
1522
            return other
1654
1523
        cache = self._fileid_to_entry_cache
1655
 
        try:
1656
 
            remaining_children = collections.deque(parent_to_children[self.root_id])
1657
 
        except:
1658
 
            import pdb; pdb.set_trace()
1659
 
            raise
 
1524
        remaining_children = collections.deque(parent_to_children[self.root_id])
1660
1525
        while remaining_children:
1661
1526
            file_id = remaining_children.popleft()
1662
1527
            ie = cache[file_id]
1671
1536
        return other
1672
1537
 
1673
1538
    @staticmethod
1674
 
    def _bytes_to_utf8name_key(bytes):
1675
 
        """Get the file_id, revision_id key out of bytes."""
 
1539
    def _bytes_to_utf8name_key(data):
 
1540
        """Get the file_id, revision_id key out of data."""
1676
1541
        # We don't normally care about name, except for times when we want
1677
1542
        # to filter out empty names because of non rich-root...
1678
 
        sections = bytes.split('\n')
1679
 
        kind, file_id = sections[0].split(': ')
1680
 
        return (sections[2], intern(file_id), intern(sections[3]))
 
1543
        sections = data.split(b'\n')
 
1544
        kind, file_id = sections[0].split(b': ')
 
1545
        return (sections[2], bytesintern(file_id), bytesintern(sections[3]))
1681
1546
 
1682
1547
    def _bytes_to_entry(self, bytes):
1683
1548
        """Deserialise a serialised entry."""
1684
 
        sections = bytes.split('\n')
1685
 
        if sections[0].startswith("file: "):
 
1549
        sections = bytes.split(b'\n')
 
1550
        if sections[0].startswith(b"file: "):
1686
1551
            result = InventoryFile(sections[0][6:],
1687
1552
                sections[2].decode('utf8'),
1688
1553
                sections[1])
1689
1554
            result.text_sha1 = sections[4]
1690
1555
            result.text_size = int(sections[5])
1691
 
            result.executable = sections[6] == "Y"
1692
 
        elif sections[0].startswith("dir: "):
 
1556
            result.executable = sections[6] == b"Y"
 
1557
        elif sections[0].startswith(b"dir: "):
1693
1558
            result = CHKInventoryDirectory(sections[0][5:],
1694
1559
                sections[2].decode('utf8'),
1695
1560
                sections[1], self)
1696
 
        elif sections[0].startswith("symlink: "):
 
1561
        elif sections[0].startswith(b"symlink: "):
1697
1562
            result = InventoryLink(sections[0][9:],
1698
1563
                sections[2].decode('utf8'),
1699
1564
                sections[1])
1700
1565
            result.symlink_target = sections[4].decode('utf8')
1701
 
        elif sections[0].startswith("tree: "):
 
1566
        elif sections[0].startswith(b"tree: "):
1702
1567
            result = TreeReference(sections[0][6:],
1703
1568
                sections[2].decode('utf8'),
1704
1569
                sections[1])
1705
1570
            result.reference_revision = sections[4]
1706
1571
        else:
1707
1572
            raise ValueError("Not a serialised entry %r" % bytes)
1708
 
        result.file_id = intern(result.file_id)
1709
 
        result.revision = intern(sections[3])
1710
 
        if result.parent_id == '':
 
1573
        result.file_id = bytesintern(result.file_id)
 
1574
        result.revision = bytesintern(sections[3])
 
1575
        if result.parent_id == b'':
1711
1576
            result.parent_id = None
1712
1577
        self._fileid_to_entry_cache[result.file_id] = result
1713
1578
        return result
1714
1579
 
1715
 
    def _get_mutable_inventory(self):
1716
 
        """See CommonInventory._get_mutable_inventory."""
1717
 
        entries = self.iter_entries()
1718
 
        inv = Inventory(None, self.revision_id)
1719
 
        for path, inv_entry in entries:
1720
 
            inv.add(inv_entry.copy())
1721
 
        return inv
1722
 
 
1723
1580
    def create_by_apply_delta(self, inventory_delta, new_revision_id,
1724
1581
        propagate_caches=False):
1725
1582
        """Create a new CHKInventory by applying inventory_delta to this one.
1738
1595
        result = CHKInventory(self._search_key_name)
1739
1596
        if propagate_caches:
1740
1597
            # Just propagate the path-to-fileid cache for now
1741
 
            result._path_to_fileid_cache = dict(self._path_to_fileid_cache.iteritems())
 
1598
            result._path_to_fileid_cache = self._path_to_fileid_cache.copy()
1742
1599
        search_key_func = chk_map.search_key_registry.get(self._search_key_name)
1743
1600
        self.id_to_entry._ensure_root()
1744
1601
        maximum_size = self.id_to_entry._root_node.maximum_size
1857
1714
                continue
1858
1715
            # This loop could potentially be better by using the id_basename
1859
1716
            # map to just get the child file ids.
1860
 
            for child in entry.children.values():
 
1717
            for child in viewvalues(entry.children):
1861
1718
                if child.file_id not in altered:
1862
1719
                    raise errors.InconsistentDelta(self.id2path(child.file_id),
1863
1720
                        child.file_id, "Child not deleted or reparented when "
1869
1726
            # re-keying, but its simpler to just output that as a delete+add
1870
1727
            # to spend less time calculating the delta.
1871
1728
            delta_list = []
1872
 
            for key, (old_key, value) in parent_id_basename_delta.iteritems():
 
1729
            for key, (old_key, value) in viewitems(parent_id_basename_delta):
1873
1730
                if value is not None:
1874
1731
                    delta_list.append((old_key, key, value))
1875
1732
                else:
1899
1756
            for.
1900
1757
        :return: A CHKInventory
1901
1758
        """
1902
 
        lines = bytes.split('\n')
1903
 
        if lines[-1] != '':
 
1759
        lines = bytes.split(b'\n')
 
1760
        if lines[-1] != b'':
1904
1761
            raise AssertionError('bytes to deserialize must end with an eol')
1905
1762
        lines.pop()
1906
 
        if lines[0] != 'chkinventory:':
 
1763
        if lines[0] != b'chkinventory:':
1907
1764
            raise ValueError("not a serialised CHKInventory: %r" % bytes)
1908
1765
        info = {}
1909
 
        allowed_keys = frozenset(['root_id', 'revision_id', 'search_key_name',
1910
 
                                  'parent_id_basename_to_file_id',
1911
 
                                  'id_to_entry'])
 
1766
        allowed_keys = frozenset((b'root_id', b'revision_id',
 
1767
                                  b'parent_id_basename_to_file_id',
 
1768
                                  b'search_key_name', b'id_to_entry'))
1912
1769
        for line in lines[1:]:
1913
 
            key, value = line.split(': ', 1)
 
1770
            key, value = line.split(b': ', 1)
1914
1771
            if key not in allowed_keys:
1915
1772
                raise errors.BzrError('Unknown key in inventory: %r\n%r'
1916
1773
                                      % (key, bytes))
1918
1775
                raise errors.BzrError('Duplicate key in inventory: %r\n%r'
1919
1776
                                      % (key, bytes))
1920
1777
            info[key] = value
1921
 
        revision_id = intern(info['revision_id'])
1922
 
        root_id = intern(info['root_id'])
1923
 
        search_key_name = intern(info.get('search_key_name', 'plain'))
1924
 
        parent_id_basename_to_file_id = intern(info.get(
1925
 
            'parent_id_basename_to_file_id', None))
1926
 
        if not parent_id_basename_to_file_id.startswith('sha1:'):
 
1778
        revision_id = bytesintern(info[b'revision_id'])
 
1779
        root_id = bytesintern(info[b'root_id'])
 
1780
        search_key_name = bytesintern(info.get(b'search_key_name', b'plain'))
 
1781
        parent_id_basename_to_file_id = bytesintern(info.get(
 
1782
            b'parent_id_basename_to_file_id', None))
 
1783
        if not parent_id_basename_to_file_id.startswith(b'sha1:'):
1927
1784
            raise ValueError('parent_id_basename_to_file_id should be a sha1'
1928
1785
                             ' key not %r' % (parent_id_basename_to_file_id,))
1929
 
        id_to_entry = info['id_to_entry']
1930
 
        if not id_to_entry.startswith('sha1:'):
 
1786
        id_to_entry = info[b'id_to_entry']
 
1787
        if not id_to_entry.startswith(b'sha1:'):
1931
1788
            raise ValueError('id_to_entry should be a sha1'
1932
1789
                             ' key not %r' % (id_to_entry,))
1933
1790
 
1935
1792
        result.revision_id = revision_id
1936
1793
        result.root_id = root_id
1937
1794
        search_key_func = chk_map.search_key_registry.get(
1938
 
                            result._search_key_name)
 
1795
            result._search_key_name.decode("ascii"))
1939
1796
        if parent_id_basename_to_file_id is not None:
1940
1797
            result.parent_id_basename_to_file_id = chk_map.CHKMap(
1941
1798
                chk_store, StaticTuple(parent_id_basename_to_file_id,),
2001
1858
        if entry.parent_id is not None:
2002
1859
            parent_id = entry.parent_id
2003
1860
        else:
2004
 
            parent_id = ''
 
1861
            parent_id = b''
2005
1862
        return StaticTuple(parent_id, entry.name.encode('utf8')).intern()
2006
1863
 
2007
1864
    def __getitem__(self, file_id):
2013
1870
            return result
2014
1871
        try:
2015
1872
            return self._bytes_to_entry(
2016
 
                self.id_to_entry.iteritems([StaticTuple(file_id,)]).next()[1])
 
1873
                next(self.id_to_entry.iteritems([StaticTuple(file_id,)]))[1])
2017
1874
        except StopIteration:
2018
1875
            # really we're passing an inventory, not a tree...
2019
1876
            raise errors.NoSuchId(self, file_id)
2066
1923
 
2067
1924
    def iter_just_entries(self):
2068
1925
        """Iterate over all entries.
2069
 
        
 
1926
 
2070
1927
        Unlike iter_entries(), just the entries are returned (not (path, ie))
2071
1928
        and the order of entries is undefined.
2072
1929
 
2080
1937
                self._fileid_to_entry_cache[file_id] = ie
2081
1938
            yield ie
2082
1939
 
 
1940
    def _preload_cache(self):
 
1941
        """Make sure all file-ids are in _fileid_to_entry_cache"""
 
1942
        if self._fully_cached:
 
1943
            return # No need to do it again
 
1944
        # The optimal sort order is to use iteritems() directly
 
1945
        cache = self._fileid_to_entry_cache
 
1946
        for key, entry in self.id_to_entry.iteritems():
 
1947
            file_id = key[0]
 
1948
            if file_id not in cache:
 
1949
                ie = self._bytes_to_entry(entry)
 
1950
                cache[file_id] = ie
 
1951
            else:
 
1952
                ie = cache[file_id]
 
1953
        last_parent_id = last_parent_ie = None
 
1954
        pid_items = self.parent_id_basename_to_file_id.iteritems()
 
1955
        for key, child_file_id in pid_items:
 
1956
            if key == (b'', b''): # This is the root
 
1957
                if child_file_id != self.root_id:
 
1958
                    raise ValueError('Data inconsistency detected.'
 
1959
                        ' We expected data with key ("","") to match'
 
1960
                        ' the root id, but %s != %s'
 
1961
                        % (child_file_id, self.root_id))
 
1962
                continue
 
1963
            parent_id, basename = key
 
1964
            ie = cache[child_file_id]
 
1965
            if parent_id == last_parent_id:
 
1966
                parent_ie = last_parent_ie
 
1967
            else:
 
1968
                parent_ie = cache[parent_id]
 
1969
            if parent_ie.kind != 'directory':
 
1970
                raise ValueError('Data inconsistency detected.'
 
1971
                    ' An entry in the parent_id_basename_to_file_id map'
 
1972
                    ' has parent_id {%s} but the kind of that object'
 
1973
                    ' is %r not "directory"' % (parent_id, parent_ie.kind))
 
1974
            if parent_ie._children is None:
 
1975
                parent_ie._children = {}
 
1976
            basename = basename.decode('utf-8')
 
1977
            if basename in parent_ie._children:
 
1978
                existing_ie = parent_ie._children[basename]
 
1979
                if existing_ie != ie:
 
1980
                    raise ValueError('Data inconsistency detected.'
 
1981
                        ' Two entries with basename %r were found'
 
1982
                        ' in the parent entry {%s}'
 
1983
                        % (basename, parent_id))
 
1984
            if basename != ie.name:
 
1985
                raise ValueError('Data inconsistency detected.'
 
1986
                    ' In the parent_id_basename_to_file_id map, file_id'
 
1987
                    ' {%s} is listed as having basename %r, but in the'
 
1988
                    ' id_to_entry map it is %r'
 
1989
                    % (child_file_id, basename, ie.name))
 
1990
            parent_ie._children[basename] = ie
 
1991
        self._fully_cached = True
 
1992
 
2083
1993
    def iter_changes(self, basis):
2084
1994
        """Generate a Tree.iter_changes change list between this and basis.
2085
1995
 
2159
2069
 
2160
2070
    def _make_delta(self, old):
2161
2071
        """Make an inventory delta from two inventories."""
2162
 
        if type(old) != CHKInventory:
 
2072
        if not isinstance(old, CHKInventory):
2163
2073
            return CommonInventory._make_delta(self, old)
2164
2074
        delta = []
2165
2075
        for key, old_value, self_value in \
2182
2092
    def path2id(self, relpath):
2183
2093
        """See CommonInventory.path2id()."""
2184
2094
        # TODO: perhaps support negative hits?
 
2095
        if isinstance(relpath, basestring):
 
2096
            names = osutils.splitpath(relpath)
 
2097
        else:
 
2098
            names = relpath
 
2099
            if relpath == []:
 
2100
                relpath = [""]
 
2101
            relpath = osutils.pathjoin(*relpath)
2185
2102
        result = self._path_to_fileid_cache.get(relpath, None)
2186
2103
        if result is not None:
2187
2104
            return result
2188
 
        if isinstance(relpath, basestring):
2189
 
            names = osutils.splitpath(relpath)
2190
 
        else:
2191
 
            names = relpath
2192
2105
        current_id = self.root_id
2193
2106
        if current_id is None:
2194
2107
            return None
2218
2131
 
2219
2132
    def to_lines(self):
2220
2133
        """Serialise the inventory to lines."""
2221
 
        lines = ["chkinventory:\n"]
 
2134
        lines = [b"chkinventory:\n"]
2222
2135
        if self._search_key_name != 'plain':
2223
2136
            # custom ordering grouping things that don't change together
2224
 
            lines.append('search_key_name: %s\n' % (self._search_key_name,))
2225
 
            lines.append("root_id: %s\n" % self.root_id)
2226
 
            lines.append('parent_id_basename_to_file_id: %s\n' %
 
2137
            lines.append(b'search_key_name: %s\n' % (
 
2138
                self._search_key_name.encode('ascii')))
 
2139
            lines.append(b"root_id: %s\n" % self.root_id)
 
2140
            lines.append(b'parent_id_basename_to_file_id: %s\n' %
2227
2141
                (self.parent_id_basename_to_file_id.key()[0],))
2228
 
            lines.append("revision_id: %s\n" % self.revision_id)
2229
 
            lines.append("id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
 
2142
            lines.append(b"revision_id: %s\n" % self.revision_id)
 
2143
            lines.append(b"id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
2230
2144
        else:
2231
 
            lines.append("revision_id: %s\n" % self.revision_id)
2232
 
            lines.append("root_id: %s\n" % self.root_id)
 
2145
            lines.append(b"revision_id: %s\n" % self.revision_id)
 
2146
            lines.append(b"root_id: %s\n" % self.root_id)
2233
2147
            if self.parent_id_basename_to_file_id is not None:
2234
 
                lines.append('parent_id_basename_to_file_id: %s\n' %
 
2148
                lines.append(b'parent_id_basename_to_file_id: %s\n' %
2235
2149
                    (self.parent_id_basename_to_file_id.key()[0],))
2236
 
            lines.append("id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
 
2150
            lines.append(b"id_to_entry: %s\n" % (self.id_to_entry.key()[0],))
2237
2151
        return lines
2238
2152
 
2239
2153
    @property
2245
2159
class CHKInventoryDirectory(InventoryDirectory):
2246
2160
    """A directory in an inventory."""
2247
2161
 
2248
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
2249
 
                 'text_id', 'parent_id', '_children', 'executable',
2250
 
                 'revision', 'symlink_target', 'reference_revision',
2251
 
                 '_chk_inventory']
 
2162
    __slots__ = ['_children', '_chk_inventory']
2252
2163
 
2253
2164
    def __init__(self, file_id, name, parent_id, chk_inventory):
2254
2165
        # Don't call InventoryDirectory.__init__ - it isn't right for this
2255
2166
        # class.
2256
2167
        InventoryEntry.__init__(self, file_id, name, parent_id)
2257
2168
        self._children = None
2258
 
        self.kind = 'directory'
2259
2169
        self._chk_inventory = chk_inventory
2260
2170
 
2261
2171
    @property
2331
2241
        accessed on this platform by the normalized path.
2332
2242
    :return: The NFC normalised version of name.
2333
2243
    """
2334
 
    #------- This has been copied to bzrlib.dirstate.DirState.add, please
 
2244
    #------- This has been copied to breezy.dirstate.DirState.add, please
2335
2245
    # keep them synchronised.
2336
2246
    # we dont import normalized_filename directly because we want to be
2337
2247
    # able to change the implementation at runtime for tests.
2346
2256
    return name
2347
2257
 
2348
2258
 
2349
 
_NAME_RE = None
 
2259
_NAME_RE = lazy_regex.lazy_compile(r'^[^/\\]+$')
2350
2260
 
2351
2261
def is_valid_name(name):
2352
 
    global _NAME_RE
2353
 
    if _NAME_RE is None:
2354
 
        _NAME_RE = re.compile(r'^[^/\\]+$')
2355
 
 
2356
2262
    return bool(_NAME_RE.match(name))
2357
2263
 
2358
2264
 
2413
2319
        if item[2] is None:
2414
2320
            raise errors.InconsistentDelta(item[0] or item[1], item[2],
2415
2321
                "entry with file_id None %r" % entry)
2416
 
        if type(item[2]) != str:
 
2322
        if not isinstance(item[2], str):
2417
2323
            raise errors.InconsistentDelta(item[0] or item[1], item[2],
2418
2324
                "entry with non bytes file_id %r" % entry)
2419
2325
        yield item
2448
2354
            raise errors.InconsistentDelta(new_path, item[1],
2449
2355
                "new_path with no entry")
2450
2356
        yield item
 
2357
 
 
2358
 
 
2359
def mutable_inventory_from_tree(tree):
 
2360
    """Create a new inventory that has the same contents as a specified tree.
 
2361
 
 
2362
    :param tree: Revision tree to create inventory from
 
2363
    """
 
2364
    entries = tree.iter_entries_by_dir()
 
2365
    inv = Inventory(None, tree.get_revision_id())
 
2366
    for path, inv_entry in entries:
 
2367
        inv.add(inv_entry.copy())
 
2368
    return inv