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 k s t �|| _|| _d S )N)�AssertionError�_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 r2 � 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 r4 )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 r: � s z$RPMTransactionItemWrapper.get_reasonc C s | j jj| j S )N)r r9 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/ r0 r2 r5 r7 r8 r: 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)rD �getId)r
r r r �tid� s zTransactionWrapper.tidc C s
| j j� S )N)rD Z
getCmdline)r
r r r �cmdline� s zTransactionWrapper.cmdlinec C s
| j j� S )N)rD Z
getReleasever)r
r r r �
releasever� s z TransactionWrapper.releaseverc C s
| j j� S )N)rD Z
getDtBegin)r
r r r �
beg_timestamp� s z TransactionWrapper.beg_timestampc C s
| j j� S )N)rD ZgetDtEnd)r
r r r �
end_timestamp� s z TransactionWrapper.end_timestampc C s
| j j� S )N)rD �getRpmdbVersionBegin)r
r r r �beg_rpmdb_version� s z$TransactionWrapper.beg_rpmdb_versionc C s
| j j� S )N)rD �getRpmdbVersionEnd)r
r r r �end_rpmdb_version� s z$TransactionWrapper.end_rpmdb_versionc C s t | jj� tjjk�S )N)�intrD r1 �libdnf� transaction�TransactionItemState_DONE)r
r r r � return_code� s z TransactionWrapper.return_codec C s
| j j� S )N)rD 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)rD �getConsoleOutput�bool)r
�outputr r r � is_output� s
z TransactionWrapper.is_outputc C s
| j j� S )N)rD Z
getComment)r
r r r �comment� s zTransactionWrapper.commentc C s | j j� gS )N)rD rF )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..)rD ZgetItems)r
�resultr )r
r rV � s
zTransactionWrapper.packagesc C s dd� | j j� D �S )Nc S s g | ] }|d �qS )r r )r_ r` r r r ra � s z-TransactionWrapper.output..)rD rX )r
r r r rZ � s zTransactionWrapper.outputc C s g S )Nr )r
r r r �error� s zTransactionWrapper.errorc C s | j j� |k| _d S )N)rD rN �altered_gt_rpmdb)r
Zrpmdbvr r r �compare_rpmdbv� s z!TransactionWrapper.compare_rpmdbvN)r> r? r@ �altered_lt_rpmdbrd r rA rG rH rI rJ rK rM rO rT rU rW r[ r\ r] r^ rV rZ rc re r r r r rC � s* rC 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)rQ rR ZMergedTransactionrD )r
rE r r r r � s z!MergedTransactionWrapper.__init__c C s | j j|j � d S )N)rD �merge)r
rE r r r rh � s z MergedTransactionWrapper.mergec C s
| j j� S )N)rD Z listUserIds)r
r r r rU � s z!MergedTransactionWrapper.loginuidc C s
| j j� S )N)rD 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 )rP rQ rR rS )r_ r` r r r ra s z8MergedTransactionWrapper.return_code..)rD Z
listStates)r
r r r rT s z$MergedTransactionWrapper.return_codec C s
| j j� S )N)rD Z listCmdlines)r
r r r rH s z MergedTransactionWrapper.cmdlinec C s
| j j� S )N)rD ZlistReleasevers)r
r r r rI s z#MergedTransactionWrapper.releaseverc C s
| j j� S )N)rD 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 ra s z3MergedTransactionWrapper.output..)rD rX )r
r r r rZ s z MergedTransactionWrapper.outputN)
r> r? r@ r rh rA rU r] rT rH rI r\ rZ r r r r rg � s rg 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) �strrI �_rpm�_group�_envZ _addon_datar �_db_dir�_output)r
Zdb_dirrI 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)rk r )r
r r r r9 ( s
zSwdbInterface.rpmc C s | j d krt| �| _ | j S )N)rl r )r
r r r �group. s
zSwdbInterface.groupc C s | j d krt| �| _ | j S )N)rm r )r
r r r �env4 s
zSwdbInterface.envc C s t jj| jtjjj�S )N)�os�path�joinrn rQ rR �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 rQ rR rw rx � RuntimeErrorr rj �initTransaction)r
Zexr r r r > s
zSwdbInterface.swdbc C s t jj|| j�}|j� d S )N)rQ rR Z Transformerrx � transform)r
Z input_dirZ transformerr r r r{ 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. rk rl rm r ZcloseTransactionZ
closeDatabasero )r
r r r rp O s
zSwdbInterface.closec C s
| j j� S )N)r ZgetPath)r
r r r ru ] 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 ZgetLastTransactionrC )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 )rP )r_ r` r r r ra o s z%SwdbInterface.old..c S s g | ] }t |��qS r )rC )r_ r` r r r ra q s c s g | ]}|j � kr|�qS r )rG )r_ r` )r] r r ra t s r r T�����)r ZlistTransactions� enumeraterD rL rN rf rd )r
r] �limitr~ rb r` rE Z
prev_transr )r] r �oldm s
zSwdbInterface.oldc C s t | jj� �S )N)rC 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)
r9 Z_pkg_to_swdb_rpm_item�reporQ rR Z#TransactionItemAction_REASON_CHANGEr ZaddItemr3 rS )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
getRPMReporj )r
r; r r r r� � s zSwdbInterface.repoc C s* | j jt|��}|dkr |S t| |�}|S )z Get package data for packageN)r ZgetRPMTransactionItemrj r )r
r; rb 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 rz ZbeginTransactionrP �calendarZtimegm�timeZgmtimerj r Z getloginuidZ
setReleaseverrI r| )r
Z
rpmdb_versionZ
using_pkgsZtsisrH r\ rG 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 ro �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� rQ rR ZTransactionState_DONEr9 r2 ZTransactionItemState_ERRORZTransactionState_ERRORro r ZaddConsoleOutputLineZendTransactionrP r� rj )r
rO rT �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& rQ rR �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& rQ rR r� r� )r
r; Z first_transZrollbackrb r r r �get_erased_reason� s z SwdbInterface.get_erased_reason)r- )T)Nr F)Nr- )r- NN)T) r> r? r@ r rq rA r9 rr rs rx r r{ rp ru r} r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r ri s0
ri )r� rt r� Zlibdnf.transactionrQ Z libdnf.utilsZdnf.i18nr Zdnf.yumr Zdnf.exceptionsr rr r r r �objectr rC rg ri r r r r � s M"