File: //usr/lib/python3/dist-packages/jeepney/io/__pycache__/common.cpython-310.pyc
o
���`�
� @ s| d dl mZ d dlmZ d dlmZmZmZmZ G dd� d�Z G dd� d�Z
G dd � d �Z G d
d � d e �Z
d efd
d�ZdS )� )�contextmanager)�count)� HeaderFields�Message� MessageFlag� MessageTypec @ s e Zd Zdd� Zdd� ZdS )�MessageFiltersc C s i | _ t� | _d S �N)�filtersr �
filter_ids��self� r �3/usr/lib/python3/dist-packages/jeepney/io/common.py�__init__ s zMessageFilters.__init__c c s( � | j �� D ] }|j�|�r|V qd S r )r
�values�rule�matches)r
�message�handler r r r s � ��zMessageFilters.matchesN)�__name__�
__module__� __qualname__r r r r r r r s r c @ s2 e Zd Zdefdd�Zdd� Zdd� Zdd � Zd
S ) � FilterHandler
c C s0 || _ t|j�| _|| _|| _| | j j| j< d S r )�_filters�nextr �
_filter_idr �queuer
)r
r
r r r r r r s
zFilterHandle.__init__c C s | j j| j= d S r )r r
r r r r r �close s zFilterHandle.closec C s | j S r )r r r r r � __enter__ s zFilterHandle.__enter__c C s | � � dS )NF)r )r
�exc_type�exc_val�exc_tbr r r �__exit__ s zFilterHandle.__exit__N)r r r r r r r r# r r r r r s
r c @ s8 e Zd Zdd� Zedd� �Zdd� Zd defd d
�ZdS ) � ReplyMatcherc C s
i | _ d S r ��_futuresr r r r r & s
zReplyMatcher.__init__c c s, � || j |< z
|V W | j |= dS | j |= w )z>Context manager to capture a reply for the given serial numberNr% )r
�serial�futurer r r �catch) s
�
zReplyMatcher.catchc C s4 |j j�tjd�}|| jv r| j| �|� dS dS )zDispatch an incoming message which may be a reply
Returns True if a task was waiting for it, otherwise False.
���TF)�header�fields�getr � reply_serialr& �
set_result)r
�msg�
rep_serialr r r �dispatch3 s
zReplyMatcher.dispatchN�excc C s: |du rt d�}| ji }| _|�� D ]}|�|� qdS )z4Throw an error in any task still waiting for a replyNz(D-Bus router closed before reply arrived)� RouterClosedr& r �
set_exception)r
r3 �futures�futr r r �drop_all? s �zReplyMatcher.drop_allr ) r r r r r r) r2 � Exceptionr8 r r r r r$ % s
r$ c @ s e Zd ZdZdS )r4 a Raised in tasks waiting for a reply when the router is closed
This will also be raised if the receiver task crashes, so tasks are not
stuck waiting for a reply that can never come. The router object will not
be usable after this is raised.
N)r r r �__doc__r r r r r4 H s r4 r0 c C s< | j jtjkrtd| j j� d���tj| j j@ r td��dS )z4Raise an error if we wouldn't expect a reply for msgz,Only method call messages have replies (not �)z/This message has the no_reply_expected flag setN) r+ � message_typer � method_call� TypeErrorr �no_reply_expected�flags�
ValueError)r0 r r r �check_replyableR s
��rB N)�
contextlibr � itertoolsr �jeepneyr r r r r r r$ r9 r4 rB r r r r �<module> s #