fix bugs

arihant2math · arihant2math · commit 679af3f990fb · 2025-04-21T12:29:07.000-07:00
diff --git a/vm/src/stdlib/nt.rs b/vm/src/stdlib/nt.rs
@@ -427,14 +427,8 @@ pub(crate) mod module {
     #[pyfunction]
     fn listvolumes(vm: &VirtualMachine) -> PyResult<PyListRef> {
         let mut volumes = vec![];
-        let find;
         let mut buffer = [0u16; 257];
-        find = unsafe {
-            FileSystem::FindFirstVolumeW(
-                buffer.as_mut_ptr(),
-                buffer.len() as _,
-            )
-        };
+        let find = unsafe { FileSystem::FindFirstVolumeW(buffer.as_mut_ptr(), buffer.len() as _) };
         if find == windows_sys::Win32::Foundation::INVALID_HANDLE_VALUE {
             return Err(errno_err(vm));
         }
@@ -447,11 +441,7 @@ pub(crate) mod module {
             }
             volumes.push(s.to_string());
             let ret = unsafe {
-                FileSystem::FindNextVolumeW(
-                    find,
-                    buffer.as_mut_ptr(),
-                    buffer.len() as _,
-                )
+                FileSystem::FindNextVolumeW(find, buffer.as_mut_ptr(), buffer.len() as _)
             };
             if ret == 0 {
                 err = std::io::Error::last_os_error().raw_os_error().unwrap_or(0);
@@ -463,10 +453,7 @@ pub(crate) mod module {
         if err != 0 && err != windows_sys::Win32::Foundation::ERROR_NO_MORE_FILES as i32 {
             return Err(std::io::Error::from_raw_os_error(err).to_pyexception(vm));
         }
-        let volumes: Vec<_> = volumes
-            .into_iter()
-            .map(|v| vm.new_pyobj(v))
-            .collect();
+        let volumes: Vec<_> = volumes.into_iter().map(|v| vm.new_pyobj(v)).collect();
         Ok(vm.ctx.new_list(volumes))
     }
 
@@ -481,8 +468,16 @@ pub(crate) mod module {
         if path.is_dir() {
             return Ok(false);
         }
-        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
-        Ok(metadata.is_file())
+        let metadata = fs::metadata(path);
+        match metadata {
+            Ok(meta) => Ok(meta.is_file()),
+            Err(err) => {
+                if err.kind() == io::ErrorKind::NotFound {
+                    return Ok(false);
+                }
+                Err(err.to_pyexception(vm))
+            }
+        }
     }
 
     #[pyfunction]
@@ -494,8 +489,16 @@ pub(crate) mod module {
         if path.exists() {
             return Ok(false);
         }
-        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
-        Ok(metadata.is_dir())
+        let metadata = fs::metadata(path);
+        match metadata {
+            Ok(meta) => Ok(meta.is_dir()),
+            Err(err) => {
+                if err.kind() == io::ErrorKind::NotFound {
+                    return Ok(false);
+                }
+                Err(err.to_pyexception(vm))
+            }
+        }
     }
 
     #[pyfunction]
@@ -507,8 +510,16 @@ pub(crate) mod module {
         if path.exists() {
             return Ok(false);
         }
-        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
-        Ok(metadata.is_file())
+        let metadata = fs::metadata(path);
+        match metadata {
+            Ok(meta) => Ok(meta.is_file()),
+            Err(err) => {
+                if err.kind() == io::ErrorKind::NotFound {
+                    return Ok(false);
+                }
+                Err(err.to_pyexception(vm))
+            }
+        }
     }
 
     #[pyfunction]
@@ -520,8 +531,16 @@ pub(crate) mod module {
         if path.exists() {
             return Ok(false);
         }
-        let metadata = fs::symlink_metadata(path).map_err(|err| err.to_pyexception(vm))?;
-        Ok(metadata.file_type().is_symlink())
+        let metadata = fs::symlink_metadata(path);
+        match metadata {
+            Ok(meta) => Ok(meta.file_type().is_symlink()),
+            Err(err) => {
+                if err.kind() == io::ErrorKind::NotFound {
+                    return Ok(false);
+                }
+                Err(err.to_pyexception(vm))
+            }
+        }
     }
 
     // End of functions that are not fully compatible with CPython
