3 NZOA @svdZdZdddddddd gZd d lmZmZmZd d lZd d lZd d l Z yd dl m Z Wn e k rxd dl m Z YnXd dl TdZdZedeZededZejdeZdeZd d ddZddddZddddgZiZxeD] Zd ee<qWdZejeZd d!Zd"d#Zd$d%ZGd&ddZ d-d(dZ!Gd)ddZ"Gd*dde"Z#Gd+d d e"Z$d.d,dZ%d S)/a ldif - generate and parse LDIF data (see RFC 2849) written by Michael Stroeder See http://python-ldap.sourceforge.net for details. $Id: ldif.py,v 1.3 2008/02/11 16:35:43 dwelch Exp $ Python compability note: Tested with Python 2.0+, but should work with Python 1.5.2+. The python-ldap package is distributed under Python-style license. Standard disclaimer: This software is made available by the author(s) to the public for free and "as is". All users of this free software are solely and entirely responsible for their own choice and use of this software for their own purposes. By using this software, each user agrees that the author(s) shall not be liable for damages of any kind in relation to its use or performance. The author(s) do not warrant that this software is fit for any purpose. Note: This file is part of the python-ldap package. For the complete python-ldap package, please visit: http://sourceforge.net/projects/python-ldap/ It has been modified for use in HPLIP. z0.5.5 ldif_patternZAttrTypeandValueLDIF CreateLDIF ParseLDIF LDIFWriter LDIFParserLDIFRecordListLDIFCopy)urllib2_request urllib2_parse urllib2_errorN)StringIO)*z[\w;.]+(;[\w_-]+)*z(([^,]|\\,)+|".*?")z [ ]*=[ ]*z ([ ]*,[ ]*z)*[ ]*z^%s$z>^((dn(:|::) %(dn_pattern)s)|(%(attrtype_pattern)s(:|::) .*)$)+)adddeletereplacerrr)r rrmodifyZmodrdnu"(^(| | | |:|<)|[ €-ÿ]+|[ ]+$)cCs,|dkr dStj|}|dko*|jd|kS)z% returns 1 if s is a LDAP DN rNr )dn_regexmatchgroup)sZrmr/usr/share/hplip/base/ldif.pyis_dnQs rcCstj|dk S)zK returns 1 if s has to be base-64 encoded because of special chars N)safe_string_research)rrrr needs_base64[srcCstdd|DS)zR return a dictionary with all items of l being the keys of the dictionary cSsg|] }|dfqS)Nr).0irrr fszlist_dict..)dict)lrrr list_dictbsr$c@sBeZdZdZdddZddZd d Zd d Zd dZddZ dS)rz Write LDIF entry or change records to file object Copy LDIF input to a file output object containing all data retrieved via URLs NL cCs4||_tddtpgD|_||_||_d|_dS)aI output_file file object for output 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 cSsg|] }|jqSr)lower)rarrrr!}sz'LDIFWriter.__init__..r N) _output_filer$Z bases64_attrs _base64_attrs_cols _line_seprecords_written)self output_file base64_attrscolsline_seprrr__init__ps zLDIFWriter.__init__cCst|}||jkr.|jj||jj|jn|j}|jj|dt||j|jj|jxX||kr|jjd|jj||t|||jd|jj|j||jd}q`WdS)z? Write string line as one or more folded lines r  rN)lenr+r)writer,min)r.lineZline_lenposrrr_unfoldLDIFLines    $zLDIFWriter._unfoldLDIFLinecCsR|j|jkst|r:|jdj|tj|jddgn|jdj||gdS)z Write a single attribute type/value pair attr_type attribute type attr_value attribute value z:: r&rz: N)r'r*rr:joinbase64Z encodestringr)r. attr_type attr_valuerrr_unparseAttrTypeandValues $z#LDIFWriter._unparseAttrTypeandValuecCsLt|jdd}|jx*|D]"}x||D]}|j||q0Wq"WdS)z? entry dictionary holding an entry N)listkeyssortr?)r.entryZ attr_typesr=r>rrr_unparseEntryRecords  zLDIFWriter._unparseEntryRecordc Cst|d}|dkrd}n|dkr(d}ntd|jd|x|D]z}|dkrX|\}}n,|dkr||\}}}|jt||ntd|rx|D]}|j||qW|dkrB|jjd |jqBWd S) z\ modlist list of additions (2-tuple) or modifications (3-tuple) r rrrzmodlist item of wrong length changetypez'Subsequent modlist item of wrong length-N)r5 ValueErrorr? MOD_OP_STRr)r6r,) r.ZmodlistZmod_lenrFmodmod_typeZmod_valsZmod_opZmod_valrrr_unparseChangeRecords&      zLDIFWriter._unparseChangeRecordcCsf|sdS|jd|t|tr*|j|nt|tr@|j|ntd|jj|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(). Ndnz*Argument record must be dictionary or listr) r? isinstancer"rDr@rLrHr)r6r,r-)r.rMrecordrrrunparses      zLDIFWriter.unparse)Nr%r&) __name__ __module__ __qualname____doc__r3r:r?rDrLrPrrrrris  r%cCs4t}t|||d}|j|||j}|j|S)a Create LDIF single formatted record including trailing empty line. This is a compability function. Use is deprecated! 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. r&)r rrPgetvalueclose)rMrOr0r1fZ ldif_writerrrrrrs  c@sBeZdZdZddZdddZd d Zd d Zd dZddZ dS)rz 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 cCs|jdjdS)zQ Strip trailing line separators from s, but no other whitespaces r& )strip)r.rrrr _stripLineSep szLDIFParser._stripLineSepNr r&cCsL||_||_tdd|pgD|_tdd|p2gD|_||_d|_dS)aK 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 cSsg|] }|jqSr)r')rrrrrr!,sz'LDIFParser.__init__..cSsg|] }|jqSr)r')rr(rrrr!-sr N) _input_file _max_entriesr$_process_url_schemes_ignored_attr_typesr, records_read)r. input_fileignored_attr_types max_entriesprocess_url_schemesr2rrrr3s zLDIFParser.__init__cCsdS)z Process a single content LDIF record. This method should be implemented by applications using LDIFParser. Nr)r.rMrCrrrhandle1szLDIFParser.handlecCsd|j|jg}|jj|_x>|jrX|jddkrX|j|j|jdd|jj|_qWdj|S)zO Unfold several folded lines with trailing space into one line r r4rNr)rZ_liner[readlineappendr;)r.Zunfolded_linesrrrr:7s  zLDIFParser._unfoldLDIFLinec Cs8|j}x|r$|ddkr$|j}q W| s<|dks<|dkr@d Sy|jd}Wn"tk rptjd|dSX|d|}|||d }|d krtj||d d}n|d kr||d dj}d}|jr0t j |}|d|jkr0t j |j }n.|d ks|dkrd}n||d dj}||fS)zj Parse a single attribute type and value pair from one or more lines of LDIF data r #r&z N:zMalformed line: %srrz::z:Zurlurrr_parseAttrTypeandValueDs2    z!LDIFParser._parseAttrTypeandValuecCs|jj|_x|jo(|j s,|j|jkrd}d}d}d}i}|j\}}x6|dk o^|dk r|j}tjdt |t |f| s| r|j\}}qP|dkr|dk rt dt |st dt ||}n|dkr|dkrd}n|dkr8|dkrt d |dk rt d |t kr2t d t ||}n>|dk rv|j|j krv||krl||j|n |g||<|j\}}qPW|r|j|||jd7_qWdS) z9 Continously read and parse LDIF records Nz %s ---> %srMz*Two lines starting with dn: in one record.z8No valid string-representation of distinguished name %s.versionrrFz/Read changetype: before getting valid dn: line.z2Two lines starting with changetype: in one record.zchangetype value %s is invalid.)r[rfrer\r_rpr'rkdebugreprrHrvalid_changetype_dictr^rgrd)r.rqrMrFZmodoprCr=r>rrrparseusR             zLDIFParser.parse)Nr Nr&) rQrRrSrTrZr3rdr:rprurrrrrs   1c@s"eZdZdZdddZddZdS) rzt Collect all records of LDIF input into a single list. of 2-tuples (dn, entry). It can be a memory hog! Nr cCstj|||||g|_dS)z See LDIFParser.__init__() Additional Parameters: all_records List instance for storing parsed records N)rr3 all_records)r.r`rarbrcrrrr3s zLDIFRecordList.__init__cCs|jj||fdS)zD Append single record to dictionary of all records. N)rvrg)r.rMrCrrrrdszLDIFRecordList.handle)Nr N)rQrRrSrTr3rdrrrrrs c@s"eZdZdZd ddZdd ZdS) rzS Copy LDIF input to LDIF output containing all data retrieved via URLs Nr r%r&c Cs&tj|||||t|||||_dS)zE See LDIFParser.__init__() and LDIFWriter.__init__() N)rr3r _output_ldif) r.r`r/rarbrcr0r1r2rrrr3szLDIFCopy.__init__cCs|jj||dS)z: Write single LDIF record to output file. N)rwrP)r.rMrCrrrrdszLDIFCopy.handle)Nr NNr%r&)rQrRrSrTr3rdrrrrrs  cCst|||dd}|j|jS)zc Parse LDIF records read from file. This is a compability function. Use is deprecated! r )rarbrc)rrurv)rWZ ignore_attrsZ maxentriesZ ldif_parserrrrrs )Nr%)Nr )&rT __version____all__Z sixext.movesr r r r<retypesior ImportErrorgZattrtype_patternZattrvalue_patternZ rdn_patternZ dn_patterncompilervarsrZMOD_OP_INTEGERrIZ CHANGE_TYPESrtcZSAFE_STRING_PATTERNrrrr$rrrrrrrrrrsR          =