U
    >vf                     @  s   d Z ddlmZ ddlmZ ddlZddlZddlZddlZddl	m
Z
 ddlmZ eejfddd	d
ddZejfddddddZddddddZdddddddZdS )zx
Helpers for configuring locale settings.

Name `localization` is chosen to avoid overlap with builtin `locale` module.
    )annotations)contextmanagerN)	Generator)optionszstr | tuple[str, str]intz,Generator[str | tuple[str, str], None, None])
new_localelc_varreturnc              
   c  sb   t |}zDt ||  t  \}}|dk	rF|dk	rF| d| V  n| V  W 5 t || X dS )aE  
    Context manager for temporarily setting a locale.

    Parameters
    ----------
    new_locale : str or tuple
        A string of the form <language_country>.<encoding>. 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)r   r   Zcurrent_localeZnormalized_codeZnormalized_encoding r   ?/tmp/pip-unpacked-wheel-vdrwu74i/pandas/_config/localization.py
set_locale   s    

r   strbool)lcr   r	   c              	   C  sB   zt | |d W 5 Q R X W n ttjfk
r8   Y dS X dS dS )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
    )r   FTN)r   
ValueErrorr   Error)r   r   r   r   r   can_set_locale6   s    r   zlist[str] | strz	list[str])locales	normalizer	   c                   s   dd  fdd| D D S )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.
    c                 S  s   g | ]}t |r|qS r   )r   .0locr   r   r   
<listcomp>b   s   z"_valid_locales.<locals>.<listcomp>c                 3  s(   | ] } rt | n| V  qd S )N)r   r   stripr   r   r   r   	<genexpr>d   s   z!_valid_locales.<locals>.<genexpr>r   )r   r   r   r   r   _valid_localesQ   s
    
r    Tz
str | None)prefixr   r	   c              
   C  s   t  dkrtddg}ng S z^|d}g }|D ]F}z|t|tjj	d W q4 t
k
rx   |t|dd Y q4X q4W n tk
r   Y nX | dkrt||S t|  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)

    )LinuxDarwinr   z-a   
)encodingzwindows-1252Nz.*
)platformsystem
subprocesscheck_outputsplitappendr   r   Zdisplayr%   UnicodeError	TypeErrorr    recompilefindalljoin)r!   r   Zraw_localesZsplit_raw_localesZout_localesxpatternfoundr   r   r   get_localesl   s$    

r6   )NT)__doc__
__future__r   
contextlibr   r   r'   r/   r)   typingr   Zpandas._config.configr   LC_ALLr   r   r    r6   r   r   r   r   <module>   s    #  