File: //usr/local/lib/python3.9/site-packages/kombu/transport/__pycache__/SLMQ.cpython-39.pyc
a
X>hG � @ s� d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZ dd lmZ z dd
lmZ dd lmZ W n ey� d ZZY n0 d d
� ejD �ZG dd� dej�ZG dd� dej�ZdS )a+ SoftLayer Message Queue transport module for kombu.
Features
========
* Type: Virtual
* Supports Direct: Yes
* Supports Topic: Yes
* Supports Fanout: No
* Supports Priority: No
* Supports TTL: No
Connection String
=================
*Unreviewed*
Transport Options
=================
*Unreviewed*
� )� annotationsN)�Empty)� bytes_to_str�safe_str)�dumps�loads)�cached_property� )�virtual)�
get_client)�
ResponseErrorc C s i | ]}|d vrt |�d�qS )�_�_ )�ord)�.0�c� r �>/usr/local/lib/python3.9/site-packages/kombu/transport/SLMQ.py�
<dictcomp>) s r c s� e Zd ZdZdZdZdZi Ze� Z � fdd�Z
� fdd�Z � fd d
�Z e
fd d �Zd
d� Z� fdd�Zdd� Zdd� Z� fdd�Zdd� Zdd� Zdd � Zed d � �Zed d � �Zed!d"� �Zed#d$� �Z ed%d&� �Z � Z S )'�Channelz
SLMQ Channel.i zkombu%(vhost)sNc sD t d u rtd��t� j|i |�� | j�� }|D ]}|| j|< q0d S )Nz7SLMQ transport requires the softlayer_messaging library)r � ImportError�super�__init__�slmq�queues� _queue_cache)�self�args�kwargsr �queue�� __class__r r r 7 s �
zChannel.__init__c s, |r| j �|� t� j||g|�R i |��S �N)�
_noack_queues�addr �
basic_consume)r r Zno_ackr r r r r r% A s ��zChannel.basic_consumec s, || j v r | j| }| j�|� t� �|�S r" )Z
_consumersZ
_tag_to_queuer# �discardr � basic_cancel)r Z consumer_tagr r r r r' G s
zChannel.basic_cancelc C s t t|���|�S )z4Format AMQP queue name into a valid SLQS queue name.)�strr � translate)r �name�tabler r r � entity_nameM s zChannel.entity_namec K sx | � | j| �}z | j| W S tyr z| jj|| jd� W n tyR Y n0 | j�|� }| j|< | Y S 0 dS )z Ensure a queue exists in SLQS.)�visibility_timeoutN) r, �queue_name_prefixr �KeyErrorr Z create_queuer- r r )r r r �qr r r �
_new_queueQ s �
zChannel._new_queuec s< | � |�}| j�|d� | j�|�jdd� t� �|� dS )zDelete queue by name.NT)�force)r, r �popr r �deleter �_delete)r r r r Z
queue_namer r r r5 _ s
zChannel._deletec K s | � |�}|�t|�� dS )zPut message onto queue.N)r1 �pushr )r r �messager r0 r r r �_putf s
z Channel._putc C s� | � |�}|�d�}|d rx|d d }tt|d ��}|| jv rV|�|d ��� n |d d �|d |jd�� |S t � �d S )
z/Try to retrieve a single message off ``queue``.r �itemsr �body�idZ
properties�
delivery_info)�slmq_message_id�slmq_queue_nameN)
r1 r3 r r r# r7 r4 �updater* r )r r r0 �rs�m�payloadr r r �_getk s
�z Channel._getc sN | j �|�j}z |d }W n ty, Y n0 | �||d � t� �|� d S )Nr> r= )Zqos�getr< r/ �delete_messager � basic_ack)r Z delivery_tagr< r r r r rF z s zChannel.basic_ackc C s | � |��� d S )z)Return the number of messages in a queue.Z
message_count)r1 Zdetail)r r r r r �_size� s z
Channel._sizec C sV | � |�}d}|�d�}|d rR|d D ] }| �||d � |d7 }q(|�d�}q|S )z'Delete all current messages in a queue.r �
r9 r; r )r1 r3 rE )r r r0 �n�resultsrA r r r �_purge� s
zChannel._purgec C s | j �| �|��}|�|��� S r" )r r r, r7 r4 )r r Z
message_idr0 r r r rE � s zChannel.delete_messagec C s� | j d u r�| j}tj�d|j�}tj�d|j�}tj�d|j�}tj�d|j�}tj�d|j �}t
tj�d| j �d��pzd�}d � |r�d
nd ||r�d |� �nd
�}t
||d�| _ | j �||� | j S )NZ SLMQ_ACCOUNTZ SL_USERNAMEZ
SL_API_KEYZ SLMQ_HOSTZ SLMQ_PORTZ SLMQ_SECURE�secureTz {}://{}{}�https�http�:� )�endpoint)�_slmq�conninfo�os�environrD Z virtual_hostZuserid�password�hostname�port�bool�transport_options�formatr Z authenticate) r rS �account�userZapi_key�hostrX rL rQ r r r r � s&
�� �z Channel.slmqc C s | j jS r" )�
connection�client�r r r r rS � s zChannel.conninfoc C s
| j jjS r" )r_ r` rZ ra r r r rZ � s zChannel.transport_optionsc C s | j �d�p| jS )Nr- )rZ rD �default_visibility_timeoutra r r r r- � s �zChannel.visibility_timeoutc C s | j �dd�S )Nr. rP )rZ rD ra r r r r. � s zChannel.queue_name_prefix) �__name__�
__module__� __qualname__�__doc__rb Z
domain_formatrR r �setr# r r% r' �CHARS_REPLACE_TABLEr, r1 r5 r8 rC rF rG rK rE �propertyr rS rZ r r- r. �
__classcell__r r r r r . s8
r c @ s. e Zd ZdZeZdZdZejj e
e j f Z dS )� TransportzSLMQ Transport.r N)
rc rd re rf r Zpolling_interval� default_portr
rk Zconnection_errorsr �socket�errorr r r r rk � s ��rk )rf �
__future__r rT rm �stringr r Zkombu.utils.encodingr r Zkombu.utils.jsonr r Zkombu.utils.objectsr rP r
Zsoftlayer_messagingr Zsoftlayer_messaging.errorsr r � punctuationrh r rk r r r r �<module> s( �