cpython/Lib/test/test_pathlib/support/lexical_path.py

53 lines
1.1 KiB
Python

"""
Simple implementation of JoinablePath, for use in pathlib tests.
"""
import ntpath
import os.path
import posixpath
from . import is_pypi
if is_pypi:
from pathlib_abc import vfspath, _JoinablePath
else:
from pathlib.types import _JoinablePath
from pathlib._os import vfspath
class LexicalPath(_JoinablePath):
__slots__ = ('_segments',)
parser = os.path
def __init__(self, *pathsegments):
self._segments = pathsegments
def __hash__(self):
return hash(vfspath(self))
def __eq__(self, other):
if not isinstance(other, LexicalPath):
return NotImplemented
return vfspath(self) == vfspath(other)
def __vfspath__(self):
if not self._segments:
return ''
return self.parser.join(*self._segments)
def __repr__(self):
return f'{type(self).__name__}({vfspath(self)!r})'
def with_segments(self, *pathsegments):
return type(self)(*pathsegments)
class LexicalPosixPath(LexicalPath):
__slots__ = ()
parser = posixpath
class LexicalWindowsPath(LexicalPath):
__slots__ = ()
parser = ntpath