← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inventory.py

  • Committer: John Arbash Meinel
  • Date: 2006-10-31 21:29:02 UTC
  • mfrom: (2104 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2110.
  • Revision ID: john@arbash-meinel.com-20061031212902-4b33920b90e9ce92
[merge] bzr.dev 2104

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
27
27
# created, but it's not for now.
28
28
ROOT_ID = "TREE_ROOT"
29
29
 
30
 
 
31
 
import collections
32
 
import os.path
 
30
import os
33
31
import re
34
32
import sys
 
33
 
 
34
from bzrlib.lazy_import import lazy_import
 
35
lazy_import(globals(), """
 
36
import collections
35
37
import tarfile
36
 
import types
37
 
from warnings import warn
38
38
 
39
39
import bzrlib
40
 
from bzrlib import errors, osutils
41
 
from bzrlib.osutils import (pumpfile, quotefn, splitpath, joinpath,
42
 
                            pathjoin, sha_strings)
43
 
from bzrlib.errors import (NotVersionedError, InvalidEntryName,
44
 
                           BzrError, BzrCheckError, BinaryFile)
 
40
from bzrlib import (
 
41
    errors,
 
42
    osutils,
 
43
    symbol_versioning,
 
44
    )
 
45
""")
 
46
 
 
47
from bzrlib.errors import (
 
48
    BzrCheckError,
 
49
    BzrError,
 
50
    )
45
51
from bzrlib.trace import mutter
46
52
 
47
53
 
82
88
    InventoryDirectory('123', 'src', parent_id='TREE_ROOT', revision=None)
83
89
    >>> i.add(InventoryFile('2323', 'hello.c', parent_id='123'))
84
90
    InventoryFile('2323', 'hello.c', parent_id='123', sha1=None, len=None)
85
 
    >>> shouldbe = {0: '', 1: 'src', 2: pathjoin('src','hello.c')}
 
91
    >>> shouldbe = {0: '', 1: 'src', 2: 'src/hello.c'}
86
92
    >>> for ix, j in enumerate(i.iter_entries()):
87
93
    ...   print (j[0] == shouldbe[ix], j[1])
88
94
    ... 
246
252
 
247
253
    def get_tar_item(self, root, dp, now, tree):
248
254
        """Get a tarfile item and a file stream for its content."""
249
 
        item = tarfile.TarInfo(pathjoin(root, dp).encode('utf8'))
 
255
        item = tarfile.TarInfo(osutils.pathjoin(root, dp).encode('utf8'))
250
256
        # TODO: would be cool to actually set it to the timestamp of the
251
257
        # revision it was last changed
252
258
        item.mtime = now
281
287
        """
282
288
        assert isinstance(name, basestring), name
283
289
        if '/' in name or '\\' in name:
284
 
            raise InvalidEntryName(name=name)
 
290
            raise errors.InvalidEntryName(name=name)
285
291
        self.executable = False
286
292
        self.revision = None
287
293
        self.text_sha1 = None
311
317
        
312
318
        This is a template method - implement _put_on_disk in subclasses.
313
319
        """
314
 
        fullpath = pathjoin(dest, dp)
 
320
        fullpath = osutils.pathjoin(dest, dp)
315
321
        self._put_on_disk(fullpath, tree)
316
322
        # mutter("  export {%s} kind %s to %s", self.file_id,
317
323
        #         self.kind, fullpath)
514
520
        self.parent_id = None
515
521
        self.name = u''
516
522
        self.revision = None
517
 
        warn('RootEntry is deprecated as of bzr 0.10.  Please use '
518
 
             'InventoryDirectory instead.',
519
 
            DeprecationWarning, stacklevel=2)
 
523
        symbol_versioning.warn('RootEntry is deprecated as of bzr 0.10.'
 
524
                               '  Please use InventoryDirectory instead.',
 
525
                               DeprecationWarning, stacklevel=2)
520
526
 
521
527
    def __eq__(self, other):
522
528
        if not isinstance(other, RootEntry):
645
651
            else:
646
652
                text_diff(to_label, to_text,
647
653
                          from_label, from_text, output_to)
648
 
        except BinaryFile:
 
654
        except errors.BinaryFile:
649
655
            if reverse:
650
656
                label_pair = (to_label, from_label)
651
657
            else:
677
683
 
678
684
    def _put_on_disk(self, fullpath, tree):
679
685
        """See InventoryEntry._put_on_disk."""
680
 
        pumpfile(tree.get_file(self.file_id), file(fullpath, 'wb'))
 
686
        osutils.pumpfile(tree.get_file(self.file_id), file(fullpath, 'wb'))
681
687
        if tree.is_executable(self.file_id):
682
688
            os.chmod(fullpath, 0755)
683
689
 
849
855
    ['', u'hello.c']
850
856
    >>> inv = Inventory('TREE_ROOT-12345678-12345678')
851
857
    >>> inv.add(InventoryFile('123-123', 'hello.c', ROOT_ID))
 
858
    Traceback (most recent call last):
 
859
    BzrError: parent_id {TREE_ROOT} not in inventory
 
860
    >>> inv.add(InventoryFile('123-123', 'hello.c', 'TREE_ROOT-12345678-12345678'))
852
861
    InventoryFile('123-123', 'hello.c', parent_id='TREE_ROOT-12345678-12345678', sha1=None, len=None)
853
862
    """
854
863
    def __init__(self, root_id=ROOT_ID, revision_id=None):
861
870
        The inventory is created with a default root directory, with
862
871
        an id of None.
863
872
        """
864
 
        # We are letting Branch.create() create a unique inventory
865
 
        # root id. Rather than generating a random one here.
866
 
        #if root_id is None:
867
 
        #    root_id = bzrlib.branch.gen_file_id('TREE_ROOT')
868
873
        if root_id is not None:
869
874
            self._set_root(InventoryDirectory(root_id, '', None))
870
875
        else:
871
876
            self.root = None
872
877
            self._byid = {}
873
 
        # FIXME: this isn't ever used, changing it to self.revision may break
874
 
        # things. TODO make everything use self.revision_id
875
878
        self.revision_id = revision_id
876
879
 
877
880
    def _set_root(self, ie):
898
901
    def iter_entries(self, from_dir=None):
899
902
        """Return (path, entry) pairs, in order by name."""
900
903
        if from_dir is None:
901
 
            assert self.root
 
904
            if self.root is None:
 
905
                return
902
906
            from_dir = self.root
903
907
            yield '', self.root
904
908
        elif isinstance(from_dir, basestring):
951
955
        # TODO? Perhaps this should return the from_dir so that the root is
952
956
        # yielded? or maybe an option?
953
957
        if from_dir is None:
954
 
            assert self.root
 
958
            if self.root is None:
 
959
                return
955
960
            from_dir = self.root
956
961
            yield '', self.root
957
962
        elif isinstance(from_dir, basestring):
982
987
            kids = dir_ie.children.items()
983
988
            kids.sort()
984
989
            for name, ie in kids:
985
 
                child_path = pathjoin(dir_path, name)
 
990
                child_path = osutils.pathjoin(dir_path, name)
986
991
                accum.append((child_path, ie))
987
992
                if ie.kind == 'directory':
988
993
                    descend(ie, child_path)
1001
1006
            kids.sort()
1002
1007
 
1003
1008
            for name, child_ie in kids:
1004
 
                child_path = pathjoin(parent_path, name)
 
1009
                child_path = osutils.pathjoin(parent_path, name)
1005
1010
                descend(child_ie, child_path)
1006
1011
        descend(self.root, u'')
1007
1012
        return accum
1057
1062
            assert self.root is None and len(self._byid) == 0
1058
1063
            self._set_root(entry)
1059
1064
            return entry
1060
 
        if entry.parent_id == ROOT_ID:
1061
 
            assert self.root is not None, self
1062
 
            entry.parent_id = self.root.file_id
1063
 
 
1064
1065
        try:
1065
1066
            parent = self._byid[entry.parent_id]
1066
1067
        except KeyError:
1068
1069
 
1069
1070
        if entry.name in parent.children:
1070
1071
            raise BzrError("%s is already versioned" %
1071
 
                    pathjoin(self.id2path(parent.file_id), entry.name))
 
1072
                    osutils.pathjoin(self.id2path(parent.file_id), entry.name))
