Add DirEntry.{is_file}

Author: palaviv

tests/snippets/stdlib_os.py

@@ -85,12 +85,16 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 	names = set()
 	paths = set()
 	dirs = set()
+	files = set()
 	for dir_entry in os.scandir(tmpdir):
 		names.add(dir_entry.name)
 		paths.add(dir_entry.path)
 		if dir_entry.is_dir():
 			dirs.add(dir_entry.name)
+		if dir_entry.is_file():
+			files.add(dir_entry.name)
 
 	assert names == set([FILE_NAME, FILE_NAME2, FOLDER])
 	assert paths == set([fname, fname2, folder])
 	assert dirs == set([FOLDER])
+	assert files == set([FILE_NAME, FILE_NAME2])

vm/src/stdlib/os.rs

@@ -222,6 +222,14 @@ impl DirEntryRef {
             .map_err(|s| vm.new_os_error(s.to_string()))?
             .is_dir())
     }
+
+    fn is_file(self, vm: &VirtualMachine) -> PyResult<bool> {
+        Ok(self
+            .entry
+            .file_type()
+            .map_err(|s| vm.new_os_error(s.to_string()))?
+            .is_file())
+    }
 }
 
 #[derive(Debug)]
@@ -284,6 +292,7 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
          "name" => ctx.new_property(DirEntryRef::name),
          "path" => ctx.new_property(DirEntryRef::path),
          "is_dir" => ctx.new_rustfunc(DirEntryRef::is_dir),
+         "is_file" => ctx.new_rustfunc(DirEntryRef::is_file),
     });
 
     py_module!(vm, "_os", {