File: //home/arjun/projects/env/lib/python3.10/site-packages/kombu/__pycache__/pidbox.cpython-310.pyc
o
'we�9 � @ s" d Z ddlmZ ddlZddlZddlmZmZ ddlm Z ddl
m
Z
ddl m Z ddl
m
Z
d d
lmZmZmZmZ d d lmZ d d lmZmZ d d
lmZ d dlmZ d dl m Z d dl m Z m Z d dl!m"Z" d dl#m$Z$ dZ%dZ&dZ'ee(�Z)e)j*e)j+Z*Z+G dd� d�Z,G dd� d�Z-dS )zGeneric process mailbox.� )� annotationsN)� defaultdict�deque)�contextmanager)�copy)�count)�time� )�Consumer�Exchange�Producer�Queue)� LamportClock)�
maybe_declare�oid_from)�InconsistencyError)�
get_logger)�match)�maybe_evaluate�reprcall)�cached_property)�uuid�
z�A node named {node.hostname} is already using this process mailbox!
Maybe you forgot to shutdown the other node or did not do so properly?
Or if you meant to start multiple nodes on the same host please make sure
you give each node a unique node name!
)�Node�Mailboxc @ s� e Zd ZdZdZdZdZdZdZ ddd�Z ddd�Z
dd � Z d
d � Z d d d
�Z
d dd�Zd dd�Zdd� Zdd� Zd dd�ZeZdd� ZdS ) r z
Mailbox node.Nc C s: || _ || _|| _|| _| jjj| _|d u ri }|| _d S �N)�channel�mailbox�hostname�state�clock�adjust� adjust_clock�handlers)�selfr r r r# r � r% �E/home/arjun/projects/env/lib/python3.10/site-packages/kombu/pidbox.py�__init__8 s
z
Node.__init__Tc sP � j �� j�}� fdd�}||_t|p� j|gf||d u r!� j jn|d�|��S )Nc s |r
t �tj� d�� d S d S )N)�node)�warnings�warn�W_PIDBOX_IN_USE�format)�name�messages� consumers�r$ r% r& �verify_exclusiveF s �z'Node.Consumer.<locals>.verify_exclusive)�no_ack�accept)r � get_queuer � on_declaredr
r r3 )r$ r r2 r3 �options�queuer1 r% r0 r& r
C s ���z
Node.Consumerc C s || j |j< |S r )r# �__name__)r$ �funr% r% r& �handlerQ s z Node.handlerc C s t d|dd� d S )NzCannot decode message: %rr ��exc_info)�error)r$ �message�excr% r% r& �on_decode_errorU s zNode.on_decode_errorc C s& | j ||p| jg| jd�}|�� |S )N)r � callbacksr@ )r
�handle_messager@ �consume)r$ r �callback�consumerr% r% r& �listenX s
�z Node.listenc
K s� |pi }t dt|d|d�||� |r| jp| j}z|||�}W n& ty' � tyE } ztd|dd� dt|�i}W Y d }~nd }~ww |rX| j| j |i|d |d |d
� |S ) Nz1pidbox received method %s [reply_to:%s ticket:%s]r% )�kwargszpidbox command error: %rr r; r= �exchange� routing_key)rH rI �ticket)
�debugr � handle_call� handle_cast�
SystemExit� Exceptionr= �repr�replyr ) r$ �method� arguments�reply_torJ rG �handlerQ r? r% r% r& �dispatch_ s* � �� �z
Node.dispatchc C s$ |si n|}| j | | jfi |��S r )r# r �r$ rR rS r% r% r& rU t s z Node.handlec C � | � ||�S r �rU rW r% r% r& rL x � zNode.handle_callc C rX r rY rW r% r% r& rM { rZ zNode.handle_castc C s� |� d�}|� d�}|� d�}|r | �|j� d�pd� | j}d}|r*||v r)d}n|r7|r7t|||�r6d}nd}|rC| jdi |��S d S ) N� destination�pattern�matcherr r FTr% )�getr" �headersr r rV )r$ �bodyr> r[ r\ r] r � run_dispatchr% r% r& rB ~ s&
� ��zNode.handle_messagec K s" | j j||||| j| j jd� d S )N)r �
serializer)r �_publish_replyr rb )r$ �datarH rI rJ rG r% r% r& rQ � s
�z
Node.reply�NNNN)NTN�NN)NNNr )r8 �
__module__� __qualname__�__doc__r r r# r r r' r
r: r@ rF rV rU rL rM rB �dispatch_from_messagerQ r% r% r% r& r & s.
�
�
r c @ s e Zd ZdZeZdZdZdZdZ dZ
dZ dZ dgZ
dZ d0dd �Zd
d � Zd1d d
�Z d1dd�Zd2dd�Zd2dd�Z d3dd�Zdd� Zedd� �Zdd � Zed4d d ��Z d4d d �Z d5d!d"�Z d6d$d%�Z d3d&d'�Z d(d)� Z d*d+� Z e!d,d-� �Z"ed.d/� �Z#dS )7r zProcess Mailbox.z %s.pidboxzreply.%s.pidboxN�direct�json� $@c C s� || _ || _|| _|d u rt� n|| _| �| j | j�| _| �| j �| _t t
�| _ |d u r/| j n|| _ |d u r9| j
n|| _
|| _| | _|
| _| | _|| _d S r )� namespace�
connection�typer r �
_get_exchangerH �_get_reply_exchange�reply_exchanger r � unclaimedr3 rb � queue_ttl�
queue_expires�reply_queue_ttl�reply_queue_expires�_producer_pool) r$ rn rp ro r r3 rb �
producer_poolru rv rw rx r% r% r& r' � s
zMailbox.__init__c C s t | �}||_|S r )r ro )r$ ro �boundr% r% r& �__call__� s zMailbox.__call__c C s |pt �� }| j||||| d�S )N)r )�socket� gethostname�node_cls)r$ r r r r# r% r% r& r � s z Mailbox.Nodec C s$ |si n|}| j |||d|||d�S )NT)rQ �timeoutrD r ��
_broadcast)r$ r[ �commandrG r� rD r r% r% r& �call� �
�z Mailbox.callc C s |si n|}| j |||dd�S �NF)rQ r� )r$ r[ r� rG r% r% r& �cast� s z Mailbox.castc C s |si n|}| j ||dd�S r� r� )r$ r� rG r% r% r& �abcast� s zMailbox.abcastr c C s$ |si n|}| j ||d||||d�S )NT)rQ r� �limitrD r r� )r$ r� rG r� r� rD r r% r% r& �
multi_call� r� zMailbox.multi_callc C s0 | j }t|� d| jj� �| j|dd| j| jd�S )N�.FT)rH rI �durable� auto_delete�expires� message_ttl)�oidr
rs r- rx rw )r$ r� r% r% r& �get_reply_queue� s �zMailbox.get_reply_queuec C s | � � S r )r� r0 r% r% r& � reply_queue� � zMailbox.reply_queuec C s( t |� d| j� d�| jdd| j| jd�S )Nr� z.pidboxFT)rH r� r� r� r� )r
rn rH rv ru )r$ r r% r% r& r4 � s �zMailbox.get_queuec c s^ � |r|V d S | j r&| j �� � }|V W d � d S 1 s w Y d S t|dd�V d S )NF)� auto_declare)rz �acquirer )r$ �producerr r% r% r& �producer_or_acquire s �
"�zMailbox.producer_or_acquirec K s� |p| j j}t|dddd�}| �||��3}z|j|f|||g|| j�� d�dd�|�� W n ty6 Y n w W d � d S W d � d S 1 sJw Y d S )Nrk � transientF)�
exchange_type�
delivery_moder� )rJ r T)rH rI �declarer_ �retry)ro �default_channelr r� �publishr �forwardr ) r$ rQ rH rI rJ r r� �opts�chanr% r% r& rc
s2 ��
��
� ��� "�zMailbox._publish_replyc C s� |||| |
d�} |p
| j j} | j}
|r't| �| �� | j|| jj| jd�d� |p+| j }| �
|| ��#}|j | |
j|
g| j �
� |rEt� | ndd�|dd� W d � d S 1 sXw Y d S )N)rR rS r[ r\ r] )rH rI )rJ rT r )r r� T)rH r� r_ rb r� )ro r� rH r r� �updaters r- r� rb r� r� r r� r )r$ rp rS r[ � reply_ticketr r� rb r� r\ r] r> r� rH r% r% r& �_publish s2 � ��
��"�zMailbox._publishFc C s� |d urt |ttf�std�t|����|
d ur2t |
t�s2| d ur2t | t�s2td�t|
�t| ����|p5i }|r;t� p<d } |pB| jj }
|d u rQ|rQ|rOt
|�pPd }| pU| j } | j |||| |
|| |
| d� | rp| j
| ||||
d�S d S )Nz'destination must be a list/tuple not {}z.pattern and matcher must be strings not {}, {})r[ r� r r� rb r\ r] )r� r� rD r )�
isinstance�list�tuple�
ValueErrorr, rp �strr ro r� �lenrb r� �_collect)r$ r� rS r[ rQ r� r� rD r rb r\ r] r� r� r% r% r& r� 8 sF ����
���zMailbox._broadcastc
s |d u r| j }|p | jj}| j}t||g|dd�} g �| j�| jj� z����W S t y1 Y nw � ����fdd�}
| �
|
� zD| �1 |rKt |�pMt � D ]} z | jj
|d� W qN tjyd Y nw �W d � W |�|j� S 1 sxw Y W |�|j� d S |�|j� w )NT)r3 r2 c sp |j j}� |d�p
d� |d�}|rt� |krd S |d��}|�kr/�r(�| � ��| � d S �| �| � d S )Nr r r� rJ )r_ r^ r �append)r` r> �headerr� �this_id�r" rD � responsesrJ rt r% r& �
on_messagep s
z$Mailbox._collect.<locals>.on_message)r� )r3 ro r� r� r
rt r r! �pop�KeyError�register_callback�ranger � drain_eventsr} r� � after_reply_message_receivedr- ) r$ rJ r� r� rD r r3 r� r7 rE r� �ir% r� r& r� _ s8 �
��� zMailbox._collectc C s t | j| |ddd�S )NFr� �rp r� r� )r � exchange_fmt)r$ rn rp r% r% r& rq � �
�zMailbox._get_exchangec C s t | j| dddd�S )Nrk Fr� r� )r �reply_exchange_fmt)r$ rn r% r% r& rr � r� zMailbox._get_reply_exchangec C s t | �S r )r r0 r% r% r& r� � r� z Mailbox.oidc C s
t | j�S r )r ry r0 r% r% r& rz � s
zMailbox.producer_pool)
rk NNNNNNNNrm re r )Nr NNNrf )NNNNNNNN)
NNFr NNNNNN)$r8 rg rh ri r r r� r� rn ro rp rH rs r3 rb r' r| r� r� r� r� r� r r� r4 r r� rc r� r� r� rq rr �propertyr� rz r% r% r% r& r � sj
�
�
�
�
�
�(
�,
r ).ri �
__future__r r} r) � collectionsr r �
contextlibr r � itertoolsr r � r
r r r
�clocksr �commonr r �
exceptionsr �logr r] r �utils.functionalr r �
utils.objectsr �
utils.uuidr �REPLY_QUEUE_EXPIRESr+ �__all__r8 �loggerrK r= r r r% r% r% r& �<module> s2 r