File: //usr/local/lib/python3.9/site-packages/kombu/transport/__pycache__/pyro.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZ ddlZddlmZmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ z$ddlZdd
lmZ dd lmZ W n ey� d Z ZZY n0 d Zd
ZdZe
e�ZG dd� dej�ZG dd� dej �Z edu�re� ddd� � ej ej dd�G dd� d���Z edk�r�e!d� e�"� ��Z#e!d �$ej%j&ej%j'�� e�(� �8Z)e!d �$ej%j&�� e#�*e �Z+e)�*d e+� W d � n1 �s�0 Y e#�,� W d � n1 �s�0 Y dS ) a� Pyro transport module for kombu.
Pyro transport, and Kombu Broker daemon.
Requires the :mod:`Pyro4` library to be installed.
Features
========
* Type: Virtual
* Supports Direct: Yes
* Supports Topic: Yes
* Supports Fanout: No
* Supports Priority: No
* Supports TTL: No
Connection String
=================
To use the Pyro transport with Kombu, use an url of the form:
.. code-block::
pyro://localhost/kombu.broker
The hostname is where the transport will be looking for a Pyro name server,
which is used in turn to locate the kombu.broker Pyro service.
This broker can be launched by simply executing this transport module directly,
with the command: ``python -m kombu.transport.pyro``
Transport Options
=================
� )� annotationsN)�Empty�Queue)�reraise)�
get_logger)�cached_property� )�virtual)� NamingError)�SerializerBasei�# z5Unable to locate pyro nameserver on host {0.hostname}zKUnable to lookup '{0.virtual_host}' in pyro nameserver on host {0.hostname}c s~ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zedd� �Z� ZS ) �Channelz
Pyro Channel.c s t � �� | jr| j�� d S �N)�super�close�
shared_queuesZ _pyroRelease��self�� __class__� �>/usr/local/lib/python3.9/site-packages/kombu/transport/pyro.pyr C s
z
Channel.closec C s
| j �� S r
)r �get_queue_namesr r r r �queuesH s zChannel.queuesc K s || � � vr| j�|� d S r
�r r � new_queue�r �queue�kwargsr r r �
_new_queueK s zChannel._new_queuec K s | j �|�S r
)r � has_queuer r r r �
_has_queueO s zChannel._has_queueNc C s | � |�}| j�|�S r
)�
_queue_forr �get)r r �timeoutr r r �_getR s
z Channel._getc C s || � � vr| j�|� |S r
r �r r r r r r! V s zChannel._queue_forc K s | � |�}| j�||� d S r
)r! r �put)r r �messager r r r �_put[ s
z Channel._putc C s | j �|�S r
)r �sizer% r r r �_size_ s z
Channel._sizec O s | j �|� d S r
)r �delete)r r �argsr r r r �_deleteb s zChannel._deletec C s | j �|�S r
)r �purger% r r r �_purgee s zChannel._purgec C s d S r
r r% r r r � after_reply_message_receivedh s z$Channel.after_reply_message_receivedc C s | j jS r
)�
connectionr r r r r r k s zChannel.shared_queues)N)�__name__�
__module__� __qualname__�__doc__r r r r r$ r! r( r* r- r/ r0 r r �
__classcell__r r r r r @ s
r c sT e Zd ZdZeZe�� ZeZ d Z
Z � fdd�Z dd� Z
dd� Zed d
� �Z� ZS ) � TransportzPyro Transport.�pyroc s t � j|fi |�� | j| _d S r
)r �__init__� global_state�state)r �clientr r r r r9 } s zTransport.__init__c C s� t �d� | j}ztj|j| jd�}W n0 tyV tttt �
|��t � � d � Y n0 z|�
|j�}t�|�W S ty� tttt�
|��t � � d � Y n0 d S )Nz0trying Pyro nameserver to find the broker daemon)�host�port� )�logger�debugr< r8 �locateNS�hostname� default_portr
r � E_NAMESERVER�format�sys�exc_info�lookupZ virtual_hostZProxy�E_LOOKUP)r ZconninfoZ
nameserver�urir r r �_open� s"
�
�
�zTransport._openc C s t jS r
)r8 � __version__r r r r �driver_version� s zTransport.driver_versionc C s | � � S r
)rL r r r r r � s zTransport.shared_queues)r2 r3 r4 r5 r r Z BrokerStater: � DEFAULT_PORTrD Z driver_typeZ driver_namer9 rL rN r r r6 r r r r r7 p s r7 z queue.Emptyc C s t � S r
)r )�cls�datar r r �<lambda>� � rR �single)Z
instance_modec @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )� KombuBrokerzmKombu Broker used by the Pyro transport.
You have to run this as a separate (Pyro) service.
c C s
i | _ d S r
�r r r r r r9 � s zKombuBroker.__init__c C s
t | j�S r
)�listr r r r r r � s zKombuBroker.get_queue_namesc C s || j v rd S t� | j |< d S r
)r r r% r r r r � s
zKombuBroker.new_queuec C s
|| j v S r
rV r% r r r r � s zKombuBroker.has_queuec C s | j | jdd�S )NF)�block)r r" r% r r r r" � s zKombuBroker.getc C s | j | �|� d S r
)r r& )r r r' r r r r& � s zKombuBroker.putc C s | j | �� S r
)r �qsizer% r r r r) � s zKombuBroker.sizec C s | j |= d S r
rV r% r r r r+ � s zKombuBroker.deletec C s4 z| j | jdd� W q ty, Y q0Y q 0 q d S )NF)�blocking)r r" r r% r r r r. � s zKombuBroker.purgeN)
r2 r3 r4 r5 r9 r r r r" r& r) r+ r. r r r r rU � s rU �__main__z,Launching Broker for Kombu's Pyro transport.z'(Expecting a Pyro name server at {}:{})zAYou can connect with Kombu using the url 'pyro://{}/kombu.broker'z kombu.broker)-r5 �
__future__r rG r r r Zkombu.exceptionsr Z kombu.logr Zkombu.utils.objectsr � r ZPyro4r8 Z Pyro4.errorsr
Z
Pyro4.utilr � ImportErrorrO rE rJ r2 r@ r r7 Zregister_dict_to_classZexposeZbehaviorrU �printZDaemon�daemonrF �configZNS_HOSTZNS_PORTrB �ns�registerrK Z requestLoopr r r r �<module> sN " 0*
�
*
�
�
,