File: //usr/lib64/python3.9/site-packages/rhsmlib/dbus/objects/__pycache__/entitlement.cpython-39.pyc
a
X�Zh�& � @ s� d dl Z d dlmZmZ d dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZ e� e�e�ZG dd � d e j�ZG d
d � d e j�ZdS ) � N)�List�Union)�
UEPConnection)� constants� base_object�util�
dbus_utils)�EntitlementService)�init_dep_injection)�Localec @ s� e Zd Zeed�dd�Zeeed�dd�Zeed�dd �Zee eee d
�d d �Z ee eee d
�dd�Z
ee j d�dd�Z dS )� EntitlementDBusImplementation)�on_date�returnc
C s~ |dkr dn| � |�}| ji d�}t|�}z|j|dd�}W n< tyx } z$t�|� t�t |���W Y d}~n
d}~0 0 |S )zGet status of entitlements.� N)�optionsT)r
�force)
� _parse_date� build_uepr �
get_status� Exception�log� exception�dbus�
DBusException�str)�selfr
�uep�service�status�exc� r �F/usr/lib64/python3.9/site-packages/rhsmlib/dbus/objects/entitlement.pyr " s
$z(EntitlementDBusImplementation.get_status)r �
proxy_optionsr c C s� |� dd�}|dkr"| �|�|d< |� dd�}|dkrD| �|�|d< |� dd�}|dkr`||d< | j|dd�}t|�}|jf i |��}|S )zDGet pools that are installed, available and consumed by this system.r
r �
after_date�futureT�Z
proxy_only)�
setdefaultr r r � get_pools) r r r" r
r# r$ r r �poolsr r r! r' 0 s z'EntitlementDBusImplementation.get_pools)r" r c C s" | j |dd�}t|�}|�� }|S )NTr% )r r �remove_all_entitlements)r r" r r �resultr r r! r) D s z5EntitlementDBusImplementation.remove_all_entitlements)�pool_idsr" r c C s* | j |dd�}t|�}|�|�\}}}|S )zSRemove entitlements by Pool IDs
:return: List of removed serials.
Tr% )r r � remove_entitlements_by_pool_ids)r r+ r" r r �_�removed_serialsr r r! r, K s z=EntitlementDBusImplementation.remove_entitlements_by_pool_ids)�serialsr" r c C s( | j |dd�}t|�}|�|�\}}|S )zSRemove entitlements by serials.
:return: List of removed serials.
Tr% )r r � remove_entitlements_by_serials)r r/ r" r r r. r- r r r! r0 V s z<EntitlementDBusImplementation.remove_entitlements_by_serials)� date_stringr c
C sB zt �|�}W n. ty< } zt�|��W Y d}~n
d}~0 0 |S )ah
Return new datetime parsed from date.
This is a wrapper around EntitlementService.parse_date that raises
a D-Bus exception instead of ValueError.
:param date_string: String representing date
:return: datetime.datetime structure representing date
:raises dbus.DBusException: String cannot be converted
N)r Z
parse_date�
ValueErrorr r )r r1 �date�errr r r! r a s
z)EntitlementDBusImplementation._parse_dateN)
�__name__�
__module__� __qualname__r �dictr r' r) r r, r0 �datetimer r r r r! r ! s r c s e Zd ZdZejZejZd� fdd� Z e
j ejddd�e
j e
j
d dd ����Ze
jejd
d �e
j
d d
� ��Ze
j ejddd�e
j e
j
d dd����Ze
j ejddd�e
j e
j
d dd����Ze
j ejddd�e
j e
j
d dd����Ze
j ejddd�e
j e
j
d dd����Zedd� �Z� ZS )!�EntitlementDBusObjectzh
A D-Bus object interacting with subscription-manager to list, get status
and remove pools.
Nc s" t t| �j|||d� t� | _d S )N)�conn� object_path�bus_name)�superr: �__init__r �impl)r r; r< r= �� __class__r r! r? | s z EntitlementDBusObject.__init__�ss�s)Z in_signatureZ
out_signaturec C s< t j|td�}t j|td�}t�|� | j�|�}t�|�S )z�
Get status of entitlements
:param on_date: Date
:param locale: String with locale (e.g. de_DE.UTF-8)
:param sender: Not used argument
:return: String with JSON dump
�Z
expected_type) r �dbus_to_pythonr r �setr@ r �json�dumps)r r
�locale�senderr r r r! � GetStatus� s
z EntitlementDBusObject.GetStatusr )Z signaturec C s t �dtj � dS )za
Signal fired, when entitlement is created/deleted/changed
:return: None
zD-Bus signal %s emittedN)r �debugr �ENTITLEMENT_INTERFACE)r r r r! �EntitlementChanged� s z(EntitlementDBusObject.EntitlementChangedz a{sv}a{sv}sc C sL t j|td�}t j|td�}t j|td�}t�|� | j�||�}t� |�S )a]
Try to get pools installed/available/consumed at this system
:param options: D-Bus object storing options of query
:param proxy_options: D-Bus object with proxy configuration
:param locale: String with locale (e.g. de_DE.UTF-8)
:param sender: Not used argument
:return: String with JSON dump
rE )
r rF r8 r r rG r@ r' rH rI )r r r" rJ rK r( r r r! �GetPools� s
z EntitlementDBusObject.GetPoolsza{sv}sc C s< t j|td�}t j|td�}t�|� | j�|�}t� |�S )a
Try to remove all entitlements (subscriptions) from the system
:param proxy_options: Settings of proxy
:param locale: String with locale (e.g. de_DE.UTF-8)
:param sender: Not used argument
:return: Json string containing response
rE )
r rF r8 r r rG r@ r) rH rI )r r" rJ rK r* r r r! �RemoveAllEntitlements� s
z+EntitlementDBusObject.RemoveAllEntitlementszasa{sv}sc C sL t j|td�}t j|td�}t j|td�}t�|� | j�||�}t �
|�S )aI
Try to remove entitlements (subscriptions) by pool_ids
:param pool_ids: List of pool IDs
:param proxy_options: Settings of proxy
:param locale: String with locale (e.g. de_DE.UTF-8)
:param sender: Not used argument
:return: Json string representing list of serial numbers
rE ) r rF �listr8 r r rG r@ r, rH rI )r r+ r" rJ rK r. r r r! �RemoveEntitlementsByPoolIds� s
z1EntitlementDBusObject.RemoveEntitlementsByPoolIdsc C sL t j|td�}t j|td�}t j|td�}t�|� | j�||�}t �
|�S )a^
Try to remove entitlements (subscriptions) by serials
:param serials: List of serial numbers of subscriptions
:param proxy_options: Settings of proxy
:param locale: String with locale (e.g. de_DE.UTF-8)
:param sender: Not used argument
:return: Json string representing list of serial numbers
rE ) r rF rR r8 r r rG r@ r0 rH rI )r r/ r" rJ rK r. r r r! �RemoveEntitlementsBySerials� s
z1EntitlementDBusObject.RemoveEntitlementsBySerialsc C s t � } | j�� | �� d S )N)r �identity�reload)Zentitlement_servicer r r! rV s
z EntitlementDBusObject.reload)NNN)N)N)N)N)N)r5 r6 r7 �__doc__r ZENTITLEMENT_DBUS_PATHZdefault_dbus_pathrN Zinterface_namer? r Zdbus_service_methodZdbus_handle_senderZdbus_handle_exceptionsrL Zdbus_service_signalrO rP rQ rS rT � staticmethodrV �
__classcell__r r rA r! r: s sh �� ����r: )r9 �typingr r r rH ZloggingZrhsm.connectionr Z rhsmlib.dbusr r r r Z rhsmlib.services.entitlementr Z"subscription_manager.injectioninitr
Zsubscription_manager.i18nr Z getLoggerr5 r ZBaseImplementationr Z
BaseObjectr: r r r r! �<module>
s
R