File: //usr/local/lib/python3.9/site-packages/kombu/transport/__pycache__/librabbitmq.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
ddlmZ dd lmZ d
Zd Zd Zd
ZG dd� dej�ZG dd� dejej�ZG dd� dej�ZG dd� dej�ZdS )zT`librabbitmq`_ transport.
.. _`librabbitmq`: https://pypi.org/project/librabbitmq/
� )� annotationsN)� ChannelError�ConnectionError)� get_manager)�version_string_as_tuple� )�base)�to_rabbitmq_queue_argumentsz�
librabbitmq version too old to detect RabbitMQ version information
so make sure you are using librabbitmq 1.5 when using rabbitmq > 3.3
i( i' zAssl not supported by librabbitmq, please use pyamqp:// or stunnelc s e Zd ZdZ� fdd�Z� ZS )�MessagezAMQP Message (librabbitmq).c s8 t � j|||||�d�|�d�|�d�|�d�d� d S )N� delivery_tag� content_type�content_encoding�headers)�channel�bodyZ
delivery_info�
propertiesr r r
r )�super�__init__�get)�selfr �props�infor �� __class__� �E/usr/local/lib/python3.9/site-packages/kombu/transport/librabbitmq.pyr $ s �zMessage.__init__)�__name__�
__module__� __qualname__�__doc__r �
__classcell__r r r r r
! s r
c @ s& e Zd ZdZeZddd�Zdd� ZdS )�ChannelzAMQP Channel (librabbitmq).Nc C s: |dur |ni }|� |||d�� |dur2||d< ||fS )z%Encapsulate data into a AMQP message.N)r r
r �priority)�update)r r r" r r
r r r r r �prepare_message5 s �zChannel.prepare_messagec K s" t |fi |��}dd� |�� D �S )Nc S s i | ]\}}|� d �|�qS )�utf8)�encode)�.0�k�vr r r �
<dictcomp>F � z3Channel.prepare_queue_arguments.<locals>.<dictcomp>)r �items)r � arguments�kwargsr r r �prepare_queue_argumentsD s z Channel.prepare_queue_arguments)NNNNN)r r r r r
r$ r/ r r r r r! 0 s �
r! c @ s e Zd ZdZeZeZdS )�
Connectionz AMQP Connection (librabbitmq).N)r r r r r! r
r r r r r0 I s r0 c @ s� e Zd ZdZeZeZeZe j
j e e
jeef Z e j
jef ZdZdZe j
jjddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd � Z!e"d d � �Z#d S ) � Transportz AMQP Transport (librabbitmq).�amqp� librabbitmqTF)Z asynchronousZ
heartbeatsc K s4 || _ |�d�p| j| _|�d�p&| j| _d | _d S )N� default_port�default_ssl_port)�clientr r4 r5 Z_Transport__reader)r r6 r. r r r r g s
�zTransport.__init__c C s t jS �N)r2 � __version__�r r r r �driver_versionn s zTransport.driver_versionc C s |� � S r7 )r �r �
connectionr r r �create_channelq s zTransport.create_channelc K s |j f i |��S r7 )� drain_events)r r< r. r r r r> t s zTransport.drain_eventsc
C s� | j }| j�� D ] \}}t||d�st|||� q|jr@tt��t|j |j
|j |j |j
|j|j|jd�fi |jpri ��}| jf i |��}| j |_ |j| j _|S )z(Establish connection to the AMQP broker.N)�host�userid�password� virtual_host� login_method�insist�ssl�connect_timeout)r6 �default_connection_paramsr, �getattr�setattrrE �NotImplementedError� NO_SSL_ERROR�dictr? r@ rA rB rC rD rF Ztransport_optionsr0 r> )r Zconninfo�name�
default_value�opts�connr r r �establish_connectionw s, � �
z Transport.establish_connectionc C s d| j _|�� dS )z!Close the AMQP broker connection.N)r6 r> �closer; r r r �close_connection� s zTransport.close_connectionc C sn |d ur\|j �� D ]
}d |_qzt�|�� � W n ttfyF Y n0 |j �� |j �� d | j
_ d | _
d S r7 ) Zchannels�valuesr< �osrR �fileno�OSError�
ValueError�clear� callbacksr6 r> )r r< r r r r �_collect� s
zTransport._collectc C s |j S r7 )� connectedr; r r r �verify_connection� s zTransport.verify_connectionc C s |� |�� | j||� d S r7 )�
add_readerrV Z on_readable)r r< �loopr r r �register_with_event_loop� s �z"Transport.register_with_event_loopc O s t | jg|�R i |��S r7 )r r6 )r �argsr. r r r r � s zTransport.get_managerc C sN z
|j }W n ty* t�tt�� Y n 0 |�d�dkrJt|d �dk S dS )N�productZRabbitMQ�version)� rd T)Zserver_properties�AttributeError�warnings�warn� UserWarning� W_VERSIONr r )r r< r r r r �qos_semantics_matches_spec� s
z$Transport.qos_semantics_matches_specc C s dd| j jr| jn| jddd�S )NZguest� localhostZPLAIN)r@ rA �port�hostnamerC )r6 rE r5 r4 r9 r r r rG � s �z#Transport.default_connection_paramsN)$r r r r r0 � DEFAULT_PORTr4 �DEFAULT_SSL_PORTr5 r r1 Zconnection_errorsr �socket�error�IOErrorrW Zchannel_errorsr Z driver_typeZ driver_nameZ
implements�extendr r: r= r> rQ rS r[ r] r` r rj �propertyrG r r r r r1 P s:
�� ��
r1 )r �
__future__r rU rp rf r3 r2 r r Zkombu.utils.amq_managerr Zkombu.utils.textr � r r ri rn ro rK r
r! Z
StdChannelr0 r1 r r r r �<module> s$