← 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-08 06:54:58 UTC
  • Revision ID: mbp@sourcefrog.net-20050708065458-2af06c3659faf1d8
- use new path-based hashcache for WorkingTree- squash mtime/ctime to whole seconds- update and if necessary write out hashcache when WorkingTree object is   created.

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.
17
18
 
18
19
import os
19
20
    
30
31
    It is possible for a `WorkingTree` to have a filename which is
31
32
    not listed in the Inventory and vice versa.
32
33
    """
33
 
    _statcache = None
34
 
    
35
34
    def __init__(self, basedir, inv):
 
35
        from bzrlib.hashcache import HashCache
 
36
        from bzrlib.trace import note, mutter
 
37
 
36
38
        self._inventory = inv
37
39
        self.basedir = basedir
38
40
        self.path2id = inv.path2id
39
 
        self._update_statcache()
 
41
 
 
42
        # update the whole cache up front and write to disk if anything changed;
 
43
        # in the future we might want to do this more selectively
 
44
        hc = self._hashcache = HashCache(basedir)
 
45
        hc.read()
 
46
        for path, ie in inv.iter_entries():
 
47
            hc.get_sha1(path)
 
48
 
 
49
        if hc.needs_write:
 
50
            mutter("write hc")
 
51
            hc.write()
 
52
 
 
53
 
40
54
 
41
55
    def __iter__(self):
42
56
        """Iterate through file_ids for this tree.
45
59
        and the working file exists.
46
60
        """
47
61
        inv = self._inventory
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
 
 
 
62
        for path, ie in inv.iter_entries():
 
63
            if os.path.exists(self.abspath(path)):
 
64
                yield ie.file_id
58
65
 
59
66
 
60
67
    def __repr__(self):
61
68
        return "<%s of %s>" % (self.__class__.__name__,
62
69
                               self.basedir)
63
70
 
 
71
 
 
72
 
64
73
    def abspath(self, filename):
65
74
        return os.path.join(self.basedir, filename)
66
75
 
80
89
                
81
90
    def has_id(self, file_id):
82
91
        # files that have been deleted are excluded
83
 
        if not self.inventory.has_id(file_id):
 
92
        inv = self._inventory
 
93
        if not inv.has_id(file_id):
84
94
            return False
85
 
        if file_id in self._statcache:
86
 
            return True
87
 
        return os.path.exists(self.abspath(self.id2path(file_id)))
 
95
        path = inv.id2path(file_id)
 
96
        return os.path.exists(self.abspath(path))
88
97
 
89
98
 
90
99
    __contains__ = has_id
91
100
    
92
101
 
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
 
 
98
102
    def get_file_size(self, file_id):
99
 
        import os, stat
100
 
        return os.stat(self._get_store_filename(file_id))[stat.ST_SIZE]
 
103
        # is this still called?
 
104
        raise NotImplementedError()
101
105
 
102
106
 
103
107
    def get_file_sha1(self, file_id):
104
 
        from bzrlib.statcache import SC_SHA1
105
 
        return self._statcache[file_id][SC_SHA1]
 
108
        path = self._inventory.id2path(file_id)
 
109
        return self._hashcache.get_sha1(path)
106
110
 
107
111
 
108
112
    def file_class(self, filename):
127
131
        from osutils import appendpath, file_kind
128
132
        import os
129
133
 
130
 
        inv = self.inventory
 
134
        inv = self._inventory
131
135
 
132
136
        def descend(from_dir_relpath, from_dir_id, dp):
133
137
            ls = os.listdir(dp)