Mini Shell
3
C��e�f � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z d dl
mZmZmZ d dlmZ d dlmZ ee�Zddd hZG d
d � d e�ZG d d
� d
e�Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Z!dEd d �Z"d d � Z#d!d"� Z$d#d$� Z%dFd%d&�Z&d'd(� Z'd)d*� Z(d+d,� Z)d-d.� Z*d/d0� Z+d1d2� Z,d3d4� Z-dGd5d6�Z.e d7d8d9g�Z/dHd;d<�Z0d=d>� Z1dId?d@�Z2dAdB� Z3dCdD� Z4dS )J� N)�
namedtuple)� constants�util)�upcase_first_letter�setenv�execWithRedirect)�BOSS)�get_module_loggerZLatnZCyrlZGrekc @ s e Zd ZdZdS )�LocalizationConfigErrorz?Exception class for localization configuration related problemsN)�__name__�
__module__� __qualname__�__doc__� r r �$/usr/lib64/python3.6/localization.pyr
' s r
c @ s e Zd ZdZdS )�InvalidLocaleSpecz>Exception class for the errors related to invalid locale specsN)r r r
r r r r r r - s r c C s t j| �}t|j�S )zCheck if the given locale has a language specified.
:return: whether the language or locale is valid
:rtype: bool
)� langtable� parse_locale�bool�language)�langcodeZparsedr r r �is_valid_langcode3 s
r c C s t | �stdj| ���dS )zOHelper to abort when a locale is not valid.
:raise: InvalidLocaleSpec
z'{}' is not a valid localeN)r r �format)�argr r r �raise_on_invalid_locale= s r c C s t j| �jS )z6Return language id without territory or anything else.)r r r )�localer r r �get_language_idF s r c C s t | �}t|�S )a~ Function that tells if the given locale is supported by the Anaconda or
not. We consider locales supported by the langtable as supported by the
Anaconda.
:param locale: locale to test
:type locale: str
:return: whether the given locale is supported or not
:rtype: bool
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)�get_english_namer )r Zen_namer r r �is_supported_localeK s r c C s t | �}t|�jt�S )a� Function that tells if the given locale can be displayed by the Linux console.
The Linux console can display Latin, Cyrillic and Greek characters reliably,
but others such as Japanese, can't be correctly installed.
:param str locale: locale to test
:return: whether the given locale is supported by the console or not
:rtype: bool
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)�get_locale_scripts�set�issubset� SCRIPTS_SUPPORTED_BY_CONSOLE)r Zlocale_scriptsr r r �locale_supported_in_consoleZ s r# c C sX t | � st |� rdS tj| �}tj|�}x&t||�D ]\}}|r8||kr8dS q8W dS )a� Function that tells if the given langcode matches the given locale. I.e. if
all parts of appearing in the langcode (language, territory, script and
encoding) are the same as the matching parts of the locale.
:param langcode: a langcode (e.g. en, en_US, en_US@latin, etc.)
:type langcode: str
:param locale: a valid locale (e.g. en_US.UTF-8 or sr_RS.UTF-8@latin, etc.)
:type locale: str
:return: whether the given langcode matches the given locale or not
:rtype: bool
FT)r r r �zip)r r �langcode_parsed�
locale_parsedZlang_valZloc_valr r r �langcode_matches_localei s
r' c C s� dd� }t | �sdS g }x�|D ]�}t |�s:|j|df� q tj| �}tj|�}||j|jd�||j|jd� ||j|jd� ||j|jd� ||j|jd� }|j||f� q W t |d d
� d d �}|r�|d d dkr�|d d S dS dS )
aB Find the best match for the locale in a list of langcodes. This is useful
when e.g. pt_BR is a locale and there are possibilities to choose an item
(e.g. rnote) for a list containing both pt and pt_BR or even also pt_PT.
:param locale: a valid locale (e.g. en_US.UTF-8 or sr_RS.UTF-8@latin, etc.)
:type locale: str
:param langcodes: a list or generator of langcodes (e.g. en, en_US, en_US@latin, etc.)
:type langcodes: list(str) or generator(str)
:return: the best matching langcode from the list of None if none matches
:rtype: str or None
c S s0 | r |r | |kr|S | S n|r,| r,| S dS )Nr r )Z locale_valueZlangcode_valueZweightr r r �score_value_pair� s
z0find_best_locale_match.<locals>.score_value_pairNr i� �d �
� c S s | d S )Nr+ r )Z
item_scorer r r �<lambda>� s z(find_best_locale_match.<locals>.<lambda>T)�key�reverse)
r �appendr r r � territoryZscriptZvariant�encoding�sorted) r Z langcodesr( Zscoresr r& r% ZscoreZsorted_langcodesr r r �find_best_locale_match� s
> r3 Fc C sN |r|j | � |r�t| �}tjd| t| �� t| �}tjd| |� d}|rj|rjx|D ]}t|�rTd}P qTW |s�tjd| � |s�tjd| � |s�tjd| � |r�|r�tjd| � tjd � tj } | t
j d
< tjd | � t d
| � yt
jt
j| � W nP t
jk
�r@ } z0tjd |� tj } t d
| � t
jt
j| � W Y d
d
}~X nX t| � | S )a� Procedure setting the system to use the given locale and store it in to the
localization module (if given). DOES NOT PERFORM ANY CHECKS OF THE GIVEN
LOCALE.
$LANG must be set by the caller in order to set the language used by gettext.
Doing this in a thread-safe way is up to the caller.
We also try to set a proper console font for the locale in text mode.
If the font for the locale can't be displayed in the Linux console,
we fall back to the English locale.
This function returns the locale that was used in the setlocale call, which,
depending on what the environment and interface is able to support, may be
different from the locale requested.
:param str locale: locale to setup
:param localization_proxy: DBus proxy of the localization module or None
:param bool text_mode: if the locale is being setup for text mode
:return: the locale that was actually set
:rtype: str
z scripts found for locale %s: %sz%console fonts found for locale %s: %sFTz$can't set console font for locale %sz-script not supported by console for locale %sz#no console font found for locale %sz4none of the suggested fonts can be set for locale %sz"falling back to the English locale�LANGzsetting locale to: %szsetlocale failed: %sN)Z SetLanguager# �log�debugr �get_locale_console_fonts�set_console_fontZwarningr � DEFAULT_LANG�os�environr �
locale_mod� setlocale�LC_ALL�Error�set_modules_locale)r Zlocalization_proxy� text_modeZscript_supportedZ
console_fontsZfont_set�font�er r r � setup_locale� sF
rD c C s t j� }|j| � dS )zSet locale of all modules.N)r Z get_proxyZ SetLocale)r Z
boss_proxyr r r r@ s r@ c C s t | � tj| dd�}t|�S )a- Function returning english name for the given locale.
:param locale: locale to return english name for
:type locale: str
:return: english name for the locale or empty string if unknown
:rtype: st
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
�en)�
languageId�languageIdQuery)r r �
language_namer )r �namer r r r s r c C s t | � tj| d�}t|�S )a+ Function returning native name for the given locale.
:param locale: locale to return native name for
:type locale: str
:return: english name for the locale or empty string if unknown
:rtype: st
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r rH r )r rI r r r �get_native_name& s rJ c c sz | pt j} ttj| d �dg �}dd� |D �}t� }x>|D ]6}t|�}|r<||kr<|j|� t|�}|slq<|V q<W dS )z�Method that generates (i.e. returns a generator) available translations
for the installer in the given localedir.
:type localedir: str
:return: generator yielding available translations (languages)
:rtype: generator yielding strings
z/*/LC_MESSAGES/anaconda.mozblob/en/blob/blobc s s | ]}|j tjj�d V qdS )� N���)�splitr: �path�sep)�.0rN r r r � <genexpr>B s z-get_available_translations.<locals>.<genexpr>N)�gettextZ_default_localedirr2 �globr r �add�get_language_locales)Z localedirZ messagefilesZ trans_genZlangsZtrans�lang�localesr r r �get_available_translations5 s
rX c C s t | � tj| d�S )a* Function returning all locales available for the given language.
:param lang: language to get available locales for
:type lang: str
:return: a list of available locales
:rtype: list of strings
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r � list_locales)rV r r r rU R s rU c C s t j| d�S )a; Function returning list of locales for the given territory. The list is
sorted from the most probable locale to the least probable one (based on
langtable's ranking.
:param territory: territory to return locales for
:type territory: str
:return: list of locales
:rtype: list of strings
)Z territoryId)r rY )r0 r r r �get_territory_locales` s
rZ c C s t | � tj| d�S )a Function returning preferred keyboard layouts for the given locale.
:param locale: locale string
:type locale: str
:return: list of preferred keyboard layouts
:rtype: list of strings
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r Zlist_keyboards)r r r r �get_locale_keyboardsm s r[ c C s t | � tj| d�S )a Function returning preferred timezones for the given locale.
:param locale: locale string
:type locale: str
:return: list of preferred timezones
:rtype: list of strings
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r Zlist_timezones)r r r r �get_locale_timezones{ s r\ c C s t | � tj| d�S )a Function returning preferred console fonts for the given locale.
:param str locale: locale string
:return: list of preferred console fonts
:rtype: list of strings
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r Zlist_consolefonts)r r r r r7 � s r7 c C s t | � tj| d�S )a Function returning preferred scripts (writing systems) for the given locale.
:param locale: locale string
:type locale: str
:return: list of preferred scripts
:rtype: list of strings
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
)rF )r r Z list_scripts)r r r r r � s r c C s* t jjdtj�}t|� tj| |d�}|S )a� Function returning translated name of a region, city or complete timezone
name according to the current value of the $LANG variable.
:param tz_spec_part: a region, city or complete timezone name
:type tz_spec_part: str
:return: translated name of the given region, city or timezone
:rtype: str
:raise InvalidLocaleSpec: if an invalid locale is given (see is_valid_langcode)
r4 )rG )r: r; �getr r9 r r Z
timezone_name)Z tz_spec_partr Zxlatedr r r �get_xlated_timezone� s
r^ c C s y&d}t |d��}|j� }W dQ R X W n tk
r: dS X t|�dk rVtjd� dS |dd� }|d dkr|tjd � dS |dd� d
|d d� }|jd
�r�d}n|jd�r�d}|d7 }t|�s�tjd|� dS t|�}|s�tjd|� dS tjd||d � |d S )z�Procedure that returns the firmware language information (if any).
:param boot text_mode: if the locale is being setup for text mode
:return: the firmware language translated into a locale string, or None
:rtype: str
zK/sys/firmware/efi/efivars/PlatformLang-8be4df61-93ca-11d2-aa0d-00e098032b8c�rNr* zPlatformLang was too short� � �-zPlatformLang was malformed�_rK r+ Zzh_chsZzh_HansZzh_chtZzh_Hantz.UTF-8z,PlatformLang was '%s', which is unsupported.z+No locales found for the PlatformLang '%s'.z4Using UEFI PlatformLang '%s' ('%s') as our language.r ���) �open�read�IOError�lenr5 r6 �
startswithr rU )rA �n�f�drW r r r �get_firmware_language� s8
rm Z
DateFieldSpecr �suffixTc sv d}� ��fdd�}t jt j�}| s.d|krD|r8|}n td| ��y||�S tk
rp |sd� n||�S Y nX dS )a. Puts the year, month and day objects in the right order according to the
currently set locale and provides format specification for each of the
fields.
:param year: any object or value representing year
:type year: any
:param month: any object or value representing month
:type month: any
:param day: any object or value representing day
:type day: any
:param bool fail_safe: whether to fall back to default in case of invalid
format or raise exception instead
:returns: a pair where the first field contains a tuple with the year, month
and day objects/values put in the right order and where the second
field contains a tuple with three :class:`_DateFieldSpec` objects
specifying formats respectively to the first (year, month, day)
field, e.g. ((year, month, day), (y_fmt, m_fmt, d_fmt))
:rtype: tuple
:raise ValueError: in case currently set locale has unsupported date
format and fail_safe is set to False
z%Y-%m-%dc
s� t jd�}| jdd�} | jd�dd � }g }g }x�|D ]x}|j|�}|sJq6|j� \}}} |dkrl|j� � n&|dkr�|j�� n|dkr�|j�� |jtd| | | j� �� q6W t |�d
ks�t |�d
kr�t
d| ��t |�t |�fS )Nz([-_0OE^#]*)([yYmbBde])(.*)z%Fz%Y-%m-%d�%r+ rl rC �Y�y�m�b�BrK z9Not all fields successfully identified in the format '%s')rl rC )rp rq )rr rs rt ) �re�compile�replacerM �match�groupsr/ �_DateFieldSpec�striprh �
ValueError�tuple)
�fmt_strZ
field_spec_reZfieldsZ
ordered_termsZordered_formatsZfieldrx �prefix�itemrn )�day�month�yearr r �order_terms_formats s(
z0resolve_date_format.<locals>.order_terms_formatsro z3Invalid date format string for current locale: '%s'N)r< � nl_langinfo�D_FMTr| )r� r� r� Z fail_safeZFAIL_SAFE_DEFAULTr� r~ r )r� r� r� r �resolve_date_format� s & r� c C sD t jd| � td| g�}|dkr0t jd| � dS t jd| � dS dS ) z�Try to set console font to the given value.
:param str font: console font name
:returns: True on success, False on failure
:rtype: bool
zsetting console font to %sZsetfontr z#console font set successfully to %sTz!setting console font to %s failedFN)r5 r6 r �error)rB �rcr r r r8 = s r8 c C s� | s
|r8x,dD ]$}|t jkrt j| rt j| } P qW | sDt|�} | r�yt| �}|d } W n( ttfk
r� tjd| � d} Y nX | s�tj } | t jd< x d D ]}|t jkr�t j|= q�W dS )
aU Clean and configure the local environment variables.
This function will attempt to determine the desired locale and configure
the process environment (os.environ) in the least surprising way. If a
locale argument is provided, it will be attempted first. After that, this
function will attempt to use the language environment variables in a manner
similar to gettext(3) (in order, $LANGUAGE, $LC_ALL, $LC_MESSAGES, $LANG),
followed by the UEFI PlatformLang, followed by a default.
When this function returns, $LANG will be set, and $LANGUAGE, $LC_ALL,
and $LC_MESSAGES will not be set, because they get in the way when changing
the language after startup.
This function must be run before any threads are started. This function
modifies the process environment, which is not thread-safe.
:param str locale: locale to setup if provided
:param bool text_mode: if the locale is being setup for text mode
:param bool prefer_environment: whether the process environment, if available, overrides the locale parameter
:return: None
:rtype: None
�LANGUAGEr> � LC_MESSAGESr4 r zCInvalid locale '%s' given on command line, kickstart or environmentN)r� r> r� r4 )r� r> r� )
r: r; rm rU r �
IndexErrorr5 r� r r9 )r rA Zprefer_environmentZvarnameZ env_langsr r r �setup_locale_environmentN s(
r� c
C s ddgi}| j d�s"tjd� dS | td�d� } tjtjj| d�� tj ddd | |d
d d | g�}|d
krztj
d� dS xltj ddtjj| d�dg�D ]L}t |�}d|kr�|j
dd�\}}n|}||kr�|g||< q�|| j|� q�W tjtjj| d�� |S )z�Get a map of languages and locales from the given OSTree.
For example: {"en": ["en_US"]}
:param repo: the OSTree repository url
:param ref: the name of branch inside the repository
:return: a map of languages and locales
rE Zen_USzfile://z-ostree repo is not local; defaulting to en_USNz tmp/usr/libz/usr/bin/ostreeZcheckoutz--repoz --subpathz /usr/lib/locale/locale-archivez%s/tmp/usr/lib/localer z5failed to check out locale-archive; check program.logz/usr/bin/localedefz--prefixZtmpz--list-archiverc r+ ztmp/usr)ri r5 �inforh r Z
mkdirChainr: rN �joinr r� Z
execReadlines�strip_codeset_and_modifierrM r/ �shutilZrmtree)Zrepo�refZ
locale_mapr� �linerV Z
_territoryr r r �get_locale_map_from_ostree� s2
r� c C s8 d| kr| d| j d�� } d| kr4| d| j d�� } | S )zzReturn a striped version of the given locale.
:param locale: a name of the locale
:return: a stripped locale
�@N�.)�find)r r r r r� � s
r� )NF)N)F)T)NFF)5rR r: ru r� r r r< rS � collectionsr Zpyanaconda.corer r Zpyanaconda.core.utilr r r Z,pyanaconda.modules.common.constants.servicesr Zpyanaconda.anaconda_loggersr r r5 r" � Exceptionr
r r r r r r# r' r3 rD r@ r rJ rX rU rZ r[ r\ r7 r r^ rm rz r� r8 r� r� r� r r r r �<module> sR
7
S
2
R
@2