1072
1073
 
1073
1074
        self._byid[entry.file_id] = entry
1074
1075
        parent.children[entry.name] = entry
1093
1094
            parent_path = parts[:-1]
1094
1095
            parent_id = self.path2id(parent_path)
1095
1096
            if parent_id is None:
1096
 
                raise NotVersionedError(path=parent_path)
 
1097
                raise errors.NotVersionedError(path=parent_path)
1097
1098
        ie = make_entry(kind, parts[-1], parent_id, file_id)
1098
1099
        return self.add(ie)
1099
1100
 
1193
1194
 
1194
1195
        Returns None IFF the path is not found.
1195
1196
        """
1196
 
        if isinstance(name, types.StringTypes):
1197
 
            name = splitpath(name)
 
1197
        if isinstance(name, basestring):
 
1198
            name = osutils.splitpath(name)
1198
1199
 
1199
1200
        # mutter("lookup path %r" % name)
1200
1201
 
1201
1202
        parent = self.root
 
1203
        if parent is None:
 
1204
            return None
1202
1205
        for f in name:
1203
1206
            try:
1204
 
                cie = parent.children[f]
 
1207
                children = getattr(parent, 'children', None)
 
1208
                if children is None:
 
1209
                    return None
 
1210
                cie = children[f]
1205
1211
                assert cie.name == f
1206
1212
                assert cie.parent_id == parent.file_id
1207
1213
                parent = cie
1264
1270
        file_ie.name = new_name
1265
1271
        file_ie.parent_id = new_parent_id
1266
1272
 
 
1273
    def is_root(self, file_id):
 
1274
        return self.root is not None and file_id == self.root.file_id
 
1275
 
1267
1276
 
1268
1277
def make_entry(kind, name, parent_id, file_id=None):
1269
1278
    """Create an inventory entry.