Mercurial > public > mercurial-scm > hg-stable
diff mercurial/osutil.py @ 5396:5105b119edd2
Add osutil module, containing a listdir function.
This is similar to os.listdir, only it returns a sorted list of tuples.
author | Bryan O'Sullivan <bos@serpentine.com> |
---|---|
date | Fri, 05 Oct 2007 15:01:06 -0700 |
parents | |
children | 0d513661d6c2 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mercurial/osutil.py Fri Oct 05 15:01:06 2007 -0700 @@ -0,0 +1,37 @@ +import os, stat + +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): + '''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 + os.sep + names = os.listdir(path) + names.sort() + for fn in names: + st = os.lstat(prefix + fn) + if stat: + result.append((fn, _mode_to_kind(st.st_mode), st)) + else: + result.append((fn, _mode_to_kind(st.st_mode))) + return result