B
AdS @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl
m Z d dl m
Z
d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ d dlm Z d dlm Z d dl Z d dl Z dZ d!ed d!ed d!ed d!ed d!ed dZ"dZ#d d Z$d d Z%d d! Z&edZ'e'd"d#d$ Z(G d%d& d&e)Z*G d'd( d(e)Z+G d)d* d*e+Z,G d+d, d,e-Z.G d-d. d.e+Z/dS )/ )format_datetime)deepcopy)datetime) CONTENT_TYPE)RELATIONSHIP_TYPE)serialize_part_xml)PackURI)Part) parse_xml)CT_CoreProperties)NS)word_to_python_date_format)xpath)FunctionNamespace)QName) binary_type) text_typeNz&{D5CDD505-2E9C-101B-9397-08002B2CF9AE}zvtzzz z)textintboolr float c C s t | tr&ttd }| r dnd|_nt | trHttd }t| |_nt | trjttd }t| |_n|t | trttd }| d|_nXt | trttd }| |_n:t | t
r| d } ttd }| |_nt d
t| |S ) Nr truefalser r r z%Y-%m-%dT%H:%M:%SZr zutf-8zUnsupported type {})
isinstancer r
CUSTOM_PROPERTY_TYPESr r r r r strftimer decode TypeErrorformattype)valueel r$ H/var/www/html/venv/lib/python3.7/site-packages/docxcompose/properties.pyvalue2vt! s*
r& c C s t | j}|dkr*| j dkr$dS dS nV|dkrY s z lower_case..r$ )contextar$ r$ r%
lower_caseW s r8 c @ s e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z d+ddZ
dd Zdd Zdd Zd d Zd d Zd,d!d"Zd-d#d$Zd%d& Zd'd( Zd)d* ZdS ).CustomPropertieszvCustom doc properties stored in ``/docProps/custom.xml``.
Allows updating of doc properties in a document.
c C sj || _ d | _d | _| | _y|jjtj}W n" t k
rR t
| | _Y nX || _t
|j | _d S )N)
docpart_elementget_doc_languagelanguagepackageZpart_related_byRTCUSTOM_PROPERTIESKeyErrorr
_part_templateblob)selfr: r; r$ r$ r% __init__` s
zCustomProperties.__init__c C s t ddS )NZ docxcomposeztemplates/custom.xml)
pkg_resourcesresource_string)rE r$ r$ r% rC n s z CustomProperties._part_templatec C sh | j d krVtd}t|tjt| j| jj j| _ | jj j | j t
j t | j j
| _nt| j| j _d S )Nz/docProps/custom.xml)r; r r CTZOFC_CUSTOM_PROPERTIESr r< r: r? Z relate_tor@ rA r
rD Z_blob)rE Zpartnamer$ r$ r% _update_partr s
z CustomProperties._update_partc C s2 t | jd| }|s"t|t|d d S )z Get the value of a property.z&.//cp:property[lower-case(@name)="{}"]r )r r< r r. rB r1 )rE keypropsr$ r$ r% __getitem__~ s z CustomProperties.__getitem__c C sZ t | jd| }|s*| || dS |d d }t|}| || | dS )z Set the value of a property.z&.//cp:property[lower-case(@name)="{}"]Nr ) r r< r r. addr& getparentreplacerJ )rE rK r" rL value_elZ new_value_elr$ r$ r% __setitem__ s z CustomProperties.__setitem__c C sp t | jd| }|s"t||d |d t}x&| jD ] }|dt | |d7 }qDW |
dS )zDelete a property.z&.//cp:property[lower-case(@name)="{}"]r pid N) r r< r r. rB rO removeMIN_PIDsetr rJ )rE rK rL rS propr$ r$ r% __delitem__ s z CustomProperties.__delitem__c sf t | jjd}|t | jjjd dtd fdd|D }|rb|d j }|ddS d S )
zWe actually should determine the correct language for each field.
Instead we simply determine the language from the first w:lang tag in
the document, and if None are found, from the w:lang tag in the default
style.
z .//w:langz {{{}}}valwc s g | ]} | kr|qS r$ )keys)r4 r0 )latin_lang_keyr$ r% r5 s z5CustomProperties.get_doc_language..r -_N) r r: r/ extendstylesr r attribrP )rE Z lang_tagsr> r$ )r\ r% r= s z!CustomProperties.get_doc_languagec C sF t | jd| }|s"t|t|d d r.z.//cp:property/@pidrT z cpZfmtidnamerS N)
r r< maxrV r
r r rW CUSTOM_PROPERTY_FMTIDr r& appendrJ )rE rh r" ZpidsrS rX rQ r$ r$ r% rN s
zCustomProperties.addc C s( | j d krg S t| j d}dd |D S )Nz.//cp:propertyc S s g | ]}| d qS )rh )rf )r4 rX r$ r$ r% r5 s z)CustomProperties.keys..)r< r )rE rL r$ r$ r% r[ s
zCustomProperties.keysc C s( | j d krg S t| j d}dd |D S )Nz.//cp:propertyc S s g | ]}t |d qS )r )r1 )r4 rX r$ r$ r% r5 s z+CustomProperties.values..)r< r )rE rL r$ r$ r% values s
zCustomProperties.valuesc C s( | j d krg S t| j d}dd |D S )Nz.//cp:propertyc S s" g | ]}| d t|d fqS )rh r )rf r1 )r4 rX r$ r$ r% r5 s z*CustomProperties.items..)r< r )rE rL r$ r$ r% items s
zCustomProperties.itemsc C s& x | D ]\}}| || q
W d S )N)rm rW )rE
propertiesrh r" r$ r$ r% set_properties s z CustomProperties.set_propertiesc C s g }x^| j jD ]R}|j|j|j|j|j|jg}x0|D ](}|jr4|j s4|
| j |j j
|d q4W qW |
| j | j j
j|d |S )zpThis method searches for all doc-properties in the document and
in section headers and footers.
)rh )r: sectionsZfirst_page_headerheaderZeven_page_headerZfirst_page_footerfooterZeven_page_footerZ_has_definitionZis_linked_to_previousr_ _find_docprops_inr; r/ body)rE rh docpropssectionZall_header_footers containerr$ r$ r% find_docprops_in_document s
z*CustomProperties.find_docprops_in_documentc sT t |d}dd |D }t |d}|dd |D d k rPt fdd|}|S )Nz2.//w:fldSimple[contains(@w:instr, 'DOCPROPERTY ')]c S s g | ] }t |qS r$ ) SimpleField)r4 Z sfield_noder$ r$ r% r5 * s z6CustomProperties._find_docprops_in..z*.//w:instrText[contains(.,'DOCPROPERTY ')]c S s g | ] }t |qS r$ ) ComplexField)r4 Z cfield_noder$ r$ r% r5 1 s c s
| j kS )N)rh )rX )rh r$ r% 4 z4CustomProperties._find_docprops_in..)r r_ filter)rE r/ rh Z sfield_nodesru Z cfield_nodesr$ )rh r% rs $ s z"CustomProperties._find_docprops_inc C sL | }t| }x2|D ]*}||j}|dkr4q|j|| jd qW dS )z)Update all the document's doc-properties.N)r> )rx dictrm rf rh updater> )rE ru Zavailable_docpropsdocpropr" r$ r$ r%
update_all7 s
zCustomProperties.update_allc C s, | |}x |D ]}|j|| jd qW dS )z=Update all instances of a given doc-property in the document.)r> N)rx r r> )rE rh r" ru r r$ r$ r% r B s
zCustomProperties.updatec C s$ | |}x|D ] }| qW dS )z0Remove the property fields but keep their value.N)rx replace_field_with_value)rE rh ru r r$ r$ r% dissolve_fieldsH s
z CustomProperties.dissolve_fields)N)N)N)__name__
__module__ __qualname____doc__rF rC rJ rM rR rY r= rb rd rf rN r[ rl rm ro rx rs r r r r$ r$ r$ r% r9 \ s*
r9 c @ sT e Zd ZdZejdejdZdd ZdddZ dd d
Z
d d Z d
d Z dd Z
dS ) FieldBasezEClass used to represent a docproperty field in the document.xml.
zZDOCPROPERTY +"{0,1}([^\\]*?)"{0,1} +(?:\\\@ +"{0,1}([^\\]*?)"{0,1} +){0,1}\\\* MERGEFORMAT)flagsc C s4 || _ | \| _| _| jr*t| j| _nd| _d S )Nshort)node_parse_fieldname_and_formatrh date_formatr
)rE
field_noder$ r$ r% rF W s
zFieldBase.__init__Nc C sP t |tr|rdS dS t |trD|d k r8t|| j|dS t|| jS t|S d S )NYN)locale)r r r r r r )rE r" r> r$ r$ r%
_format_value_ s
zFieldBase._format_valuec C s
t dS )z; Sets the value of the docproperty in the document
N)NotImplementedError)rE r" r> r$ r$ r% r i s zFieldBase.updatec C s
t dS )zS Removes the field from the document, replacing it with
its value.
N)r )rE r$ r$ r% r n s z"FieldBase.replace_field_with_valuec C s
t d S )N)r )rE r$ r$ r% _get_fieldname_stringt s z FieldBase._get_fieldname_stringc C s$ | j | }|d kr dS | S )N)NN) fieldname_and_format_search_exprsearchr groups)rE matchr$ r$ r% r w s
z%FieldBase._parse_fieldname_and_format)N)N)r r r r recompileUNICODEr rF r r r r r r$ r$ r$ r% r P s
r c @ s8 e Zd ZdZded Zdd Zd ddZd d
Z dS ) ry z Represents a simple field, i.e. node in the
document.xml, its body containing the value of the field.
self.node here is the node.
z {{{}}}instrrZ c C s | j j| j S )N)r ra attr_name)rE r$ r$ r% r s z!SimpleField._get_fieldname_stringNc C s( t | jd}|r$| j||d|d _d S )Nz.//w:t)r> r )r r r r )rE r" r> r r$ r$ r% r s zSimpleField.updatec C sD | j }t|| j }t| j d }|| j ||| d S )Nr )r rO listindexr rU insert)rE parentr w_rr$ r$ r% r s
z$SimpleField.replace_field_with_value)N)
r r r r r r r r r r r$ r$ r$ r% ry s
ry c @ s e Zd ZdZdS )InvalidComplexFieldzNThis exception is raised when a complex field cannot
be handled correctly.N)r r r r r$ r$ r$ r% r s r c s e Zd ZdZdZdZdZdZ fddZdd Z e
d
d Z e
d d
Z dd Z
e
dd Zdd Zdd ZdddZdd Z ZS ) rz a2 Represents a complex field, i.e. a several nodes delimited by runs
containing and .
In these fields, the actual value is stored in nodes that come after a
node.
z=./preceding-sibling::w:r/w:fldChar[@w:fldCharType="begin"]/..z;./following-sibling::w:r/w:fldChar[@w:fldCharType="end"]/..z@./following-sibling::w:r/w:fldChar[@w:fldCharType="separate"]/..z w:instrTextc s* | | _| j | _tt| | d S )N)rO r w_psuperrz rF )rE r ) __class__r$ r% rF s
zComplexField.__init__c sx }|dk r j|n jj fddjD }g }x |D ]}|t|j qJW ddd |D S )zThe field name can be split up in several instrText runs
so we look for all the instrText nodes between the begin and either
separate or end runs
Nc s g | ]}j | k r|qS r$ )r r )r4 run)lastrE r$ r% r5 s z6ComplexField._get_fieldname_string..r, c S s g | ]
}|j qS r$ )r )r4 Zeachr$ r$ r% r5 s ) get_separate_runr r end_run_runsr_ r XPATH_TEXTSjoin)rE separate_runrunsZtextsr r$ )r rE r% r s
z"ComplexField._get_fieldname_stringc C s& t | j| j}|s d}t||d S )Nz1Complex field without begin node is not supported)r r XPATH_PRECEDING_BEGINSr )rE Zbeginsmsgr$ r$ r% begin_run s
zComplexField.begin_runc C s8 t | ds2t| j| j}|s(d}t||d | _| jS )N_end_runz/Complex field without end node is not supportedr )hasattrr r XPATH_FOLLOWING_ENDSr r )rE Zendsr r$ r$ r% r s
zComplexField.end_runc C s@ t | j| j}|sdS |d }| j|| j| jk s
Nc s0 g | ](}j |krj | k r|qS r$ )r r )r4 r ) end_indexrE separate_indexr$ r% r5 s z4ComplexField.get_runs_for_update..)r r r r r )rE r r$ )r rE r r% get_runs_for_update s z ComplexField.get_runs_for_updatec sv }|dkr6jj fddjD }n"j|fddjD }|dj |j |S )z
Get all nodes between
and including boundaries,
plus the node
Nc s g | ]}j | k r|qS r$ )r r )r4 r )r rE r$ r% r5 s zEComplexField.get_runs_to_replace_field_with_value..c s g | ]} j |kr|qS r$ )r r )r4 r )rE r r$ r% r5 s r )r r r r r r r rk )rE r r r$ )r rE r r% $get_runs_to_replace_field_with_value s z1ComplexField.get_runs_to_replace_field_with_valueNc C s6 | }|r|d }t|d}|r6| j||d|d _x |dd D ]}|| qDW x|dd D ] }t|d}|rd| j| qdW nt| j}t|dd } | d t
d d t| j}
|
t|
dd t d t
d }| j||d|_|
| | j
| j} | j| d | | j| d
|
d S ) Nr z.//w:t)r> rT z w:fldCharz{{{}}}fldCharTyperZ r zr )r r r r rU r r r rW r r r
rk r r r ) rE r" r> Zruns_after_separateZ first_w_rr Zunnecessary_w_tr r Z
w_fld_charZ value_runZ
docprop_indexr$ r$ r% r s.
zComplexField.updatec C s& | }x|D ]}| j| qW d S )N)r r rU )rE Zruns_to_remover r$ r$ r% r > s
z%ComplexField.replace_field_with_value)N)r r r r r r r r rF r r2 r r r r r r r r
__classcell__r$ r$ )r r% rz s
8rz )0Z babel.datesr copyr r Zdocx.opc.constantsr rI r r@ Z
docx.opc.oxmlr Zdocx.opc.packurir Z
docx.opc.partr Z docx.oxmlr
Zdocx.oxml.corepropsr Zdocxcompose.utilsr r
r Z
lxml.etreer r sixr r rG r rj r r rV r& r1 r3 nsr8 objectr9 r ry Exceptionr rz r$ r$ r$ r% sH u/