a
    !cd                     @   s   d dl mZ d dl mZ d dl mZ d dlZd dlZd dlmZ d dlZ	d dl
Z
d Zd Zdeef ZG dd de	jjZG d	d
 d
e	jjZG dd deZdd Zdd ZG dd deZdS )    )absolute_import)print_function)unicode_literalsN)_%s.%sc                       s   e Zd Z fddZ  ZS )TransactionErrorc                    s   t t| | d S N)superr   __init__)selfmsg	__class__ 6/usr/lib/python3.9/site-packages/dnf/transaction_sr.pyr
   /   s    zTransactionError.__init____name__
__module____qualname__r
   __classcell__r   r   r   r   r   .   s   r   c                       s   e Zd Z fddZ  ZS )TransactionReplayErrorc                    sr   || _ t|ttfr|| _n|g| _|r:tdj|d}ntd}| jD ]}|dt| 7 }qHtt	| 
| dS )z
        :param filename: The name of the transaction file being replayed
        :param errors: a list of error classes or a string with an error description
        zWThe following problems occurred while replaying the transaction from file "{filename}":)filenamez<The following problems occurred while running a transaction:z
  N)r   
isinstancelisttupleerrorsr   formatstrr	   r   r
   )r   r   r   r   errorr   r   r   r
   4   s    
zTransactionReplayError.__init__r   r   r   r   r   r   3   s   r   c                       s   e Zd Z fddZ  ZS )#IncompatibleTransactionVersionErrorc                    s   t t| || d S r   )r	   r   r
   )r   r   r   r   r   r   r
   M   s    z,IncompatibleTransactionVersionError.__init__r   r   r   r   r   r   L   s   r   c              
   C   s   |  d\}}zt|}W n: tyT } z"t|tdj|dW Y d }~n
d }~0 0 zt| W n: ty } z"t|tdj|dW Y d }~n
d }~0 0 |tkrt|tdj|tdd S )N.z1Invalid major version "{major}", number expected.)majorz1Invalid minor version "{minor}", number expected.)minorzPIncompatible major version "{major}", supported major version is "{major_supp}".)r!   Z
major_supp)splitint
ValueErrorr   r   r   VERSION_MAJORr   )versionr   r!   r"   er   r   r   _check_versionQ   s,    r)   c              
   C   st  dt i}g }g }g }| du r |S |  D ]}| r\||j|jtj|j	|j
d q(| r| }|j| g tj| d}| D ].}|d | | tj| d q|| q(| r(| }	|j|	 g tj|	 d}
|	 D ]0}|
d | | tj| d	 q||
 q(|rT||d
< |rb||d< |rp||d< |S )z
    Serializes a transaction to a data structure that is equivalent to the stored JSON format.
    :param transaction: the transaction to serialize (an instance of dnf.db.history.TransactionWrapper)
    r'   N)actionnevrareasonrepo_id)r*   idpackagespackage_typesr/   )name	installedpackage_type)r*   r.   groupsr0   r4   )r.   r2   
group_typerpmsenvironments)VERSIONr/   
is_packageappendZaction_namer+   libdnftransactionZTransactionItemReasonToStringr,   Z	from_repoZis_groupZ	get_groupZ
getGroupIdZcompsPackageTypeToStringZgetPackageTypesZgetPackagesZgetNameZgetInstalledZgetPackageTypeZis_environmentZget_environmentZgetEnvironmentIdZ	getGroupsZgetGroupType)r<   datar6   r4   r7   tsigroup
group_datapkgenvenv_datagrpr   r   r   serialize_transactionl   sd    

rE   c                   @   s   e Zd ZdZd/ddZdd Zd	d
 Zdd Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. ZdS )0TransactionReplaya  
    A class that encapsulates replaying a transaction. The transaction data are
    loaded and stored when the class is initialized. The transaction is run by
    calling the `run()` method, after the transaction is created (but before it is
    performed), the `post_transaction()` method needs to be called to verify no
    extra packages were pulled in and also to fix the reasons.
     NFc                 C   sv   || _ || _|| _|| _|| _| j jjs.d| _t | _i | _	g | _
