Your IP : 216.73.216.52
U
8�-d�f � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZmZmZmZmZmZ d dlmZmZ d dlmZmZ d dl m Z m Z d d l m Z d d
l!m"Z"m#Z#m$Z$m%Z% d d l&m'Z' d d l(m)Z) d d
l*m+Z+ e�,� Z-e�.e/�Z0dZ1dZ2dZ3e%j4j5e+j6e%j4j5 e+j7 e%j8j5e+j9e%j8j5 e+j7 e%j:j5e+j;e%j:j5 e+j7 e%j<j5e+j;e%j<j5 e+j7 e%j=j5e+j>e%j=j5 e+j7 e"j?j5e+j6e"j?j5 e+j7 e"j@j5e+j;e"j@j5 e+j7 e1e+j6e1 e+j7 e2e+j6e2 e+j7 e3e+j6e3 e+j7 i
ZAdZBdZCdZDdZEdde�F� dddddg e$j8j5ejGi g dddg e%j:j5d�dddg d�dd�ZHd d � ZIe eJe
f d �d d �ZKee eJe
f d!�d"d#�ZLee eJe
f d!�d$d%�ZMe eJe
f d �d&d'�ZNe eJe
f d �d(d)�ZOdDeePe eJe
f d*�d+d,�ZQe e eJe
f eJe eJe
f d-�d.d/�ZRdEeJePee eJe
f eSf d0�d1d2�ZTeJeJd3�d4d5�ZUe e eJ eJd6�d7d8�ZVdFe eeJeJf e
eJ e eJ d9�d:d;�ZWe
e eJd<�d=d>�ZXe eJe
f eJd?�d@dA�ZYdBdC� ZZdS )G� N)� OrderedDict)�datetime�timezone)�Any�Dict�List�Optional�Tuple)� event_logger�
exceptions� livepatch�messages�util�version)�UA_CONFIGURABLE_KEYS�UAConfig)�get_available_resources�get_contract_information)�ATTACH_FAIL_DATE_FORMAT�PRINT_WRAP_WIDTH)�entitlement_factory)�ContractStatus�UserFacingAvailability�UserFacingConfigStatus�UserFacingStatus)�notices)�Notice)�TxtColorZ essentialZstandardZadvancedz({name: <17}{available: <11}{description}zJ{name: <17}{available: <11}{entitled: <11}{auto_enabled: <14}{description}z0SERVICE ENTITLED STATUS DESCRIPTIONz4{name: <17}{entitled: <19}{status: <19}{description}zUContent provided in json response is currently considered Experimental and may changez0.1F� ��id�name�
created_at�products�tech_support_level�r! r r" Zexternal_account_ids)Z_docZ_schema_versionr �
machine_id�attached� effective�expires�origin�services�execution_status�execution_details�featuresr �contract�account� simulatedc C s dd� | � � D �S )Nc S s$ g | ] }|j j|jj|jjd ��qS ))r! Z reason_code�reason)� entitlementr! Z named_msg�msg��.0�service� r8 �1/usr/lib/python3/dist-packages/uaclient/status.py�
<listcomp>v s
��z,_get_blocked_by_services.<locals>.<listcomp>)Z blocking_incompatible_services)�entr8 r8 r9 �_get_blocked_by_servicesu s �r< )�returnc
C s� d }d}| � � }| �� }| j|kr&dnd}|tjkr<tj}nZ| j|krXtj}|| j }n>| �� \}}|tj kr~|j|j
d�}n
|r�|j
}|tjkr�d}t | �} | j | j
|j|j|||| |d� S )Nr �no�yes)�code�message) r! � description�entitled�status�status_details�description_override� available�
blocked_by�warning)�status_description_override�contract_statusr! r �
UNENTITLEDr � UNAVAILABLE� INAPPLICABLEZuser_facing_status�WARNINGr4 r< Zpresentation_namerB �value)
r; �inapplicable_resourcesrI rE rF rK rG Z
ent_statusZdetailsrH r8 r8 r9 �_attached_service_status� s<
�
�rR )�cfgr= c C s� t �tj� t �tj� t�t�}| jd }|d }t j
j }|� |d d|�
d�t �� pZg |d |d |�
dd �|�
d
g �|d �| jjd | jjd | jj�
dd �| jj�
d g �d
�d�� |�
d�r�| jj|d< |�
d�r�|d |d< | j�
d�}|�st| �}dd� t|dd� d�D �}|D ]Z}zt| |�
dd �d�}W n tjk
�rZ Y �q Y nX || �} |d �t| |�� �q |d jdd� d� | jj�
d i ��
d �}
|
�r�|
�
d i ��
d �} | �r�| |d d!< |S )"z8Return configuration of attached status as a dictionary.�machineTokenInfo� contractInfoZ machineIdTr* r r! � createdAtr r# r �externalAccountIDsr% )r& r' r* r r/ r0 � effectiveTor) �
effectiveFromr( ZavailableResourcesc S s& i | ] }|� d �s|d |� d��qS )rG r! rB ��get�r6 �resourcer8 r8 r9 �
<dictcomp>� s
� z$_attached_status.<locals>.<dictcomp>c S s | � dd�S �Nr! r rZ ��xr8 r8 r9 �<lambda>� � z"_attached_status.<locals>.<lambda>��key�rS r! r+ c S s | � dd�S r_ rZ r` r8 r8 r9 rb � rc �supportr3 � affordances� supportLevelr/ r$ )r �remover Z AUTO_ATTACH_RETRY_FULL_NOTICEZ AUTO_ATTACH_RETRY_TOTAL_FAILURE�copy�deepcopy�DEFAULT_STATUSZ
machine_tokenr rN rP �updater[ �listZmachine_token_filer0 Zcontract_expiry_datetimer �sortedr r �EntitlementNotFoundError�appendrR �sort� entitlements) rS �responserT rU r$ � resourcesrQ r] �ent_clsr; rg ri r8 r8 r9 �_attached_status� sz
�
� ����
�
�
�� rx c C s� t �t�}t| �}|D ]�}|�d�r.tjj}ntjj}zt | |�dd�d�}W n6 t
j k
r� t �
tjj|�dd�d�� Y qY nX |jdkr�t�� dkr�|| �}|�� }nd }|d
�|�d |d �|j||d �� q|d
jd
d� d� |S )z#Return unattached status as a dict.rG r! r rf zwithout a 'name' key)r7 r FNr+ � presentedAs)r! rB rF rG c S s | � dd�S r_ rZ r` r8 r8 r9 rb rc z$_unattached_status.<locals>.<lambda>rd )rk rl rm r r[ r � AVAILABLErP rM r r rq �LOG�debugr
Z!AVAILABILITY_FROM_UNKNOWN_SERVICE�formatr! r �on_supported_kernelrJ rr rB rs )rS ru rv r] rG rw Zlp�descr_overrider8 r8 r9 �_unattached_status� sD
�
��
�
�
��r� c
C s� t j| jdd�}||O }|r |S t�|�}g }|�di �D ]j}|d }zt| |d�}W n$ tjk
rz |� |� Y q:Y nX t
j j } |j
r�|�dd�| kr:|� |� q:|r�||d< |S )z1Remove beta services from response dict if neededzfeatures.allow_beta)�configZ
path_to_valuer+ r! rf rD r )r Zis_config_value_truerS rk rl r[ r r rq rr r �ACTIVErP �is_beta)
rS �show_allru Zconfig_allow_betaZ new_responseZreleased_resourcesr] �
resource_namerw Zenabled_statusr8 r8 r9 �_handle_beta_resources! s, �
r� c C s� t }|jj}tj}| �� \}}t�� p(g }|dkrN|jj}tj j
||d�j }n,t j
�| �d��rz|jj}d}tjj
|d�}|||| j| j| jd�}i } tD ] }
t| |
�r�t| |
�| |
< q�| |d d< |S ) aG Return a dict with execution_status, execution_details and notices.
Values for execution_status will be one of UserFacingConfigStatus
enum:
inactive, active, reboot-required
execution_details will provide more details about that state.
notices is a list of tuples with label and description items.
r )�pid� lock_holderzmarker-reboot-cmdszconfiguration changes)� operation)r, r- r Z config_pathr� r. r� � ua_config)r �INACTIVErP r
�NO_ACTIVE_OPERATIONSZcheck_lock_infor ro r� Z LOCK_HELDr} r4 �os�path�existsZ data_pathZREBOOTREQUIREDZENABLE_REBOOT_REQUIRED_TMPLZcfg_pathrS r. r �hasattr�getattr) rS Z
userStatusZ
status_valZ status_descZlock_pidr� Z notices_listr� �retr� re r8 r8 r9 �_get_config_statusE s< �
��
r� )rS r� r= c C sl | j rt| �}nt| �}|�t| �� t�� r:| �d|� t| ||�}|shdd� |� dg �D �}||d< |S )a Return status as a dict, using a cache for non-root users
When unattached, get available resources from the contract service
to report detailed availability of different resources for this
machine.
Write the status-cache when called by root.
� status-cachec S s g | ]}|� d d�dkr|�qS �rG r? rZ r5 r8 r8 r9 r: � s �zstatus.<locals>.<listcomp>r+ )
� is_attachedrx r� rn r� r Zwe_are_currently_rootZ write_cacher� r[ )rS r� ru �available_servicesr8 r8 r9 rD p s
�rD )rt �entitlement_namer= c C s` | D ]N}|� d�|kr|� d�r$dnd|� di �� d�r<dnd|� di �d� S qddi d�S ) z0Extract information from the entitlements array.�typerC r? r> Z obligationsZenableByDefaultrh )rC � auto_enabledrh rZ )rt r� r3 r8 r8 r9 � _get_entitlement_information� s �
� r� )�tokenr� r= c
C sj d}t �t�}zt| |�}W nR tjk
rn } z2t|d�rZ|jdkrZtjt j
j t j
j d��|�W 5 d}~X Y nX |�
di �}|�
di �}|�|�
dd �|�
d
d �|�
d d �|�
d g �d
�|�
d
d �|�
d�|�
d d �|�
dg �d�dd�� t�tj�} |�
d��rx|�
d�|d< |d }
|
| } | �� dk�rxt jj|d d |
�t�d�} tj| j | j d� t�d| j d � d}|�
d��r�|�
d�|d< |d }
| |
} | �� dk�r�t jj|d d |
�t�d�} tj| j | j d� t�d| j d � d}| � d �}|�r|�
d �}nt | �}|�
d g �}d d � t |d!d"� d#�D �}|D ]�}|�
d
d �}zt | |d$�}W n tj k
�r� Y �qFY nX || d%�}t!||�}|d �"|�
d&|j �|j#|d' |d( |j |k�r�d)nd*d+�� �qF|d j$d,d"� d#� t!|d-�}|d' �r$|d. �
d/�}|�r$||d d0< |�t%| �� t&| ||�}|�sbd1d � |�
d g �D �}||d < ||fS )2z�Get a status dictionary based on a token.
Returns a tuple with the status dictionary and an integer value - 0 for
success, 1 for failure
r r@ i� )r4 Zmsg_codeNrU Z accountInfor r r! rV r# )r r! r" r# rW r% T)r/ r0 r1 rX r) r/ )Z contract_id�date)Z error_msgZ
error_codezThis token is not valid.
�
� rY r( r� r+ ZresourceEntitlementsc S s g | ]}|d s|d �qS )rG r! r8 r\ r8 r8 r9 r: � s �z#simulate_status.<locals>.<listcomp>c S s | d S )Nr! r8 r` r8 r8 r9 rb � rc z!simulate_status.<locals>.<lambda>rd rf )rS ry rC r� r? r> )r! rB rC r� rG c S s | � dd�S r_ rZ r` r8 r8 r9 rb rc rg rh ri r$ c S s g | ]}|� d d�dkr|�qS r� rZ r5 r8 r8 r9 r: s �)'rk rl rm r r ZContractAPIErrorr� r@ �UserFacingErrorr
ZATTACH_INVALID_TOKENr4 r! r[ rn r �nowr ZutcZ
total_secondsZATTACH_FORBIDDEN_EXPIREDr} �strftimer �event�error�infoZATTACH_FORBIDDEN_NOT_YETZ
read_cacher rp r rq r� rr rB rs r� r� )rS r� r� r� ru Zcontract_information�eZ
contract_infoZ account_infor� Zexpiration_datetimeZdeltarA Zeffective_datetimeZ status_cacherv rt rQ r] r� rw r; Zentitlement_informationrg ri r� r8 r8 r9 �simulate_status� s�
�
�
����
�
�
�
� ���
�r� )�stringr= c C s t j�� rt�| | �S | S )z=Return colorized string if using a tty, else original string.)�sys�stdout�isatty� STATUS_COLORr[ )r� r8 r8 r9 �colorize s r� )�commandsr= c C st d}| D ] }|r|d7 }|d� |�7 }qd� tj|td dd��}d|krTd }d
}nd }d }d
jtj|||tjd�S )Nr z && � z \
� z )�widthZsubsequent_indentr� z{
z
}z{ z }z%{color}{prefix}{content}{suffix}{end})Zcolor�prefix�content�suffix�end)�join�textwrapZwrapr r} r � DISABLEGREY�ENDC)r� r� �cmdZwrapped_contentr� r� r8 r8 r9 �colorize_commands$ s0 ���r� )� column_data�headerr= c sh g }|r|� |� tdd� | D ��}|dkrPd�|�� |�� fdd�| D �� n|�dd� | D �� |S )z�Return a list of content lines to print to console for a section
Content lines will be center-aligned based on max value length of first
column.
c S s g | ]}t |d ��qS )r )�len�r6 Zpairr8 r8 r9 r: K s z.get_section_column_content.<locals>.<listcomp>r z{{:>{}}}: {{}}c s g | ]}� j |� �qS r8 )r} r� ��templater8 r9 r: N s c S s g | ] }|d �qS )r� r8 r� r8 r8 r9 r: Q s )rr �maxr} �extend)r� r� r� Ztemplate_lengthr8 r� r9 �get_section_column_content@ s
r� )r) r= c C s8 | d kr dS z | � � } W n tk
r, Y nX | �d�S )NzUnknown/Expiredz%c %Z)Z
astimezone� Exceptionr� )r) r8 r8 r9 �format_expiresU s r� )rD r= c C sN | � d��sL| � d�rVtjdddddd�g}| d D ]}|�tjf |�� q4d
�|�S tjdddd �g}| d D ]<}|� d �}|r�|n|d
}|�tj|d |d |d �� qp| � d�}|r�|�d� |�|� | � d��r|�d� t| d �� �D ]\}}|�d�||�� q�|�dt j
j g� t �
� dk�rB|�dt jg� d
�|�S g }tg}| d D ]�} | d }
| � d �}|�r~|n| d
}| d t|
�t| d �|d�} | � dd�} | dk �r�| � d d�}
|
dk �r�|�|
� |�tjf | �� �q^| d d }| � d��st|�d k�r\|�d� |�d� | � d�}|�rD|�|� t|�d k�r\|�|� | � d��r�|�d� t| d �� �D ] \}}|�d�||�� �q�|�d � g }| d! d }|�r�|�d"|f� | d d }|�r�|�d#|f� | d$ d%k�r$|�d&t| d' �f� |�d(t|�f� |�rD|�d� |�t|d)�� d
�|�S )*z&Format status dict for tabular output.r' r1 ZSERVICErz �ENTITLEDZ AUTO_ENABLEDZ DESCRIPTION)r! rG rC r� rB r+ r� )r! rG rB rF rB r! rG r ZNOTICESr. z
FEATURESz{}: {}r FrC rD )r! rC rD rB rI NrA r/ r$ r z+
Enable services with: pro enable <service>r0 ZAccountZ Subscriptionr* Zfreez Valid untilr) zTechnical support level)r� )r[ �STATUS_SIMULATED_TMPLr} rr r� �STATUS_UNATTACHED_TMPLr� rp �itemsr
Z
UNATTACHEDr4 r r~ Z)LIVEPATCH_KERNEL_NOT_SUPPORTED_UNATTACHED�
STATUS_HEADERr� � STATUS_TMPLr� r� r� )rD r� r7 r rB r re rP Zservice_warnings�service_statusrC Zfmt_argsrI Zwarning_messager$ ZpairsZ account_nameZ
contract_namer8 r8 r9 �format_tabular_ s�
��
��
���
�
�
�
r� c C s
t | �}d}t� }||d< |D ]^}|d |ks<|�d�|kr zt| |d d�}W n tjk
rj Y q Y nX |}|| �} q~q |dkr�t�d�|���| jr�t |i �}|d } |d |d< | |d< | dkr�|j
r�d |d
< n"|d r�t j j
}
nt jj
}
|
|d < |j|d < |S )
z�Return help information from an uaclient service as a dict
:param name: Name of the service for which to return help data.
:raises: UserFacingError when no help is available.
Nr! ry rf zNo help available for '{}'rD rC �enabledTZbetarG �help)r r r[ r r rq r� r} r� rR r� r rz rP rM Z help_info) rS r! rv Z
help_resourceZ
response_dictr] Z help_ent_clsZhelp_entr� Z
status_msgrG r8 r8 r9 r� � sB �
�
r� )F)F)N)[rk Zloggingr� r� r� � collectionsr r r �typingr r r r r Zuaclientr
r r r
r r Zuaclient.configr r Zuaclient.contractr r Zuaclient.defaultsr r Zuaclient.entitlementsr Z(uaclient.entitlements.entitlement_statusr r r r Zuaclient.filesr Zuaclient.files.noticesr Zuaclient.messagesr Zget_event_loggerr� Z getLogger�__name__r{ Z ESSENTIALZSTANDARDZADVANCEDr� rP ZOKGREENr� r� ZFAILrN r� rM rO Z
WARNINGYELLOWr� rL r� r� r� r� r� Z get_versionr� rm r< �strrR rx r� r� r� �boolrD r� �intr� r� r� r� r� r� r� r8 r8 r8 r9 �<module> s�
���� � ��� )H0$+ !
� � � � �
p