U md@sdZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z eej fddd d d d Zej fd dddddZddddddZdddddddZdS)zx Helpers for configuring locale settings. Name `localization` is chosen to avoid overlap with builtin `locale` module. ) annotations)contextmanagerN) Generator)optionszstr | tuple[str, str]intz,Generator[str | tuple[str, str], None, None]) new_localelc_varreturnc csbt|}zDt||t\}}|dk rF|dk rF|d|Vn|VW5t||XdS)aE Context manager for temporarily setting a locale. Parameters ---------- new_locale : str or tuple A string of the form .. For example to set the current locale to US English with a UTF8 encoding, you would pass "en_US.UTF-8". lc_var : int, default `locale.LC_ALL` The category of the locale being set. Notes ----- This is useful when you want to run a particular block of code under a particular locale, without globally setting the locale. This probably isn't thread-safe. N.)locale setlocale getlocale)rrZcurrent_localeZnormalized_codeZnormalized_encodingrT/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/_config/localization.py set_locales    rstrbool)lcrr c CsBzt||dW5QRXWnttjfk r8YdSXdSdS)aj Check to see if we can set a locale, and subsequently get the locale, without raising an Exception. Parameters ---------- lc : str The locale to attempt to set. lc_var : int, default `locale.LC_ALL` The category of the locale being set. Returns ------- bool Whether the passed locale can be set )rFTN)r ValueErrorr Error)rrrrrcan_set_locale6s rzlist[str] | strz list[str])locales normalizer csddfdd|DDS)ar Return a list of normalized locales that do not throw an ``Exception`` when set. Parameters ---------- locales : str A string where each locale is separated by a newline. normalize : bool Whether to call ``locale.normalize`` on each locale. Returns ------- valid_locales : list A list of valid locales. cSsg|]}t|r|qSr)r.0locrrr bsz"_valid_locales..c3s(|] }rt|n|VqdS)N)r rstriprrrr dsz!_valid_locales..r)rrrrr_valid_localesQs  r Tz str | None)prefixrr c Cstdkrtddg}ngSz^|d}g}|D]F}z|t|tjj dWq4t k rx|t|ddYq4Xq4Wnt k rYnX|dkrt ||St |d}|d |}t ||S) a) Get all the locales that are available on the system. Parameters ---------- prefix : str If not ``None`` then return only those locales with the prefix provided. For example to get all English language locales (those that start with ``"en"``), pass ``prefix="en"``. normalize : bool Call ``locale.normalize`` on the resulting list of available locales. If ``True``, only locales that can be set without throwing an ``Exception`` are returned. Returns ------- locales : list of strings A list of locale strings that can be set with ``locale.setlocale()``. For example:: locale.setlocale(locale.LC_ALL, locale_string) On error will return an empty list (no locale available, e.g. Windows) )LinuxDarwinr z-a )encodingz windows-1252Nz.* )platformsystem subprocess check_outputsplitappendrrdisplayr% UnicodeError TypeErrorr recompilefindalljoin)r!rZ raw_localesZsplit_raw_localesZ out_localesxpatternfoundrrr get_localesls$   r7)NT)__doc__ __future__r contextlibrr r'r0r)typingrZpandas._config.configrLC_ALLrrr r7rrrrs     #