HEX
Server: Apache/2.4.52 (Ubuntu)
System: Linux spn-python 5.15.0-89-generic #99-Ubuntu SMP Mon Oct 30 20:42:41 UTC 2023 x86_64
User: arjun (1000)
PHP: 8.1.2-1ubuntu2.20
Disabled: NONE
$ pwd: /home/arjun/projects/env/lib/python3.10/site-packages/kombu/__pycache__/
Upload Files
File: //home/arjun/projects/env/lib/python3.10/site-packages/kombu/__pycache__/messaging.cpython-310.pyc
o

'we�^�@s�dZddlmZddlmZddlmZddlmZddl	m
Z
ddlmZm
Z
dd	lmZmZmZdd
lmZddlmZmZddlmZmZerPdd
lmZdZGdd�d�ZGdd�d�ZdS)zSending and receiving messages.�)�annotations)�count)�
TYPE_CHECKING���
maybe_declare)�compress)�
is_connection�
maybe_channel)�Exchange�Queue�maybe_delivery_mode)�ContentDisallowed)�dumps�prepare_accept_content)�ChannelPromise�
maybe_list)�
TracebackType)rr�Producer�Consumerc@s�eZdZdZdZdZdZdZdZdZ	dZ
			d1dd�Zdd�Zd	d
�Z
dd�Zd
d�Zd2dd�Zdeefdd�Z					d3dd�Z	d4dd�Zdd�Zdd�Zeee�Zdd�Zdd �Zd5d)d*�Zd+d,�ZeZ		d6d-d.�Zed/d0��Z dS)7ra�Message Producer.

    Arguments:
    ---------
        channel (kombu.Connection, ChannelT): Connection or channel.
        exchange (kombu.entity.Exchange, str): Optional default exchange.
        routing_key (str): Optional default routing key.
        serializer (str): Default serializer. Default is `"json"`.
        compression (str): Default compression method.
            Default is no compression.
        auto_declare (bool): Automatically declare the default exchange
            at instantiation. Default is :const:`True`.
        on_return (Callable): Callback to call for undeliverable messages,
            when the `mandatory` or `immediate` arguments to
            :meth:`publish` is used. This callback needs the following
            signature: `(exception, exchange, routing_key, message)`.
            Note that the producer needs to drain events to use this feature.
    N�TcCs~||_||_|p
|j|_|p|j|_|p|j|_|p|j|_d|_|jdur+td�|_|dur2||_|jr=|�	|j�dSdS)Nr)
�_channel�exchange�routing_key�
serializer�compression�	on_return�_channel_promiser�auto_declare�revive)�self�channelrrrrrr�r"�H/home/arjun/projects/env/lib/python3.10/site-packages/kombu/messaging.py�__init__As

�zProducer.__init__cCsd|j�d�S)Nz<Producer: �>�r�r r"r"r#�__repr__S�zProducer.__repr__cCs|j|��fS�N)�	__class__�__reduce_args__r'r"r"r#�
__reduce__Vr)zProducer.__reduce__cCsd|j|j|j|j|jfSr*)rrrrrr'r"r"r#r,Ys�zProducer.__reduce_args__cCs|jjr|j��dSdS)z�Declare the exchange.

        Note:
        ----
            This happens automatically at instantiation when
            the :attr:`auto_declare` flag is enabled.
        N)r�name�declarer'r"r"r#r/]s�zProducer.declareFcKs|r