|rX|rXttdn|rh| | n
| | dS )a  
        :param base: the dnf base
        :param filename: the filename to load the transaction from (conflicts with the 'data' argument)
        :param data: the dictionary to load the transaction from (conflicts with the 'filename' argument)
        :param ignore_extras: whether to ignore extra package pulled into the transaction
        :param ignore_installed: whether to ignore installed versions of packages
        :param skip_unavailable: whether to skip transaction packages that aren't available
        TzKConflicting TransactionReplay arguments have been specified: filename, dataN)_base	_filename_ignore_installed_ignore_extras_skip_unavailableconfstrictset_nevra_cache_nevra_reason_cache	_warningsr%   r   _load_from_file_load_from_data)r   baser   r=   Zignore_extrasZignore_installedZskip_unavailabler   r   r   r
      s    
zTransactionReplay.__init__c                 C   s   || _ t|dZ}zt|}W n: tjjyZ } zt|t|d W Y d }~n
d }~0 0 W d    n1 sp0    Y  z| | W n. t	y } zt||W Y d }~n
d }~0 0 d S )Nrr    )
rI   openjsonloaddecoderZJSONDecodeErrorr   r   rT   r   )r   fnfreplay_datar(   r   r   r   rS      s    Fz!TransactionReplay._load_from_filec                 C   s|   || _ | | j  | j dg | _| | jtdd | j dg | _| | jtdd | j dg | _| | jtdd d S )Nr6   Zarrayr4   r7   )_replay_data_verify_toplevel_jsonget_rpms_assert_typer   _groups_environments)r   r=   r   r   r   rT      s    z!TransactionReplay._load_from_datac                 C   s   |r| j | nt|d S r   )rR   r:   r   )r   Z	warn_onlyr   r   r   r   _raise_or_warn  s    z TransactionReplay._raise_or_warnc                 C   s$   t ||s ttdj||dd S )Nz*Unexpected type of "{id}", {exp} expected.)r.   Zexp)r   r   r   r   )r   valuetr.   Zexpectedr   r   r   rb     s    
zTransactionReplay._assert_typec                 C   sJ   | j }d|vr$t|tdjdd| |d tdd t|d | d S )Nr'   zMissing key "{key}".keystring)rI   r   r   r   rb   r   r)   )r   r]   r[   r   r   r   r_     s
    z'TransactionReplay._verify_toplevel_jsonc              
   C   s@  z,|d }|d }|d }t j|d }W nz tyj } z&ttdj|jd dW Y d }~nFd }~0  ty } z&ttdj|d |d	W Y d }~n
d }~0 0 t	
|}|jt	jgd
}t|dkrttdj|d|d }	d|	j|	jf }
| jj j|	j|	jd}|	jd ur&|	jnd}|j||	j|	jd}|rd|j|d}|rd|| }|s| | jtdj|d d S |dkr| j| |dv r|| j|< |dv r0|dkr| r| j|s| | jtdj|
|d t j!"| jjj#|d}| jj$j%|| jj&j' d n|dkr|( }|sh| | jtdj||d d S t j!"| jjj#|d}| jj$j%|| jj&j' d n|dv r| }|s| | jtdj||d d S | jr|d kr<|D ]}| jj$j)|d!d" qn6|dkr&| jj*+|d | nttd#j||d$d S )%Nr*   r+   r-   r,   z%Missing object key "{key}" in an rpm.r   rh   zFUnexpected value of package reason "{reason}" for rpm nevra "{nevra}".)r,   r+   )Zforms   z)Cannot parse NEVRA for package "{nevra}".r+   r   )r1   arch)epochr'   release)Zreponamez Cannot find rpm nevra "{nevra}".zReason Change)InstallUpgrade	Downgrade	ReinstallRemoved)rp   rq   rr   rp   z:Package "{na}" is already installed for action "{action}".)nar*   )rA   )ZselectZoptionalrs   zLPackage nevra "{nevra}" not available in repositories for action "{action}".)r+   r*   )ZUpgraded
DowngradedZReinstalledrt   Z	Obsoletedz<Package nevra "{nevra}" not installed for action "{action}".rt   F)Z
clean_depszFUnexpected value of package action "{action}" for rpm nevra "{nevra}".)r*   r+   ),r;   r<   ZStringToTransactionItemReasonKeyErrorr   r   r   args
IndexErrorhawkeyZSubjectZget_nevra_possibilitiesZ
FORM_NEVRAlenr1   rm   rH   Zsackqueryfilterrn   r'   ro   unionr2   re   rL   rP   addrQ   Z_get_installonly_queryrJ   dnfselectorZSelectorrO   ZgoalinstallrM   rN   Z	availableZerasehistoryZ
set_reason)r   pkg_datar*   r+   r-   r,   r(   ZsubjZparsed_nevrasZparsed_nevraru   Zquery_narn   r|   Z
query_repoZsltrrA   r   r   r   _replay_pkg_action  s    





"


z$TransactionReplay._replay_pkg_actionc                 C   s2  | j j|}|s,| | jtd|  d S | j jj||j	|j
|}z|D ]}|d }| |tdd |d }| |tdd |d }	| |	td	d z|||tj|	 W qL tjjy }
 ztt|
