~bzr-pqm/bzr/bzr.dev

1739.2.6 by Robert Collins
Merge bzr.dev
1
# Copyright (C) 2006, 2008 Canonical Ltd
1739.2.3 by Robert Collins
Add a replacement for os.listdir which returns file kind information from readdir when it is available. This drops our osutils.walkdirs time further, down to 77ms.
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1739.2.3 by Robert Collins
Add a replacement for os.listdir which returns file kind information from readdir when it is available. This drops our osutils.walkdirs time further, down to 77ms.
16
1739.2.6 by Robert Collins
Merge bzr.dev
17
"""Python implementation of readdir interface."""
1739.2.3 by Robert Collins
Add a replacement for os.listdir which returns file kind information from readdir when it is available. This drops our osutils.walkdirs time further, down to 77ms.
18
19
3696.3.5 by Robert Collins
Streamline _walkdirs_utf8 for utf8 file systems, reducing time to traverse a mozilla tree from 1s to .6 seconds. (Robert Collins)
20
import stat
21
22
23
_directory = 'directory'
24
_chardev = 'chardev'
25
_block = 'block'
26
_file = 'file'
27
_fifo = 'fifo'
28
_symlink = 'symlink'
29
_socket = 'socket'
30
_unknown = 'unknown'
31
32
_formats = {
33
    stat.S_IFDIR:'directory',
34
    stat.S_IFCHR:'chardev',
35
    stat.S_IFBLK:'block',
36
    stat.S_IFREG:'file',
37
    stat.S_IFIFO:'fifo',
38
    stat.S_IFLNK:'symlink',
39
    stat.S_IFSOCK:'socket',
40
}
41
42
43
def _kind_from_mode(stat_mode, _formats=_formats, _unknown='unknown'):
44
    """Generate a file kind from a stat mode. This is used in walkdirs.
45
3696.3.10 by Robert Collins
Review feedback.
46
    It's performance is critical: Do not mutate without careful benchmarking.
1739.2.3 by Robert Collins
Add a replacement for os.listdir which returns file kind information from readdir when it is available. This drops our osutils.walkdirs time further, down to 77ms.
47
    """
3696.3.5 by Robert Collins
Streamline _walkdirs_utf8 for utf8 file systems, reducing time to traverse a mozilla tree from 1s to .6 seconds. (Robert Collins)
48
    try:
49
        return _formats[stat_mode & 0170000]
50
    except KeyError:
51
        return _unknown