3
��g�< � @ s� d dl Zd dlZd dlZd dlZd dlmZ d dlm Z d dl
Z
G dd� de �Z G dd� de �Z
G dd � d e �ZG d
d � d e �ZdS ) � N)�_)�loggerc @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
PersistorBasec C s" || _ i | _i | _i | _i | _d S )N)�history�
_installed�_removed� _upgraded� _downgraded)�selfr � r �/usr/lib/python3.6/group.py�__init__ s
zPersistorBase.__init__c C s( t | j�t | j� t | j� t | j� S )N)�lenr r r r )r
r r r �__len__( s zPersistorBase.__len__c C s i | _ i | _i | _i | _d S )N)r r r r )r
r r r �clean+ s zPersistorBase.cleanc C s t �d S )N)�NotImplementedError)r
�objr r r � _get_obj_id1 s zPersistorBase._get_obj_idc C s* | j jj|d|tjj�}|jtjj� d S )N� )r �swdb�addItem�libdnf� transaction�TransactionItemReason_USERZsetStateZTransactionItemState_DONE)r
�item�action�tir r r �_add_to_history4 s z PersistorBase._add_to_historyc C s$ || j | j|�< | j|tjj� d S )N)r r r r r � TransactionItemAction_INSTALL)r
r r r r �install8 s zPersistorBase.installc C s$ || j | j|�< | j|tjj� d S )N)r r r r r � TransactionItemAction_REMOVE)r
r r r r �remove< s zPersistorBase.removec C s$ || j | j|�< | j|tjj� d S )N)r r r r r � TransactionItemAction_UPGRADE)r
r r r r �upgrade@ s zPersistorBase.upgradec C s$ || j | j|�< | j|tjj� d S )N)r r r r r � TransactionItemAction_DOWNGRADE)r
r r r r � downgradeD s zPersistorBase.downgradec C s t �d S )N)r )r
�obj_id�name�translated_name� pkg_typesr r r �newH s zPersistorBase.newc C s t �d S )N)r )r
r& r r r �getK s zPersistorBase.getc C s t �d S )N)r )r
�patternr r r �search_by_patternN s z PersistorBase.search_by_patternN)�__name__�
__module__� __qualname__r
r r r r r r! r# r% r* r+ r- r r r r r s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�GroupPersistorc C s" | j jj� }dd� |D �}t|�S )Nc S s g | ]}|j � r|�qS r )�getCompsGroupItem)�.0�ir r r �
V s z+GroupPersistor.__iter__..)r r �getItems�iter)r
�itemsr r r �__iter__T s zGroupPersistor.__iter__c C s |j � S )N)�
getGroupId)r
r r r r r Y s zGroupPersistor._get_obj_idc C sH | j jj� }|j|� |d k r(|j|� |d k r:|j|� |j|� |S )N)r r ZcreateCompsGroupItemZ
setGroupId�setName�setTranslatedName�setPackageTypes)r
r& r' r( r) �
swdb_groupr r r r* \ s
zGroupPersistor.newc C s" | j jj|�}|sd S |j� }|S )N)r r r2 )r
r& r> r r r r+ f s
zGroupPersistor.getc C s | j jj|�S )N)r r ZgetCompsGroupItemsByPattern)r
r, r r r r- m s z GroupPersistor.search_by_patternc C s | j jj|�S )N)r r ZgetPackageCompsGroups)r
�pkg_namer r r �get_package_groupsp s z!GroupPersistor.get_package_groupsc C s� | j jj|dd�}|tjjkr"dS t| j|��}xJ| jj � D ]<\}}x2|j
� D ]&}|j � |kr`qN|j � sjqN|j
|� qNW q.)r r r6 r7 )r
r8 r r r r9 � s z EnvironmentPersistor.__iter__c C s |j � S )N)ZgetEnvironmentId)r
r r r r r � s z EnvironmentPersistor._get_obj_idc C sH | j jj� }|j|� |d k r(|j|� |d k r:|j|� |j|� |S )N)r r ZcreateCompsEnvironmentItemZsetEnvironmentIdr; r< r= )r
r& r' r( r) �swdb_envr r r r* � s
zEnvironmentPersistor.newc C s" | j jj|�}|sd S |j� }|S )N)r r rM )r
r& rN r r r r+ � s
zEnvironmentPersistor.getc C s | j jj|�S )N)r r Z!getCompsEnvironmentItemsByPattern)r
r, r r r r- � s z&EnvironmentPersistor.search_by_patternc C s | j jj|�S )N)r r ZgetCompsGroupEnvironments)r
rH r r r �get_group_environments� s z+EnvironmentPersistor.get_group_environmentsc C s� | j jj|�}|sdS t| j|��}xJ| jj� D ]<\}}x2|j� D ]&}|j� |krTqB|j � s^qB|j
|� qBW q0W xJ| j j� D ]<\}}x2|j� D ]&}|j� |kr�q�|j � s�q�|j |� q�W q|W |r�dS dS )NFT)
r rI r+ rD rO r r8 Z getGroupsr: rE r! r rF )r
rH r> Zgroup_environmentsZenv_id�envrI r r r �is_removable_group� s* z'EnvironmentPersistor.is_removable_groupN)
r. r/ r0 r9 r r* r+ r- rO rQ r r r r rL � s
rL c @ s� e Zd Zd,dd�Zdd� Zdd� Zdd � Zd-d
d �Zd d
� Zdd� Z dd� Z
d.dd�Z d/dd�Z d0dd�Z
d1dd�Zd2dd�Zd3d d �Zd4d d �Zd d!� Zd"d#� Zed$d%� �Zed&d'� �Zd(d)� Zd*d+� ZdS )5�RPMTransactionNc C s: || _ || _| js0y| j jj� W n Y nX i | _d S )N)r r r ZinitTransaction� _swdb_ti_pkg)r
r r r r r r
� s zRPMTransaction.__init__c s8 � j r� j j� }n � jjj� }� fdd�|D �}t|�S )Nc s& g | ] }|j � rtjjj� j|��qS r )�
getRPMItem�dnf�dbr �RPMTransactionItemWrapper)r3 r4 )r
r r r5 � s z+RPMTransaction.__iter__..)r r6 r r r7 )r
r8 r )r
r r9 � s
zRPMTransaction.__iter__c s8 � j r� j j� }n � jjj� }� fdd�|D �}t|�S )Nc s& g | ] }|j � rtjjj� j|��qS r )rT rU rV r rW )r3 r4 )r
r r r5 � s z*RPMTransaction.__len__..)r r6 r r r )r
r8 r )r
r r � s
zRPMTransaction.__len__c C sP | j jj� }|j|j� |j|jp$d� |j|j� |j |j
� |j |j � |S )Nr )
r r Z
createRPMItemr; r' ZsetEpochZepochZ
setVersion�versionZ
setRelease�releaseZsetArch�arch)r
rJ �rpm_itemr r r �_pkg_to_swdb_rpm_item� s z$RPMTransaction._pkg_to_swdb_rpm_itemc C sV | j |�}| j|�}|d kr&| j|�}| jjj||||�}|rH|j|� || j|< |S )N)r\ �
get_repoid�
get_reasonr r r �
addReplacedByrS )r
rJ r rG � replaced_byr[ Zrepoid�resultr r r r* � s
zRPMTransaction.newc C s t |dd �}|r|S |jS )NZ_force_swdb_repoid)�getattrZreponame)r
rJ ra r r r r] � s zRPMTransaction.get_repoidc C s | j jj|j|jd�S )zGet reason for package� �����)r r rC r' rZ )r
rJ r r r r^ s zRPMTransaction.get_reasonc C s t jj| j|��S )zGet reason for package)r r Z TransactionItemReasonToStringr^ )r
rJ r r r �get_reason_name s z RPMTransaction.get_reason_namec C s8 |pg }x*|D ]"}| j |tjj�}|r|j|� qW d S )N)r* r r � TransactionItemAction_OBSOLETEDr_ )r
� obsoletedr` Zobsr r r r �_add_obsoleted
s
z RPMTransaction._add_obsoletedc C s6 | j |tjj�}| j |tjj|d�}| j||d� d S )N)r` )r* r r r$ � TransactionItemAction_DOWNGRADEDrh )r
r* �oldrg �ti_new�ti_oldr r r �
add_downgrade s z RPMTransaction.add_downgradec C s | j ||� d S )N)�
add_remove)r
rj rG r r r � add_erase s zRPMTransaction.add_erasec C s4 |d krt jj}| j|t jj|�}| j||d� d S )N)r` )r r r r* r rh )r
r* rg rG rk r r r � add_install s zRPMTransaction.add_installc C s6 | j |tjj�}| j |tjj|d�}| j||d� d S )N)r` )r* r r � TransactionItemAction_REINSTALL�!TransactionItemAction_REINSTALLEDrh )r
r* rj rg rk rl r r r �
add_reinstall s z RPMTransaction.add_reinstallc C s" |p
t jj}| j|t jj|�}d S )N)r r r r* r )r
rj rG rl r r r rn $ s zRPMTransaction.add_removec C s6 | j |tjj�}| j |tjj|d�}| j||d� d S )N)r` )r* r r r" � TransactionItemAction_UPGRADEDrh )r
r* rj rg rk rl r r r � add_upgrade( s zRPMTransaction.add_upgradec C s^ |j r
dS |jjrdS y.|d rB|j� rBtjtd�j|�� dS W n tk
rX dS X dS )Nr Zmodularitylabelz\No available modular metadata for modular package '{}', it cannot be installed on the systemrc ) Z
_from_cmdlineZrepoZmodule_hotfixesZ_is_in_active_moduler Zcriticalr �format�
ValueError)r
�hdrrJ r r r �_test_fail_safe- s z RPMTransaction._test_fail_safec C sR d}�x0| D �]&}�y�|j tjjkrP|jj}|| j||j�7 }|j||d� �n�|j tjjkrp|j |jj
� �n�|j tjj kr�|jj}|| j||j�7 }|j||d� �nL|j tjj kr�|jj}|| j||j�7 }|j||d� �n|j tjj
k�r|j |jj
� n�|j tjjk�r<|jj}|| j||j�7 }|j||� n�|j tjjk�r\|j |jj
� n�|j tjjk�r||j |jj
� nz|j tjjk�r�|jj}|| j||j�7 }|j||d� n@|j tjjk�r�|j |jj
� n |j tjjk�r�ntd|j ��W q tjk
�r2 } ztjjtd| ���W Y dd}~X q X q W |�rNtjjtd���|S )z!Populate the RPM transaction set.r �ur4 z%TransactionItemAction not handled: %sz An rpm exception occurred: %sNz1No available modular metadata for modular package) r r r r$ rJ Z_headerry Z
addInstallri ZaddErase�idxr Z TransactionItemAction_OBSOLETErf rq Z addReinstallrr r r" rt Z#TransactionItemAction_REASON_CHANGE� RuntimeError�rpm�errorrU �
exceptions�Errorr )r
ZtsZmodular_problems�tsirx �er r r �_populate_rpm_ts; sR *z RPMTransaction._populate_rpm_tsc C sX t � }xL| D ]D}|jtjjkr y|j|j� W q tk
rN td| ��Y q X q W |S )Nz*TransactionItem is has no RPM attached: %s) rD r rU r ZFORWARD_ACTIONSrF rJ �KeyErrorr| )r
ra r� r r r � install_setm s
zRPMTransaction.install_setc C sb t � }xV| D ]N}|jtjjtjjg kr y|j|j� W q t k
rX t
d| ��Y q X q W |S )Nz*TransactionItem is has no RPM attached: %s) rD r rU r ZBACKWARD_ACTIONSr rr rF rJ r� r| )r
ra r� r r r �
remove_sety s
zRPMTransaction.remove_setc C s, dd� | j D �}t|�r(td�|d S dS )zj Ensures all the members can be passed to rpm as they are to perform
the transaction.
c S s g | ]}|j d kr|�qS )�src)rZ )r3 rJ r r r r5 � s z3RPMTransaction._rpm_limitations..z+Will not install a source rpm package (%s).r N)r� r r )r
Z src_installsr r r �_rpm_limitations� s
z RPMTransaction._rpm_limitationsc s � fdd�| D �S )Nc s g | ]}|j � kr|�qS r )r )r3 r� )r r r r5 � s z-RPMTransaction._get_items..r )r
r r )r r �
_get_items� s zRPMTransaction._get_items)N)NN)N)N)N)NN)N)N)N)r. r/ r0 r
r9 r r\ r* r] r^ re rh rm ro rp rs rn ru ry r� �propertyr� r� r� r� r r r r rR � s*
2
rR )Zlibdnf.transactionr Zdnf.db.historyrU Zdnf.transactionZdnf.exceptionsZdnf.i18nr Zdnf.utilr r} �objectr r1 rL rR r r r r � s 3<<