Python 3.4+:扩展pathlib.Path

Jos*_*hua 7 python path python-3.x pathlib

下面的代码是我首先尝试的,但some_path.with_suffix('.jpg')显然返回一个pathlib.PosixPath对象(我在Linux上)而不是我的版本PosixPath,因为我没有重新定义with_suffix.我是否必须复制所有内容pathlib或有更好的方法吗?

import os
import pathlib
from shutil import rmtree


class Path(pathlib.Path):

    def __new__(cls, *args, **kwargs):
        if cls is Path:
            cls = WindowsPath if os.name == 'nt' else PosixPath
        self = cls._from_parts(args, init=False)
        if not self._flavour.is_supported:
            raise NotImplementedError("cannot instantiate %r on your system"
                                      % (cls.__name__,))
        self._init()
        return self

    def with_stem(self, stem):
        """
        Return a new path with the stem changed.

        The stem is the final path component, minus its last suffix.
        """
        if not self.name:
            raise ValueError("%r has an empty name" % (self,))
        return self._from_parsed_parts(self._drv, self._root,
                                       self._parts[:-1] + [stem + self.suffix])

    def rmtree(self, ignore_errors=False, onerror=None):
        """
        Delete the entire directory even if it contains directories / files.
        """
        rmtree(str(self), ignore_errors, onerror)


class PosixPath(Path, pathlib.PurePosixPath):
    __slots__ = ()


class WindowsPath(Path, pathlib.PureWindowsPath):
    __slots__ = ()
Run Code Online (Sandbox Code Playgroud)

ZZY*_*ZZY 3

some_path您的版本的实例吗Path

我在您的代码中附加了以下两行进行了测试:

p = Path('test.foo')
print(type(p.with_suffix('.bar')))
Run Code Online (Sandbox Code Playgroud)

结果正确:<class '__main__.PosixPath'>

仅当使用 时p = pathlib.Path('test.foo'),结果为<class 'pathlib.PosixPath'>