t||j|fi|��SdS)z=Declare exchange if not already declared during this session.N)rr!)r �entity�retry�retry_policyr"r"r#rhs�zProducer.maybe_declarecCs2t||�r|j||p|j�fS|||p|jj�fSr*)�
isinstancer.�
delivery_moder)r rr4r
rr"r"r#�_delivery_detailsms
�
�zProducer._delivery_detailsrc
Ks|j}|dur	gn|}|
durin|
}
|durin|}|dur"|jn|}|dur+|jn|}|�|p3|j|�\}|d<|durHtt|d��|d<|�||	||||
�\}}}|jrg|jj	rg|j|vrg|�
|j�|
rt|jj||fi|��}||||||
|||||||�S)a�Publish message to the specified exchange.

        Arguments:
        ---------
            body (Any): Message body.
            routing_key (str): Message routing key.
            delivery_mode (enum): See :attr:`delivery_mode`.
            mandatory (bool): Currently not supported.
            immediate (bool): Currently not supported.
            priority (int): Message priority. A number between 0 and 9.
            content_type (str): Content type. Default is auto-detect.
            content_encoding (str): Content encoding. Default is auto-detect.
            serializer (str): Serializer to use. Default is auto-detect.
            compression (str): Compression method to use.  Default is none.
            headers (Dict): Mapping of arbitrary headers to pass along
                with the message body.
            exchange (kombu.entity.Exchange, str): Override the exchange.
                Note that this exchange must have been declared.
            declare (Sequence[EntityT]): Optional list of required entities
                that must have been declared before publishing the message.
                The entities will be declared using
                :func:`~kombu.common.maybe_declare`.
            retry (bool): Retry publishing, or declaring entities if the
                connection is lost.
            retry_policy (Dict): Retry configuration, this is the keywords
                supported by :meth:`~kombu.Connection.ensure`.
            expiration (float): A TTL in seconds can be specified per message.
                Default is no expiration.
            timeout (float): Set timeout to wait maximum timeout second
                for message to publish.
            **properties (Any): Additional message properties, see AMQP spec.
        Nr4i��
expiration)
�_publishrrr5r�str�int�_preparerr.�append�
connection�ensure)r �bodyrr4�	mandatory�	immediate�priority�content_type�content_encodingr�headersrrr1r2r/r6�timeout�
propertiesr7�
exchange_namer"r"r#�publishzs2&
�
�

�zProducer.publishc
sj|j}
|
�||||||�}|r|j��fdd�|D�|�d�}t|t�r*|j|d<|
j||
|||	|d�S)Ncsg|]}�|��qSr"r")�.0r0rr"r#�
<listcomp>��z%Producer._publish.<locals>.<listcomp>�reply_to)rrr?r@rE)r!�prepare_messager�getr3rr.�
basic_publish)r r>rArBrCrDrFrr?r@rr/rEr!�messagerLr"rr#r7�s"�


�zProducer._publishcCsD|j}t|t�r |�}|_|j�|�|jr |jd�|j�|S)N�basic_return)rr3rrrr�events�add�r r!r"r"r#�_get_channel�s
zProducer._get_channelcCs
||_dSr*r&rTr"r"r#�_set_channel�s
zProducer._set_channelcsrt|�r|��|_t�fdd��}t|t�r!||_|�|�|_dS||_|jr1|jjd�|j�|�|�|_dS)z*Revive the producer after connection loss.cs�jSr*)�default_channelr"�r<r"r#�<lambda>�sz!Producer.revive.<locals>.<lambda>rQN)	r	�__connection__rr3rrrrRrSrTr"rXr#r�s
zProducer.revivecCs|Sr*r"r'r"r"r#�	__enter__��zProducer.__enter__�exc_type�type[BaseException] | None�exc_val�BaseException | None�exc_tb�TracebackType | None�return�NonecCs|��dSr*)�release)r r]r_rar"r"r#�__exit__�szProducer.__exit__cCsdSr*r"r'r"r"r#rer\zProducer.releasecCsh|s|p|j}t||d�\}}}nt|t�r |sd}|�|�}n|s$d}|r/t||�\}|d<|||fS)N)rzutf-8�binaryr)rrr3r8�encoder)r r>rrBrCrrDr"r"r#r:s

�

zProducer._preparecCs(z	|jp|jjjWStyYdSwr*)rZr!r<�client�AttributeErrorr'r"r"r#r<s
�zProducer.connection)NNNNNN�F)NNFFrNNNNNNFNNNNr*�r]r^r_r`rarbrcrd)NNNNN)!�__name__�
__module__�__qualname__�__doc__rrrrrrrZr$r(r-r,r/rr
rr5rHr7rUrV�propertyr!rr[rfre�closer:r<r"r"r"r#rsV
�

�

�H
�	


�rc@s(eZdZdZeZdZdZdZdZdZ	dZ
dZdZdZ
dZed�Z			d<dd�Zedd��Zejd	d��Zd
d�Zdd
�Zdd�Zdd�Zd=dd�Zdd�Zd>dd�Zd d!�ZeZd"d#�Zd$d%�Zd&d'�Zd(d)�Z d?d,d-�Z!d@d.d/�Z"d0d1�Z#dedfd2d3�Z$d>d4d5�Z%d6d7�Z&d8d9�Z'ed:d;��Z(dS)AraMessage consumer.

    Arguments:
    ---------
        channel (kombu.Connection, ChannelT): see :attr:`channel`.
        queues (Sequence[kombu.Queue]): see :attr:`queues`.
        no_ack (bool): see :attr:`no_ack`.
        auto_declare (bool): see :attr:`auto_declare`
        callbacks (Sequence[Callable]): see :attr:`callbacks`.
        on_message (Callable): See :attr:`on_message`
        on_decode_error (Callable): see :attr:`on_decode_error`.
        prefetch_count (int): see :attr:`prefetch_count`.
    NTrcCs�||_t|pg�|_|dur|jn||_|dur|jpgn||_||_|
|_i|_|dur0||_|dur7||_	t
|�|_|	|_|jrJ|�
|j�dSdSr*)r!r�queues�no_ack�	callbacks�
on_message�
tag_prefix�_active_tagsr�on_decode_errorr�accept�prefetch_countr)r r!rsrtrruryrvrzr{rwr"r"r#r$s&��
�zConsumer.__init__cCst|j���Sr*)�list�_queues�valuesr'r"r"r#rs�szConsumer.queuescCsdd�|D�|_dS)NcSsi|]}|j|�qSr")r.)rI�qr"r"r#�
<dictcomp>�rKz#Consumer.queues.<locals>.<dictcomp>)r})r rsr"r"r#rs�scCs�|j��t|�}|_t|j���D]\}}|j�|d�||j�}|j|j<|�	|�q|j
r6|��|jdurD|j
|jd�dSdS)z&Revive consumer after connection loss.N)r{)rx�clearr
r!r|r}�items�popr.rrr/r{�qos)r r!�qname�queuer"r"r#r�s

�zConsumer.revivecCs|j��D]}|��qdS)z�Declare queues, exchanges and bindings.

        Note:
        ----
            This is done automatically at instantiation
            when :attr:`auto_declare` is set.
        N)r}r~r/�r r�r"r"r#r/�s
�zConsumer.declarecC�|j�|�dS)a%Register a new callback to be called when a message is received.

        Note:
        ----
            The signature of the callback needs to accept two arguments:
            `(body, message)`, which is the decoded message body
            and the :class:`~kombu.Message` instance.
        N)rur;)r �callbackr"r"r#�register_callback�s	zConsumer.register_callbackcCs|��|Sr*)�consumer'r"r"r#r[�szConsumer.__enter__r]r^r_r`rarbrcrdcCsT|jr$|jjr&|jjjj}t||�s(z|��WdSty#YdSwdSdSdSr*)r!r<ri�connection_errorsr3�cancel�	Exception)r r]r_ra�conn_errorsr"r"r#rf�s
��zConsumer.__exit__cCs(||j�}|jr|��||j|j<|S)z�Add a queue to the list of queues to consume from.

        Note:
        ----
            This will not start consuming from the queue,
            for that you will have to call :meth:`consume` after.
        )r!rr/r}r.r�r"r"r#�	add_queue�s

zConsumer.add_queuecCslt|j���}|r4|dur|jn|}|dd�|d}}|D]
}|j||dd�q|j||dd�dSdS)aVStart consuming messages.

        Can be called multiple times, but note that while it
        will consume from new queues added since the last call,
        it will not cancel consuming from removed queues (
        use :meth:`cancel_by_queue`).

        Arguments:
        ---------
            no_ack (bool): See :attr:`no_ack`.
        N���T�rt�nowaitF)r|r}r~rt�_basic_consume)r rtrs�H�Tr�r"r"r#r��s�zConsumer.consumecCs.|jj}|j��D]}||�q	|j��dS)z�End all active queue consumers.

        Note:
        ----
            This does not affect already delivered messages, but it does
            mean the server will not send any more messages for this consumer.
        N)r!�basic_cancelrxr~r�)r r��tagr"r"r#r��s
zConsumer.cancelc
Cs~t|t�r|jn|}z,z|j�|�}Wn	tyYnw|j�|�W|j�|d�dSW|j�|d�dS|j�|d�w)zCancel consumer by queue name.N)	r3rr.rxr��KeyErrorr!r�r})r r�r�r�r"r"r#�cancel_by_queues��"zConsumer.cancel_by_queuecCs|}t|t�r
|j}||jvS)z8Return :const:`True` if currently consuming from queue'.)r3rr.rx)r r�r.r"r"r#�consuming_froms

