Current File : //lib/python3.6/site-packages/pyzor/__pycache__/forwarder.cpython-36.pyc
3
�2�^
� @ sR d Z ddlZddlZy ddlZW n ek
r< ddlZY nX G dd� de�ZdS )z Manage the forwarder process.� Nc @ s< e Zd ZdZddd�Zdd� Zddd �Zd
d � Zd d
� ZdS )� Forwarderz'Forwards digest to remote pyzor servers�' c C s( t jd�| _|| _tj|�| _|| _dS )a
forward_client: a pyzor.client.Client instance to use as
forwarding client
remote_servers: a list of (hostname,port) tuples where digests should
be forwarded to
max_queue_size: max amount of queued digests
ZpyzordN)�loggingZ getLogger�log�forwarding_client�Queue�
forward_queue�remote_servers)�selfr r Zmax_queue_size� r � /usr/lib/python3.6/forwarder.py�__init__ s zForwarder.__init__c C s� x�y| j jddd�\}}W n& tjk
r@ | jdkr:dS wY nX xj| jD ]`}y&|rd| jj||� n| jj||� W qJ tk
r� } z| j j
d|||� W Y dd}~X qJX qJW qW dS )z'read forwarding requests from the queueT� )�blockZtimeoutNz%Forwarding digest %s to %s failed: %s) r �getr ZEmptyr r � whitelistZreport� Exceptionr �warn)r
�digestr ZserverZexr r r �
_forward_loop s
zForwarder._forward_loopFc C s> | j dkrdS y| jj||f� W n tjk
r8 Y nX dS )z�If forwarding is enabled, insert a digest into the forwarding queue
if whitelist is True, the digest will be forwarded as whitelist request
if the queue is full, the digest is dropped
N)r r Z
put_nowaitr ZFull)r
r r r r r �queue_forward_request5 s
z Forwarder.queue_forward_requestc C s t j| jd�j� dS )zstart the forwarding thread)�targetN)� threadingZThreadr �start)r
r r r �start_forwardingB s zForwarder.start_forwardingc C s
d| _ dS )z?disable forwarding and tell the forwarding thread to end itselfN)r )r
r r r �stop_forwardingF s zForwarder.stop_forwardingN)r )F) �__name__�
__module__� __qualname__�__doc__r
r r r r r r r r r s
r )r r r r � ImportErrorZqueue�objectr r r r r �<module> s
Mini Shell