← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Martin Pool
  • Date: 2005-07-06 10:07:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050706100731-ce25f8bf569ad1ae
- start adding refactored/simplified hash cache
  not used yet

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/workingtree.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
# TODO: Don't allow WorkingTrees to be constructed for remote branches.
18
 
 
19
 
# FIXME: I don't know if writing out the cache from the destructor is really a
20
 
# good idea, because destructors are considered poor taste in Python, and
21
 
# it's not predictable when it will be written out.
22
17
 
23
18
import os
24
 
import fnmatch
25
 
        
 
19
    
26
20
import bzrlib.tree
27
 
from bzrlib.osutils import appendpath, file_kind, isdir, splitpath
28
 
from bzrlib.errors import BzrCheckError
29
 
from bzrlib.trace import mutter
 
21
from errors import BzrCheckError
 
22
from trace import mutter
30
23
 
31
24
class WorkingTree(bzrlib.tree.Tree):
32
25
    """Working copy tree.
37
30
    It is possible for a `WorkingTree` to have a filename which is
38
31
    not listed in the Inventory and vice versa.
39
32
    """
 
33
    _statcache = None
 
34
    
40
35
    def __init__(self, basedir, inv):
41
 
        from bzrlib.hashcache import HashCache
42
 
        from bzrlib.trace import note, mutter
43
 
 
44
36
        self._inventory = inv
45
37
        self.basedir = basedir
46
38
        self.path2id = inv.path2id
47
 
 
48
 
        # update the whole cache up front and write to disk if anything changed;
49
 
        # in the future we might want to do this more selectively
50
 
        hc = self._hashcache = HashCache(basedir)
51
 
        hc.read()
52
 
        hc.scan()
53
 
 
54
 
        if hc.needs_write:
55
 
            mutter("write hc")
56
 
            hc.write()
57
 
            
58
 
            
59
 
    def __del__(self):
60
 
        if self._hashcache.needs_write:
61
 
            self._hashcache.write()
62
 
 
 
39
        self._update_statcache()
63
40
 
64
41
    def __iter__(self):
65
42
        """Iterate through file_ids for this tree.
68
45
        and the working file exists.
69
46
        """
70
47
        inv = self._inventory
71
 
        for path, ie in inv.iter_entries():
72
 
            if os.path.exists(self.abspath(path)):
73
 
                yield ie.file_id
 
48
        for file_id in self._inventory:
 
49
            # TODO: This is slightly redundant; we should be able to just
 
50
            # check the statcache but it only includes regular files.
 
51
            # only include files which still exist on disk
 
52
            ie = inv[file_id]
 
53
            if ie.kind == 'file':
 
54
                if ((file_id in self._statcache)
 
55
                    or (os.path.exists(self.abspath(inv.id2path(file_id))))):
 
56
                    yield file_id
 
57
 
74
58
 
75
59
 
76
60
    def __repr__(self):
77
61
        return "<%s of %s>" % (self.__class__.__name__,
78
 
                               getattr(self, 'basedir', None))
79
 
 
80
 
 
 
62
                               self.basedir)
81
63
 
82
64
    def abspath(self, filename):
83
65
        return os.path.join(self.basedir, filename)
98
80
                
99
81
    def has_id(self, file_id):
100
82
        # files that have been deleted are excluded
101
 
        inv = self._inventory
102
 
        if not inv.has_id(file_id):
 
83
        if not self.inventory.has_id(file_id):
103
84
            return False
104
 
        path = inv.id2path(file_id)
105
 
        return os.path.exists(self.abspath(path))
 
85
        if file_id in self._statcache:
 
86
            return True
 
87
        return os.path.exists(self.abspath(self.id2path(file_id)))
106
88
 
107
89
 
108
90
    __contains__ = has_id
109
91
    
110
92
 
 
93
    def _update_statcache(self):
 
94
        if not self._statcache:
 
95
            from bzrlib.statcache import update_cache
 
96
            self._statcache = update_cache(self.basedir, self.inventory)
 
97
 
111
98
    def get_file_size(self, file_id):
112
 
        # is this still called?
113
 
        raise NotImplementedError()
 
99
        import os, stat
 
100
        return os.stat(self._get_store_filename(file_id))[stat.ST_SIZE]
114
101
 
115
102
 
116
103
    def get_file_sha1(self, file_id):
117
 
        path = self._inventory.id2path(file_id)
118
 
        return self._hashcache.get_sha1(path)
 
104
        from bzrlib.statcache import SC_SHA1
 
105
        return self._statcache[file_id][SC_SHA1]
119
106
 
120
107
 
121
108
    def file_class(self, filename):
137
124
 
138
125
        Skips the control directory.
139
126
        """
140
 
        inv = self._inventory
 
127
        from osutils import appendpath, file_kind
 
128
        import os
 
129
 
 
130
        inv = self.inventory
141
131
 
142
132
        def descend(from_dir_relpath, from_dir_id, dp):
143
133
            ls = os.listdir(dp)
204
194
        Currently returned depth-first, sorted by name within directories.
205
195
        """
206
196
        ## TODO: Work from given directory downwards
 
197
        from osutils import isdir, appendpath
 
198
        
207
199
        for path, dir_entry in self.inventory.directories():
208
200
            mutter("search for unknowns in %r" % path)
209
201
            dirabs = self.abspath(path)
266
258
        # Eventually it should be replaced with something more
267
259
        # accurate.
268
260
        
 
261
        import fnmatch
 
262
        from osutils import splitpath
 
263
        
269
264
        for pat in self.get_ignore_list():
270
265
            if '/' in pat or '\\' in pat:
271
266
                
285
280
        else:
286
281
            return None
287
282
        
 
283
 
 
284
        
 
285
        
 
286