o ðù‹i#ã@sdddlmZddlZddlmZddlmZddlmZm Z dddd œZ d Z d Z Gd d „d e ƒZdS)é)ÚdeepcopyN)ÚElement)Úetree)ÚZipFileÚ ZIP_DEFLATEDz9óz&MailMerge.__init__..z7{%(w)s}r/{%(w)s}fldChar[@{%(w)s}fldCharType="begin"]/..crrrrrrrr;rz5{%(w)s}r/{%(w)s}fldChar[@{%(w)s}fldCharType="end"]/..cSsg|]}|‘qSrrrrrrr=sz{%(w)s}r/{%(w)s}instrTextÚ MergeTextcsg|] }ˆˆ|df‘qS)r r)rÚi)ÚparentrrrMsÿz{%(w)s}mailMerge) rÚzipÚpartsÚsettingsÚ_settings_infoÚremove_empty_tablesrÚparseÚopenÚfindallÚ NAMESPACESÚattribÚCONTENT_TYPES_PARTSÚ_MailMerge__get_tree_of_fileÚCONTENT_TYPE_SETTINGSÚreÚcompileÚIÚvaluesÚ enumerateÚtagÚmatchrÚgroupÚlistÚtextÚ getparentÚextendÚrangeÚremoveÚgetrootÚfind)ÚselfÚfiler"Ú content_typesÚtypeÚziÚ to_deleteÚrÚpartÚidxÚchildÚinstrÚmÚfieldsÚ idx_beginÚidx_endÚblockÚ settings_rootÚ mail_merger)rrrÚ__init__sz € ø   ÿ  ÿ ÿû   ÿôõ   üzMailMerge.__init__cCs<|jdt dd¡d}|j |¡}|t |j |¡¡fS)NÚPartNameú/r )r'r&ÚsplitrÚgetinforr#r$)r;r<Úfnr?rrrÚ__get_tree_of_fileZs zMailMerge.__get_tree_of_filecCs¶| ¡D] }|jdi|di¤Žqt|dtƒ}|jjD]9}||jvr4t |j|  ¡¡}|  |j |¡q||j krIt |j   ¡¡}|  |j |¡q|  |j |j |¡¡q| ¡dS)NÚrr)Úget_merge_fieldsÚmergerrrÚfilelistrrÚtostringr9ÚwritestrÚfilenamer!r ÚreadÚclose)r;r<ÚfieldÚoutputr?ÚxmlrrrÚwrite_s      zMailMerge.writeNcCsB|s|j ¡}tƒ}|D]}| d¡D] }| |jd¡qq |S)Nz .//MergeFieldr)rr.Úsetr%Úaddr')r;rrGrBÚmfrrrrUps ÿzMailMerge.get_merge_fieldsc CsØ|j ¡D]d}| ¡}|j}|dtks|dtkrqg}|D] }| |¡| |¡qt|ƒD]8\}}|dkrJtdtƒ} d| j dt<| | ¡g} |D]}t |ƒ} | | ¡|  | ¡qN|j | fi|¤Žq0qdS)z¡ Duplicate template page. Creates a copy of the template for each item in the list, does a merge, and separates the them by page breaks. z {%(w)s}ftrz {%(w)s}hdrrú {%(w)s}brÚpagez {%(w)s}typeN) rr.r9r0r&r8Úappendr/rr'rrV) r;Ú replacementsrBÚrootr0rrDrÚreplÚ pagebreakrÚ child_copyrrrÚ merge_pagesys,      ôôzMailMerge.merge_pagescKsT|s|j ¡}| ¡D]\}}t|tƒr| ||¡q |D] }| |||¡qq dS©N)rr.ÚitemsÚ isinstancer3Ú merge_rowsÚ_MailMerge__merge_field)r;rrgr]Ú replacementrBrrrrV˜s  ÿüzMailMerge.mergecCs | d|¡D]{}t|ƒ}| ¡dt|_| |¡|dur!|nt|ƒ}g}|p*d}| dd¡ d¡}t |ƒD]#\}} t dtƒ} | | _ |  | ¡|t |ƒdkr[|  t dtƒ¡q8| d ¡} | dur}| | ¡} t|ƒD]} | | | ¡qn| | ¡q| |¡qdS) Nú.//MergeField[@name="%s"]z{%(w)s}rrTú Ú z{%(w)s}tr rdr)r%r3Úclearr&r0r6ÚstrÚreplacerPr/rr4rfÚlenr:rÚreversedÚinsertr8)r;rBr]r4rcrÚnodesÚ text_partsrÚ text_partÚ text_nodeÚphrÚnoderrrÚ __merge_field£s0    €     äzMailMerge.__merge_fieldc Cs| |¡\}}}|durDt|ƒdkr6||=t|ƒD]\}}t|ƒ}|j|gfi|¤Ž| |||¡qdS|jrF| ¡} |  |¡dSdSdS)Nr) Ú_MailMerge__find_row_anchorryr/rrVr{r"r5r8) r;ÚanchorÚrowsÚtablerCÚtemplaterÚrow_dataÚrowrrrrrpÂs ýô zMailMerge.merge_rowscCsj|s|j ¡}|D])}| dt¡D]}t|ƒD]\}}| d|¡dur0|||fSqqq dS)Nz .//{%(w)s}tblrs)NNN)rr.r%r&r/r:)r;r]rrBr†rCr‰rrrÚ__find_row_anchorÒs ÿÿÿzMailMerge.__find_row_anchor)Frm) Ú__name__Ú __module__Ú __qualname__rMr)r`rUrlrVrqrprƒrrrrr s C   r )Úcopyrr+Ú lxml.etreerÚlxmlrÚzipfilerrr&r(r*Úobjectr rrrrÚs   ý