# osutil.py - pure Python version of osutil.c # # Copyright 2009 Matt Mackall and others # # This software may be used and distributed according to the terms of the # GNU General Public License version 2, incorporated herein by reference. import os import stat as _stat posixfile = open 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