Implmentation for os.pathconf_names

mmednick · mmednick · commit d8fe7787bf96 · 2023-02-15T17:05:27.000-08:00
diff --git a/vm/src/stdlib/posix.rs b/vm/src/stdlib/posix.rs
@@ -50,8 +50,10 @@ pub mod module {
         ffi::{CStr, CString},
         fs, io,
         os::unix::{ffi as ffi_ext, io::RawFd},
+        str::FromStr,
     };
-    use strum_macros::EnumString;
+    use strum::VariantNames;
+    use strum_macros::{EnumString, EnumVariantNames};
 
     #[pyattr]
     use libc::{PRIO_PGRP, PRIO_PROCESS, PRIO_USER};
@@ -1691,7 +1693,7 @@ pub mod module {
 
     // Copy from [nix::unistd::PathconfVar](https://docs.rs/nix/0.21.0/nix/unistd/enum.PathconfVar.html)
     // Change enum name to fit python doc
-    #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq, EnumString)]
+    #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq, EnumString, EnumVariantNames)]
     #[repr(i32)]
     #[allow(non_camel_case_types)]
     pub enum PathconfVar {
@@ -1891,6 +1893,22 @@ pub mod module {
         pathconf(PathOrFd::Fd(fd), name, vm)
     }
 
+    #[pyattr]
+    fn pathconf_names(vm: &VirtualMachine) -> PyDictRef {
+        let pathname = vm.ctx.new_dict();
+        for variant in PathconfVar::VARIANTS {
+            // get the name of variant as a string to use as the dictionary key
+            let key: PyObjectRef = vm.ctx.new_str(variant.to_string()).into();
+            // get the enum from the string and convert it to an integer for the dictionary value
+            let value: PyObjectRef = vm
+                .ctx
+                .new_int(PathconfVar::from_str(variant).unwrap() as u8)
+                .into();
+            pathname.set_item(&*key, value, vm).unwrap();
+        }
+        pathname
+    }
+
     #[cfg(any(target_os = "linux", target_os = "macos"))]
     #[derive(FromArgs)]
     struct SendFileArgs {