W Y d }
~
qLd }
~
0 0 qLW n@ ty, }
 z&ttd
j|
jd dW Y d }
~
n
d }
~
0 0 |S )NzGroup id '%s' is not available.r1   zgroups.packages.namerj   r2   zgroups.packages.installedbooleanr3   zgroups.packages.package_typez.Missing object key "{key}" in groups.packages.r   rh   )rH   compsZ_group_by_idre   rL   r   r   r?   newr1   ui_namerb   r   boolZ
addPackager;   r<   stringToCompsPackageTyper   Errorr   rw   r   rx   )r   group_id	pkg_typespkgsZcomps_group
swdb_grouprA   r1   r2   r3   r(   r   r   r   _create_swdb_groupv  s,    (z$TransactionReplay._create_swdb_groupc                 C   s*   |  |||}|d ur&| jjj| d S r   )r   rH   r   r?   r   r   r   r   r   r   r   r   r   _swdb_group_install  s    z%TransactionReplay._swdb_group_installc                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S NzGroup id '%s' is not installed.)	rH   r   r?   r`   re   rJ   r   r   upgrader   r   r   r   _swdb_group_upgrade  s    z%TransactionReplay._swdb_group_upgradec                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S r   )	rH   r   r?   r`   re   rJ   r   r   	downgrader   r   r   r   _swdb_group_downgrade  s    z'TransactionReplay._swdb_group_downgradec                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S r   )	rH   r   r?   r`   re   rJ   r   r   remover   r   r   r   _swdb_group_remove  s    z$TransactionReplay._swdb_group_removec                 C   sd  | j j|}|s,| | jtd|  d S | j jj||j	|j
|}z|D ]}|d }| |tdd |d }| |tdd |d }	| |	td	d ztj|	}	W n4 tjjy }
 ztt|
W Y d }
~
n
d }
~
0 0 |	tjjtjjfvrttd
j|d d||||	 qLW n@ ty^ }
 z&ttdj|
jd dW Y d }
~
n
d }
~
0 0 |S )Nz%Environment id '%s' is not available.r.   zenvironments.groups.idrj   r2   zenvironments.groups.installedr   r5   zenvironments.groups.group_typezlInvalid value "{group_type}" of environments.groups.group_type, only "mandatory" or "optional" is supported.)r5   z2Missing object key "{key}" in environments.groups.r   rh   )rH   r   Z_environment_by_idre   rL   r   r   rB   r   r1   r   rb   r   r   r;   r<   r   r   r   r   ZCompsPackageType_MANDATORYZCompsPackageType_OPTIONALr   ZaddGrouprw   rx   )r   env_idr   r4   Z	comps_envswdb_envrD   r.   r2   r5   r(   r   r   r   _create_swdb_environment  s@    "z*TransactionReplay._create_swdb_environmentc                 C   s*   |  |||}|d ur&| jjj| d S r   )r   rH   r   rB   r   r   r   r   r4   r   r   r   r   _swdb_environment_install  s    z+TransactionReplay._swdb_environment_installc                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S Nz%Environment id '%s' is not installed.)	rH   r   rB   r`   re   rJ   r   r   r   r   r   r   r   _swdb_environment_upgrade  s    z+TransactionReplay._swdb_environment_upgradec                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S r   )	rH   r   rB   r`   re   rJ   r   r   r   r   r   r   r   _swdb_environment_downgrade  s    z-TransactionReplay._swdb_environment_downgradec                 C   sT   | j jj|s*| | jtd|  d S | |||}|d urP| j jj| d S r   )	rH   r   rB   r`   re   rJ   r   r   r   r   r   r   r   _swdb_environment_remove  s    z*TransactionReplay._swdb_environment_removec                 C   s   | j S )z>
        :returns: the loaded data of the transaction
        )r^   r   r   r   r   get_data  s    zTransactionReplay.get_datac                 C   s   | j S )zW
        :returns: an array of warnings gathered during the transaction replay
        )rR   r   r   r   r   get_warnings
  s    zTransactionReplay.get_warningsc                 C   s  | j }g }| jD ]B}z| | W q tyP } z|| W Y d}~qd}~0 0 q| jD ]}z|d }|d }ztj|d }W nH tj	j
