~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Martin Pool
  • Date: 2005-05-11 06:12:44 UTC
  • Revision ID: mbp@sourcefrog.net-20050511061244-b0eed343857f10d3
- remove dead code for cmd_compare_trees
- compare_trees new parameter want_unchanged to avoid allocating
  strings if not needed

Show diffs side-by-side

added added

removed removed

Lines of Context:
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
19
    
26
21
from errors import BzrCheckError
27
22
from trace import mutter
28
23
 
 
24
 
29
25
class WorkingTree(bzrlib.tree.Tree):
30
26
    """Working copy tree.
31
27
 
35
31
    It is possible for a `WorkingTree` to have a filename which is
36
32
    not listed in the Inventory and vice versa.
37
33
    """
 
34
    _statcache = None
 
35
    
38
36
    def __init__(self, basedir, inv):
39
 
        from bzrlib.hashcache import HashCache
40
 
        from bzrlib.trace import note, mutter
41
 
 
42
37
        self._inventory = inv
43
38
        self.basedir = basedir
44
39
        self.path2id = inv.path2id
45
40
 
46
 
        # update the whole cache up front and write to disk if anything changed;
47
 
        # in the future we might want to do this more selectively
48
 
        hc = self._hashcache = HashCache(basedir)
49
 
        hc.read()
50
 
        hc.scan()
51
 
 
52
 
        if hc.needs_write:
53
 
            mutter("write hc")
54
 
            hc.write()
55
 
            
56
 
            
57
 
    def __del__(self):
58
 
        if self._hashcache.needs_write:
59
 
            self._hashcache.write()
60
 
 
61
 
 
62
41
    def __iter__(self):
63
42
        """Iterate through file_ids for this tree.
64
43
 
65
44
        file_ids are in a WorkingTree if they are in the working inventory
66
45
        and the working file exists.
67
46
        """
 
47
        self._update_statcache()
68
48
        inv = self._inventory
69
 
        for path, ie in inv.iter_entries():
70
 
            if os.path.exists(self.abspath(path)):
71
 
                yield ie.file_id
 
49
        for file_id in self._inventory:
 
50
            # TODO: This is slightly redundant; we should be able to just
 
51
            # check the statcache but it only includes regular files.
 
52
            # only include files which still exist on disk
 
53
            ie = inv[file_id]
 
54
            if ie.kind == 'file':
 
55
                if ((file_id in self._statcache)
 
56
                    or (os.path.exists(self.abspath(inv.id2path(file_id))))):
 
57
                    yield file_id
 
58
 
72
59
 
73
60
 
74
61
    def __repr__(self):
75
62
        return "<%s of %s>" % (self.__class__.__name__,
76
 
                               getattr(self, 'basedir', None))
77
 
 
78
 
 
 
63
                               self.basedir)
79
64
 
80
65
    def abspath(self, filename):
81
66
        return os.path.join(self.basedir, filename)
96
81
                
97
82
    def has_id(self, file_id):
98
83
        # files that have been deleted are excluded
99
 
        inv = self._inventory
100
 
        if not inv.has_id(file_id):
 
84
        if not self.inventory.has_id(file_id):
101
85
            return False
102
 
        path = inv.id2path(file_id)
103
 
        return os.path.exists(self.abspath(path))
 
86
        self._update_statcache()
 
87
        if file_id in self._statcache:
 
88
            return True
 
89
        return os.path.exists(self.abspath(self.id2path(file_id)))
104
90
 
105
91
 
106
92
    __contains__ = has_id
107
93
    
108
94
 
 
95
    def _update_statcache(self):
 
96
        import statcache
 
97
        if not self._statcache:
 
98
            self._statcache = statcache.update_cache(self.basedir, self.inventory)
 
99
 
109
100
    def get_file_size(self, file_id):
110
 
        # is this still called?
111
 
        raise NotImplementedError()
 
101
        import os, stat
 
102
        return os.stat(self._get_store_filename(file_id))[stat.ST_SIZE]
112
103
 
113
104
 
114
105
    def get_file_sha1(self, file_id):
115
 
        path = self._inventory.id2path(file_id)
116
 
        return self._hashcache.get_sha1(path)
 
106
        import statcache
 
107
        self._update_statcache()
 
108
        return self._statcache[file_id][statcache.SC_SHA1]
117
109
 
118
110
 
119
111
    def file_class(self, filename):
138
130
        from osutils import appendpath, file_kind
139
131
        import os
140
132
 
141
 
        inv = self._inventory
 
133
        inv = self.inventory
142
134
 
143
135
        def descend(from_dir_relpath, from_dir_id, dp):
144
136
            ls = os.listdir(dp)
290
282
                    return pat
291
283
        else:
292
284
            return None
293
 
        
 
 
b'\\ No newline at end of file'
 
285
        
 
286
 
 
287
        
 
288
        
 
289