OwlCyberSecurity - MANAGER
Edit File: status.cpython-38.pyc
U ��g r � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dlm Z d dl m Z m Z m Z mZmZ d dlmZmZmZmZmZmZmZ d dlmZ d dlmZm Z d d l m Z m Z d d l 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+m,Z,m-Z-m.Z. d dl/m0Z0 d dl1m2Z2 e�3� Z4e�5e�6e7��Z8dZ9dZ:dZ;e)j<j=e2j>ej? e2j@ e)jAj=e2jBejC e2j@ e)jDj=e2jEejF e2j@ e)jGj=e2jEejH e2j@ e)jIj=e2jJejK e2j@ e&jLj=e2j>ejM e2j@ e&jNj=e2jEejO e2j@ e9e2j>ejP e2j@ e:e2j>ejQ e2j@ e;e2j>ejR e2j@ i ZSdZTdZUdjVejWejXejYejZd�Z[dZ\dZ]dde�^� dddddg e(jAj=ej_i g d d d g e)jDj=d �d d d g d �dd �Z`d d!� Zae ebe f d"�d#d$�Zce e ebe f d%�d&d'�Zde e ebe f d%�d(d)�Zee ebe f d"�d*d+�ZfdFe ege ebe f d,�d-d.�Zhe e ebe f ebe ebe f d/�d0d1�ZidGebegee ebe f ejf d2�d3d4�Zkebebd5�d6d7�Zle e eb ebd8�d9d:�ZmdHe eebebf eeb e eb d;�d<d=�Znee ebd>�d?d@�ZodIe ebe f egebdA�dBdC�ZpdDdE� ZqdS )J� N)� OrderedDict)�datetime�timezone)�Enum)�Any�Dict�List�Optional�Tuple)� event_logger� exceptions� livepatch�lock�messages�util�version)� _is_attached)�UA_CONFIGURABLE_KEYS�UAConfig)�get_available_resources�get_contract_information)�ATTACH_FAIL_DATE_FORMAT�PRINT_WRAP_WIDTH)�entitlement_factory)�ContractStatus�UserFacingAvailability�UserFacingConfigStatus�UserFacingStatus)� machine_token�notices� state_files�user_config_file)�Notice)�TxtColorZ essentialZstandardZadvancedz({name: <17}{available: <11}{description}zJ{name: <17}{available: <11}{entitled: <11}{auto_enabled: <16}{description}z4{name: <17}{entitled: <10}{status: <13}{description}��name�entitled�status� descriptionz4{name: <17}{entitled: <19}{status: <22}{description}z={marker} {name: <15}{entitled: <19}{status: <22}{description}zUContent provided in json response is currently considered Experimental and may changez0.1F� ��idr% � 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 s4 g | ],}|j js|j jn|j j|jj|jjd ��qS ))r% Z reason_code�reason)� entitlement� is_variantr% � variant_nameZ named_msg�msg��.0�service� rD �1/usr/lib/python3/dist-packages/uaclient/status.py� <listcomp>� s �� �z,_get_blocked_by_services.<locals>.<listcomp>)Z blocking_incompatible_services)�entrD rD rE �_get_blocked_by_services� s �rH )�returnc s� d }d}| � � }| �� }| j�kr&dnd}i }|tjkr@tj} nz| j�kr\tj} �| j }n^| �� \} } | tj kr�| j| j d�}n | r�| j }| tjkr�d}| j r�� �fdd�| j � � D �}t | �} | j| j|j| j|||| |d� } | js�|| d< | S ) Nr) �no�yes)�code�messagec s$ i | ] \}}|t |� d ��� ��qS ))�cfg)�_attached_service_status)rB r? Z variant_cls�rN �inapplicable_resourcesrD rE � <dictcomp>� s � �z,_attached_service_status.<locals>.<dictcomp>) r% r( r& r' �status_details�description_override� available� blocked_by�warning�variants)�status_description_override�contract_statusr% r � UNENTITLEDr � UNAVAILABLE� INAPPLICABLEZuser_facing_status�WARNINGr@ rX �itemsrH Zpresentation_namer( �valuer> ) rG rQ rN rW rS rT rZ rU rX Z ent_statusZdetailsrV �service_statusrD rP rE rO � sL � � � rO )rN rI c C s� t �tj� t �tj� t| �jr.t �tj� t� t �}t � | �}|j d }|d }t jj}|�|d d|�d�t �� pzg |d |d |�dd �|�d g �|d �|jd |jd |j�dd �|j�d g �d �d�� |�d�r�|j|d< |�d�r�|d |d< |j �d�}|�st| �}dd� t|dd� d�D �}|D ]T}zt| |�dd �d�} W n tjk �rp Y �q6Y nX |d �t| || �� �q6|d j dd� d� |� � �d i ��d �} | �r�| �d i ��d �} | �r�| |d d!< |S )"z8Return configuration of attached status as a dictionary.�machineTokenInfo� contractInfoZ machineIdTr4 r+ r% � createdAtr) r- r* �externalAccountIDsr/ )r0 r1 r4 r r9 r: � effectiveTor3 � effectiveFromr2 ZavailableResourcesc S s& i | ] }|� d �s|d |� d��qS )rU r% r( ��get�rB �resourcerD rD rE rR s � z$_attached_status.<locals>.<dictcomp>c S s | � dd�S �Nr% r) rh ��xrD rD rE �<lambda> � z"_attached_status.<locals>.<lambda>��key�rN r% r5 c S s | � dd�S rl rh rm rD rD rE ro rp �supportr= � affordances� supportLevelr9 r. ) r �remover" Z AUTO_ATTACH_RETRY_FULL_NOTICEZ AUTO_ATTACH_RETRY_TOTAL_FAILUREr Z is_attached_and_contract_validZCONTRACT_EXPIRED�copy�deepcopy�DEFAULT_STATUSr Zget_machine_token_filer r] r` �updateri �listr: Zcontract_expiry_datetimer �sortedr r �EntitlementNotFoundError�appendrO �sort� entitlements) rN �responseZmachine_token_filerb rc r. � resourcesrQ rk rG rt rv rD rD rE �_attached_status� sp � ���� � �� r� c C s� t �t�}t| �}|D ]�}|�d�r.tjj}ntjj}zt | |�dd�d�}W n. t j k r| t � d|�dd�� Y qY nX |jdkr�t�� tjjkr�|�� }nd}|d �|�d |d �|j||d �� q|d jd d � d� |S )z#Return unattached status as a dict.rU r% r) rs z@Ignoring availability of unknown service %s from contract serverzwithout a 'name' keyr Nr5 � presentedAs)r% r( rT rU c S s | � dd�S rl rh rm rD rD rE ro G rp z$_unattached_status.<locals>.<lambda>rq )rx ry rz r ri r � AVAILABLEr` r\ r r r~ �LOG�debugr% r �on_supported_kernel�LivepatchSupport� UNSUPPORTEDrY r r( r� )rN r� r� rk rU rG �descr_overriderD rD rE �_unattached_status s> � ��� ��r� c C s� t }|jj}tj}t�� \}}t�� p(g }|dkrL|j j}tj j ||d�}n"t j jrn|jj}d}tjj |d�}|||| j| j| jd�}tjj�� } tD ]<} t| | �r�| | dkr�t| | �} t| t �r�| j} | | | < 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_holderzconfiguration changes)� operation)r6 r7 r Z config_path�configr8 Nr� � ua_config) r �INACTIVEr` r �NO_ACTIVE_OPERATIONSr Zcheck_lock_infor r| �ACTIVEZ LOCK_HELD�formatr Zreboot_cmd_marker_fileZ is_presentZREBOOTREQUIREDZENABLE_REBOOT_REQUIRED_TMPLZcfg_pathrN r8 r! Z user_configZ public_configZto_dictr �hasattr�getattr� isinstancer ) rN Z userStatusZ status_valZ status_descZlock_pidr� Z notices_listr� �retr� rr �valrD rD rE �_get_config_statusL sB ��� r� )rN �show_allrI c C sd t | �jrt| �}nt| �}|�t| �� t�� r>tj � |� |s`dd� |� 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. c S s g | ]}|� d d�dkr|�qS �rU rK rh rA rD rD rE rF � s �zstatus.<locals>.<listcomp>r5 ) r � is_attachedr� r� r{ r� r Zwe_are_currently_rootr Zstatus_cache_file�writeri )rN r� r� �available_servicesrD rD rE r' { s �r' )r� �entitlement_namerI 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.�typer& rK rJ Z obligationsZenableByDefaultru )r& � auto_enabledru rh )r� r� r= rD rD rE � _get_entitlement_information� s �� � r� )�tokenr� rI c C s6 d}t �t�}zt| |�}W nD tjk r` } z$t|d�rL|jdkrLt�� �|�W 5 d}~X Y nX |� di �}|� di �}|� |� dd�|� d d�|� d d�|� d g �d �|� d d�|� d�|� d d�|� d g �d�dd�� t � t j�} |� d��rp|� d�|d< |d } | | } | �� dk�rptjj|d d | �t�d�} tj| j| jd� t�tjd | j d � d}|� d��r�|� d�|d< |d } | | } | �� dk�r�tjj|d d | �t�d�} tj| j| jd� t�tjd | j d � d}t | �}dd� t |d d � d �D �}|� d g �}|D ]�}|� d d�}zt | |d �}W n tj k �rf Y �q(Y nX t ||�}|d! �!|� d"|j�|j"|d# |d$ |j|k�r�d%nd&d'�� �q(|d! j#d(d � d � t |d)�}|d# �r�|d* � d+�}|�r�||d d,< |� t$| �� |�s.d-d� |� d!g �D �}||d!< ||fS ).z�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 rL i� Nrc Z accountInfor+ r) r% rd r- )r+ r% r, r- re r/ T)r9 r: r; rf r3 r9 )Z contract_id�date)Z error_msgZ error_code� � rg r2 c S s g | ]}|d s|d �qS )rU r% rD rj rD rD rE rF � s �z#simulate_status.<locals>.<listcomp>c S s | d S )Nr% rD rm rD rD rE ro � rp z!simulate_status.<locals>.<lambda>rq ZresourceEntitlementsrs r5 r� r& r� rK rJ )r% r( r& r� rU c S s | � dd�S rl rh rm rD rD rE ro rp rt ru rv r. c S s g | ]}|� d d�dkr|�qS r� rh rA rD rD rE rF s �)%rx ry rz r r ZContractAPIErrorr� rL ZAttachInvalidTokenErrorri r{ r �nowr ZutcZ total_secondsr ZE_ATTACH_FORBIDDEN_EXPIREDr� �strftimer �event�errorr@ r% �infoZSTATUS_TOKEN_NOT_VALIDZE_ATTACH_FORBIDDEN_NOT_YETr r} r r~ r� r r( r� r� )rN r� r� r� r� Zcontract_information�eZ contract_infoZ account_infor� Zexpiration_datetimeZdeltarM Zeffective_datetimer� rQ r� rk r� rG Zentitlement_informationrt rv r� rD rD rE �simulate_status� s� � ���� �� ��� � �� �r� )�stringrI c C s t j�� rt�| | �S | S )z=Return colorized string if using a tty, else original string.)�sys�stdout�isatty�STATUS_HUMANIZE_COLORIZEri )r� rD rD rE �for_human_colorized% s ��r� )�commandsrI 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� rD rD rE �colorize_commands. s0 ���r� )� column_data�headerrI 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�rB ZpairrD rD rE rF U s z.get_section_column_content.<locals>.<listcomp>r z{{:>{}}}: {{}}c s g | ]}� j |� �qS rD )r� r� ��templaterD rE rF X s c S s g | ] }|d �qS )r� rD r� rD rD rE rF [ s )r �maxr� �extend)r� r� r� Ztemplate_lengthrD r� rE �get_section_column_contentJ s r� )r3 rI c C s: | d krt jS z | �� } W n tk r. Y nX | �d�S )Nz%c %Z)r Z STATUS_CONTRACT_EXPIRES_UNKNOWNZ astimezone� Exceptionr� )r3 rD rD rE �format_expires_ s r� )r' r� rI c C s@ | � d��s�| � d�rv| � dd�s(tjS tjtjtjtjtjtj d�g}| � dg �D ]}|� tjf |�� qTd� |�S | � dd�s�tjg}n�t jtjtjtj d�g}| � dg �D ]Z}|� d�}|r�|n |� d d �}|� d �d kr�tj ntj}|� t j|� d d �||d�� q�| � d�}|�r2|� tj� |�|� | � d��r�|� dtj � t| � di ��� �D ] \}} |� d�|| �� �qb|�s�|�d tjg� |�d tjjg� t�� tjjk�r�|�d tjg� d� |�S g } d} | � dd��s�tjg}�ndt g}| � dg �D �]N} | � dd �} | � d�}|�r.|n | � d d �}| � d d �t | �t | � dd ��|d�}| � dd�}|dk �r�|� dd�}|dk �r�| � |� | � d�}|�r�|�s�d} d�|d �|d < |� t jf |�� |�r|�rt |�� �D ]h\}\}}|t |�d k�rdnd }|� t!j||� d �t |� dd ��t |� dd ��|� d d �d �� �q�q| �rv|� d � |� tj"� | � d��s�t | �d k�r�|� d � |� tj� | � d�}|�r�|�|� t | �d k�r�|�| � | � d��r&|� dtj � t| � di ��� �D ] \}} |� d�|| �� �q|� d � |�sV| �rJ|� tj#� n |� tj� |� tj$jd d �� g }| � d!i �� d d"�}|�r�|� tj%|f� | � d#i �� d d"�}|�r�|� tj&|f� | � d$d�d%k�r|� tj't(| � d&��f� | � d#i �� d'd"�}|� tj)t |�f� |�r6|� d � |�t*|d(�� d� |�S ))z&Format status dict for tabular output.r1 r; r5 N)r% rU r& r� r( r� )r% rU r( rT r( r) rU rK r% r r8 z{}: {}Fr& r' r$ rW rM rX Tz{}*r� u ├u └)�markerr% r& r' r( r zpro enable <service>)Zcommandr: �unknownr9 r4 Zfreer3 r. )r� )+ri r Z STATUS_NO_SERVICES_AVAILABLE�STATUS_SIMULATED_TMPLr� �STATUS_SERVICEZSTATUS_AVAILABLE�STATUS_ENTITLEDZSTATUS_AUTO_ENABLED�STATUS_DESCRIPTIONr r� �STATUS_UNATTACHED_TMPLZSTANDALONE_YESZ STANDALONE_NOZSTATUS_NOTICESr� ZSTATUS_FEATURESr} r_ ZSTATUS_ALL_HINTZ E_UNATTACHEDr@ r r� r� r� Z)LIVEPATCH_KERNEL_NOT_SUPPORTED_UNATTACHED� STATUS_HEADERr� � STATUS_TMPL� enumerater� �VARIANT_STATUS_TMPLZSTATUS_SERVICE_HAS_VARIANTSZ STATUS_ALL_HINT_WITH_VARIANTSZ"STATUS_FOOTER_ENABLE_SERVICES_WITHZSTATUS_FOOTER_ACCOUNTZSTATUS_FOOTER_SUBSCRIPTIONZSTATUS_FOOTER_VALID_UNTILr� ZSTATUS_FOOTER_SUPPORT_LEVELr� )r' r� r� rC r� r( rU r rr r` Zservice_warningsZ has_variantsra r& Zfmt_argsrW Zwarning_messagerX �idx�_Zvariantr� ZpairsZ account_nameZ contract_namer. rD rD rE �format_tabulari s6 �� �� � � �� �� ��� � � �� � � �� �� �� ��� r� c C s� t | �}d}t� }||d< |D ]V}|d |ks<|�d�|kr zt| |d d�}W n tjk rj Y q Y nX |} qvq |dkr�tj|d��t| �jr�t |i | �}|d }|d |d< ||d< n"|d r�t j j } nt j j } | |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: UbuntuProError when no help is available. Nr% r� rs )r% r' r& rU �help)r r ri r r r~ Z NoHelpContentr r� rO r r� r` r\ Z help_info) rN r% r� Z help_resourceZ response_dictrk Zhelp_entra Z status_msgrU rD rD rE r� % s2 r� )F)F)N)F)rrx Zloggingr� r� � collectionsr r r �enumr �typingr r r r r Zuaclientr r r r r r r Z(uaclient.api.u.pro.status.is_attached.v1r Zuaclient.configr r Zuaclient.contractr r Zuaclient.defaultsr r Zuaclient.entitlementsr Z(uaclient.entitlements.entitlement_statusr r r r Zuaclient.filesr r r r! Zuaclient.files.noticesr" Zuaclient.messagesr# Zget_event_loggerr� Z getLoggerZ replace_top_level_logger_name�__name__r� Z ESSENTIALZSTANDARDZADVANCEDr� r` ZOKGREENZSTATUS_STATUS_ENABLEDr� r� ZFAILZSTATUS_STATUS_DISABLEDr] r� ZSTATUS_STATUS_INAPPLICABLEr\ ZSTATUS_STATUS_UNAVAILABLEr^ Z WARNINGYELLOWZSTATUS_STATUS_WARNINGZENTITLEDZSTATUS_ENTITLED_ENTITLEDr[ ZSTATUS_ENTITLED_UNENTITLEDZSTATUS_SUPPORT_ESSENTIALZSTATUS_SUPPORT_STANDARDZSTATUS_SUPPORT_ADVANCEDr� r� r� r� r� r� Z STATUS_STATUSr� r� r� r� Z get_versionr� rz rH �strrO r� r� r� �boolr' r� �intr� r� r� r� r� r� r� rD rD rD rE �<module> s $ ������ �� �� ���(� ���� � ;G./ � � � x � � =