~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Robert Collins
  • Date: 2005-10-06 22:15:52 UTC
  • mfrom: (1185.13.2)
  • mto: This revision was merged to the branch mainline in revision 1420.
  • Revision ID: robertc@robertcollins.net-20051006221552-9b15c96fa504e0ad
mergeĀ fromĀ upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
21
21
# it's not predictable when it will be written out.
22
22
 
23
23
import os
24
 
    
 
24
import stat
 
25
import fnmatch
 
26
        
25
27
import bzrlib.tree
26
 
from errors import BzrCheckError
27
 
from trace import mutter
 
28
from bzrlib.osutils import appendpath, file_kind, isdir, splitpath
 
29
from bzrlib.errors import BzrCheckError
 
30
from bzrlib.trace import mutter
 
31
 
 
32
class TreeEntry(object):
 
33
    """An entry that implements the minium interface used by commands.
 
34
 
 
35
    This needs further inspection, it may be better to have 
 
36
    InventoryEntries without ids - though that seems wrong. For now,
 
37
    this is a parallel hierarchy to InventoryEntry, and needs to become
 
38
    one of several things: decorates to that hierarchy, children of, or
 
39
    parents of it.
 
40
    Another note is that these objects are currently only used when there is
 
41
    no InventoryEntry available - i.e. for unversioned objects.
 
42
    Perhaps they should be UnversionedEntry et al. ? - RBC 20051003
 
43
    """
 
44
 
 
45
    def __eq__(self, other):
 
46
        # yes, this us ugly, TODO: best practice __eq__ style.
 
47
        return (isinstance(other, TreeEntry)
 
48
                and other.__class__ == self.__class__)
 
49
 
 
50
    def kind_character(self):
 
51
        return "???"
 
52
 
 
53
 
 
54
class TreeDirectory(TreeEntry):
 
55
    """See TreeEntry. This is a directory in a working tree."""
 
56
 
 
57
    def __eq__(self, other):
 
58
        return (isinstance(other, TreeDirectory)
 
59
                and other.__class__ == self.__class__)
 
60
 
 
61
    def kind_character(self):
 
62
        return "/"
 
63
 
 
64
 
 
65
class TreeFile(TreeEntry):
 
66
    """See TreeEntry. This is a regular file in a working tree."""
 
67
 
 
68
    def __eq__(self, other):
 
69
        return (isinstance(other, TreeFile)
 
70
                and other.__class__ == self.__class__)
 
71
 
 
72
    def kind_character(self):
 
73
        return ''
 
74
 
 
75
 
 
76
class TreeLink(TreeEntry):
 
77
    """See TreeEntry. This is a symlink in a working tree."""
 
78
 
 
79
    def __eq__(self, other):
 
80
        return (isinstance(other, TreeLink)
 
81
                and other.__class__ == self.__class__)
 
82
 
 
83
    def kind_character(self):
 
84
        return ''
 
85
 
28
86
 
29
87
class WorkingTree(bzrlib.tree.Tree):
30
88
    """Working copy tree.
67
125
        """
68
126
        inv = self._inventory
69
127
        for path, ie in inv.iter_entries():
70
 
            if os.path.exists(self.abspath(path)):
 
128
            if bzrlib.osutils.lexists(self.abspath(path)):
71
129
                yield ie.file_id
72
130
 
73
131
 
81
139
        return os.path.join(self.basedir, filename)
82
140
 
83
141
    def has_filename(self, filename):
84
 
        return os.path.exists(self.abspath(filename))
 
142
        return bzrlib.osutils.lexists(self.abspath(filename))
85
143
 
86
144
    def get_file(self, file_id):
87
145
        return self.get_file_byname(self.id2path(file_id))
93
151
        ## XXX: badly named; this isn't in the store at all
94
152
        return self.abspath(self.id2path(file_id))
95
153
 
 
154
 
 
155
    def id2abspath(self, file_id):
 
156
        return self.abspath(self.id2path(file_id))
 
157
 
96
158
                
97
159
    def has_id(self, file_id):
98
160
        # files that have been deleted are excluded
100
162
        if not inv.has_id(file_id):
