^Q\@sdZddlZddlZddlmZddlmZddlmZddl Z ddlZej ddkre Z ndZ iedfd 6edfd 6edfd 6ed fd 6edfd6edfd6edfd6ZGdddeZdS)z:Class to show and manipulate user fields in odf documents.N) UserFieldDecl)OFFICENS)loadzutf-8valuefloatZ percentagecurrencyz date-valuedatez time-valuetimez boolean-valueZbooleanz string-valuestringc@seZdZdZdZdZddddZddZddZd 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)selfsrcdestrB/var/www/dbchiro/venv/lib/python3.4/site-packages/odf/userfield.py__init__9s ``  zUserFields.__init__cCstjddkr:t|jts_t|jtjs_tjddkrt|jtrtj |jst d|jqn|jdkrtj |_nt |j|_ dS)Nrrz%s is no odt file.)sys version_info isinstancerstrioIOBase basestringzipfile is_zipfile TypeErrorstdinrr)rrrrloaddocGs _zUserFields.loaddoccCs9|jdkr"|jjdn|jj|jdS)N-)rrsave)rrrrsavedocRszUserFields.savedoccCsdd|jDS)zlList (extract) all known user-fields. @return list of user-field names as unicode strings. cSsg|]}|dqS)rr).0xrrr _s z*UserFields.list_fields..)list_fields_and_values)rrrr list_fieldsZszUserFields.list_fieldscCs|jg}|jjt}x|D]}|jd}|dkr\|jd}n|jd}|jd}|dks||kr)|j|||fq)q)W|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_namerrrr2as    z!UserFields.list_fields_and_valuescCsdd|j|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)rr)r/r0rrrr1s z*UserFields.list_values..)r2)rr:rrr list_values|szUserFields.list_valuescCsBt|tdkst|j|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)rrr?)rr>valuesrrrgets zUserFields.getcCsWt|tdkst|j|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)rrr2)rr>fieldsr=rrrrget_type_and_values zUserFields.get_type_and_valuecCs|j|jjt}xy|D]q}|jd}||kr#|jd}|j|}|dkr|jd|q|jd|q#q#W|jdS)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 r6r4r r5rN)r+rr7rr8rAZ setAttributer.)rdatar;r<r>r=rrrrupdates    zUserFields.update)__name__ __module__ __qualname____doc__rrrr+r.r3r2r?rArCrErrrrr 2s    r )rIr r'Zodf.textrZodf.namespacesrZodf.opendocumentrr$r!r#unicodeZ OUTENCODINGZ VALUE_TYPESobjectr rrrrs$