osutil.py
52 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
|
r8232 | # osutil.py - pure Python version of osutil.c | ||
# | ||||
# Copyright 2009 Matt Mackall <mpm@selenic.com> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r8232 | |||
|
r7704 | import os | ||
import stat as _stat | ||||
|
r9031 | posixfile = open | ||
|
r8421 | |||
|
r7704 | def _mode_to_kind(mode): | ||
if _stat.S_ISREG(mode): return _stat.S_IFREG | ||||
if _stat.S_ISDIR(mode): return _stat.S_IFDIR | ||||
if _stat.S_ISLNK(mode): return _stat.S_IFLNK | ||||
if _stat.S_ISBLK(mode): return _stat.S_IFBLK | ||||
if _stat.S_ISCHR(mode): return _stat.S_IFCHR | ||||
if _stat.S_ISFIFO(mode): return _stat.S_IFIFO | ||||
if _stat.S_ISSOCK(mode): return _stat.S_IFSOCK | ||||
return mode | ||||
def listdir(path, stat=False, skip=None): | ||||
'''listdir(path, stat=False) -> list_of_tuples | ||||
Return a sorted list containing information about the entries | ||||
in the directory. | ||||
If stat is True, each element is a 3-tuple: | ||||
(name, type, stat object) | ||||
Otherwise, each element is a 2-tuple: | ||||
(name, type) | ||||
''' | ||||
result = [] | ||||
prefix = path | ||||
if not prefix.endswith(os.sep): | ||||
prefix += os.sep | ||||
names = os.listdir(path) | ||||
names.sort() | ||||
for fn in names: | ||||
st = os.lstat(prefix + fn) | ||||
if fn == skip and _stat.S_ISDIR(st.st_mode): | ||||
return [] | ||||
if stat: | ||||
result.append((fn, _mode_to_kind(st.st_mode), st)) | ||||
else: | ||||
result.append((fn, _mode_to_kind(st.st_mode))) | ||||
return result | ||||
|
r8421 | |||