zConsumer.consuming_fromcCstdd�|j��D��S)z�Purge messages from all queues.

        Warning:
        -------
            This will *delete all ready messages*, there is no undo operation.
        css�|]}|��VqdSr*)�purge)rIr�r"r"r#�	<genexpr>!s�z!Consumer.purge.<locals>.<genexpr>)�sumr}r~r'r"r"r#r�szConsumer.purgecCr�)a�Enable/disable flow from peer.

        This is a simple flow-control mechanism that a peer can use
        to avoid overflowing its queues or otherwise finding itself
        receiving more messages than it can process.

        The peer that receives a request to stop sending content
        will finish sending the current content (if any), and then wait
        until flow is reactivated.
        N)r!�flow)r �activer"r"r#r�#sz
Consumer.flowrFcCs|j�|||�S)aSpecify quality of service.

        The client can request that messages should be sent in
        advance so that when the client finishes processing a message,
        the following message is already held locally, rather than needing
        to be sent down the channel. Prefetching gives a performance
        improvement.

        The prefetch window is Ignored if the :attr:`no_ack` option is set.

        Arguments:
        ---------
            prefetch_size (int): Specify the prefetch window in octets.
                The server will send a message in advance if it is equal to
                or smaller in size than the available prefetch size (and
                also falls within other prefetch limits). May be set to zero,
                meaning "no specific limit", although other prefetch limits
                may still apply.

            prefetch_count (int): Specify the prefetch window in terms of
                whole messages.

            apply_global (bool): Apply new settings globally on all channels.
        )r!�	basic_qos)r �
prefetch_sizer{�apply_globalr"r"r#r�0s�zConsumer.qoscCs|jj|d�S)a�Redeliver unacknowledged messages.

        Asks the broker to redeliver all unacknowledged messages
        on the specified channel.

        Arguments:
        ---------
            requeue (bool): By default the messages will be redelivered
                to the original recipient. With `requeue` set to true, the
                server will attempt to requeue the message, potentially then
                delivering it to an alternative subscriber.
        )�requeue)r!�
basic_recover)r r�r"r"r#�recoverMs
zConsumer.recovercs*|j}|s	td����fdd�|D�dS)a�Method called when a message is received.

        This dispatches to the registered :attr:`callbacks`.

        Arguments:
        ---------
            body (Any): The decoded message body.
            message (~kombu.Message): The message instance.

        Raises
        ------
            NotImplementedError: If no consumer callbacks have been
                registered.
        z$Consumer does not have any callbackscsg|]}|����qSr"r")rIr��r>rPr"r#rJnsz$Consumer.receive.<locals>.<listcomp>N)ru�NotImplementedError)r r>rPrur"r�r#�receive\szConsumer.receivecCs:|j�|j�}|dur|�||�}|j||j||d�|S)Nr�)rxrNr.�_add_tagr��_receive_callback)r r��consumer_tagrtr�r�r"r"r#r�ps
�zConsumer._basic_consumecCs(|pd�|jt|j��}||j|j<|S)Nz{}{})�formatrw�next�_tagsrxr.)r r�r�r�r"r"r#r�ys
�zConsumer._add_tagc
Cs�|j}|j|jd}}}z't|dd�}|r||�}|dur!||_|jr+|�|j�WS|r/dn|��}WntyQ}z|js@�|�||�WYd}~dSd}~ww|rX||�S|�	||�S)N�message_to_python)
rzrvr!�getattr�errors�_reraise_errorry�decoder�r�)r rPrz�on_mr!�decoded�m2p�excr"r"r#r�s$��zConsumer._receive_callbackcCsdt|�j�d|j�d�S)N�<z: r%)�typermrsr'r"r"r#r(�szConsumer.__repr__cCs"z|jjjWStyYdSwr*)r!r<rirjr'r"r"r#r<�s
�zConsumer.connection)	NNNNNNNNNrlr*)rrFrk))rmrnrorprr!rsrtrrurvryrzr{r}rr�r$rq�setterrr/r�r[rfr�r�r�rrr�r�r�r�r�r�r�r�r�r�r(r<r"r"r"r#r's\	
�




	


�
	rN)rp�
__future__r�	itertoolsr�typingr�commonrrrr<r	r
r0rrr
�
exceptionsr�
serializationrr�utils.functionalrr�typesr�__all__rrr"r"r"r#�<module>s$
@ Telegram