Function setlocale

crates/stdlib/src/locale.rs:262–296

(args: LocaleArgs, vm: &VirtualMachine)

    #[pyfunction]
    fn setlocale(args: LocaleArgs, vm: &VirtualMachine) -> PyResult {
        let error = error(vm);
        if cfg!(windows) && (args.category < LC_ALL || args.category > LC_TIME) {
            return Err(vm.new_exception_msg(error, "unsupported locale setting".into()));
        }
        unsafe {
            let result = match args.locale.flatten() {
                None => libc::setlocale(args.category, ptr::null()),
                Some(locale) => {
                    let locale_str = locale.as_str();
                    // On Windows, validate encoding name length
                    #[cfg(windows)]
                    {
                        let valid = if args.category == LC_ALL {
                            check_locale_name_all(locale_str)
                        } else {
                            check_locale_name(locale_str)
                        };
                        if !valid {
                            return Err(
                                vm.new_exception_msg(error, "unsupported locale setting".into())
                            );
                        }
                    }
                    let c_locale: CString =
                        CString::new(locale_str).map_err(|e| e.to_pyexception(vm))?;
                    libc::setlocale(args.category, c_locale.as_ptr())
                }
            };
            if result.is_null() {
                return Err(vm.new_exception_msg(error, "unsupported locale setting".into()));
            }
            pystr_from_raw_cstr(vm, result)
        }
    }

    /// Get the current locale encoding.
    #[pyfunction]