File: //lib64/python3.6/site-packages/__pycache__/ldif.cpython-36.pyc
3
�`a6O � @ sb d Z ddlmZ dZdddddd d
gZdd lZdd lmZmZ dd
l m
Z
dd l Z ddl m
Z
mZ dZdZed e Zed e d Zed e d Zejde �Zde� Zddddd�Zdd d d d �Zdd d d!gZi ZxeD ] Z d ee < q�W d"d#� Z d$Z eje �Z d%d&� Z G d'd� d�Z!d.d)d�Z"G d*d� d�Z#G d+d � d e#�Z$G d,d
� d
e#�Z%d/d-d�Z&d S )0zc
ldif - generate and parse LDIF data (see RFC 2849)
See https://www.python-ldap.org/ for details.
� )�unicode_literalsz3.3.1� ldif_pattern�
CreateLDIF� ParseLDIF�
LDIFWriter�
LDIFParser�LDIFRecordList�LDIFCopyN)� b64encode� b64decode)�StringIO)�urlparse�urlopenz[\w;.-]+(;[\w_-]+)*z(([^,]|\\,)+|".*?")z [ ]*=[ ]*z ([ ]*\+[ ]*z)*[ ]*z
([ ]*,[ ]*z^%s$z>^((dn(:|::) %(dn_pattern)s)|(%(attrtype_pattern)s(:|::) .*)$)+� � � )�add�delete�replace� incrementr r r r )r r r r �modifyZmodrdnc C s, | dkr dS t j| �}|dko*|jd�| kS )z!
returns 1 if s is a LDAP DN
� r Nr )�dn_regex�match�group)�sZrm� r � /usr/lib64/python3.6/ldif.py�is_dn8 s
r s (^( |
|
| |:|<)|[
�-�]+|[ ]+$)c C s dd� | D �S )zN
return a dictionary with all items of l being the keys of the dictionary
c S s i | ]
}d |�qS )Nr )�.0�ir r r �
<dictcomp>I s z list_dict.<locals>.<dictcomp>r )�lr r r � list_dictE s r# c @ sJ e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dS )r z�
Write LDIF entry or change records to file object
Copy LDIF input to a file output object containing all data retrieved
via URLs
N�L �
c C s4 || _ tdd� |pg D ��| _|| _|| _d| _dS )aB
output_file
file object for output; should be opened in *text* mode
base64_attrs
list of attribute types to be base64-encoded in any case
cols
Specifies how many columns a line may have before it's
folded into many lines.
line_sep
String used as line separator
c S s g | ] }|j � �qS r )�lower)r �ar r r �
<listcomp>` s z'LDIFWriter.__init__.<locals>.<listcomp>r N)� _output_filer# �
_base64_attrs�_cols�_last_line_sep�records_written)�self� output_file� base64_attrs�cols�line_sepr r r �__init__S s
zLDIFWriter.__init__c C s� t |�}|| jkr.| jj|� | jj| j� n�| j}| jj|dt|| j�� � | jj| j� xX||k r�| jjd� | jj||t||| j d �� � | jj| j� || j d }q`W dS )z7
Write string line as one or more folded lines
r � r N)�lenr+ r) �writer, �min)r. �lineZline_len�posr r r �
_unfold_linese s
$zLDIFWriter._unfold_linesc C s |j � | jkptj|�dk S )z�
returns 1 if attr_value has to be base-64 encoded because
of special chars or because attr_type is in self._base64_attrs
N)r& r* �safe_string_re�search)r. � attr_type�
attr_valuer r r �_needs_base64_encodingz s z!LDIFWriter._needs_base64_encodingc C sZ | j ||�r<t|�jd�}|jdd�}| jdj||g�� n| jdj||jd�g�� dS )z�
Write a single attribute type/value pair
attr_type
attribute type (text)
attr_value
attribute value (bytes)
�asciir% r z:: z: N)r? r
�decoder r: �join)r. r= r> Zencodedr r r �_unparseAttrTypeandValue� s z#LDIFWriter._unparseAttrTypeandValuec C s8 x2t |j� �D ]"\}}x|D ]}| j||� q W qW dS )z3
entry
dictionary holding an entry
N)�sorted�itemsrC )r. �entryr= �valuesr> r r r �_unparseEntryRecord� s
z LDIFWriter._unparseEntryRecordc C s� t |d �}|dkrd}n|dkr(d}n td| ��| jd|jd�� x�|D ]�}|dkrb|\}}n2|dkr�|\}}}| jt| |jd�� ntd ��|r�x|D ]}| j||� q�W |dkrL| jjd
| j � qLW d S ) zP
modlist
list of additions (2-tuple) or modifications (3-tuple)
r r r r r z modlist item of wrong length: %d�
changetyper@ z'Subsequent modlist item of wrong length�-N)r5 �
ValueErrorrC �encode�
MOD_OP_STRr) r6 r, ) r. ZmodlistZmod_lenrI �modZmod_typeZmod_valsZmod_opZmod_valr r r �_unparseChangeRecord� s(
z LDIFWriter._unparseChangeRecordc C sp |j d�}| jd|� t|t�r,| j|� n&t|t�rB| j|� ntdt|� ��| j j
| j � | j d | _ dS )z�
dn
string-representation of distinguished name
record
Either a dictionary holding the LDAP entry {attrtype:record}
or a list with a modify list like for LDAPObject.modify().
zutf-8�dnz8Argument record must be dictionary or list instead of %sr N)
rL rC �
isinstance�dictrH �listrO rK �reprr) r6 r, r- )r. rP �recordr r r �unparse� s
zLDIFWriter.unparse)Nr$ r% ) �__name__�
__module__� __qualname__�__doc__r3 r: r? rC rH rO rV r r r r r L s
r$ c C sD t jdtdd� t� }t|||d�}|j| |� |j� }|j� |S )a�
Create LDIF single formatted record including trailing empty line.
This is a compatibility function.
dn
string-representation of distinguished name
record
Either a dictionary holding the LDAP entry {attrtype:record}
or a list with a modify list like for LDAPObject.modify().
base64_attrs
list of attribute types to be base64-encoded in any case
cols
Specifies how many columns a line may have before it's
folded into many lines.
zhldif.CreateLDIF() is deprecated. Use LDIFWriter.unparse() instead. It will be removed in python-ldap 3.1r )�category�
stacklevelr% )�warnings�warn�DeprecationWarningr r rV �getvalue�close)rP rU r0 r1 �fZ ldif_writerr r r r r � s c @ sd e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z ddd�Z dd� Z
dS )r z�
Base class for a LDIF parser. Applications should sub-class this
class and override method handle() to implement something meaningful.
Public class attributes:
records_read
Counter for records processed so far
Nr r% c C s� || _ t| j jd�t�| _|| _tdd� |p.g D ��| _tdd� |pFg D ��| _|| _ d| _
d| _ d| _ d| _
i jtd�| _t| _y| j� | _W n tk
r� d| _Y nX dS )a
Parameters:
input_file
File-object to read the LDIF input from
ignored_attr_types
Attributes with these attribute type names will be ignored.
max_entries
If non-zero specifies the maximum number of entries to be
read from f.
process_url_schemes
List containing strings with URLs schemes to process with urllib.
An empty list turns off all URL processing and the attribute
is ignored completely.
line_sep
String used as line separator
r c S s g | ] }|j � �qS r )r& )r r r r r r( s z'LDIFParser.__init__.<locals>.<listcomp>c S s g | ] }|j � �qS r )r& )r r' r r r r( s Nr )� _input_filerQ �read�bytes�_file_sends_bytes� _max_entriesr# �_process_url_schemes�_ignored_attr_typesr, �version� line_counter� byte_counter� records_read�fromkeys� CHANGE_TYPES�changetype_counterr �
_b64decode� _readline�
_last_line�EOFError)r. �
input_file�ignored_attr_types� max_entries�process_url_schemesr2 r r r r3 � s zLDIFParser.__init__c C s dS )zw
Process a single content LDIF record. This method should be
implemented by applications using LDIFParser.
Nr )r. rP rF r r r �handle( s zLDIFParser.handlec C s~ | j j� }| jr|jd�}| jd | _| jt|� | _|s>d S |dd � dkrZ|d d� S |dd � dkrv|d d � S |S d S )
Nzutf-8r r z
r% ���rz ���r{ )rc �readlinerf rA rk rl r5 )r. r r r r rr / s
zLDIFParser._readlinec C sn | j dkr td| j| jf ��| j g}| j� }x.|r\|d dkr\|j|dd� � | j� }q0W || _ dj|�S )zG
Unfold several folded lines with trailing space into one line
Nz%EOF reached after %d lines (%d bytes)r r4 r r )rs rt rk rl rr �appendrB )r. Zunfolded_linesZ next_liner r r r: @ s
zLDIFParser._unfold_linesc C s\ | j � }x |r$|d dkr$| j � }q
W |s.dS |dkr:dS y|jd�}W n2 tk
rz } ztdt|� ��W Y dd}~X nX |d|� }|||d � }|dkr�||d d� j� }|jd �}n�|d
kr�||d d� }|jd �}| j|�}nf|d k�r>||d d� j� }d}| j�rTt |�}|d | jk�rTt
|�j � }n||d
d� jd �}||fS )z�
Parse a single attribute type and value pair from one or
more lines of LDIF data
Returns attr_type (text) and attr_value (bytes)
r �#NrJ �:zno value-spec in %sr z: zutf-8z::r@ z:<r )NN)rJ N) r: �indexrK rT �lstriprL rq �striprh r
r rd ) r. Z
unfolded_lineZ colon_pos�er= Z
value_specr> Zurl�ur r r �_next_key_and_valueQ s: "
z LDIFParser._next_key_and_valuec C sZ | j }y.|� \}}x |dkr0|dkr0|� \}}qW W n tk
rP d\}}Y nX ||fS )z�
Consume empty lines until first non-empty line.
Must only be used between full records!
Returns non-empty key-value-tuple.
N)NN)r� rt )r. �next_key_and_value�k�vr r r �_consume_empty_lines� s
z LDIFParser._consume_empty_linesc C sx | j }y4| j� \}}|dkr8t|jd��| _| j� \}}W n tk
rN dS X �x |dkon| j sr| j| jk �rr|dkr�td| j t
|�f ��|jd�}t |�s�td| j t
|�f ��|}i }|� \}}x�|dk�rJ|j � | j
k�ry|| j|� W n tk
�r |g||< Y nX y|� \}}W q� tk
�rF d \}}Y q�X q�W | j||� | jd | _| j� \}}qTW dS )
z8
Continuously read and parse LDIF entry records
rj r@ NrP z;Line %d: First line of record does not start with "dn:": %szutf-8z6Line %d: Not a valid string-representation for dn: %s.r )NN)r� r� �intrA rj rt rg rm rK rk rT r r& ri r} �KeyErrorry )r. r� r� r� rP rF r r r �parse_entry_records� s@
z LDIFParser.parse_entry_recordsc C s | j � S )zM
Invokes LDIFParser.parse_entry_records() for backward compatibility
)r� )r. r r r �parse� s zLDIFParser.parsec C s g pd}dS )z�
Process a single LDIF record representing a single modify operation.
This method should be implemented by applications using LDIFParser.
Nr )r. rP �modops�controlsr r r �
handle_modify� s zLDIFParser.handle_modifyc ! C s4 | j }| j� \}}|dkr0t|�| _| j� \}}�x�|d koN| j sR| j| jk �r.|dkrptd| jt|�f ��|j d�}t
|�s�td| jt|�f ��|}|� \}}g }x||d ko�|dk�r&|j d�}y|j dd�\}}}W n* tk
�r d }|j dd �\}}Y nX |j |||f� |� \}}q�W d } |d
k�rh|j d�}|t
k�rZtd t|� ��|} |� \}}| d k�r�g }
�y�x|d k�r�y t| } W n, tk
�r� td
| jt|�f ��Y nX |j d�}|} g }
y|� \}}W n tk
�r d\}}Y nX xH|| k�rJ|
j |� y|� \}}W n tk
�rD d\}}Y nX �qW |
j | | |
�p\d f� |� \}}|dk�r~|� \}}�q~W W n tk
�r� d\}}Y nX |
�r�| j||
|� n x|d k�r�|� \}}�q�W | j� \}}y| j| d | j| < W n tk
�r d | j| < Y nX | jd | _q4W d S )Nrj rP z;Line %d: First line of record does not start with "dn:": %szutf-8z6Line %d: Not a valid string-representation for dn: %s.Zcontrolr4 r r rI zInvalid changetype: %sr z"Line %d: Invalid mod-op string: %srJ )NN)NN)NN)r� r� r� rj rg rm rK rk rT rA r �splitr} �valid_changetype_dict�MOD_OP_INTEGERr� rt r� rp )r. r� r� r� rP r� Z control_typeZ criticalityZ
control_valuerI r� ZmodopZmodattrZ modvaluesr r r �parse_change_records� s�
z LDIFParser.parse_change_records)Nr Nr% )N)rW rX rY rZ r3 ry rr r: r� r� r� r� r� r� r r r r r � s
'06
c @ s, e Zd ZdZd
dd�Zdd� Zd dd �ZdS ) r z�
Collect all records of a LDIF file. It can be a memory hog!
Records are stored in :attr:`.all_records` as a single list
of 2-tuples (dn, entry), after calling :meth:`.parse`.
Nr c C s" t j| ||||� g | _g | _d S )N)r r3 � all_records�all_modify_changes)r. ru rv rw rx r r r r3 J s zLDIFRecordList.__init__c C s | j j||f� dS )zS
Append a single record to the list of all records (:attr:`.all_records`).
N)r� r} )r. rP rF r r r ry U s zLDIFRecordList.handlec C s g pd}| j j|||f� dS )z�
Process a single LDIF record representing a single modify operation.
This method should be implemented by applications using LDIFParser.
N)r� r} )r. rP r� r� r r r r� [ s z LDIFRecordList.handle_modify)Nr N)N)rW rX rY rZ r3 ry r� r r r r r B s
c @ s" e Zd ZdZd
dd�Zdd � ZdS ) r zM
Copy LDIF input to LDIF output containing all data retrieved
via URLs
Nr r$ r% c C s& t j| ||||� t||||�| _dS )z=
See LDIFParser.__init__() and LDIFWriter.__init__()
N)r r3 r � _output_ldif) r. ru r/ rv rw rx r0 r1 r2 r r r r3 j s zLDIFCopy.__init__c C s | j j||� dS )z2
Write single LDIF record to output file.
N)r� rV )r. rP rF r r r ry v s zLDIFCopy.handle)Nr NNr$ r% )rW rX rY rZ r3 ry r r r r r d s
c C s. t jdtdd� t| ||dd�}|j� |jS )zL
Parse LDIF records read from file.
This is a compatibility function.
zildif.ParseLDIF() is deprecated. Use LDIFRecordList.parse() instead. It will be removed in python-ldap 3.1r )r[ r\ r )rv rw rx )r] r^ r_ r r� r� )rb Z ignore_attrsZ
maxentriesZ ldif_parserr r r r } s )Nr$ )Nr )'rZ Z
__future__r � __version__�__all__�re�base64r
r �ior r] Z ldap.compatr
r Zattrtype_patternZattrvalue_patternZattrtypeandvalue_patternZ rdn_patternZ
dn_pattern�compiler �varsr r� rM ro r� �cr ZSAFE_STRING_PATTERNr; r# r r r r r r r r r r �<module> sT
T"