20
21
from sets import Set
21
22
import os.path, os, fnmatch
23
from osutils import pumpfile, compare_files, filesize, quotefn, sha_file, \
24
joinpath, splitpath, appendpath, isdir, isfile, file_kind, fingerprint_file
26
from stat import S_ISREG, S_ISDIR, ST_MODE, ST_SIZE
28
24
from inventory import Inventory
29
25
from trace import mutter, note
26
from osutils import pumpfile, compare_files, filesize, quotefn, sha_file, \
27
joinpath, splitpath, appendpath, isdir, isfile, file_kind
30
28
from errors import bailout
30
from stat import S_ISREG, S_ISDIR, ST_MODE, ST_SIZE
75
78
doc="Inventory of this Tree")
77
80
def _check_retrieved(self, ie, f):
78
fp = fingerprint_file(f)
81
if ie.text_size != None:
82
if ie.text_size != fp['size']:
81
# TODO: Test this check by damaging the store?
82
if ie.text_size is not None:
84
if fs != ie.text_size:
83
85
bailout("mismatched size for file %r in %r" % (ie.file_id, self._store),
84
86
["inventory expects %d bytes" % ie.text_size,
85
"file is actually %d bytes" % fp['size'],
87
"file is actually %d bytes" % fs,
86
88
"store is probably damaged/corrupt"])
88
if ie.text_sha1 != fp['sha1']:
92
if ie.text_sha1 != f_hash:
89
93
bailout("wrong SHA-1 for file %r in %r" % (ie.file_id, self._store),
90
94
["inventory expects %s" % ie.text_sha1,
91
"file is actually %s" % fp['sha1'],
95
"file is actually %s" % f_hash,
92
96
"store is probably damaged/corrupt"])
95
def print_file(self, fileid):
96
"""Print file with id `fileid` to stdout."""
98
pumpfile(self.get_file(fileid), sys.stdout)
101
def export(self, dest):
99
def export(self, dest):
102
100
"""Export this tree to a new directory.
104
102
`dest` should not exist, and will be created holding the
105
103
contents of this tree.
107
TODO: To handle subdirectories we need to create the
105
:todo: To handle subdirectories we need to create the
108
106
directories first.
110
108
:note: If the export fails, the destination directory will be
144
142
return "<%s of %s>" % (self.__class__.__name__,
147
def abspath(self, filename):
145
def _rel(self, filename):
148
146
return os.path.join(self.basedir, filename)
150
148
def has_filename(self, filename):
151
return os.path.exists(self.abspath(filename))
149
return os.path.exists(self._rel(filename))
153
151
def get_file(self, file_id):
154
return self.get_file_byname(self.id2path(file_id))
156
def get_file_byname(self, filename):
157
return file(self.abspath(filename), 'rb')
152
return file(self._get_store_filename(file_id), 'rb')
159
154
def _get_store_filename(self, file_id):
160
## XXX: badly named; this isn't in the store at all
161
return self.abspath(self.id2path(file_id))
163
def has_id(self, file_id):
164
# files that have been deleted are excluded
165
if not self.inventory.has_id(file_id):
167
return os.access(self.abspath(self.inventory.id2path(file_id)), os.F_OK)
155
return self._rel(self.id2path(file_id))
169
157
def get_file_size(self, file_id):
170
158
return os.stat(self._get_store_filename(file_id))[ST_SIZE]
196
193
inv = self.inventory
198
def descend(from_dir_relpath, from_dir_id, dp):
195
def descend(from_dir, from_dir_id, dp):
199
196
ls = os.listdir(dp)
202
## TODO: If we find a subdirectory with its own .bzr
203
## directory, then that is a separate tree and we
204
## should exclude it.
205
199
if bzrlib.BZRDIR == f:
208
202
# path within tree
209
fp = appendpath(from_dir_relpath, f)
203
fp = appendpath(from_dir, f)
212
206
fap = appendpath(dp, f)
238
232
for ff in descend(fp, f_ie.file_id, fap):
241
for f in descend('', inv.root.file_id, self.basedir):
235
for f in descend('', None, self.basedir):
247
for subp in self.extras():
248
if not self.is_ignored(subp):
253
"""Yield all unknown files in this WorkingTree.
240
def unknowns(self, path='', dir_id=None):
241
"""Yield names of unknown files in this WorkingTree.
255
243
If there are any unknown directories then only the directory is
256
244
returned, not all its children. But if there are unknown files
259
247
Currently returned depth-first, sorted by name within directories.
261
## TODO: Work from given directory downwards
263
for path, dir_entry in self.inventory.directories():
264
mutter("search for unknowns in %r" % path)
265
dirabs = self.abspath(path)
266
if not isdir(dirabs):
267
# e.g. directory deleted
271
for subf in os.listdir(dirabs):
273
and (subf not in dir_entry.children)):
278
subp = appendpath(path, subf)
249
for fpath, fclass, fkind, fid in self.list_files():
282
254
def ignored_files(self):
283
"""Yield list of PATH, IGNORE_PATTERN"""
284
for subp in self.extras():
285
pat = self.is_ignored(subp)
290
def get_ignore_list(self):
291
"""Return list of ignore patterns.
293
Cached in the Tree object after the first call.
295
if hasattr(self, '_ignorelist'):
296
return self._ignorelist
298
l = bzrlib.DEFAULT_IGNORE[:]
299
if self.has_filename(bzrlib.IGNORE_FILENAME):
300
f = self.get_file_byname(bzrlib.IGNORE_FILENAME)
301
l.extend([line.rstrip("\n\r") for line in f.readlines()])
255
for fpath, fclass, fkind, fid in self.list_files():
306
260
def is_ignored(self, filename):
307
r"""Check whether the filename matches an ignore pattern.
309
Patterns containing '/' or '\' need to match the whole path;
310
others match against only the last component.
312
If the file is ignored, returns the pattern which caused it to
313
be ignored, otherwise None. So this can simply be used as a
314
boolean if desired."""
316
# TODO: Use '**' to match directories, and other extended
317
# globbing stuff from cvs/rsync.
319
# XXX: fnmatch is actually not quite what we want: it's only
320
# approximately the same as real Unix fnmatch, and doesn't
321
# treat dotfiles correctly and allows * to match /.
322
# Eventually it should be replaced with something more
325
for pat in self.get_ignore_list():
326
if '/' in pat or '\\' in pat:
328
# as a special case, you can put ./ at the start of a
329
# pattern; this is good to match in the top-level
332
if (pat[:2] == './') or (pat[:2] == '.\\'):
336
if fnmatch.fnmatchcase(filename, newpat):
339
if fnmatch.fnmatchcase(splitpath(filename)[-1], pat):
261
"""Check whether the filename matches an ignore pattern."""
262
## TODO: Take them from a file, not hardcoded
263
## TODO: Use extended zsh-style globs maybe?
264
## TODO: Use '**' to match directories?
265
## TODO: Patterns without / should match in subdirectories?
266
for i in bzrlib.DEFAULT_IGNORE:
267
if fnmatch.fnmatchcase(filename, i):
added
removed
bzrlib/tree.py
