3
�ft`%: � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z ddl m
Z
mZmZ G dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�ZdS )� N)�ucd)�misc)�
DatabaseError� )�GroupPersistor�EnvironmentPersistor�RPMTransactionc @ sj e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zedd � �Zed d � �Zed d � �Zed!d"� �Zed#d$� �Zed%d&� �Zejd'd&� �Zed(d)� �Zejd*d)� �Zed+d,� �Zed-d.� �Zed/d0� �Zejd1d0� �Zed2d3� �Z d4d5� Z ed6d7� �Z d8d9� Z ed:d;� �Z edd?� �Z"d@S )A�RPMTransactionItemWrapperc C s || _ || _d S )N)�_swdb�_item)�self�swdb�item� r � /usr/lib/python3.6/history.py�__init__# s z"RPMTransactionItemWrapper.__init__c C s | j j� j� S )N)r ZgetItemZtoStr)r r r r �__str__( s z!RPMTransactionItemWrapper.__str__c C s | j |j k S )N)r )r �otherr r r �__lt__+ s z RPMTransactionItemWrapper.__lt__c C s | j |j kS )N)r )r r r r r �__eq__. s z RPMTransactionItemWrapper.__eq__c C s
| j j� S )N)r �__hash__)r r r r r 1 s z"RPMTransactionItemWrapper.__hash__c C s dS )NTr )r �patternr r r �match4 s z RPMTransactionItemWrapper.matchc C s | j j� d k S )N)r �
getRPMItem)r r r r �
is_package7 s z$RPMTransactionItemWrapper.is_packagec C s | j j� d k S )N)r �getCompsGroupItem)r r r r �is_group: s z"RPMTransactionItemWrapper.is_groupc C s | j j� d k S )N)r �getCompsEnvironmentItem)r r r r �is_environment= s z(RPMTransactionItemWrapper.is_environmentc C s
| j j� S )N)r r )r r r r � get_group@ s z#RPMTransactionItemWrapper.get_groupc C s
| j j� S )N)r r )r r r r �get_environmentC s z)RPMTransactionItemWrapper.get_environmentc C s | j j� j� S )N)r r ZgetName)r r r r �nameF s z RPMTransactionItemWrapper.namec C s | j j� j� S )N)r r ZgetEpoch)r r r r �epochJ s z RPMTransactionItemWrapper.epochc C s | j j� j� S )N)r r Z
getVersion)r r r r �versionN s z!RPMTransactionItemWrapper.versionc C s | j j� j� S )N)r r Z
getRelease)r r r r �releaseR s z!RPMTransactionItemWrapper.releasec C s | j j� j� S )N)r r ZgetArch)r r r r �archV s z RPMTransactionItemWrapper.archc C s* | j rdj| j | j| j�S dj| j| j�S )Nz{}:{}-{}z{}-{})r" �formatr# r$ )r r r r �evrZ s z RPMTransactionItemWrapper.evrc C s | j j� j� S )N)r r ZgetNEVRA)r r r r �nevra` s z RPMTransactionItemWrapper.nevrac C s
| j j� S )N)r Z getAction)r r r r �actiond s z RPMTransactionItemWrapper.actionc C s | j j|� d S )N)r Z setAction)r �valuer r r r) h s c C s
| j j� S )N)r Z getReason)r r r r �reasonl s z RPMTransactionItemWrapper.reasonc C s | j j|�S )N)r Z setReason)r r* r r r r+ p s c C s$ y
| j j� S tk
r dS X d S )N� )r Z
getActionName�AttributeError)r r r r � action_namet s
z%RPMTransactionItemWrapper.action_namec C s$ y
| j j� S tk
r dS X d S )Nr, )r ZgetActionShortr- )r r r r � action_short{ s
z&RPMTransactionItemWrapper.action_shortc C s
| j j� S )N)r �getState)r r r r �state� s z RPMTransactionItemWrapper.statec C s | j j|� d S )N)r �setState)r r* r r r r1 � s c C s
| j j� S )N)r � getRepoid)r r r r � from_repo� s z#RPMTransactionItemWrapper.from_repoc C s | j j� sdS d| j j� S )Nr, �@)r r3 )r r r r � ui_from_repo� s
z&RPMTransactionItemWrapper.ui_from_repoc C s d S )Nr )r r r r �
obsoleting� s z$RPMTransactionItemWrapper.obsoletingc C s | j jj| �S )N)r
�rpm�
get_reason)r r r r r9 � s z$RPMTransactionItemWrapper.get_reasonc C s | j jj| j S )N)r
r8 Z _swdb_ti_pkgr )r r r r �pkg� s z RPMTransactionItemWrapper.pkgc C s | j jS )N)r: �files)r r r r r; � s z RPMTransactionItemWrapper.filesc C s | j S )N)r: )r r r r �_active� s z!RPMTransactionItemWrapper._activeN)#�__name__�
__module__� __qualname__r r r r r r r r r r r �propertyr! r" r# r$ r% r' r( r) �setterr+ r. r/ r1 r4 r6 r7 r9 r: r; r< r r r r r " sB r c @ s� e Zd ZdZdZdd� Zedd� �Zedd� �Zedd � �Z ed
d � �Z
ed d
� �Z edd� �Z edd� �Z
edd� �Zedd� �Zedd� �Zedd� �Zedd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))�TransactionWrapperFc C s
|| _ d S )N)�_trans)r �transr r r r � s zTransactionWrapper.__init__c C s
| j j� S )N)rC �getId)r r r r �tid� s zTransactionWrapper.tidc C s
| j j� S )N)rC Z
getCmdline)r r r r �cmdline� s zTransactionWrapper.cmdlinec C s
| j j� S )N)rC Z
getReleasever)r r r r �
releasever� s z TransactionWrapper.releaseverc C s
| j j� S )N)rC Z
getDtBegin)r r r r �
beg_timestamp� s z TransactionWrapper.beg_timestampc C s
| j j� S )N)rC ZgetDtEnd)r r r r �
end_timestamp� s z TransactionWrapper.end_timestampc C s
| j j� S )N)rC �getRpmdbVersionBegin)r r r r �beg_rpmdb_version� s z$TransactionWrapper.beg_rpmdb_versionc C s
| j j� S )N)rC �getRpmdbVersionEnd)r r r r �end_rpmdb_version� s z$TransactionWrapper.end_rpmdb_versionc C s t | jj� tjjk�S )N)�intrC r0 �libdnf� transaction�TransactionItemState_DONE)r r r r � return_code� s z TransactionWrapper.return_codec C s
| j j� S )N)rC Z getUserId)r r r r �loginuid� s zTransactionWrapper.loginuidc C s | j S )N)�packages)r r r r �data� s zTransactionWrapper.datac C s | j j� }t|�S )N)rC �getConsoleOutput�bool)r �outputr r r � is_output� s
z TransactionWrapper.is_outputc C s
| j j� S )N)rC Z
getComment)r r r r �comment� s zTransactionWrapper.commentc C s | j j� gS )N)rC rE )r r r r �tids� s zTransactionWrapper.tidsc C s g S )Nr )r r r r �performed_with� s z!TransactionWrapper.performed_withc s � j j� }� fdd�|D �S )Nc s g | ]}t � |��qS r )r )�.0�i)r r r �
� s z/TransactionWrapper.packages..)rC ZgetItems)r �resultr )r r rU � s
zTransactionWrapper.packagesc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r )r^ r_ r r r r` � s z-TransactionWrapper.output..)rC rW )r r r r rY � s zTransactionWrapper.outputc C s g S )Nr )r r r r �error� s zTransactionWrapper.errorc C s | j j� |k| _d S )N)rC rM �altered_gt_rpmdb)r Zrpmdbvr r r �compare_rpmdbv� s z!TransactionWrapper.compare_rpmdbvN)r= r> r? �altered_lt_rpmdbrc r r@ rF rG rH rI rJ rL rN rS rT rV rZ r[ r\ r] rU rY rb rd r r r r rB � s* rB c @ sh e Zd Zdd� Zdd� Zedd� �Zdd� Zed d
� �Zed d � �Z ed
d� �Z
edd� �Z dd� Z dS )�MergedTransactionWrapperc C s t jj|j�| _d S )N)rP rQ ZMergedTransactionrC )r rD r r r r � s z!MergedTransactionWrapper.__init__c C s | j j|j � d S )N)rC �merge)r rD r r r rg � s z MergedTransactionWrapper.mergec C s
| j j� S )N)rC Z listUserIds)r r r r rT � s z!MergedTransactionWrapper.loginuidc C s
| j j� S )N)rC ZlistIds)r r r r r\ s z MergedTransactionWrapper.tidsc C s dd� | j j� D �S )Nc S s g | ]}t |tjjk��qS r )rO rP rQ rR )r^ r_ r r r r` s z8MergedTransactionWrapper.return_code..)rC Z
listStates)r r r r rS s z$MergedTransactionWrapper.return_codec C s
| j j� S )N)rC Z listCmdlines)r r r r rG s z MergedTransactionWrapper.cmdlinec C s
| j j� S )N)rC ZlistReleasevers)r r r r rH s z#MergedTransactionWrapper.releaseverc C s
| j j� S )N)rC Z listComments)r r r r r[ s z MergedTransactionWrapper.commentc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r )r^ r_ r r r r` s z3MergedTransactionWrapper.output..)rC rW )r r r r rY s z MergedTransactionWrapper.outputN)
r= r> r? r rg r@ rT r\ rS rG rH r[ rY r r r r rf � s rf c @ s� e Zd Zd6dd�Zdd� Zedd� �Zedd � �Zed
d � �Zed d
� �Z edd� �Z
dd� Z dd� Z edd� �Z
dd� Zd7dd�Zd8d d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd9d(d)�Zd*d+� Zd,d-� Zd:d.d/�Zd;d0d1�Zd2d3� Zd4d5� ZdS )<�
SwdbInterfacer, c C s8 t |�| _d | _d | _d | _d | _d | _|| _g | _d S )N) �strrH �_rpm�_group�_envZ _addon_datar
�_db_dir�_output)r Zdb_dirrH r r r r s
zSwdbInterface.__init__c C s | j � d S )N)�close)r r r r �__del__% s zSwdbInterface.__del__c C s | j d krt| �| _ | j S )N)rj r )r r r r r8 ( s
zSwdbInterface.rpmc C s | j d krt| �| _ | j S )N)rk r )r r r r �group. s
zSwdbInterface.groupc C s | j d krt| �| _ | j S )N)rl r )r r r r �env4 s
zSwdbInterface.envc C s t jj| jtjjj�S )N)�os�path�joinrm rP rQ �SwdbZdefaultDatabaseName)r r r r �dbpath: s zSwdbInterface.dbpathc C sZ | j sTytjj| j�| _ W n. tk
rH } ztt|���W Y dd}~X nX | j j� | j S )z Lazy initialize Swdb object N) r
rP rQ rv rw � RuntimeErrorr ri �initTransaction)r Zexr r r r
> s
zSwdbInterface.swdbc C s t jj|| j�}|j� d S )N)rP rQ Z Transformerrw � transform)r Z input_dirZ transformerr r r rz K s zSwdbInterface.transformc C sZ y| ` W n tk
r Y nX d | _d | _d | _| jrJ| jj� | jj� d | _g | _d S )N) �_tidr- rj rk rl r
ZcloseTransactionZ
closeDatabasern )r r r r ro O s
zSwdbInterface.closec C s
| j j� S )N)r
ZgetPath)r r r r rt ] s zSwdbInterface.pathc C s
| j j� S )N)r
Z
resetDatabase)r r r r �reset_dba s zSwdbInterface.reset_dbTc C s | j j� }|sd S t|�S )N)r
ZgetLastTransactionrB )r �complete_transactions_only�tr r r �laste s
zSwdbInterface.lastNr Fc s� � pg � dd� � D �� | j j� }dd� |D �}� rD� fdd�|D �}xJt|�D ]>\}}|dkr`qN||d }|jj� |jj� krNd|_d|_qNW |d d d� S ) Nc S s g | ] }t |��qS r )rO )r^ r_ r r r r` o s z%SwdbInterface.old..c S s g | ] }t |��qS r )rB )r^ r_ r r r r` q s c s g | ]}|j � kr|�qS r )rF )r^ r_ )r\ r r r` t s r r T�����)r
ZlistTransactions� enumeraterC rK rM re rc )r r\ �limitr} ra r_ rD Z
prev_transr )r\ r �oldm s
zSwdbInterface.oldc C s t | jj� �S )N)rB r
Z
getCurrent)r r r r � get_current� s zSwdbInterface.get_currentc C sB | j j|�}| j|�}tjj}| jj||||�}|jtjj � |S )zSet reason for package)
r8 Z_pkg_to_swdb_rpm_item�reporP rQ Z#TransactionItemAction_REASON_CHANGEr
ZaddItemr2 rR )r r: r+ �rpm_itemZrepoidr) Ztir r r �
set_reason� s
zSwdbInterface.set_reasonc C s | j jt|��S )zGet repository of package)r
Z
getRPMRepori )r r: r r r r� � s zSwdbInterface.repoc C s* | j jt|��}|dkr |S t| |�}|S )z Get package data for packageN)r
ZgetRPMTransactionItemri r )r r: ra r r r � package_data� s
zSwdbInterface.package_datac C sf y| j j� W n Y nX | j jttjtj� ��t|�|p>dtt j
� �|�}| j j | j � || _
|S )Nr, )r
ry ZbeginTransactionrO �calendarZtimegm�timeZgmtimeri r Z getloginuidZ
setReleaseverrH r{ )r Z
rpmdb_versionZ
using_pkgsZtsisrG r[ rF r r r �beg� s
zSwdbInterface.begc C sN | j j� }|j|j� |j|jp"d� |j|j� |j|j � |j
|j � |S )Nr ) r
Z
createRPMItemZsetNamer! ZsetEpochr" Z
setVersionr# Z
setReleaser$ ZsetArchr% )r Zpor� r r r �pkg_to_swdb_rpm_item� s
z"SwdbInterface.pkg_to_swdb_rpm_itemc C sD t | d�sd S |sd S x(|j� D ] }t|�}| jjd|f� q W d S )Nr{ r )�hasattr�
splitlinesr rn �append)r �msg�liner r r �log_scriptlet_output� s
z"SwdbInterface.log_scriptlet_outputc C s� t | d�sd S |d krFtjj}x&| jD ] }|jtjjkr&tjj}P q&W x | jD ]\}}| j j
||� qNW g | _| j j t t
j
� �t|�|� d S )Nr{ )r� rP rQ ZTransactionState_DONEr8 r1 ZTransactionItemState_ERRORZTransactionState_ERRORrn r
ZaddConsoleOutputLineZendTransactionrO r� ri )r rN rS �errorsZtsiZfile_descriptorr� r r r �end� s
zSwdbInterface.endc C s | j j|�S )z{ Search for history transactions which contain specified
packages al. la. "yum list". Returns transaction ids. )r
ZsearchTransactionsByRPM)r ZpatternsZ ignore_caser r r �search� s zSwdbInterface.searchc C s8 | j j|j|jd�}|tjjkr$dS |tjjkr4dS dS )z)Returns True if package is user installedr TFr� )r
� resolveRPMTransactionItemReasonr! r% rP rQ �TransactionItemReason_USER� TransactionItemReason_UNKNOWN)r r: r+ r r r �user_installed� s z SwdbInterface.user_installedc C sF |r| j j|j|j|�}n| j j|j|jd�}|tjjkrBtjj}|S )a2 Get reason of package before transaction being undone. If package
is already installed in the system, keep his reason.
:param pkg: package being installed
:param first_trans: id of first transaction being undone
:param rollback: True if transaction is performing a rollbackr r� )r
r� r! r% rP rQ r� r� )r r: Z first_transZrollbackra r r r �get_erased_reason� s z SwdbInterface.get_erased_reason)r, )T)Nr F)Nr, )r, NN)T) r= r> r? r rp r@ r8 rq rr rw r
rz ro rt r| r r� r� r� r� r� r� r� r� r� r� r� r� r r r r rh s0
rh )r� rs r� Zlibdnf.transactionrP Z libdnf.utilsZdnf.i18nr Zdnf.yumr Zdnf.exceptionsr rq r r r �objectr rB rf rh r r r r � s M"