101
163
            return False
102
164
        path = inv.id2path(file_id)
103
 
        return os.path.exists(self.abspath(path))
 
165
        return bzrlib.osutils.lexists(self.abspath(path))
104
166
 
105
167
 
106
168
    __contains__ = has_id
107
169
    
108
170
 
109
171
    def get_file_size(self, file_id):
110
 
        # is this still called?
111
 
        raise NotImplementedError()
112
 
 
 
172
        return os.path.getsize(self.id2abspath(file_id))
113
173
 
114
174
    def get_file_sha1(self, file_id):
115
175
        path = self._inventory.id2path(file_id)
116
176
        return self._hashcache.get_sha1(path)
117
177
 
118
178
 
 
179
    def is_executable(self, file_id):
 
180
        if os.name == "nt":
 
181
            return self._inventory[file_id].executable
 
182
        else:
 
183
            path = self._inventory.id2path(file_id)
 
184
            mode = os.lstat(self.abspath(path)).st_mode
 
185
            return bool(stat.S_ISREG(mode) and stat.S_IEXEC&mode)
 
186
 
 
187
    def get_symlink_target(self, file_id):
 
188
        return os.readlink(self.id2path(file_id))
 
189
 
119
190
    def file_class(self, filename):
120
191
        if self.path2id(filename):
121
192
            return 'V'
135
206
 
136
207
        Skips the control directory.
137
208
        """
138
 
        from osutils import appendpath, file_kind
139
 
        import os
140
 
 
141
209
        inv = self._inventory
142
210
 
143
211
        def descend(from_dir_relpath, from_dir_id, dp):
172
240
                                            "now of kind %r"
173
241
                                            % (fap, f_ie.kind, f_ie.file_id, fk))
174
242
 
175
 
                yield fp, c, fk, (f_ie and f_ie.file_id)
 
243
                # make a last minute entry
 
244
                if f_ie:
 
245
                    entry = f_ie
 
246
                else:
 
247
                    if fk == 'directory':
 
248
                        entry = TreeDirectory()
 
249
                    elif fk == 'file':
 
250
                        entry = TreeFile()
 
251
                    elif fk == 'symlink':
 
252
                        entry = TreeLink()
 
253
                    else:
 
254
                        entry = TreeEntry()
 
255
                
 
256
                yield fp, c, fk, (f_ie and f_ie.file_id), entry
176
257
 
177
258
                if fk != 'directory':
178
259
                    continue
194
275
            if not self.is_ignored(subp):
195
276
                yield subp
196
277
 
 
278
    def iter_conflicts(self):
 
279
        conflicted = set()
 
280
        for path in (s[0] for s in self.list_files()):
 
281
            stem = get_conflicted_stem(path)
 
282
            if stem is None:
 
283
                continue
 
284
            if stem not in conflicted:
 
285
                conflicted.add(stem)
 
286
                yield stem
197
287
 
198
288
    def extras(self):
199
289
        """Yield all unknown files in this WorkingTree.
205
295
        Currently returned depth-first, sorted by name within directories.
206
296
        """
207
297
        ## TODO: Work from given directory downwards
208
 
        from osutils import isdir, appendpath
209
 
        
210
298
        for path, dir_entry in self.inventory.directories():
211
299
            mutter("search for unknowns in %r" % path)
212
300
            dirabs = self.abspath(path)
269
357
        # Eventually it should be replaced with something more
270
358
        # accurate.
271
359
        
272
 
        import fnmatch
273
 
        from osutils import splitpath
274
 
        
275
360
        for pat in self.get_ignore_list():
276
361
            if '/' in pat or '\\' in pat:
277
362
                
290
375
                    return pat
291
376
        else:
292
377
            return None
293
 
        
 
 
b'\\ No newline at end of file'
 
378
 
 
379
CONFLICT_SUFFIXES = ('.THIS', '.BASE', '.OTHER')
 
380
def get_conflicted_stem(path):
 
381
    for suffix in CONFLICT_SUFFIXES:
 
382
        if path.endswith(suffix):
 
383
            return path[:-len(suffix)]