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 dlZejdZejjZ 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*ddZ(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)reducednfc C s" | j | j| jf}ttjtt|S )N)
categoriesgroups environmentsr operator__add__maplen)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 utilfirst)seqr r r _first_if_iterable; s r c s t jj fdd|D }|r&|S |r>tjtj j}ntjtj tjdj}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 )nameid).0g)patternr r E s z _by_pattern..)flagsN)r Zi18nr recompilefnmatch translatematchIsetr addr ui_name)r case_sensitiveZsqnexactr% retr r )r r _by_patternA s
r- c C s | j d krtjS | j S )N)Z
display_ordersysmaxsize)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_idtypesexcludestrictexclude_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_localecache)selfr r r __init__o s z_Langs.__init__c C s" t jt j} | dkrdS dj| S )NC.)NN)localeZ getlocale LC_MESSAGESjoin)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 AVAILABLEupdate INSTALLEDZgetCompsGroupItemr(
getGroupId)r: available installedresultrQ 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| jkrtdt| }n.| j| jkrtdt| }ntdt| }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[ extendGROUPSgroups_by_patternr0 rY rP rT r r rR r ) r: ZpatternsresZpatenvsgrpsrV 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: iobjlangsr 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| jS )N)rn ZdescZ desc_by_lang)r: r r r ui_description s zForwarder.ui_descriptionc C s | j | j| jS )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 fddZdd 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_idgrprd 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 fddZdd 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 loggererror)r: Zidsr Zgier r r
_build_groups s
zEnvironment._build_groupsc c s\ xVt j| j| jD ]B}y| j|V W q tk
rR } ztj| W Y d d }~X qX qW d S )N) itertoolschainrz
option_idsry rv r r )r: rw r r r r r{ s
zEnvironment.groups_iterc C s | j | jS )N)r rz )r: r r r mandatory_groups s z Environment.mandatory_groupsc C s | j | jS )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 fddZdd 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 tjS )N)r libcompsPACKAGE_TYPE_CONDITIONAL)r: r r r conditional_packages* s zGroup.conditional_packagesc C s | j tjS )N)r r PACKAGE_TYPE_DEFAULT)r: r r r default_packages. s zGroup.default_packagesc C s t | j| jS )N)r r r )r: r r r
packages_iter2 s zGroup.packages_iterc C s | j tjS )N)r r PACKAGE_TYPE_MANDATORY)r: r r r mandatory_packages6 s zGroup.mandatory_packagesc C s | j tjS )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-ddZ d.ddZ
dd Ze
dd Zdd Zd/d d Zd0d d Zd d! Ze
d"d# Zd$d% Zd1d&d'Zd2d(d)Zd*d+ Zd,S )3Compsc 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 | jS )N)r rf )r: r r r __len__` s z
Comps.__len__c C s t || j| jS )N)rq rg _group_by_id)r: Z icategoryr r r _build_categoryc s zComps._build_categoryc C s t || j| jS )N)r~ rg r )r: Z ienvironmentr r r _build_environmentf s zComps._build_environmentc C s t || j| jS )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$ t jj|st| j||}t|S )N)r r is_string_typeAssertionErrorcategories_by_patternr )r: r r* Zcatsr r r category_by_pattern} s zComps.category_by_patternc C s t jj|stt||| jS )N)r r r r 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 tdS )N)key)sortedenvironments_iterr1 )r: r r r r s zComps.environmentsc s. t jj stt 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 )r r _environment_by_id s zComps._environment_by_idc C s$ t jj|st| j||}t|S )N)r r r r r] r )r: r r* rb r r r environment_by_pattern s z Comps.environment_by_patternc C s4 t jj|stt| j }t|||}t|tdS )N)r ) r r r 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 tdS )N)r )r r{ r1 )r: r r r r s z Comps.groupsc s. t jj stt 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 r r s zComps._group_by_idc C s$ t jj|st| j||}t|S )N)r r r r r` r )r: r r* rc r r r group_by_pattern s zComps.group_by_patternc C s0 t jj|stt||t| j }t|tdS )N)r ) r r r 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 r basearchonlyr optionalrequires
isinstancelibdnf transactionZCompsGroupPackagegetNameZgetPackageTyper 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| jfS )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 | jt | j t | j t | j S )N)r install install_optupgraderemove)r: r r r r s zTransactionBunch.__len__c C s6 x0|D ](}t |tr | j| q| jt| qW d S )N)r r r( )Zparamvalitemr 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 fddZd d
Zd d Z d ddZ
dd Z dd Z d ddZ
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 t jj|st| jjj|S )N)r r r r rN r0 Zis_removable_pkg)r: Zpkg_namer r r _removable_pkgK s zSolver._removable_pkgc C s t jj|st| jjj|S )N)r r r r rN rZ Zis_removable_group)r: group_idr r r _removable_grpO s zSolver._removable_grpNTc
C s t jj|st| jj|}|s4ttdt| | j j
j ||j |j
|}| j j
j| t }xD|jD ]:} |r|| j|kr|qh|| j| j|||7 }|j| jdt qhW x.|jD ]$} |r| j|krq|j| jdt qW |S )Nz#Environment id '%s' does not exist.TF)r r r r r r r r r rN rZ newr r) r r r r _group_installaddGroupr r r )
r: env_idr r3 r4 r5 comps_envswdb_envtrans comps_groupr r r _environment_installS s" zSolver._environment_installc C s t jj|dkst| jjj|}|s6ttd| | jjj | t
}t dd |j D }x&|D ] }| j
|svqf|| j|7 }qfW |S )NTz%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..)r r r r 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 sN t jj|st| jj|}| jjj|}|s>t t
d| |sRt t
d| t dd |j D }|j
}| jjj|j|j|j|}t }x\|jD ]R}|j|kr| jjj|jr|| j|j7 }n|| j|j|7 }|j|jdt qW xL|jD ]B}|j|kr(| jjj|jr(|| j|j7 }|j|jdt qW | 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 r r 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 t jj|st| jj|}|s4ttdt| | j j
j ||j |j
|}x(|j D ] }|j|j dtj|j qVW | j j
j| 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 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 sb t jj|st jjj|}|s2ttd| jjj | 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..) r r r r rN r0 rE r r r r getPackages)r: r r r r )r: r r s zSolver._group_removec s& t jj|st| jj|}| jjj|}g }|sP|r<|j n|}t
t d| |sdt
t d| |j }t
dd |j D | j||| | jjj||j|j |}x(|j D ] }|j|jdtj|j qW | jjj| t }fdd D |_ fddD |_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 r r 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|jt|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