← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/_readdir_py.py

Major code cleanup.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/_readdir_py.py
 
1
# Copyright (C) 2006, 2008 Canonical Ltd
 
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
"""Python implementation of readdir interface."""
 
18
 
 
19
 
 
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
 
 
46
    It's performance is critical: Do not mutate without careful benchmarking.
 
47
    """
 
48
    try:
 
49
        return _formats[stat_mode & 0170000]
 
50
    except KeyError:
 
51
        return _unknown