y } z,|tt| W Y d}~W qZW Y d}~n
d}~0 0 |dkr| |||d  nv|dkr| |||d  nX|dkr*| |||d  n:|d	krH| |||d  n|ttd
j||d W qZ ty } z,|ttdj|jd d W Y d}~qZd}~0  ty } z|| W Y d}~qZd}~0 0 qZ| jD ]}	z
|	d }|	d }
ztj|	d }W nL tj	j
y^ } z.|tt| W Y d}~W qW Y d}~n
d}~0 0 |dkr~| |
||	d  nv|dkr| |
||	d  nX|dkr| |
||	d  n:|d	kr| |
||	d  n|ttdj||
d W nt ty< } z,|ttdj|jd d W Y d}~n8d}~0  tyj } z|| W Y d}~n
d}~0 0 q|rt||dS )z*
        Replays the transaction.
        Nr*   r.   r0   rp   r/   rq   rv   rt   z@Unexpected value of group action "{action}" for group "{group}".)r*   r?   z&Missing object key "{key}" in a group.r   rh   r4   zJUnexpected value of environment action "{action}" for environment "{env}".)r*   rB   z-Missing object key "{key}" in an environment.)rI   ra   r   r   r:   rc   r;   r<   r   r   r   r   r   r   r   r   r   r   rw   rx   rd   r   r   r   r   r   )r   r[   r   r   r(   r@   r*   r   r   rC   r   r   r   r   run  s    
"$



"&




$zTransactionReplay.runc                 C   s>  | j jsdS g }| j jD ]}z
|j}W n0 tyX } zW Y d}~qW Y d}~n
d}~0 0 t|}|| jvr| jr|jtjj	tjj
tjjfvrtdj|d}| js|t| n| j| z>| j| }|jtjjtjjfv stj||jdkr||_W q ty$ } zW Y d}~qd}~0 0 q|r:t| j|dS )z
        Sets reasons in the transaction history to values from the stored transaction.

        Also serves to check whether additional packages were pulled in by the
        transaction, which results in an error (unless ignore_extras is True).
        NzgPackage nevra "{nevra}", which is not present in the transaction file, was pulled into the transaction.rl   r   )rH   r<   rA   rw   r   rP   rJ   r*   r;   ZTransactionItemAction_UPGRADEDZ TransactionItemAction_DOWNGRADEDZ!TransactionItemAction_REINSTALLEDr   r   rK   r:   r   rR   rQ   ZTransactionItemAction_INSTALLZTransactionItemAction_REMOVEZTransactionItemReasonComparer,   r   rI   )r   r   r>   rA   r(   r+   r   Zreplay_reasonr   r   r   post_transaction`  sD    
"



z"TransactionReplay.post_transaction)rG   NFFF)r   r   r   __doc__r
   rS   rT   re   rb   r_   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rF      s6        
'
a


(


OrF   )Z
__future__r   r   r   r;   rz   Zdnf.i18nr   Zdnf.exceptionsr   rX   r&   ZVERSION_MINORr8   
exceptionsr   r   r   r   r)   rE   objectrF   r   r   r   r   <module>   s    K