implement more nt

arihant2math · arihant2math · commit 15558e027843 · 2025-04-21T11:44:06.000-07:00
diff --git a/vm/src/stdlib/nt.rs b/vm/src/stdlib/nt.rs
@@ -424,6 +424,108 @@ pub(crate) mod module {
         Ok(vm.ctx.new_list(drives))
     }
 
+    #[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 _,
+            )
+        };
+        if find == windows_sys::Win32::Foundation::INVALID_HANDLE_VALUE {
+            return Err(errno_err(vm));
+        }
+        let mut err = 0;
+        while err == 0 {
+            let s = unsafe { widestring::WideCString::from_ptr_str(buffer.as_mut_ptr()) };
+            let s = s.to_string_lossy();
+            if s.is_empty() {
+                break;
+            }
+            volumes.push(s.to_string());
+            let ret = unsafe {
+                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);
+            }
+        }
+        if find != windows_sys::Win32::Foundation::INVALID_HANDLE_VALUE {
+            unsafe { FileSystem::FindVolumeClose(find) };
+        }
+        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();
+        Ok(vm.ctx.new_list(volumes))
+    }
+
+    // TOOD: these functions are not fully compataible with CPython
+    // they exist for some compatibility
+    #[pyfunction]
+    fn _path_exists(path: OsPath, vm: &VirtualMachine) -> PyResult<bool> {
+        let path = path.as_ref();
+        if path.exists() {
+            return Ok(true);
+        }
+        if path.is_dir() {
+            return Ok(false);
+        }
+        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
+        Ok(metadata.is_file())
+    }
+
+    #[pyfunction]
+    fn _path_isdir(path: OsPath, vm: &VirtualMachine) -> PyResult<bool> {
+        let path = path.as_ref();
+        if path.is_dir() {
+            return Ok(true);
+        }
+        if path.exists() {
+            return Ok(false);
+        }
+        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
+        Ok(metadata.is_dir())
+    }
+
+    #[pyfunction]
+    fn _path_isfile(path: OsPath, vm: &VirtualMachine) -> PyResult<bool> {
+        let path = path.as_ref();
+        if path.is_file() {
+            return Ok(true);
+        }
+        if path.exists() {
+            return Ok(false);
+        }
+        let metadata = fs::metadata(path).map_err(|err| err.to_pyexception(vm))?;
+        Ok(metadata.is_file())
+    }
+
+    #[pyfunction]
+    fn _path_islink(path: OsPath, vm: &VirtualMachine) -> PyResult<bool> {
+        let path = path.as_ref();
+        if path.is_symlink() {
+            return Ok(true);
+        }
+        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())
+    }
+
+    // End of functions that are not fully compatible with CPython
+
     #[pyfunction]
     fn set_handle_inheritable(
         handle: intptr_t,
