File: //proc/self/root/lib64/python3.9/site-packages/ldap/__pycache__/asyncsearch.cpython-39.pyc
a
�.d� � @ 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 || _ || _t�| � d S �N)�receivedResultType�expectedResultTypes� Exception�__init__)�selfr r � r
�6/usr/lib64/python3.9/site-packages/ldap/asyncsearch.pyr s zWrongResultType.__init__c C s d� | jd�| j��S )Nz3Received wrong result type {} (expected one of {}).z, )�formatr �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 ���c
C s( | 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> | � � d}|| }d}d}d| _|| _z�d\}} |r�|du rr| sr| j�| jd|�\}} }
} | jr6| �� d| _q6| sxq�|tvr�t |t��| D ]L} ||k r�| jd | _n*|dks�||k r�| �
|| � n d}d} q�|d }q�d\}} || _q2W |�r2| jdk�r2| j� | j� n"|�r0| jdk�r0| j� | j� 0 | � � |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 )NNN)
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 sH
�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 NN)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 �| |� g | _d S r )r r �
allResultsr r
r
r r � s z
List.__init__c C s | j �||f� d S r )r) �appendr$ 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 �| |� i | _d S r )r r �
allEntriesr r
r
r r � s z
Dict.__init__c C s |t v r|\}}|| j|< d S r )�ENTRY_RESULT_TYPESr- �r r% r& �dn�entryr
r
r r! � s zDict._processSingleResultNr+ 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 �| |� |pd| _i �| ji �| _d S )Nr
)r, r �
indexed_attrs�fromkeys�index)r r r3 r
r
r r � s
zIndexedDict.__init__c
C s| |t v rx|\}}|| j|< | jD ]V}||v r || D ]@}z| j| | �|� W q4 tyr |g| j| |< Y q40 q4q d S r )r. r- r3 r5 r* �KeyError)r r% r& r0 r1 �a�vr
r
r r! � s
z IndexedDict._processSingleResult)Nr+ r
r
r
r r2 � s
r2 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 �| |� || _|| _|| _d S r )r r �_f� headerStr� footerStr)r r �fr< r= r
r
r r � s zFileWriter.__init__c C s | j �| j� dS )zs
The headerStr is written to output after starting search but
before receiving and processing results.
N)r; �writer< r r
r
r r � s zFileWriter.preProcessingc C s | j �| j� dS )zX
The footerStr is written to output after receiving and
processing results.
N)r; r? r= r r
r
r r � s zFileWriter.postProcessingN)r: r: )r r r r' r r r r
r
r
r r9 � s
r9 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
r: c C s: t |tj�r|| _n t�|�| _t�| || jj||� d S r )�
isinstance�ldifr@ � _ldif_writerr9 r Z _output_file)r r Z
writer_objr< r= r
r
r r s zLDIFWriter.__init__c C s" |t v r |\}}| j�||� d S r )r. rC Zunparser/ r
r
r r! s z LDIFWriter._processSingleResultN)r: r: r+ r
r
r
r r@ s
r@ )r' Zldapr rB ZRES_SEARCH_ENTRYZRES_SEARCH_RESULTZRES_SEARCH_REFERENCEr r. r r r r( r, r2 r9 r@ r
r
r
r �<module> s$ ��