o \Rp@sTdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z zddl m Z Wn ey;ddl Z YnwzddlmZWn eyOiZYnwddlmZddlmZddlZeeZe ejed Zejes}e ejed Zid d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1Zd2d3Zd4d5Z  dhd6d7Z!did:d;Z"djd?d@Z#dAdBZ$dkdDdEZ%dFddGddHidfdIdJZ&  F dldKdLZ'dMdNZ(dOdPZ)dQdRZ*dSdTZ+dmdVdWZ,dmdXdYZ-dZd[Z.dnd\d]Z/d^d_Z0d`daZ1dbdcZ2dddeZ3 dndfdgZ4dS)oz Open and modify Microsoft Word 2007 docx files (called 'OpenXML' and 'Office OpenXML' by Microsoft) Part of Python's docx module - http://github.com/mikemaccana/python-docx See LICENSE for licensing information. N)etree)abspathbasenamejoin)Image)TAGS)PendingDeprecationWarning)warnz docx-templatetemplatemoz8http://schemas.microsoft.com/office/mac/office/2008/mainoz'urn:schemas-microsoft-com:office:officevez;http://schemas.openxmlformats.org/markup-compatibility/2006wzz7application/vnd.openxmlformats-officedocument.theme+xmlzLapplication/vnd.openxmlformats-officedocument.wordprocessingml.fontTable+xmlz:application/vnd.openxmlformats-package.core-properties+xmlzEapplication/vnd.openxmlformats-officedocument.extended-properties+xmlzPapplication/vnd.openxmlformats-officedocument.wordprocessingml.document.main+xmlzKapplication/vnd.openxmlformats-officedocument.wordprocessingml.settings+xmlzLapplication/vnd.openxmlformats-officedocument.wordprocessingml.numbering+xmlzIapplication/vnd.openxmlformats-officedocument.wordprocessingml.styles+xmlzNapplication/vnd.openxmlformats-officedocument.wordprocessingml.webSettings+xml) z/word/theme/theme1.xmlz/word/fontTable.xmlz/docProps/core.xmlz/docProps/app.xmlz/word/document.xmlz/word/settings.xmlz/word/numbering.xmlz/word/styles.xmlz/word/webSettings.xmlOverride)PartName ContentTyper>r?z image/gifz image/jpegz image/pngz8application/vnd.openxmlformats-package.relationships+xmlzapplication/xml)gifjpegjpgpngrelsxml) ExtensionrwDefault)rr%r0r/)typespartspart filetypes extensionattrs default_elmr*r*r+ contenttypessB   renc Csvddd}td}td}tdd||t|id}td }td |d }|||||||||S) z.Make a new heading, return the heading elementHeadingTitolo)ritrJrNr`rarMrr]r^)r/strr0) headingtext headinglevellanglmaprorr`rZr;r*r*r+headings      rTdxaautoc# Cstd}t|d} td} tdddid} | | tdt|t|d d} | | t|rwtd } d D]8}||vsGd |vrqd |vrOd n|}i}||D] }t|||||<qYt||d}| |q9| | td ddid}| ||| td}t| D]}d|rt||ndi}|td|dq||td}td}tdddid}|||||rDd}|dD]m}td}td}|rt|||d }nddd }td|d}tdddddd d!d}||||||t|tt fs|g}|D]}t|t j r*||q|t |d"d#q|||d$7}q||||rJd$ndd%D]}td}d}|D]s} td}td}|rpt|||d }nddd }td|d}||||t| tt fs| g} | D]/}!t|!t j r||!q|rd&||vr||d&}"nd'}"|t |!|"d#q|||d$7}qY||qO|S)(a/ Return a table element based on specified parameters @param list contents: A list of lists describing contents. Every item in the list can be a string or a valid XML element itself. It can also be a list. In that case all the listed elements will be merged into the cell. @param bool heading: Tells whether first line should be treated as heading or not @param list colw: list of integer column widths specified in wunitS. @param str cwunit: Unit used for column width: 'pct' : fiftieths of a percent 'dxa' : twentieths of a point 'nil' : no width 'auto' : automagically determined @param int tblw: Table width @param str twunit: Unit used for table width. Same possible values as cwunit. @param dict borders: Dictionary defining table border. Supported keys are: 'top', 'left', 'bottom', 'right', 'insideH', 'insideV', 'all'. When specified, the 'all' key has precedence over others. Each key must define a dict of border attributes: color : The color of the border, in hex or 'auto' space : The space, measured in points sz : The size of the border, in eighths of a point val : The style of the border, see http://www.schemacentral.com/sc/ooxml/t-w_ST_Border.htm @param list celstyle: Specify the style for each colum, list of dicts. supported keys: 'align' : specify the alignment, see paragraph documentation. @return lxml.etree: Generated XML etree element tblrtblPrtblStylerar2rMtblW)rrL tblBorders)topr\bottomrightinsideHinsideValltblLook0400tblGridr2390gridColtrtrPrcnfStyle 000000100000tctcPr0rtcWshdclearFFFFFFtext299)racolorfill themeFill themeFillTintcenter)rbNalignr\) r/rjr0rkeysunicoderanger6r7rir_Elementro)#contentsrcolwcwunittblwtwunitborderscelstyletablecolumns tableprops tablestyle tablewidth tablebordersrdkrr borderelem tablelook tablegridrerowrowpropsrcell cellpropswattr cellwidth cellstylerS contentrowcontentcrr*r*r+r)s'                            rc, Cs||dur tdtd}t|} |dur8| |vr3dtt|d} | || <|dd| t| fgn2|| } n-dtt|d} |dd|gttd d } t j | s\t | t |t| |t| } z| } | duruin| } Wni} Yi}| D] \}}||t||<q|d d}d d d d ddddd|}|dvrdnd}|dkrdnd}|r|s| jd d\}}|dvr||}}d}t||}t||}tddd}|tdddd| idtd dd}|td!dd|td"dd||td#dd}td$dd%d&|d'd(}||td)dd}|td*dtt|tt|d+d(||td,dd-d.id(}td/dt|d0||d1d(}|td2dd%d%d3d(|td4d||d5d(td6dd7d8id(}|td9dd||||tddd} | || || |td:dd;ddd?d@id(}#tdAdBd}$|$|#tdCdB|d&|d'd(}%tdDdBdEd%d%d%dFd(}&tdGdB||d5d(}'tdHd%d%d%d%dIdBdJ}(|(|'|(|&|(|%|(|$|(|"tdK})|)|(td}*|*|)tdL}+|+|*|dur:||+|fS||+fS)Mz Take a relationshiplist, picture file name, and return a paragraph containing the image and an updated relationshiplist NzQUsing picture() without imagefiledict parameter will be deprecated in the future.2rIdrzIhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/imagez media/%s_%szmedia/wordmedia OrientationrZi)r)rrrtruefalserr)rrrri1blipFillrr>bliprrembed)r>r@r?stretchfillRectsrcRectnvPicPrcNvPrrz Picture 1)idnamedescrrxcNvPicPrpicLocks)noChangeAspectnoChangeArrowheadsspPrbwModerxfrmi`)rotflipHflipVoff)xyext)cxcyprstGeomprstrectavLst graphicDataurirgraphicgraphicFrameLocksr1cNvGraphicFramePrrdocPr effectExtent25400)lr]rrdextentinline)distTdistBdistLdistR)r?r>drawingrJ)r rrrrjr0rr template_dirospathisdirmkdirshutilcopyfileropen_getexifitemsrgetsizer/int),relationshiplistpicnamepicdescription pixelwidth pixelheightnochangeaspectnochangearrowheads imagefiledictpicidpicpathpicrelid media_dirimageexif imageExiftagvalueimageOrientation imageAngle imageFlipH imageFlipV emuperpixelwidthheightblipfillrnvpicprcnvprcnvpicprspprrprstgeomr graphicdatar  framelocksframeprdocpr effectextentrrrrZror*r*r+pictures                                             rIcCsHd}t|}|D]}|jdtdkr!|jr!||jr!d}q |S)z;Search a document for a regex, return success / fail resultF{%s}trT)recompileiterr5r8r;search)r)rNresultsearchreelementr*r*r+rNis   rNcCsV|}t|}|D]}|jdtdkr(|jr(||jr(t|||j|_q |S)z_ Replace all occurences of string with a different string, return updated document rJr)rKrLrMr5r8r;rNsub)r)rNreplacer1rPrQr*r*r+rSus   rScCsj|}dD].}g}|D]}|jdtd|fkr%|js%t|s%||q |D] }||q(q|S)zV Perform misc cleaning operations on documents. Returns cleaned document. )r]rz{%s}%sr)rMr5r8r;rjr0 getparentremove)r)r1r]rmlistrQr*r*r+cleans  rWcCs|} |}|j|kr|Sq)z Finds fist parent of element of the given type @param object element: etree element @param string the tag parent to search for @return object element: the found parent or None when not found )rTr5)rQr5rJr*r*r+findTypeParents   rXrcCst|}g}g}|D]j}|jdtdkrw|jrw||t||kr+|dd}t dt|dD]@}|r<n;t t|D]3} |rHn.| |t|krut | | |} d} | D] } | || j7} q[| | } | ru|| d}qBq6q t |S)aWReturn set of all regex matches This is an advanced version of python-docx.search() that takes into account blocks of elements at a time. What it does: It searches the entire document body for text blocks. Since the text to search could be spawned across multiple text blocks, we need to adopt some sort of algorithm to handle this situation. The smaller matching group of blocks (up to bs) is then adopted. If the matching group has more than one block, blocks other than first are cleared and all the replacement text is put on first block. Examples: original text blocks : [ 'Hel', 'lo,', ' world!' ] search : 'Hello,' output blocks : [ 'Hello,' ] original text blocks : [ 'Hel', 'lo', ' __', 'name', '__!' ] search : '(__[a-z]+__)' output blocks : [ '__name__' ] @param instance document: The original document @param str search: The text to search for (regexp) append, or a list of etree elements @param int bs: See above @return set All occurences of search string rJrrFrr2T) rKrLrMr5r8r;r0rjpoprrNgroupr:)r)rNbsrPmatches searchelsrQfoundrse txtsearchrmatchr*r*r+ AdvSearchs6 !     rcc Csd}|}t|}g}|D]6}|jdtdkrF|jrF||t||kr0|dd} t dt|dD] } | rCnt t|D]} | rOn| | t|krDt | | | } d} | D] }| ||j7} qc| | }|rDd} |rt dt d |j t d |t d | t d td d|t d|t d| t|tjrt dnt|ttrt dn t dt||| d}d}| D]o}|t||j7}||kr>|s>t|tjr|g}t|ttfr+t||dtd}t|d| ||_||d}|D]}||||d7}qn t||| ||_d}t d|qd||_qqIq;q|S)a  Replace all occurences of string with a different string, return updated document This is a modified version of python-docx.replace() that takes into account blocks of elements at a time. The replace element can also be a string or an xml etree element. What it does: It searches the entire document body for text blocks. Then scan thos text blocks for replace. Since the text to search could be spawned across multiple text blocks, we need to adopt some sort of algorithm to handle this situation. The smaller matching group of blocks (up to bs) is then adopted. If the matching group has more than one block, blocks other than first are cleared and all the replacement text is put on first block. Examples: original text blocks : [ 'Hel', 'lo,', ' world!' ] search / replace: 'Hello,' / 'Hi!' output blocks : [ 'Hi!', '', ' world!' ] original text blocks : [ 'Hel', 'lo,', ' world!' ] search / replace: 'Hello, world' / 'Hi!' output blocks : [ 'Hi!!', '', '' ] original text blocks : [ 'Hel', 'lo,', ' world!' ] search / replace: 'Hel' / 'Hal' output blocks : [ 'Hal', 'lo,', ' world!' ] @param instance document: The original document @param str search: The text to search for (regexp) @param mixed replace: The replacement text or lxml.etree element to append, or a list of etree elements @param int bs: See above @return instance The document with replacement applied FrJrrrr2TzFound element!zSearch regexp: %szRequested replacement: %szMatched text: %szMatched text (splitted): %scSs|jSN)r;)rer*r*r+VszadvReplace..zMatched at position: %szmatched in elements: %szWill replace with XML CODEz"Will replace with LIST OF ELEMENTSzWill replace with:z{%s}pzReplacing in element #: %s)rKrLrMr5r8r;r0rjrYrrNlogdebugpatternmapstartr6rrr7rirRrXrTindexinsert)r)rNrSr[DEBUGr1rPr]rQr^rr_r`rarrbcurlenreplacedrerJinsindexrr*r*r+ advReplaces)                  rqcCsg}g}|D]}|jdtddkr||q|D]9}d}|D]%}|jdtddkr;|jr:||j}q%|jdtddkrJ|d}q%t|dksV||q|S) z;Return the raw text of a document, as a list of paragraphs.r4rz}pr2z}tz}tab r)rMr5r8r0r;rj)r) paratextlistparalistrQpararlr*r*r+getdocumenttexts(      rvc Cstddd}|td|dd|td|dd|td|dd|td d |dd|s5|}|td |dd|td d dd|tdddd|tddddtd}dD]}d|||f}|t|qd|S)z Create core properties (common document properties referred to in the 'Dublin Core' specification). See appproperties() for other stuff. corePropertiesrrtitlerr=r>subjectcreatorkeywords,lastModifiedByrevisionr categoryExamples descriptionz%Y-%m-%dT%H:%M:%SZ)createdmodifiedz%s)r/r0rtimestrftimerr%) rxrzr{r|lastmodifiedby coreprops currenttimedoctimeelm_strr*r*r+corepropertiess4     rcCs`tddd}td}ddddd d d d d ddddddd}|D]}|t|||ddq|S)zg Create app-specific properties. See docproperties() for more common document properties. Propertiesrrzz Normal.dotm6r 83475zMicrosoft Word 12.0.0r128r583z12.0000)Template TotalTimePagesWords Characters Application DocSecurityLines Paragraphs ScaleCrop LinksUpToDateCharactersWithSpaces SharedDocHyperlinksChanged AppVersionNry)r/rr%r0)apppropspropspropr*r*r+ apppropertiess0 rcCs(td}|td|td|S)zGenerate websettings webSettingsallowPNGdoNotSaveAsSingleFiler.)webr*r*r+ websettingssrcCs,ddgddgddgddgd d gd d gg}|S) NzMhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/numberingz numbering.xmlzJhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/stylesz styles.xmlzLhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/settingsz settings.xmlzOhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/webSettingszwebSettings.xmlzMhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/fontTablez fontTable.xmlzIhttp://schemas.openxmlformats.org/officeDocument/2006/relationships/themeztheme/theme1.xmlr*)r&r*r*r+r&s* r&cCsXtd}d}|D] }tdddt|d|d|ddd}|||d7}q |S) z"Generate a Word relationships filezdr RelationshipNrr)IdTypeTargetrx)rr%r/rr0)r& relationshipscount relationshiprel_elmr*r*r+wordrelationshipss  rc Csb|dur tdttjtsJtj|dtjd}tj d} t t|d|d|d|d |d |d i} | D]} t d | | t j| d d} || | | q5|durq|D]\} }d|t| f}t d ||| |qXdg}tdD]&\}}}|D]}||vrqt||}|dd}t d ||||qqyt d||t | dS)z" Save a modified document NzRUsing savedocx() without imagefiledict parameter will be deprecated in the future.r)mode compression.r!zdocProps/core.xmlzdocProps/app.xmlz[Content_Types].xmlzword/webSettings.xmlzword/_rels/document.xml.relsz Saving: %sT) pretty_printzword/media/%s_%sz .DS_StorerzSaved new file to: %r)r rrrrrr"r# ZIP_DEFLATEDrchdirrfinfortostringwritestrr"rwritewalkrclose)r)rrrrroutputr-docxfileprev_dir treesandfilestree treestring imagepathr0 archivenamefiles_to_ignoredirpathdirnames filenamesfilename templatefiler*r*r+savedocxsR        r)NrNN)rGrH)r[Fr\)r)NNTTN)rrd)5__doc__rrKrrr"lxmlros.pathrrrPILr ImportError PIL.ExifTagsr exceptionsrwarningsr logging getLogger__name__rfrdirname__file__rrr8r,r1r/rYrorrrrIrNrSrWrXrcrqrvrrrr&rrr*r*r*r+s             " ' E .   8   L #!