37 lines
978 B
Python
37 lines
978 B
Python
|
"""
|
||
|
|
Python polyfills for os
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
from typing import AnyStr
|
||
|
|
|
||
|
|
from ..decorators import substitute_in_graph
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = ["fspath"]
|
||
|
|
|
||
|
|
|
||
|
|
# Copied from os.py in the standard library
|
||
|
|
@substitute_in_graph(os.fspath, can_constant_fold_through=True)
|
||
|
|
def fspath(path: AnyStr | os.PathLike[AnyStr]) -> AnyStr:
|
||
|
|
if isinstance(path, (str, bytes)):
|
||
|
|
return path
|
||
|
|
|
||
|
|
path_type = type(path)
|
||
|
|
try:
|
||
|
|
path_repr = path_type.__fspath__(path) # type: ignore[arg-type]
|
||
|
|
except AttributeError:
|
||
|
|
if hasattr(path_type, "__fspath__"):
|
||
|
|
raise
|
||
|
|
raise TypeError(
|
||
|
|
f"expected str, bytes or os.PathLike object, not {path_type.__name__}",
|
||
|
|
) from None
|
||
|
|
if isinstance(path_repr, (str, bytes)):
|
||
|
|
return path_repr # type: ignore[return-value]
|
||
|
|
raise TypeError(
|
||
|
|
f"expected {path_type.__name__}.__fspath__() to return str or bytes, "
|
||
|
|
f"not {type(path_repr).__name__}",
|
||
|
|
)
|