3
��g�` � @ s� d dl mZ d dl mZ d dl mZ d dlZd dlmZ d dlm Z m
Z
d dl m Z d dlZ
d dlZ
d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZejd�ZejjZ ejj Z ejj Z ejj!Z"e e B e B e"B Z#d d
� Z$d d � Z%d
d� Z&dd� Z'd*dd�Z(G dd� de)�Z*G dd� de)�Z+G dd� de)�Z,G dd� de,�Z-G d d � d e,�Z.G d d � d e,�Z/G d d!� d!e,�Z0G d"d#� d#e)�Z1G d$d%� d%e)�Z2G d&d'� d'e)�Z3G d(d)� d)e)�Z4dS )+� )�absolute_import)�print_function)�unicode_literalsN)�
CompsError)�_�ucd)�reduce�dnfc C s" | j | j| jf}ttjtt|��S )N)�
categories�groups� environmentsr �operator�__add__�map�len)�comps� collections� r �/usr/lib/python3.6/comps.py�_internal_comps_length6 s r c C s | d kr d S t jj| �S )N)r �util�first)�seqr r r �_first_if_iterable; s r c s� t jj� �� � fdd�|D �}|r&|S |r>tjtj� ��j}ntjtj� �tjd�j}t � }x`|D ]X}||j
�r||j |� qb|j dk r�||j �r�|j |� qb|j
dk rb||j
�rb|j |� qbW |S )z;Return items from sqn matching either exactly or glob-wise.c s$ h | ] }|j � ks |j� kr|�qS r )�name�id)�.0�g)�patternr r � E s z _by_pattern..)�flagsN)r Zi18nr �re�compile�fnmatch� translate�match�I�setr �addr �ui_name)r �case_sensitiveZsqn�exactr% �retr r )r r � _by_patternA s
r- c C s | j d krtjS | j S )N)Z
display_order�sys�maxsize)�groupr r r �_fn_display_orderZ s r1 Tc C s | |||||�S )aF
Installs a group or an environment identified by grp_or_env_id.
This method is preserved for API compatibility. It used to catch an
exception thrown when a gorup or env was already installed, which is no
longer thrown.
`install_fnc` has to be Solver._group_install or
Solver._environment_install.
r )Z install_fncZ
grp_or_env_id�types�exclude�strict�exclude_groupsr r r �install_or_skip^ s
r6 c @ s, e Zd ZdZdd� Zedd� �Zdd� ZdS ) �_Langsz6Get all usable abbreviations for the current language.c C s d | _ d | _d S )N)� last_locale�cache)�selfr r r �__init__o s z_Langs.__init__c C s" t jt j�} | dkrdS dj| �S )N�C�.)NN)�localeZ getlocale� LC_MESSAGES�join)Zlclr r r �_dotted_locale_strs s z_Langs._dotted_locale_strc C sz | j � }| j|kr| jS g | _|g}|dkr6|jd� x6|D ].}x(tj|�D ]}|| jkrL| jj|� qLW q.)r' rP � AVAILABLE�update� INSTALLEDZgetCompsGroupItemr( �
getGroupId)r: � available� installed�resultrQ r0 r r r � _get_groups� s
zCompsQuery._get_groupsc C s` t � }| j| j@ r&|jdd� |D �� | j| j@ r\x(|D ] }|j� }|sJq8|j|j� � q8W |S )Nc S s h | ]
}|j �qS r )r )r rQ r r r r � s z'CompsQuery._get_envs..)r' rP rR rS rT ZgetCompsEnvironmentItemr( ZgetEnvironmentId)r: rV rW rX rQ �envr r r � _get_envs� s
zCompsQuery._get_envsc G s t jj� }g |_g |_x�|D ]�}g }}| j| j@ rf| jj|�}| j j
j |�}| j ||�}|jj
|� | j| j@ r�| jj|�}| j jj |�}| j||�}|jj
|� | o�| r | j| jkr�td�t|� }n.| j| jkr�td�t|� }ntd�t|� }t|��q W |S )Nz&Module or Group '%s' is not installed.z&Module or Group '%s' is not available.z$Module or Group '%s' does not exist.)r r ZBunchr r rO � ENVIRONMENTSr �environments_by_patternrN rZ Zsearch_by_patternr[ �extend�GROUPS�groups_by_patternr0 rY rP rT r r rR r ) r: Zpatterns�resZpat�envs�grpsrV rW �msgr r r rE � s.
zCompsQuery.getN) rF rG rH rR rT r\ r_ r; rY r[ rE r r r r rK � s rK c @ s< e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Zd S ) � Forwarderc C s || _ || _d S )N)�_i�_langs)r: �iobj�langsr r r r; � s zForwarder.__init__c C s t | j|�S )N)�getattrrf )r: r r r r � __getattr__� s zForwarder.__getattr__c C s. x(| j j� D ]}|j|�}|d k r |S q W |S )N)rg rE )r: �defaultZdctrD �tr r r �_ui_text� s
zForwarder._ui_textc C s | j | j| j�S )N)rn ZdescZ desc_by_lang)r: r r r �ui_description� s zForwarder.ui_descriptionc C s | j | j| j�S )N)rn r Z name_by_lang)r: r r r r) � s zForwarder.ui_nameN) rF rG rH r; rk rn �propertyro r) r r r r re � s
re c s8 e Zd Z� fdd�Zdd� Zdd� Zedd� �Z� ZS ) �Categoryc s t t| �j||� || _d S )N)�superrq r; �_group_factory)r: rh ri �
group_factory)� __class__r r r; � s zCategory.__init__c C s0 | j |j�}|d kr,d}t||j| jf ��|S )Nz no group '%s' from category '%s')rs r �
ValueErrorr )r: �grp_id�grprd r r r � _build_group� s
zCategory._build_groupc c s x| j D ]}| j|�V qW d S )N)� group_idsry )r: rw r r r � groups_iter� s zCategory.groups_iterc C s t | j� �S )N)�listr{ )r: r r r r � s zCategory.groups) rF rG rH r; ry r{ rp r �
__classcell__r r )ru r rq � s rq c sL e Zd Z� fdd�Zdd� Zdd� Zdd� Zed d
� �Zed d � �Z � Z
S )
� Environmentc s t t| �j||� || _d S )N)rr r~ r; rs )r: rh ri rt )ru r r r; � s zEnvironment.__init__c C s0 | j |j�}|d kr,d}t||j| jf ��|S )Nz#no group '%s' from environment '%s')rs r rv r )r: rw rx rd r r r ry s
zEnvironment._build_groupc C sX g }xN|D ]F}y|j | j|�� W q
tk
rN } ztj|� W Y d d }~X q
X q
W |S )N)rB ry rv �logger�error)r: Zidsr Zgi�er r r �
_build_groups s
zEnvironment._build_groupsc c s\ xVt j| j| j�D ]B}y| j|�V W q tk
rR } ztj|� W Y d d }~X qX qW d S )N)� itertools�chainrz �
option_idsry rv r r� )r: rw r� r r r r{ s
zEnvironment.groups_iterc C s | j | j�S )N)r� rz )r: r r r �mandatory_groups s z Environment.mandatory_groupsc C s | j | j�S )N)r� r� )r: r r r �optional_groups s zEnvironment.optional_groups) rF rG rH r; ry r� r{ rp r� r� r} r r )ru r r~ � s
r~ c sh e Zd Z� fdd�Zdd� Zedd� �Zedd� �Zd d
� Zed d � �Z ed
d� �Z
edd� �Z � Z S )�Groupc s$ t t| �j||� || _|j| _d S )N)rr r� r; � _pkg_factoryrl Zselected)r: rh ri Z pkg_factory)ru r r r; " s zGroup.__init__c s � fdd�| j D �S )Nc s g | ]}|j � kr|�qS r )�type)r �pkg)�type_r r �
( s z+Group._packages_of_type..)�packages)r: r� r )r� r �_packages_of_type' s zGroup._packages_of_typec C s | j tj�S )N)r� �libcomps�PACKAGE_TYPE_CONDITIONAL)r: r r r �conditional_packages* s zGroup.conditional_packagesc C s | j tj�S )N)r� r� �PACKAGE_TYPE_DEFAULT)r: r r r �default_packages. s zGroup.default_packagesc C s t | j| j�S )N)r r� r� )r: r r r �
packages_iter2 s zGroup.packages_iterc C s | j tj�S )N)r� r� �PACKAGE_TYPE_MANDATORY)r: r r r �mandatory_packages6 s zGroup.mandatory_packagesc C s | j tj�S )N)r� r� �PACKAGE_TYPE_OPTIONAL)r: r r r �optional_packages: s zGroup.optional_packagesc C s | j jS )N)rf Z uservisible)r: r r r �visible> s z
Group.visible)
rF rG rH r; r� rp r� r� r� r� r� r� r} r r )ru r r� s r� c @ sL e Zd ZdZejeejeej e
ej e iZ
dd� Zedd� �Zedd� �ZdS ) �Packagez#Represents comps package data. :apic C s
|| _ d S )N)rf )r: �ipkgr r r r; L s zPackage.__init__c C s | j jS )N)rf r )r: r r r r O s z Package.namec C s | j | j S )N)�_OPT_MAPr� )r: r r r � option_typeT s zPackage.option_typeN)rF rG rH rI r� r� � CONDITIONALr� �DEFAULTr� � MANDATORYr� �OPTIONALr� r; rp r r� r r r r r� B s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
dd� �Z d-dd�Z d.dd�Z
dd� Ze
dd� �Zdd� Zd/d d �Zd0d d �Zd d!� Ze
d"d#� �Zd$d%� Zd1d&d'�Zd2d(d)�Zd*d+� Zd,S )3�Compsc C s t j� | _t� | _d S )N)r� r� rf r7 rg )r: r r r r; \ s
zComps.__init__c C s
t | j�S )N)r rf )r: r r r �__len__` s z
Comps.__len__c C s t || j| j�S )N)rq rg � _group_by_id)r: Z icategoryr r r �_build_categoryc s zComps._build_categoryc C s t || j| j�S )N)r~ rg r� )r: Z ienvironmentr r r �_build_environmentf s zComps._build_environmentc C s t || j| j�S )N)r� rg �_build_package)r: Zigroupr r r ry i s zComps._build_groupc C s t |�S )N)r� )r: r� r r r r� l s zComps._build_packagec C sV t j� }y|j|� W n, t jk
rB |j� }tdj|���Y nX | j|7 _d S )N� )r� r� Z fromxml_fZ ParserErrorZget_last_errorsr r@ rf )r: �fnr �errorsr r r �_add_from_xml_filenameo s z Comps._add_from_xml_filenamec C s t | j� �S )N)r| �categories_iter)r: r r r r
x s zComps.categoriesFc C s | j ||�}t|�S )N)�categories_by_patternr )r: r r* Zcatsr r r �category_by_pattern} s zComps.category_by_patternc C s t ||| j�S )N)r- r
)r: r r* r r r r� � s zComps.categories_by_patternc s � fdd�� j jD �S )Nc 3 s | ]}� j |�V qd S )N)r� )r �c)r: r r � � s z(Comps.categories_iter..)rf r
)r: r )r: r r� � s zComps.categories_iterc C s t | j� td�S )N)�key)�sorted�environments_iterr1 )r: r r r r � s zComps.environmentsc s t jj� fdd�| j� D ��S )Nc 3 s | ]}|j � kr|V qd S )N)r )r r )r r r r� � s z+Comps._environment_by_id..)r r r r� )r: r r )r r �_environment_by_id� s zComps._environment_by_idc C s | j ||�}t|�S )N)r] r )r: r r* rb r r r �environment_by_pattern� s z Comps.environment_by_patternc C s$ t | j� �}t|||�}t|td�S )N)r� )r| r� r- r� r1 )r: r r* rb Z
found_envsr r r r] � s z Comps.environments_by_patternc s � fdd�� j jD �S )Nc 3 s | ]}� j |�V qd S )N)r� )r r� )r: r r r� � s z*Comps.environments_iter..)rf r )r: r )r: r r� � s zComps.environments_iterc C s t | j� td�S )N)r� )r� r{ r1 )r: r r r r � s z Comps.groupsc s t jj� fdd�| j� D ��S )Nc 3 s | ]}|j � kr|V qd S )N)r )r r )�id_r r r� � s z%Comps._group_by_id..)r r r r{ )r: r� r )r� r r� � s zComps._group_by_idc C s | j ||�}t|�S )N)r` r )r: r r* rc r r r �group_by_pattern� s zComps.group_by_patternc C s t ||t| j� ��}t|td�S )N)r� )r- r| r{ r� r1 )r: r r* rc r r r r` � s zComps.groups_by_patternc s � fdd�� j jD �S )Nc 3 s | ]}� j |�V qd S )N)ry )r r )r: r r r� � s z$Comps.groups_iter..)rf r )r: r )r: r r{ � s zComps.groups_iterN)F)F)F)F)F)F)rF rG rH r; r� r� r� ry r� r� rp r
r� r� r� r r� r� r] r� r r� r� r` r{ r r r r r� Y s*
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
CompsTransPkgc C s� t jj|�r&d| _|| _d| _d | _n\t|tj j
�r\d| _|j � | _|j � t
j@ | _d | _n&|j| _|j| _|jt
j@ | _|j| _d S )NFT)r r Zis_string_type� basearchonlyr �optional�requires�
isinstance�libdnf� transactionZCompsGroupPackage�getNameZgetPackageTyper� r� r� )r: Z pkg_or_namer r r r; � s
zCompsTransPkg.__init__c C s0 | j |j ko.| j| jko.| j| jko.| j| jkS )N)r r� r� r� )r: �otherr r r �__eq__� s zCompsTransPkg.__eq__c C s | j S )N)r )r: r r r �__str__� s zCompsTransPkg.__str__c C s t | j| j| j| jf�S )N)�hashr r� r� r� )r: r r r �__hash__� s zCompsTransPkg.__hash__N)rF rG rH r; r� r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Z e j
d d
� �Z ed d
� �Z e j
dd
� �Z edd� �Z e j
dd� �Z edd� �Z
e
j
dd� �Z
dS )�TransactionBunchc C s$ t � | _t � | _t � | _t � | _d S )N)r' �_install� _install_opt�_remove�_upgrade)r: r r r r; � s zTransactionBunch.__init__c C sN | j j|j � | jj|j� | jj|j� | j|jB | j | j | j | _| S )N)r� rS r� r� r� )r: r� r r r �__iadd__� s
zTransactionBunch.__iadd__c C s( t | j�t | j� t | j� t | j� S )N)r �install� install_opt�upgrade�remove)r: r r r r� � s zTransactionBunch.__len__c C s6 x0|D ](}t |t�r | j|� q| jt|�� qW d S )N)r� r� r( )Zparam�val�itemr r r �
_set_value� s
zTransactionBunch._set_valuec C s | j S )z�
Packages to be installed with strict=True - transaction will
fail if they cannot be installed due to dependency errors etc.
)r� )r: r r r r� s zTransactionBunch.installc C s | j | j|� d S )N)r� r� )r: �valuer r r r� s c C s | j S )zw
Packages to be installed with strict=False - they will be
skipped if they cannot be installed
)r� )r: r r r r�
s z TransactionBunch.install_optc C s | j | j|� d S )N)r� r� )r: r� r r r r� s c C s | j S )N)r� )r: r r r r� s zTransactionBunch.removec C s | j | j|� d S )N)r� r� )r: r� r r r r� s c C s | j S )N)r� )r: r r r r� ! s zTransactionBunch.upgradec C s | j | j|� d S )N)r� r� )r: r� r r r r� % s N)rF rG rH r; r� r� rJ r� rp r� �setterr� r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zedd� �Zedd� �Zeg fdd��Zd d
� Zd d � Z d dd�Z
dd� Z dd� Z d dd�Z
dd� Zdd� Zdd � Zd
S ) �Solverc C s || _ || _|| _d S )N)rN r Z
_reason_fn)r: rN r Z reason_fnr r r r; + s zSolver.__init__c C s dd� | j D �S )Nc S s h | ]
}|j �qS r )r )r rx r r r r 2 s z.Solver._mandatory_group_set..)r� )rZ r r r �_mandatory_group_set0 s zSolver._mandatory_group_setc C s" dd� | j | j | j | j D �S )Nc S s h | ] }|j � �qS r )r� )r r� r r r r 6 s z+Solver._full_package_set..)r� r� r� r� )rx r r r �_full_package_set4 s zSolver._full_package_setc sv � fdd�}t � }|t@ r*|j|| j�� |t@ rB|j|| j�� |t@ rZ|j|| j�� |t@ rr|j|| j �� |S )Nc s � fdd�| D �S )Nc s g | ]}|j � kr|�qS r )r )r r� )r3 r r r� = s z8Solver._pkgs_of_type..filter..r )�pkgs)r3 r r �filter< s z$Solver._pkgs_of_type..filter)
r' r� rS r� r� r� r� r� r� r� )r0 � pkg_typesr3 r� r� r )r3 r �
_pkgs_of_type: s zSolver._pkgs_of_typec C s | j jj|�S )N)rN r0 Zis_removable_pkg)r: Zpkg_namer r r �_removable_pkgK s zSolver._removable_pkgc C s | j jj|�S )N)rN rZ Zis_removable_group)r: �group_idr r r �_removable_grpO s zSolver._removable_grpNTc
C s� | j j|�}|s$ttd�t|� ��| jjj||j|j |�}| jjj
|� t � }xD|j D ]:} |rl| j
|krlqX|| j| j
|||�7 }|j| j
dt� qXW x.|jD ]$} |r�| j
|kr�q�|j| j
dt� q�W |S )Nz#Environment id '%s' does not exist.TF)r r� r r r rN rZ �newr r) r� r� r� r �_group_install�addGroupr� r� r� )
r: �env_idr� r3 r4 r5 � comps_env�swdb_env�trans� comps_groupr r r �_environment_installS s zSolver._environment_installc C sx | j jj|�}|s"ttd�| ��| j jj|� t� }tdd� |j� D ��}x&|D ] }| j |�sbqR|| j
|�7 }qRW |S )Nz%Environment id '%s' is not installed.c S s g | ] }|j � �qS r )rU )r rQ r r r r� s s z.Solver._environment_remove..) rN rZ rE r r r� r� r' � getGroupsr� �
_group_remove)r: r� r� r� rz r� r r r �_environment_removej s
zSolver._environment_removec C s> | j j|�}| jjj|�}|s.ttd�| ��|sBttd�| ��tdd� |j� D ��}|j � }| jjj
|j |j |j
|�}t� }x\|jD ]R}|j |kr�| jjj|j �r�|| j|j �7 }n|| j|j |�7 }|j|j dt� q�W xL|jD ]B}|j |k�r| jjj|j ��r|| j|j �7 }|j|j dt� q�W | jjj|� |S )Nz"Environment '%s' is not installed.z"Environment '%s' is not available.c S s g | ] }|j � �qS r )rU )r rQ r r r r� � s z/Solver._environment_upgrade..TF)r r� rN rZ rE r r r' r� �getPackageTypesr� r r r) r� r� r0 �_group_upgrader� r� r� r� r� r� )r: r� r� r� �old_setr� r� r� r r r �_environment_upgradez s,
zSolver._environment_upgradec
C s� | j j|�}|s$ttd�t|� ��| jjj||j|j |�}x(|j
� D ] }|j |jdt j
|j � qFW | jjj|� t� } |r�| jj| j||g d�� n| jj| j||g d�� | S )Nz Group id '%s' does not exist.F)r3 )r r� r r r rN r0 r� r r) r� �
addPackager� r� r� r� r� rS r� r� )
r: r� r� r3 r4 r5 r� �
swdb_grouprQ r� r r r r� � s zSolver._group_installc sR � j jj|�}|s"ttd�| ��� j jj|� t� }� fdd�|j� D �|_|S )Nz&Module or Group '%s' is not installed.c s h | ]}� j |j� �r|�qS r )r� r� )r r� )r: r r r � s z'Solver._group_remove..)rN r0 rE r r r� r� � getPackages)r: r� r� r� r )r: r r� � s zSolver._group_removec s | j j|�}| jjj|�}g }|s@|r,|jn|}ttd�| ��|sTttd�| ��|j� }t dd� |j
� D ���| j |||�� | jjj ||j
|j|�}x(|j� D ] }|j|j
dtj|j � q�W | jjj|� t� }�fdd�� D �|_� fdd��D �|_�fd d�� D �|_|S )
Nz&Module or Group '%s' is not installed.z&Module or Group '%s' is not available.c S s g | ] }|j � �qS r )r� )r rQ r r r r� � s z)Solver._group_upgrade..Fc s h | ]}|j � kr|�qS r )r )r r� )r� r r r � s z(Solver._group_upgrade..c s" h | ]}|d d� � D �kr|�qS )c S s g | ]
}|j �qS r )r )r r� r r r r� � s z3Solver._group_upgrade...r )r r )�new_setr r r � s c s h | ]}|j � kr|�qS r )r )r r� )r� r r r � s )r r� rN r0 rE r) r r r� r' r� r� r� r r� r� r� r� r� r� r� r� r� ) r: r� r� r� r3 Zargumentr� rQ r� r )r� r� r r� � s( zSolver._group_upgradec C sl xf| j jD ]Z}| j j|�}|jr
t|j�t|j� }|jj� j� j |d�}x|D ]}|j
j |� qPW q
W d S )N)r ) Z persistorr r0 rW r' Z full_listZ pkg_excludeZsackZqueryZfiltermZ_goalr� )r: �baser0 Zp_grpZinstalled_pkg_namesZinstalled_pkgsr� r r r �'_exclude_packages_from_installed_groups� s
z.Solver._exclude_packages_from_installed_groups)NTN)NTN)rF rG rH r; rJ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� * s
#
r� )NTN)5Z
__future__r r r Zlibdnf.transactionr� Zdnf.exceptionsr Zdnf.i18nr r � functoolsr r Zdnf.utilr# rC r� r� r> Zloggingr
r! r. Z getLoggerr r� Z CompsPackageType_CONDITIONALr� ZCompsPackageType_DEFAULTr� ZCompsPackageType_MANDATORYr� ZCompsPackageType_OPTIONALr� Z ALL_TYPESr r r- r1 r6 �objectr7 rK re rq r~ r� r� r� r� r� r� r r r r � sP
!A'"f(C