XylotrechusZ
3
�ft`~ � @ st d dl mZ d dl mZ d dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z ej
d�ZG dd � d ej�ZdS )
� )�absolute_import)�unicode_literalsN)�commands)� OptionParser)�_�dnfc @ sX e Zd ZdZdZed�Zed d
� �Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� ZdS )�UpgradeCommandzTA class containing methods needed by the cli to execute the
update command.
�upgrade�update�
upgrade-to� update-to� localupdate�upz,upgrade a package or packages on your systemc C s" | j ddtd�tjtd�d� d S )NZpackages�*zPackage to upgradeZPACKAGE)�nargs�help�action�metavar)� add_argumentr r ZParseSpecGroupFileCallback)�parser� r � /usr/lib/python3.6/upgrade.py�
set_argparser* s z UpgradeCommand.set_argparserc C sZ | j j}d|_d|_d|_d|_tj| j| j � | j j
sDtj | j� d| _ d| _
d| _dS )z�Verify that conditions are met so that this command can run.
These include that there are enabled repositories with gpg
keys, and that this command is being run by the root user.
TN)�cli�demandsZsack_activationZavailable_reposZ resolvingZ root_userr Z _checkGPGKey�base�opts� filenamesZ_checkEnabledRepo�upgrade_minimal� all_security�skipped_grp_specs)�selfr r r r � configure0 s zUpgradeCommand.configurec C s� | j r
dnd}| jj| j|| jd� | jjs<| jjs<| jjrzd}|| j� O }|| j � O }|| j
� O }|| j � O }|r�d S n| j j
� d S tjjtd���d S )N�eqZgte)�cmp_type�allFz No packages marked for upgrade.)r r Z _populate_update_security_filterr r r � pkg_specs� grp_specs�_update_modules�
_update_files�_update_packages�_update_groupsr Z upgrade_allr �
exceptions�Errorr )r! r$ �resultr r r �runB s
zUpgradeCommand.runc C sN t | jj�}tjjr6tjjj| j�}|j | jj�| _
n
| jj| _
t | j
�|kS )N) �lenr r' r r Z WITH_MODULES�module� module_baseZ
ModuleBaser r )r! Zgroup_specs_numr2 r r r r( V s
z UpgradeCommand._update_modulesc C s� d}| j jr�x~| jj| j jd| jjjd�D ]^}y| jj|� d}W q* tjj k
r� } z$t
j t d�| jjj
j|j�� W Y d d }~X q*X q*W |S )NF)�strict�progressTzNo match for argument: %s)r r r Zadd_remote_rpms�outputr4 Zpackage_upgrader r, � MarkingError�logger�infor �term�bold�location)r! �successZpkg�er r r r) ` s
*z UpgradeCommand._update_filesc C sr d}xh| j jD ]\}y| jj|� d}W q tjjk
rh } z"tjt d�| jj
j j |�� W Y d d }~X qX qW |S )NFTzNo match for argument: %s)
r r& r r r r, r6 r7 r8 r r5 r9 r: )r! r< Zpkg_specr= r r r r* m s
(z UpgradeCommand._update_packagesc C s | j r| jj| j � dS dS )NTF)r r Zenv_group_upgrade)r! r r r r+ x s z UpgradeCommand._update_groupsN)r r
r r r
r )�__name__�
__module__� __qualname__�__doc__�aliasesr Zsummary� staticmethodr r" r/ r( r) r* r+ r r r r r # s
r )Z
__future__r r ZloggingZdnf.exceptionsr Zdnf.baseZdnf.clir Zdnf.cli.option_parserr Zdnf.i18nr Z getLoggerr7 ZCommandr r r r r �<module> s