| Server IP : 209.38.156.173 / Your IP : 216.73.216.122 [ Web Server : Apache/2.4.52 (Ubuntu) System : Linux lakekumayuhotel 5.15.0-136-generic #147-Ubuntu SMP Sat Mar 15 15:53:30 UTC 2025 x86_64 User : root ( 0) PHP Version : 8.1.2-1ubuntu2.22 Disable Function : NONE Domains : 2 Domains MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib/python3/dist-packages/uaclient/__pycache__/ |
Upload File : |
o
F��c�` � @ 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 dlmZ d dlmZmZ d dlm Z m Z d d l m Z d d
l m!Z!m"Z"m#Z#m$Z$ d d l%m&Z& e�'� Z(e�)e*�Z+d Z,d
Z-dZ.e$j/j0e&j1e$j/j0 e&j2 e$j3j0e&j4e$j3j0 e&j2 e$j5j0e&j6e$j5j0 e&j2 e$j7j0e&j6e$j7j0 e&j2 e!j8j0e&j1e!j8j0 e&j2 e!j9j0e&j6e!j9j0 e&j2 e,e&j1e, e&j2 e-e&j1e- e&j2 e.e&j1e. e&j2 i Z:dZ;dZ<dZ=dZ>i dd�dd�de�?� �dd�dd�dd�d d�d d�d g �d e#j3j0�d ej@�d!i �d"g �d#d$d$d$g e$j5j0d%��d&d$d$d$g d'��d(d�ZAd)d*� ZBd+e eCe
f fd,d-�ZDd.ed+e eCe
f fd/d0�ZEd.ed+e eCe
f fd1d2�ZFd+e eCe
f fd3d4�ZGd+e eCe
f fd5d6�ZHdQd.ed7eId+e eCe
f fd8d9�ZJd:e e eCe
f d;eCd+e eCe
f fd<d=�ZK dQd>eCd7eId+ee eCe
f eLf fd?d@�ZMdAeCd+eCfdBdC�ZNdDe e eC d+eCfdEdF�ZO dRdGe eeCeCf dHe
eC d+e eC fdIdJ�ZPd e
e d+eCfdKdL�ZQd9e eCe
f d+eCfdMdN�ZRdOdP� ZSdS )S� N)� OrderedDict)�datetime�timezone)�Any�Dict�List�Optional�Tuple)� event_logger�
exceptions�messages�system�util�version)�UAConfig)�get_available_resources�get_contract_information)�ATTACH_FAIL_DATE_FORMAT�PRINT_WRAP_WIDTH)�entitlement_factory)�ContractStatus�UserFacingAvailability�UserFacingConfigStatus�UserFacingStatus)�TxtColor� essential�standard�advancedz({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}�_doczUContent provided in json response is currently considered Experimental and may change�_schema_versionz0.1r �
machine_id�attachedF� effective�expires�origin�services�execution_status�execution_details�features�notices�contract� ��id�name�
created_at�products�tech_support_level�account�r. r- r/ �external_account_ids� simulatedc C s dd� | � � D �S )Nc S s$ g | ]}|j j|jj|jjd ��qS ))r. � reason_code�reason)� entitlementr. � named_msg�msg��.0�service� r> �1/usr/lib/python3/dist-packages/uaclient/status.py�
<listcomp>j s ���z,_get_blocked_by_services.<locals>.<listcomp>)� blocking_incompatible_services)�entr> r> r? �_get_blocked_by_servicesi s �rC �returnc C s� d}d }| � � }| j|v rdnd}|tjkrtj}n | j|v r(tj}|| j }n| �� \}}|r3|j}|tjkr:d}t | �}| j
| j |j |j ||||d�S )Nr+ �no�yes)r. � description�entitled�status�status_details�description_override� available�
blocked_by)
�contract_statusr. r �
UNENTITLEDr � UNAVAILABLE� INAPPLICABLE�user_facing_statusr: rC �presentation_namerG �value) rB �inapplicable_resourcesrJ rK rN rL �
ent_status�detailsrM r> r> r? �_attached_service_statust s0
�rX �cfgc C s� | j �dtj� t�t�}| jd }|d }tj j
}|� |d d|� d�| j �
� p*g |d |d |� d d�|� d
g �|d �| jjd | jjd | jj� d d�| jj� d g �d
�d�� |� d�rf| jj|d< |� d�rq|d |d< | j� d�}|s}t| �}dd� t|dd� d�D �}|D ]'}z t| |� dd�d�}W n
tjy� Y q�w || �} |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.r+ �machineTokenInfo� contractInfo� machineIdTr$ r- r. � createdAtr0 r, �externalAccountIDsr3 )r r! r$ r) r* r2 � effectiveTor# �
effectiveFromr" �availableResourcesc S s& i | ]}|� d �s|d |� d��qS )rL r. rG ��get�r< �resourcer> r> r? �
<dictcomp>� s ��z$_attached_status.<locals>.<dictcomp>c S � | � dd�S �Nr. r+ rb ��xr> r> r? �<lambda>� � z"_attached_status.<locals>.<lambda>��key�rY r. r% c S rg rh rb ri r> r> r? rk � rl �supportr8 � affordances� supportLevelr* r1 )� notice_file�
try_remover � AUTO_ATTACH_RETRY_NOTICE_PREFIX�copy�deepcopy�DEFAULT_STATUS�
machine_tokenr rQ rT �updaterc �read�machine_token_filer2 �contract_expiry_datetimer �sortedr r �EntitlementNotFoundError�appendrX �sort� entitlements) rY �responserZ r[ r1 � resourcesrU re �ent_clsrB rp rr r> r> r? �_attached_status� st
�
�����
�
���� r� c C s� t �t�}t| �}|D ]G}|�d�rtjj}ntjj}z t | |�dd�d�}W n t
j y? t �
tjj|�dd�d�� Y q w |d �|�d|d �|j|d �� q |d jd
d � d � |S )
z#Return unattached status as a dict.rL r. r+ ro zwithout a 'name' key)r= r% � presentedAs)r. rG rL c S rg rh rb ri r> r> r? rk � rl z$_unattached_status.<locals>.<lambda>rm )rv rw rx r rc r � AVAILABLErT rP r r r �LOG�debugr �!AVAILABILITY_FROM_UNKNOWN_SERVICE�formatr� rG r� )rY r� r� re rL r� r> r> r? �_unattached_status� s4
�
�����r� c
C s� t j| jdd�}||O }|r|S t�|�}g }|�di �D ]3}|d }zt| |d�}W n tjy; |� |� Y q w t
j j } |j
rK|�dd�| krP|� |� q |rW||d< |S )z1Remove beta services from response dict if neededzfeatures.allow_beta)�config�
path_to_valuer% r. ro rI r+ )r �is_config_value_truerY rv rw rc r r r r� r �ACTIVErT �is_beta)
rY �show_allr� �config_allow_beta� new_response�released_resourcesre �
resource_namer� �enabled_statusr> r> r? �_handle_beta_resources s0 �
�
�r� c
C s� t }|jj}tj}| �� \}}| j�� pg }|dkr(|jj}tj j
||d�j }n%t j
�| �d��rM|jj}d}|D ] \}} |dkrE| } nq9tjj
|d�}|||| j| j| jd�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 changeszReboot required�� operation)r& r' r) � config_pathr� r( )r �INACTIVErT r �NO_ACTIVE_OPERATIONS�check_lock_infors r{ r� � LOCK_HELDr� r: �os�path�exists� data_path�REBOOTREQUIRED�ENABLE_REBOOT_REQUIRED_TMPL�cfg_pathrY r( )
rY �
userStatus�
status_val� status_desc�lock_pidr� r) r� �labelrG r> r> r? �_get_config_status$ s< �� ���r� r� c C s� | j rt| �}nt| �}|�t| �� | jr,| �d|� t�� s,| j �
dt j j
dd�� t| ||�}|sCdd� |�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-cacher+ z
fix operationr� c S � g | ] }|� d d�dkr|�qS �rL rF rb r; r> r> r? r@ j � ��zstatus.<locals>.<listcomp>r% )� is_attachedr� r� rz r� � root_mode� write_cacher
�
should_rebootrs �remover r� r� r� rc )rY r� r� �available_servicesr> r> r? rI K s(
��
�rI r� �entitlement_namec C s` | D ]'}|� d�|kr)|� d�rdnd|� di �� d�r dnd|� di �d� S qddi d�S ) z0Extract information from the entitlements array.�typerH rF rE � obligations�enableByDefaultrq )rH � auto_enabledrq rb )r� r� r8 r> r> r? � _get_entitlement_informationt s �
�� r� �tokenc
C s^ d}t �t�}zt| |�}W n$ tjy2 } zt|d�r,|jdkr,tjt j
j t j
j d��|�d}~ww |�
di �}|�
di �}|�|�
dd �|�
d
d �|�
d d �|�
d g �d
�|�
d
d �|�
d�|�
d d �|�
dg �d�dd�� t�tj�} |�
d�r�|�
d�|d< t�|d �}
|
| } | �� dkr�t jj|d d |
�t�d�} tj| j | j d� t� d| j d � d}|�
d�r�|�
d�|d< t�|d �}
| |
} | �� dkr�t j j|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 ]G}|�
d
d �}zt!| |d$�}W n tj"�y< Y �q w || d%�}t#||�}|d �$|�
d&|j �|j%|d' |d( |j |v�rad)nd*d+�� �q |d j&d,d"� d#� t#|d-�}|d' �r�|d. �
d/�}|�r�||d d0< |�t'| �� t(| ||�}|�s�d1d � |�
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 �codei� )r: �msg_codeNr[ � accountInfor- r+ r. r] r0 )r- r. r/ r0 r^ r3 T)r* r2 r5 r_ r# r* )� contract_id�date)� error_msg�
error_codezThis token is not valid.
�
� r` r" r� r% �resourceEntitlementsc S s g | ]
}|d s|d �qS )rL r. r> rd r> r> r? r@ � s ��z#simulate_status.<locals>.<listcomp>c S s | d S )Nr. r> ri r> r> r? rk � s z!simulate_status.<locals>.<lambda>rm ro )rY r� rH r� rF rE )r. rG rH r� rL c S rg rh rb ri r> r> r? rk � rl rp rq rr r1 c S r� r� rb r; r> r> r? r@ � r� ))rv rw rx r r �ContractAPIError�hasattrr� �UserFacingErrorr �ATTACH_INVALID_TOKENr: r. rc rz r �nowr �utcr �parse_rfc3339_date�
total_seconds�ATTACH_FORBIDDEN_EXPIREDr� �strftimer �event�error�info�ATTACH_FORBIDDEN_NOT_YET�
read_cacher r~ r r r� r� rG r� r� r� )rY r� r� �retr� �contract_information�e�
contract_info� account_infor� �expiration_datetime�delta�message�effective_datetime� status_cacher� r� rU re r� r� rB �entitlement_informationrp rr r� r> r> r? �simulate_status� s�
���
�
����
�
�
� �
� ���
�r� �stringc C s t j�� r t�| | �S | S )z=Return colorized string if using a tty, else original string.)�sys�stdout�isatty� STATUS_COLORrc )r� r> r> r? �colorize s r� �commandsc C st d}| D ]}|r |d7 }|d� |�7 }qd� tj|td dd��}d|v r*d }d
}nd }d }d
jtj|||tjd�S )Nr+ z && � z \
� z )�width�subsequent_indentr� z{
z
}z{ z }z%{color}{prefix}{content}{suffix}{end})�color�prefix�content�suffix�end)�join�textwrap�wrapr r� r � DISABLEGREY�ENDC)r� r� �cmd�wrapped_contentr� r� r> r> r? �colorize_commands s,
���r� � column_data�headerc sl dg}|r
|� |� tdd� | D ��}|dkr*d�|�� |�� fdd�| D �� |S |�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.
r+ c S s g | ]}t |d ��qS )r )�len�r< �pairr> r> r? r@ / s z.get_section_column_content.<locals>.<listcomp>r z{{:>{}}}: {{}}c s g | ]}� j |� �qS r> )r� r ��templater> r? r@ 2 s c S s g | ]}|d �qS )r� r> r r> r> r? r@ 5 s )r� �maxr� �extend)r� r� r� �template_lengthr> r r? �get_section_column_content$ s
�r c C s6 | d u rdS z| � � } W n ty Y nw | �d�S )NzUnknown/Expiredz%c %Z)�
astimezone� Exceptionr� )r# r> r> r? �format_expires9 s �
r c C s� | � d�s�| � d�r,tjdddddd�g}| d D ]
}|�tjd&i |��� qd
�|�S tjdddd �g}| d D ]
}|�tjd&i |��� q9| � d �rZ|�t| � d �pUg d
d�� | � d�rz|�d� t| d � � �D ]
\}}|�d�||�� ql|�dt
j j g� d
�|�S t
g}| d D ]-}|d }|� d�}|r�|n|d }|d t|�t|d �|d�} |�tjd&i | ��� q�| d d }
| � d �r�|�t| � d �p�g d
d�� | � d�r�|�d� t| d � � �D ]
\}}|�d�||�� q�|�d� g } | d d } | �r
| �d | f� | d d }
|
�r | �d |
f� | d d k�r8| �d!t| d" �f� | �d#t|
�f� | �rC|�t| d$�� d
�|�S )'z&Format status dict for tabular output.r! r5 �SERVICEr� �ENTITLED� AUTO_ENABLED� DESCRIPTION)r. rL rH r� rG r% r� )r. rL rG r) �NOTICES)r� r( z
FEATURESz{}: {}r+ rH rK rG r. rI )r. rH rI rG r* r1 z+
Enable services with: pro enable <service>r2 �Account� Subscriptionr$ �freez Valid untilr# zTechnical support level)r� Nr> )rc �STATUS_SIMULATED_TMPLr� r� r� �STATUS_UNATTACHED_TMPLr r r~ �itemsr �
UNATTACHEDr: �
STATUS_HEADERr� � STATUS_TMPLr )rI r� r= rn rT �service_statusrH �descr_overriderG �fmt_argsr1 �pairs� account_name�
contract_namer> r> r? �format_tabularC s�
��
��
��
�
�
��
r c C s t | �}d}t� }||d< |D ],}|d |ks |�d�|kr;z
t| |d d�}W n
tjy2 Y qw |}|| �} nq|du rHt�d�|���| jrjt |i �}|d } |d |d< | |d< | dkri|j
rid |d
< n|d rst 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. r� ro zNo help available for '{}'rI rH �enabledT�betarL �help)r r rc r r r r� r� r� rX r� r r� rT rP � help_info) rY r. r� �
help_resource�
response_dictre � help_ent_cls�help_entr �
status_msgrL r> r> r? r# � sF
��� �
�
r# )F)N)Trv �loggingr� r� r� � collectionsr r r �typingr r r r r �uaclientr
r r r
r r �uaclient.configr �uaclient.contractr r �uaclient.defaultsr r �uaclient.entitlementsr �(uaclient.entitlements.entitlement_statusr r r r �uaclient.messagesr �get_event_loggerr� � getLogger�__name__r� � ESSENTIAL�STANDARD�ADVANCEDr� rT �OKGREENr� r� �FAILrQ r� rP r
rO r� r r r r � get_versionr� rx rC �strrX r� r� r� r� �boolrI r� �intr� r� r� r r r r# r> r> r> r? �<module> s�
����� ������� �
� � �
������ � "G#$ ')��
����
� ���
�
[