diff -r 59a52a9dd0dc Lib/os.py --- a/Lib/os.py Thu Jun 02 15:06:09 2016 -0700 +++ b/Lib/os.py Thu Jun 02 16:03:49 2016 -0700 @@ -877,29 +877,35 @@ def fsencode(filename): """ - Encode filename to the filesystem encoding with 'surrogateescape' error - handler, return bytes unchanged. On Windows, use 'strict' error handler if - the file system encoding is 'mbcs' (which is the default encoding). + Encode filename (an os.PathLike, bytes, or str) to the filesystem + encoding with 'surrogateescape' error handler, return bytes unchanged. + On Windows, use 'strict' error handler if the file system encoding is + 'mbcs' (which is the default encoding). """ + filename = fspath(filename) if isinstance(filename, bytes): return filename elif isinstance(filename, str): return filename.encode(encoding, errors) else: - raise TypeError("expect bytes or str, not %s" % type(filename).__name__) + raise TypeError("expected str, bytes or os.PathLike object, not " + + path_type.__name__) def fsdecode(filename): """ - Decode filename from the filesystem encoding with 'surrogateescape' error - handler, return str unchanged. On Windows, use 'strict' error handler if - the file system encoding is 'mbcs' (which is the default encoding). + Decode filename (an os.PathLike, bytes, or str) from the filesystem + encoding with 'surrogateescape' error handler, return str unchanged. On + Windows, use 'strict' error handler if the file system encoding is + 'mbcs' (which is the default encoding). """ + filename = fspath(filename) if isinstance(filename, str): return filename elif isinstance(filename, bytes): return filename.decode(encoding, errors) else: - raise TypeError("expect bytes or str, not %s" % type(filename).__name__) + raise TypeError("expected str, bytes or os.PathLike object, not " + + path_type.__name__) return fsencode, fsdecode diff -r 59a52a9dd0dc Lib/test/test_os.py --- a/Lib/test/test_os.py Thu Jun 02 15:06:09 2016 -0700 +++ b/Lib/test/test_os.py Thu Jun 02 16:03:49 2016 -0700 @@ -3106,6 +3106,21 @@ for s in 'hello', 'goodbye', 'some/path/and/file': self.assertEqual(s, os.fspath(s)) + def test_return_pathlike(self): + class Pathlike: + def __init__(self, path): + self.path = path + + def __fspath__(self): + return self.path + + for p in "path/like/object", b"path/like/object": + pathlike = Pathlike(p) + + self.assertEqual(p, os.fspath(pathlike)) + self.assertEqual(b"path/like/object", os.fsencode(pathlike)) + self.assertEqual("path/like/object", os.fsdecode(pathlike)) + def test_garbage_in_exception_out(self): vapor = type('blah', (), {}) for o in int, type, os, vapor():