python - 使用 Python os 模块获取 unix 文件类型

Question

我想获取由路径指定的文件的 unix 文件类型(找出它是否是常规文件、命名管道、 block 设备...)

我在文档 os.stat(path).st_type 中找到，但在 Python 3.6 中，这似乎不起作用。

另一种方法是使用 os.DirEntry 对象(例如通过 os.listdir(path))，但只有方法 is_dir()、is_file() 和 is_symlink()。

有什么想法吗？

Answer 1

您使用 stat解释 os.stat(path).st_mode 的结果的模块。

>>> import os
>>> import stat
>>> stat.S_ISDIR(os.stat('/dev/null').st_mode)
False
>>> stat.S_ISCHR(os.stat('/dev/null').st_mode)
True

您可以制作一个通用函数来返回确定的类型。这适用于 Python 2 和 3。

import enum
import os
import stat

class PathType(enum.Enum):
    dir = 0  # directory
    chr = 1  # character special device file
    blk = 2  # block special device file
    reg = 3  # regular file
    fifo = 4  # FIFO (named pipe)
    lnk = 5  # symbolic link
    sock = 6  # socket
    door = 7  # door  (Py 3.4+)
    port = 8  # event port  (Py 3.4+)
    wht = 9  # whiteout (Py 3.4+)

    unknown = 10

    @classmethod
    def get(cls, path):
        if not isinstance(path, int):
            path = os.stat(path).st_mode
        for path_type in cls:
            method = getattr(stat, 'S_IS' + path_type.name.upper())
            if method and method(path):
                return path_type
        return cls.unknown

PathType.__new__ = (lambda cls, path: cls.get(path))

>>> PathType('/dev/null')
<PathType.chr: 1>
>>> PathType('/home')
<PathType.dir: 0>