U >Ke@sdZddlZddlZddlmZddlmZddlmZddl Z ddlZej ddkrZe Z dZ edfedfedfed fed fed fed fd ZGdddeZdS)z:Class to show and manipulate user fields in odf documents.N) UserFieldDecl)OFFICENS)loadzutf-8valuez date-valuez time-valuez boolean-valuez string-value)float percentagecurrencydatetimebooleanstringc@sdeZdZdZdZdZdddZddZddZd d Z dd d Z d dZ ddZ ddZ ddZdS) UserFieldsz&List, view and manipulate user fields.NcCs|dks@dt|ks@dt|ks@dt|ks@t|tdks@t|dksdt|ksdt|ksdt|kst|tdkst||_||_d|_dS)aConstructor @param src open file in binary mode: source document, or filename as a unicode string, or None for stdin. @param dest opendile in binary mode: destination document, or filename as a unicode string, or None for stdout. NrbBufferedReaderBytesIOwbBufferedWriter)reprtypeAssertionErrorsrc_file dest_filedocument)selfsrcdestr1/tmp/pip-unpacked-wheel-awqfhqe2/odf/userfield.py__init__9s @@zUserFields.__init__cCstjddkr(t|jtsBt|jtjsBtjddkr^t|jtr^t |jspt d|jn|jdkrptj |_t |j|_ dS)Nrrz%s is no odt file.)sys version_info isinstancerstrioIOBase basestringzipfile is_zipfile TypeErrorstdinrrrrrrloaddocGs B  zUserFields.loaddoccCs*|jdkr|jdn|j|jdS)N-)rrsaver-rrrsavedocRs zUserFields.savedoccCsdd|DS)zlList (extract) all known user-fields. @return list of user-field names as unicode strings. cSsg|] }|dqS)rr.0xrrr _sz*UserFields.list_fields..list_fields_and_valuesr-rrr list_fieldsZszUserFields.list_fieldscCsx|g}|jt}|D]V}|d}|dkr>|d}n |d}|d}|dksb||kr||||fq|S)a(List (extract) user-fields with type and value. @param field_names list of field names as unicode strings to show, or None for all. @return list of tuples (, , ) as type (unicode string, stringified type, unicode string). valuetyper stringvaluernameN)r.rgetElementsByTyper getAttributeappend)r field_namesZ found_fields all_fieldsf value_typer field_namerrrr7as      z!UserFields.list_fields_and_valuescCsdd||DS)zExtract the contents of given field names from the file. @param field_names list of field names as unicode strings @return list of field values as unicode strings. cSsg|] }|dqS)r!rr2rrrr5sz*UserFields.list_values..r6)rr?rrr list_values|szUserFields.list_valuescCs0t|tdkst||g}|s(dS|dS)zExtract the contents of this field from the file. @param field_name unicode string: name of a field @return field value as a unicode string or None if field does not exist. rNr)rrrD)rrCvaluesrrrgets  zUserFields.getcCs>t|tdkst||g}|s(dS|d\}}}||fS)zExtract the type and contents of this field from the file. @param field_name unicode string: name of a field @return tuple (, ) as a pair of unicode strings or None if field does not exist. rNr)rrr7)rrCfieldsrBrrrrget_type_and_values  zUserFields.get_type_and_valuecCsr||jt}|D]L}|d}||kr|d}||}|dkrX|d|q|d|q|dS)zSet the value of user fields. The field types will be the same. data ... dict, with field name as key, field value as value Returns None r;r9r r:rN)r.rr<rr=rFZ setAttributer1)rdatar@rArCrBrrrrupdates    zUserFields.update)NN)N)__name__ __module__ __qualname____doc__rrr r.r1r8r7rDrFrHrJrrrrr2s     r)rNr"r)Zodf.textrZodf.namespacesrZodf.opendocumentrr&r#r%unicodeZ OUTENCODINGZ VALUE_TYPESobjectrrrrrs$