File: //usr/lib64/python3.9/site-packages/rhsmlib/services/__pycache__/entitlement.cpython-39.pyc
a
X�ZhF_ � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl
m Z d dlm
Z
mZ d dlmZ d dlmZ d dlmZmZ d dlmZ e�e�ZG d d
� d
�ZdS ) � N)�Union�Callable)� injection)�ugettext)�
managerlib�utils)�EntCertActionInvoker)� certificate)�
exceptions�productsc @ s< e Zd Zd*ejdd�dd�Zeeejd�dd��Z d+ee
e d �d
d �Z d,e
eef ee
e
e
ee
ejeejeee e d
�dd�Zd-eee
ed�dd�Zed.e
eje
e
eeeejeee
e d� dd��Ze dd�dd�Zeeeed�dd�Ze d �d d �Zeed �d d!�Zeed"�d#d$�Zedd �d%d&��Zd/e
e
dd'�d(d)�Z dS )0�EntitlementServiceN)�cp�returnc C s< || _ t�tj�| _t�tj�| _t�tj�| _t � | _
d S )N) r
�inj�require�IDENTITY�identityZPROD_DIRZ product_dirZENT_DIR�entitlement_dirr �
entcertlib)�selfr
� r �B/usr/lib64/python3.9/site-packages/rhsmlib/services/entitlement.py�__init__! s
zEntitlementService.__init__)�on_dater c C sp zt j �|d�}W n6 tyH t�dt�� �}ttd�j|d���Y n0 |�� t j � � �� k rlttd���|S )z�
Return new datetime parsed from date
:param on_date: String representing date
:return It returns datetime.datetime structure representing date
z%Y-%m-%dzNDate entered is invalid. Date should be in YYYY-MM-DD format (example: {date}))�datezPast dates are not allowed)
�datetime�strptime�
ValueError�time�strftime� localtime�_�formatr �now)�clsr Z
local_dater r r �
parse_date( s ��
z EntitlementService.parse_dateF)r �forcer c C s� t �t j|�}|j|kr ||_|du rJt�d� t �t j�}d |_|�� |� � | j
� � r�|� � }|�
� }|j�� }|j�� }|� � } ||||| d�}
ntd�di i dd�}
t�dt|
� � |
S )NTz'Deleting cache entitlement status cache)�statusZ status_id�reasons�
reason_ids�validZUnknown�unknownFzentitlement status: %s)r r � CERT_SORTERr �log�debug�ENTITLEMENT_STATUS_CACHE�
server_status� delete_cache�loadr �is_validZget_system_statusZget_system_status_idr( Zget_name_message_mapZget_reason_ids_mapr! �str) r r r& �sorter� status_cacheZoverall_statusZoverall_status_idr( r) r* r' r r r �
get_status< s:
� �z EntitlementService.get_statusr )� pool_subsets�matches� pool_only�match_installed�
no_overlap�
service_level�show_allr �future�
after_date�page�items_per_page�kwargsr c
K s( |
rt �td�|
�� ��t|t�r*|g}|s6g d�}||||||||| |
d�
}| �|� i }d|v r�t�| j �j
|dd�}dd� |D �|d< d |v r�| j ||dd
�}|r�d d� |D �|d < nd d� |D �|d < d
|v �r$| j ||||||| |
t
| �t
| �dd� }|�r dd� |D �|d
< n||d
< |S )NzUnknown arguments: %s�� installed�consumed� available)
r8 r9 r: r; r< r= r> r r? r@ rE T)� iso_datesc S s g | ] }|� � �qS r ��_asdict��.0�xr r r �
<listcomp>� � z0EntitlementService.get_pools.<locals>.<listcomp>rF )r= r9 rH c S s g | ]}|� � d �qS )�pool_idrI rK r r r rN � rO c S s g | ] }|� � �qS r rI rK r r r rN � rO rG ) r> r r< r; r9 r= r? r@ rA rB rH c S s g | ] }|d �qS )�idr rK r r r rN � rO )r
�ValidationErrorr! �keys�
isinstancer4 �validate_optionsr ZInstalledProductsr
�list�get_consumed_product_pools�get_available_pools�int)r r8 r9 r: r; r< r= r> r r? r@ rA rB rC �options�resultsrE rF rG r r r � get_poolsi s^
�
�
�
z EntitlementService.get_pools)r= r9 rH r c C sL t �dg d��}t �dg d��}t�tj�}|j�� }t�tj�}g } | j� � | j�
� }
t j ||d�} |d usx|d ur�t
t
| j|
��}
|r�tj} ntj} |
D �]�}
d}d}d}d}d}d}d}d}d}d}d}d}|
j}|�r�|jp�d}|jp�d}|
jjdk�r6|
jjd k�r6|j�pd}|j�p&d}|j�p2d}n d }d }d }|j}|j }|j �pXd}|j �pdd}|j }|j �r~t!d
�}nt!d �}|j"�r�t!d �}nt!d�}t!d
�}t#|
j$d��r�|
j$j%}dd� |
j&D �}g } d} t�tj��'� �r�|
j(�r$d|
j(v �r$|
j(d |v �r||
j(d } |�)|�} | �s�|
|j*v �rF| �+t!d�� n:|
j,�-� t.j.�/t0�1� �k �rr| �+t!d�� n| �+t!d�� n| �+t!d�� |d u �r�|d u �r�|d u �r�| �+|||||||
j2|||
�3� |||| | | |
j,�4� �| |
j,�-� �|�� q�| �+|||||||
j2|||
�3� ||||||| | | |
j,�4� �| |
j,�-� �|�� q�| S )N�OldConsumedStatus)�subscription_name�provides�sku�contract�account�serialrP �provides_management�active�
quantity_used� service_typer= �status_details�subscription_type�starts�ends� system_type�ConsumedStatus)r^ r_ r` ra rb rc rP rd re rf rg �rolesr= �usage�addonsrh ri rj rk rl )�
filter_stringr= � ZNo� � ZVirtualZPhysicalZYesz
Not AvailablerQ c S s i | ]}|j |j�qS r )rQ �name)rL �pr r r �
<dictcomp>. rO zAEntitlementService.get_consumed_product_pools.<locals>.<dictcomp>ZCNzSubscription is currentzSubscription is expiredzSubscription has not begunz?Subscription management service doesn't support Status Details.)5� collections�
namedtupler r r, r( Z get_subscription_reasons_mapZPOOLTYPE_CACHEr �refreshrV r Z EntitlementCertificateFilter�filter�matchr Zformat_iso8601_dateZ format_date�orderrg r= �version�major�minorrn ro rp ru r` ra rb rf Z virt_onlyr! rd �hasattrZpoolrQ r Zare_reasons_supportedZsubject�getZvalid_entitlement_certs�appendZ valid_range�endr r# r ZGMTrc r3 Zbegin) r r= r9 rH r] rm r5 Zcert_reasons_mapZpooltype_cacheZconsumed_statusesZcertsZ cert_filterZdate_formatterZcertru r` ra rb rf rg rn ro rp rl rd r} rP Zprovided_productsr( Z pool_typer r r rW � s� ��
�� ��z-EntitlementService.get_consumed_product_pools) r> r r< r; r9 r= r? r@ rA rB rH r c sP | dur t |d �} d| } nd }} d } } | ||||� ||| | d�
}
t�tj�}t�tj�}|�||
�}t|�dkr�tj| ||||||| | |
d�
}|� � }|j
|
|t � � | d�i}||_ |�
� ttd�� fdd � }� d
ur�tt||��}| dk�rL|| | }|d | | }|dk�r$| }|||� }|D ]}||d < | |d
< �q4|S )a`
Get list of available pools
:param show_all:
:param on_date:
:param no_overlap:
:param match_installed:
:param matches:
:param service_level:
:param future:
:param after_date:
:param page:
:param items_per_page:
:param iso_dates:
:return:
Trt r )
r> r r< r; r9 r= r? r@ rA rB )
Zget_allZ active_onZ overlappingZ uninstalledrq r? r@ rA rB rH )�filter_optionsZpools�timeout)� pool_datar c s$ d}| d r| d }� � � |� � kS )Nrr r= )�lower)r� Z
pool_level�r= r r � filter_pool_by_service_level� s zLEntitlementService.get_available_pools.<locals>.filter_pool_by_service_levelN� rA rB )rY r r r ZAVAILABLE_ENTITLEMENT_CACHEZget_not_obsolete_data�lenr Zget_available_entitlementsr� �uuidr Zavailable_entitlementsZ write_cache�dict�boolrV r{ )r> r r< r; r9 r= r? r@ rA rB rH Z_pageZ_items_per_pager� r �cacheZavailable_poolsr� �datar� Zlo_idxZhi_idx�itemr r� r rX x sl
� �
��
z&EntitlementService.get_available_pools)rZ r c C sD t g d���|d �s$t�td���|d rHd|d vrHt�td���n�|d rld|d vrlt�td���n�|d d ur�t d
dg��|d �s�t�td ���n�|d r�d|d vr�t�td
���n~|d r�d|d vr�t�td���nZ|d �rt d
dg��|d ��st�td���n(| j�� �s@d|d v �r@t�td���d S )NrD r8 z_Error: invalid listing type provided. Only "installed", "consumed", or "available" are allowedr> rG z0Error: --all is only applicable with --availabler z3Error: --ondate is only applicable with --availabler= rF zGError: --servicelevel is only applicable with --available or --consumedr; z<Error: --match-installed is only applicable with --availabler< z7Error: --no-overlap is only applicable with --availabler: zHError: --pool-only is only applicable with --available and/or --consumedz$Error: this system is not registered)�set�
issupersetr
rR r! � intersectionr r3 )r rZ r r r rU � s< �������z#EntitlementService.validate_options)�
unbind_method�
consumer_uuid�idsr c C sz g }g }|D ]d}z|||� |� |� W q tjyn } z,|jdkrF� |� |� t�|� W Y d}~q d}~0 0 q ||fS )a(
Method for unbinding entitlements
:param unbind_method: unbindByPoolId or unbindBySerial
:param consumer_uuid: UUID of consumer
:param ids: List of serials or pool_ids
:return: Tuple of two lists containing unbinded and not-unbinded subscriptions
i� N)r� �
connectionZRestlibException�coder- �error)r r� r� r� �successZfailureZid_�rer r r � _unbind_ids s
"z EntitlementService._unbind_ids)r c C s | j �| jj�}| j�� |S )zY
Try to remove all entitlements
:return: Result of REST API call
)r
Z unbindAllr r� r �update)r Zresponser r r �remove_all_entitlements s
z*EntitlementService.remove_all_entitlements)�pool_idsr c C sn g }t �|�}| j�� | j�|�}| �| jj| jj |�\}}|rZ|D ]}|�
|| � qF| j � � |||fS )z�
Try to remove entitlements by pool IDs
:param pool_ids: List of pool IDs
:return: List of serial numbers of removed subscriptions
)
r �unique_list_itemsr rz Zlist_serials_for_pool_idsr� r
ZunbindByPoolIdr r� �extendr r� )r r� �removed_serialsZ _pool_idsZpool_id_to_serialsZ
removed_poolsZunremoved_poolsrP r r r � remove_entitlements_by_pool_ids) s
�
z2EntitlementService.remove_entitlements_by_pool_ids)�serialsr c C s6 t �|�}| �| jj| jj|�\}}| j�� ||fS )z�
Try to remove pools by Serial numbers
:param serials: List of serial numbers
:return: Tuple of two items: list of serial numbers of already removed subscriptions and list
of not removed serials
) r r� r� r
ZunbindBySerialr r� r r� )r r� Z_serialsr� Zunremoved_serialsr r r � remove_entitlements_by_serials@ s
�
z1EntitlementService.remove_entitlements_by_serialsc C s> t jt jdd�} t �t j�}t�d|j � d|_| �� dS )z�
This callback function is called, when there is detected any change in directory with entitlement
certificates (e.g. certificate is installed or removed)
N)r z#Clearing in-memory cache of file %s) r r r, r/ r- r. Z
CACHE_FILEr0 r2 )r5 r6 r r r �reloadP s
zEntitlementService.reload)� remove_cacher& r c C sn |du r@t �t j�}|�� r$|�� t �t j�}|�� r@|�� |du rd| j�| j j
d�sdt � d� | j
�� S )z�
Try to refresh entitlement certificate(s) from candlepin server
:return: Report of EntCertActionInvoker
TzOWarning: Unable to refresh entitlement certificates; service likely unavailable)r r ZCONTENT_ACCESS_CACHE�exists�removeZCONTENT_ACCESS_MODE_CACHEr1 r
Z regenEntitlementCertificatesr r� r- r. r r� )r r� r& Zcontent_accessZcontent_access_moder r r rz \ s
zEntitlementService.refresh)N)NF) NNNNNNNNNNr r )NNF) NNNNNNNNr r F)FF) �__name__�
__module__� __qualname__r� Z
UEPConnectionr � classmethodr4 r r% r� r� r7 r rV rY r\ rW � staticmethodrX rU r �tupler� r� r� r� r� rz r r r r r s� / �
� K �� F ��o! r ) rx r Zloggingr �typingr r Zsubscription_managerr r Zsubscription_manager.i18nr r! r r Z subscription_manager.entcertlibr Zrhsmr Zrhsmlib.servicesr
r Zrhsm.connectionr� Z getLoggerr� r- r r r r r �<module> s