File: //lib/python3/dist-packages/uaclient/cli/__pycache__/cves.cpython-38.pyc
U
��Jh � @ s� d dl Z d dlmZmZ d dlmZmZ d dlmZ d dl m
Z
m Z m Z d dl
mZmZ d dlmZ d dlmZ ejejd �ed
�d d ��Zd
d� Zdd� Zed
�dd�Zed
�dd�Ze dejej eej de e
dej dd�e
dej dd�gd�gd �Z dS ) � N)�
exceptions�messages)� CVEsOptions�_cves)�cli_util)� ProArgument�ProArgumentGroup�
ProCommand)�Table� create_link)� HelpCategory)�UAConfig)�msg��cfgc K s@ t | j| jd�}zt||d�}W n tjk
r: d }Y nX |S )N)�fixable� unfixable)Zoptionsr )r r r r r �VulnerabilityDataNotFound)�argsr �kwargsZ cve_options�result� r �3/usr/lib/python3/dist-packages/uaclient/cli/cves.py�_get_cve_vulnerabilities s �
r c C sV g }| j �� D ]B\}}|jD ]2}| j�|j�}|r |�||j|jpDd|jf� q q|S )N�-)�packages�items�cves�get�name�appendZpriorityZ
fix_origin)�cve_vulnerabilities�rowsZ package_nameZ package_infoZcveZcve_infor r r �_get_cve_table_rows s
�� r# c C sX g }t | dd� d�D ]>}|�|d t�|d �|d t|d d�|d �d �f� q|S )
Nc S s | d d� | d �fS )Nr )ZcriticalZhighZmediumZlowZ
negligible� )�index)�rowr r r �<lambda>3 s �z"_format_cve_rows.<locals>.<lambda>)�keyr r$ � � z https://ubuntu.com/security/{})�textZurl)�sortedr r Zcolorize_priorityr �format)Zcve_rowsZformatted_rowsr& r r r �_format_cve_rows/ s �
��� r. c C s� t | |d�}|st�� �|jrxtt|��}|rHttddddg|d�� q�| jrZtt j
� q�| j rltt j � q�tt j
� n.| jr�tt j
� n | j r�tt j � n
tt j
� d S )Nr ZPackageZPriorityZOriginZ
Vulnerability)Zheadersr" )r r r r r. r# �printr
r r Z CLI_UNFIXABLE_CVES_NOT_AFFECTEDr Z CLI_FIXABLE_CVES_NOT_AFFECTEDZCLI_CVES_NOT_AFFECTED)r r r! r" r r r �
_list_cvesG s,
�� r0 c K sN | j r| jrtjddd��zt| |d� W n tk
rH tj�� Y nX d S )Nr r )Zoption1Zoption2)r r ) r r r ZInvalidOptionCombinationr0 �BrokenPipeError�sys�stderr�close)r r r r r r � action_cvesg s �r5 r Tz --unfixable�
store_true)�help�actionz --fixable)Z arguments)r7 Z descriptionr8 Z
help_categoryZpreserve_descriptionZargument_groups)!r2 Zuaclientr r Z#uaclient.api.u.pro.security.cves.v1r r Z uaclient.clir Zuaclient.cli.commandsr r r Zuaclient.cli.formatterr
r Zuaclient.cli.parserr Zuaclient.configr
Z with_spinnerZCLI_CVES_SPINNER_MSGr r# r. r0 r5 ZCLI_CVESZ
CLI_CVES_DESCZSECURITYZCLI_CVES_UNFIXABLEZCLI_CVES_FIXABLEZ cves_commandr r r r �<module> sF ������