File: //lib64/python3.6/site-packages/ldap/__pycache__/asyncsearch.cpython-36.pyc
3
�`a� � @ s� d Z ddlZddlmZ ddlZejejejhZejejhZG dd� de �Z
G dd� d�Z G dd � d e �Z G d
d � d e �Z
G d d
� d
e
�ZG dd� de �ZG dd� de�ZdS )zg
ldap.asyncsearch - handle async LDAP search operations
See https://www.python-ldap.org/ for details.
� N)� __version__c @ s e Zd Zdd� Zdd� ZdS )�WrongResultTypec C s || _ || _tj| � d S )N)�receivedResultType�expectedResultTypes� Exception�__init__)�selfr r � r �#/usr/lib64/python3.6/asyncsearch.pyr s zWrongResultType.__init__c C s d| j dj| j�f S )Nz3Received wrong result type %s (expected one of %s).z, )r �joinr )r r r r
�__str__ s zWrongResultType.__str__N)�__name__�
__module__� __qualname__r r r r r r
r s r c @ sL e Zd ZdZdd� Zddd�Zd d
� Zd d � Zd
d� Zddd�Z dd� Z
dS )�AsyncSearchHandlerz`
Class for stream-processing LDAP search results
Arguments:
l
LDAPObject instance
c C s || _ d | _d| _d S )N� )�_l�_msgId�_afterFirstResult)r �lr r r
r 1 s zAsyncSearchHandler.__init__Nr r c
C s( | j j||||||| ||� | _d| _dS )a�
searchRoot
See parameter base of method LDAPObject.search()
searchScope
See parameter scope of method LDAPObject.search()
filterStr
See parameter filter of method LDAPObject.search()
attrList=None
See parameter attrlist of method LDAPObject.search()
attrsOnly
See parameter attrsonly of method LDAPObject.search()
timeout
Maximum time the server shall use for search operation
sizelimit
Maximum number of entries a server should return
(request client-side limit)
serverctrls
list of server-side LDAP controls
clientctrls
list of client-side LDAP controls
r N)r Z
search_extr r )
r Z
searchRootZ searchScopeZ filterStrZattrListZ attrsOnly�timeoutZ sizelimitZ serverctrlsZ clientctrlsr r r
� startSearch6 s
!z AsyncSearchHandler.startSearchc C s dS )zd
Do anything you want after starting search but
before receiving and processing results
Nr )r r r r
�
preProcessing^ s z AsyncSearchHandler.preProcessingc C s dS )zh
Do anything you want right after successfully receiving but before
processing first result
Nr )r r r r
�afterFirstResultd s z#AsyncSearchHandler.afterFirstResultc C s dS )zI
Do anything you want after receiving and processing all results
Nr )r r r r
�postProcessingj s z!AsyncSearchHandler.postProcessingc
C s, | j � d}|| }d}d}d| _|| _z�d\}} x�|� r�x@|dkrz| rz| jj| jd|�\}} }
} | jr<| j� d| _q<W | s�P |tkr�t |t��xR| D ]J} ||k r�| jd | _n(|dks�||k r�| j
|| � n
d}d}P |d }q�W d\}} || _q4W W d|�r | jdk�r | jj | j� X | j � |S )a1
ignoreResultsNumber
Don't process the first ignoreResultsNumber results.
processResultsCount
If non-zero this parameters indicates the number of results
processed is limited to processResultsCount.
timeout
See parameter timeout of ldap.LDAPObject.result()
r r N)NN)NN)
r ZbeginResultsDroppedZendResultBreakr Zresult3r r r �SEARCH_RESULT_TYPESr �_processSingleResultZabandonr )
r ZignoreResultsNumberZprocessResultsCountr Zresult_counterZend_result_counterZgo_ahead�partialZ result_typeZ result_listZ result_msgidZresult_serverctrlsZ result_itemr r r
�processResultso sD
z!AsyncSearchHandler.processResultsc C s dS )zv
Process single entry
resultType
result type
resultItem
Single item of a result list
Nr )r �
resultType�
resultItemr r r
r � s z'AsyncSearchHandler._processSingleResult���)Nr r! r NNr! )r r r! ) r
r r �__doc__r r r r r r r r r r r
r ' s
0r c @ s e Zd ZdZdd� Zdd� ZdS )�Listz�
Class for collecting all search results.
This does not seem to make sense in the first place but think
of retrieving exactly a certain portion of the available search
results.
c C s t j| |� g | _d S )N)r r �
allResults)r r r r r
r � s z
List.__init__c C s | j j||f� d S )N)r$ �append)r r r r r r
r � s zList._processSingleResultN)r
r r r" r r r r r r
r# � s r# c @ s e Zd ZdZdd� Zdd� ZdS )�DictzJ
Class for collecting all search results into a dictionary {dn:entry}
c C s t j| |� i | _d S )N)r r �
allEntries)r r r r r
r � s z
Dict.__init__c C s |t kr|\}}|| j|< d S )N)�ENTRY_RESULT_TYPESr' )r r r �dn�entryr r r
r � s zDict._processSingleResultN)r
r r r" r r r r r r
r& � s r& c @ s" e Zd ZdZddd�Zdd� ZdS )� IndexedDictz�
Class for collecting all search results into a dictionary {dn:entry}
and maintain case-sensitive equality indexes to entries
Nc C s* t j| |� |pf | _i j| ji �| _d S )N)r& r �
indexed_attrs�fromkeys�index)r r r, r r r
r � s
zIndexedDict.__init__c
C s� |t kr�|\}}|| j|< xf| jD ]\}||kr"xN|| D ]B}y| j| | j|� W q8 tk
rx |g| j| |< Y q8X q8W q"W d S )N)r( r' r, r. r% �KeyError)r r r r) r* �a�vr r r
r � s
z IndexedDict._processSingleResult)N)r
r r r" r r r r r r
r+ � s
r+ c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �
FileWriterz�
Class for writing a stream of LDAP search results to a file object
Arguments:
l
LDAPObject instance
f
File object instance where the LDIF data is written to
� c C s" t j| |� || _|| _|| _d S )N)r r �_f� headerStr� footerStr)r r �fr5 r6 r r r
r � s zFileWriter.__init__c C s | j j| j� dS )zs
The headerStr is written to output after starting search but
before receiving and processing results.
N)r4 �writer5 )r r r r
r � s zFileWriter.preProcessingc C s | j j| j� dS )zX
The footerStr is written to output after receiving and
processing results.
N)r4 r8 r6 )r r r r
r � s zFileWriter.postProcessingN)r3 r3 )r
r r r" r r r r r r r
r2 � s
r2 c @ s" e Zd ZdZddd�Zdd� ZdS ) �
LDIFWriterz�
Class for writing a stream LDAP search results to a LDIF file
Arguments:
l
LDAPObject instance
writer_obj
Either a file-like object or a ldif.LDIFWriter instance used for output
r3 c C s: t |tj�r|| _n tj|�| _tj| || jj||� d S )N)�
isinstance�ldifr9 � _ldif_writerr2 r Z _output_file)r r Z
writer_objr5 r6 r r r
r s zLDIFWriter.__init__c C s" |t kr |\}}| jj||� d S )N)r( r< Zunparse)r r r r) r* r r r
r s z LDIFWriter._processSingleResultN)r3 r3 )r
r r r" r r r r r r
r9 s
r9 )r" Zldapr r; ZRES_SEARCH_ENTRYZRES_SEARCH_RESULTZRES_SEARCH_REFERENCEr r( r r r r# r& r+ r2 r9 r r r r
�<module> s