RustPython · youknowone · Feb 28, 2023 · Feb 24, 2023 · Feb 24, 2023 · Feb 24, 2023
diff --git a/Lib/test/test_locale.py b/Lib/test/test_locale.py
@@ -6,7 +6,7 @@
 import sys
 import codecs
 
-
+@unittest.SkipTest(reason="Library not completed")
 class BaseLocalizedTest(unittest.TestCase):
     #
     # Base class for tests using a real locale

diff --git a/stdlib/src/lib.rs b/stdlib/src/lib.rs
@@ -15,6 +15,7 @@ mod dis;
 mod gc;
 mod hashlib;
 mod json;
+mod locale;
 mod math;
 #[cfg(unix)]
 mod mmap;
@@ -159,5 +160,9 @@ pub fn get_module_inits() -> impl Iterator<Item = (Cow<'static, str>, StdlibInit
         {
             "_uuid" => uuid::make_module,
         }
+        #[cfg(any(target_os = "linux", target_os = "macos"))]
+        {
+            "_locale" => locale::make_module,
+        }
     }
 }
diff --git a/stdlib/src/locale.rs b/stdlib/src/locale.rs
@@ -0,0 +1,117 @@
+#[cfg(any(target_os = "linux", target_os = "macos"))]
+pub(crate) use _locale::make_module;
+
+#[cfg(any(target_os = "linux", target_os = "macos"))]
+#[pymodule]
+mod _locale {
+    extern crate libc;
+    use rustpython_vm::{
+        builtins::{PyDictRef, PyIntRef, PyListRef},
+        PyObjectRef, PyResult, VirtualMachine,
+    };
+
+    #[pyattr]
+    use libc::{
+        ABDAY_1, ABDAY_2, ABDAY_3, ABDAY_4, ABDAY_5, ABDAY_6, ABDAY_7, ABMON_1, ABMON_10, ABMON_11,
+        ABMON_12, ABMON_2, ABMON_3, ABMON_4, ABMON_5, ABMON_6, ABMON_7, ABMON_8, ABMON_9,
+        ALT_DIGITS, AM_STR, CODESET, CRNCYSTR, DAY_1, DAY_2, DAY_3, DAY_4, DAY_5, DAY_6, DAY_7,
+        D_FMT, D_T_FMT, ERA, ERA_D_FMT, ERA_D_T_FMT, ERA_T_FMT, LC_ALL, LC_COLLATE, LC_CTYPE,
+        LC_MESSAGES, LC_MONETARY, LC_NUMERIC, LC_TIME, MON_1, MON_10, MON_11, MON_12, MON_2, MON_3,
+        MON_4, MON_5, MON_6, MON_7, MON_8, MON_9, NOEXPR, PM_STR, RADIXCHAR, THOUSEP, T_FMT,
+        T_FMT_AMPM, YESEXPR,
+    };
+
+    use std::ffi::{c_char, CStr};
+
+    #[pyattr(name = "CHAR_MAX")]
+    fn char_max(vm: &VirtualMachine) -> PyIntRef {
+        vm.ctx.new_int(libc::c_char::MAX)
+    }
+
+    unsafe fn _get_grouping(group: *mut c_char, vm: &VirtualMachine) -> PyListRef {
+        let mut group_vec: Vec<PyObjectRef> = Vec::new();
+        let mut ptr = group;
+
+        while *ptr != (u8::MIN as i8) && *ptr != i8::MAX {
+            let val = vm.ctx.new_int(*ptr);
+            group_vec.push(val.into());
+            ptr = ptr.offset(1);
+        }
+        vm.ctx.new_list(group_vec)
+    }
+
+    unsafe fn _parse_ptr_to_str(vm: &VirtualMachine, raw_ptr: *mut c_char) -> PyResult {
+        let slice = unsafe { CStr::from_ptr(raw_ptr) };
+        let cstr = slice
+            .to_str()
+            .map(|s| s.to_owned())
+            .map_err(|e| vm.new_unicode_decode_error(format!("unable to decode: {e}")))?;
+
+        Ok(vm.new_pyobj(cstr))
+    }
+
+    #[pyfunction]
+    fn localeconv(vm: &VirtualMachine) -> PyResult<PyDictRef> {
+        let result = vm.ctx.new_dict();
+        macro_rules! set_string_field {
+            ($field:expr) => {{
+                result.set_item(stringify!($field), _parse_ptr_to_str(vm, $field)?, vm)?
+            }};
+        }
+
+        macro_rules! set_int_field {
+            ($field:expr) => {{
+                result.set_item(stringify!($field), vm.new_pyobj($field), vm)?
+            }};
+        }
+
+        macro_rules! set_group_field {
+            ($field:expr) => {{
+                result.set_item(stringify!($field), _get_grouping($field, vm).into(), vm)?
+            }};
+        }
+
+        unsafe {

629A
+            let lc = libc::localeconv();
+
+            let mon_grouping = (*lc).mon_grouping;
+            let int_frac_digits = (*lc).int_frac_digits;
+            let frac_digits = (*lc).frac_digits;
+            let p_cs_precedes = (*lc).p_cs_precedes;
+            let p_sep_by_space = (*lc).p_sep_by_space;
+            let n_cs_precedes = (*lc).n_cs_precedes;
+            let p_sign_posn = (*lc).p_sign_posn;
+            let n_sign_posn = (*lc).n_sign_posn;
+            let grouping = (*lc).grouping;
+            let decimal_point = (*lc).decimal_point;
+            let thousands_sep = (*lc).thousands_sep;
+            let int_curr_symbol = (*lc).int_curr_symbol;
+            let currency_symbol = (*lc).currency_symbol;
+            let mon_decimal_point = (*lc).mon_decimal_point;
+            let mon_thousands_sep = (*lc).mon_thousands_sep;
+            let n_sep_by_space = (*lc).n_sep_by_space;
+            let positive_sign = (*lc).positive_sign;
+            let negative_sign = (*lc).negative_sign;
+
+            set_group_field!(mon_grouping);
+            set_group_field!(grouping);
+            set_int_field!(int_frac_digits);
+            set_int_field!(frac_digits);
+            set_int_field!(p_cs_precedes);
+            set_int_field!(p_sep_by_space);
+            set_int_field!(n_cs_precedes);
+            set_int_field!(p_sign_posn);
+            set_int_field!(n_sign_posn);
+            set_string_field!(decimal_point);
+            set_string_field!(thousands_sep);
+            set_string_field!(int_curr_symbol);
+            set_string_field!(currency_symbol);
+            set_string_field!(mon_decimal_point);
+            set_string_field!(mon_thousands_sep);
+            set_int_field!(n_sep_by_space);
+            set_string_field!(positive_sign);
+            set_string_field!(negative_sign);
+        }
+        Ok(result)
+    }
+}