3
���h� � @ s� d Z ddlmZ ddlZddlZddlZddlmZ eejfe d�dd��Z
ejfe e e d�d d
�Z
d d � Zd
d� Zddefdd�ZdS )zx
Helpers for configuring locale settings.
Name `localization` is chosen to avoid overlap with builtin `locale` module.
� )�contextmanagerN)�options)�lc_varc c sZ t j� }z>t j|| � t j� }tdd� |D ��r>dj|�V n| V W dt j||� X dS )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.
c s s | ]}|d k V qd S )N� )�.0�xr r �=/tmp/pip-build-5_djhm0z/pandas/pandas/_config/localization.py� <genexpr>' s z set_locale.<locals>.<genexpr>�.N)�locale� getlocale� setlocale�all�join)Z
new_localer Zcurrent_localeZnormalized_localer r r �
set_locale s
r )�lcr �returnc C s@ y t | |d�� W dQ R X W n ttjfk
r6 dS X dS dS )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 NFT)r �
ValueErrorr �Error)r r r r r �can_set_locale/ s r c C s* |rdd� }ndd� }t ttt|| ���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 t j| j� �S )N)r � normalize�strip)r r r r �<lambda>\ s z _valid_locales.<locals>.<lambda>c S s | j � S )N)r )r r r r r ^ s )�list�filterr �map)Zlocalesr Z
normalizerr r r �_valid_localesJ s
r c C s t jdgdd�} | S )Nz locale -aT)�shell)�
subprocess� check_output)� raw_localesr r r �_default_locale_getterc s r! Tc C s� y
|� }W n t jk
r dS X yb|jd�}g }xN|D ]F}y|jt|tjjd�� W q8 tk
r| |jt|dd�� Y q8X q8W W n t k
r� Y nX | dkr�t
||�S t j | � d��}|j
dj|��}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.
locale_getter : callable
The function to use to retrieve the current locales. This should return
a string with each locale separated by a newline character.
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 None (no locale available, e.g. Windows)
N�
)�encodingz windows-1252z.*�
)r �CalledProcessError�split�append�strr �displayr# � UnicodeError� TypeErrorr �re�compile�findallr )�prefixr Z
locale_getterr Z out_localesr �pattern�foundr r r � get_localesh s&
r2 )�__doc__�
contextlibr r r, r Zpandas._config.configr �LC_ALL�intr r( �boolr r r! r2 r r r r �<module> s |