21
20
from sets import Set
22
21
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
24
28
from inventory import Inventory
25
29
from trace import mutter, note
26
from osutils import pumpfile, compare_files, filesize, quotefn, sha_file, \
27
joinpath, splitpath, appendpath, isdir, isfile, file_kind
28
30
from errors import bailout
30
from stat import S_ISREG, S_ISDIR, ST_MODE, ST_SIZE
78
75
doc="Inventory of this Tree")
80
77
def _check_retrieved(self, ie, f):
81
# TODO: Test this check by damaging the store?
82
if ie.text_size is not None:
84
if fs != ie.text_size:
78
fp = fingerprint_file(f)
81
if ie.text_size != None:
82
if ie.text_size != fp['size']:
85
83
bailout("mismatched size for file %r in %r" % (ie.file_id, self._store),
86
84
["inventory expects %d bytes" % ie.text_size,
87
"file is actually %d bytes" % fs,
85
"file is actually %d bytes" % fp['size'],
88
86
"store is probably damaged/corrupt"])
92
if ie.text_sha1 != f_hash:
88
if ie.text_sha1 != fp['sha1']:
93
89
bailout("wrong SHA-1 for file %r in %r" % (ie.file_id, self._store),
94
90
["inventory expects %s" % ie.text_sha1,
95
"file is actually %s" % f_hash,
91
"file is actually %s" % fp['sha1'],
96
92
"store is probably damaged/corrupt"])
99
def export(self, dest):
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):
100
102
"""Export this tree to a new directory.
102
104
`dest` should not exist, and will be created holding the
103
105
contents of this tree.
105
:todo: To handle subdirectories we need to create the
107
TODO: To handle subdirectories we need to create the
106
108
directories first.
108
110
:note: If the export fails, the destination directory will be
142
144
return "<%s of %s>" % (self.__class__.__name__,
145
def _rel(self, filename):
147
def abspath(self, filename):
146
148
return os.path.join(self.basedir, filename)
148
150
def has_filename(self, filename):
149
return os.path.exists(self._rel(filename))
151
return os.path.exists(self.abspath(filename))
151
153
def get_file(self, file_id):
152
return file(self._get_store_filename(file_id), 'rb')
154
return self.get_file_byname(self.id2path(file_id))
156
def get_file_byname(self, filename):
157
return file(self.abspath(filename), 'rb')
154
159
def _get_store_filename(self, file_id):
155
return self._rel(self.id2path(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)
157
169
def get_file_size(self, file_id):
158
170
return os.stat(self._get_store_filename(file_id))[ST_SIZE]
193
196
inv = self.inventory
195
def descend(from_dir, from_dir_id, dp):
198
def descend(from_dir_relpath, from_dir_id, dp):
196
199
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.
199
205
if bzrlib.BZRDIR == f:
202
208
# path within tree
203
fp = appendpath(from_dir, f)
209
fp = appendpath(from_dir_relpath, f)
206
212
fap = appendpath(dp, f)
232
238
for ff in descend(fp, f_ie.file_id, fap):
235
for f in descend('', None, self.basedir):
241
for f in descend('', inv.root.file_id, self.basedir):
240
def unknowns(self, path='', dir_id=None):
241
"""Yield names of unknown files in this WorkingTree.
247
for subp in self.extras():
248
if not self.is_ignored(subp):
253
"""Yield all unknown files in this WorkingTree.
243
255
If there are any unknown directories then only the directory is
244
256
returned, not all its children. But if there are unknown files
247
259
Currently returned depth-first, sorted by name within directories.
249
for fpath, fclass, fkind, fid in self.list_files():
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)
254
282
def ignored_files(self):
255
for fpath, fclass, fkind, fid in self.list_files():
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()])
260
306
def is_ignored(self, filename):
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):
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):
added
removed
bzrlib/tree.py
