Merge branch 'move-fspath' of https://github.com/palaviv/RustPython into palaviv-move-fspath

Lib/os.py

@@ -133,34 +133,3 @@ def getenv(key, default=None):
     The optional second argument can specify an alternate default.
     key, default and the result are str."""
     return environ.get(key, default)
-
-
-def fspath(path):
-    """Return the path representation of a path-like object.
-
-    If str or bytes is passed in, it is returned unchanged. Otherwise the
-    os.PathLike interface is used to get the path representation. If the
-    path representation is not str or bytes, TypeError is raised. If the
-    provided path is not str, bytes, or os.PathLike, TypeError is raised.
-    """
-    if isinstance(path, (str, bytes)):
-        return path
-
-    # Work from the object's type to match method resolution of other magic
-    # methods.
-    path_type = type(path)
-    try:
-        path_repr = path_type.__fspath__(path)
-    except AttributeError:
-        if hasattr(path_type, '__fspath__'):
-            raise
-        else:
-            raise TypeError("expected str, bytes or os.PathLike object, "
-                            "not " + path_type.__name__)
-    if isinstance(path_repr, (str, bytes)):
-        return path_repr
-    else:
-        raise TypeError("expected {}.__fspath__() to return str or bytes, "
-                        "not {}".format(path_type.__name__,
-                                        type(path_repr).__name__))
-

tests/snippets/stdlib_os.py

@@ -49,6 +49,10 @@ else:
 	assert os.altsep == None
 	assert os.pathsep == ":"
 
+assert os.fspath("Testing") == "Testing"
+assert os.fspath(b"Testing") == b"Testing"
+assert_raises(TypeError, lambda: os.fspath([1,2,3]))
+
 class TestWithTempDir():
 	def __enter__(self):
 		if os.name == "nt":

vm/src/stdlib/os.rs

@@ -16,7 +16,7 @@ use crate::obj::objset::PySet;
 use crate::obj::objstr::{self, PyString, PyStringRef};
 use crate::obj::objtype::PyClassRef;
 use crate::pyobject::{
-    ItemProtocol, PyClassImpl, PyObjectRef, PyRef, PyResult, PyValue, TryIntoRef,
+    ItemProtocol, PyClassImpl, PyObjectRef, PyRef, PyResult, PyValue, TryIntoRef, TypeProtocol,
 };
 use crate::vm::VirtualMachine;
 
@@ -617,6 +617,19 @@ fn os_chdir(path: PyStringRef, vm: &VirtualMachine) -> PyResult<()> {
     env::set_current_dir(&path.value).map_err(|s| vm.new_os_error(s.to_string()))
 }
 
+fn os_fspath(path: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+    if objtype::issubclass(&path.class(), &vm.ctx.str_type())
+        || objtype::issubclass(&path.class(), &vm.ctx.bytes_type())
+    {
+        Ok(path)
+    } else {
+        Err(vm.new_type_error(format!(
+            "expected str or bytes object, not {}",
+            path.class()
+        )))
+    }
+}
+
 pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
     let ctx = &vm.ctx;
 
@@ -725,6 +738,8 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
         "DirEntry" => dir_entry,
         "stat_result" => stat_result,
         "getcwd" => ctx.new_rustfunc(os_getcwd),
+        "chdir" => ctx.new_rustfunc(os_chdir),
+        "fspath" => ctx.new_rustfunc(os_fspath),
         "O_RDONLY" => ctx.new_int(0),
         "O_WRONLY" => ctx.new_int(1),
         "O_RDWR" => ctx.new_int(2),