U cqeã@s†dZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZdd lmZGd d „d e ƒZGd d „d ƒZdS)z3WordprocessingML Package class and related objects.é)Ú annotations)ÚIO)ÚImage)ÚRELATIONSHIP_TYPE)Ú OpcPackage©ÚPackURI)Ú ImagePart)Ú lazypropertyc@sBeZdZdZdd„Zdddœdd„Zed d œd d „ƒZd d„ZdS)ÚPackagez6Customizations specific to a WordprocessingML package.cCs | ¡dS)z¢Called by loading code after all parts and relationships have been loaded. This method affords the opportunity for any required post-processing. N)Ú_gather_image_parts©Úself©rú0/tmp/pip-unpacked-wheel-gx5hpqp3/docx/package.pyÚafter_unmarshalszPackage.after_unmarshalústr | IO[bytes]r ©Úimage_descriptorÚreturncCs |j |¡S)zºReturn |ImagePart| containing image specified by `image_descriptor`. The image-part is newly created if a matching one is not already present in the collection. )Ú image_partsÚget_or_add_image_part)rrrrrrszPackage.get_or_add_image_partÚ ImageParts)rcCstƒS)z0|ImageParts| collection object for this package.)rr rrrr!szPackage.image_partscCsD| ¡D]6}|jrq|jtjkr"q|j|jkr0q|j |j¡qdS)zCLoad the image part collection with all the image parts in package.N)Z iter_relsZ is_externalZreltypeÚRTZIMAGEZ target_partrÚappend)rÚrelrrrr &s   zPackage._gather_image_partsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr rr rrrrr s r c@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d dœdd„Z dd„Z dd„Z dd„Z dS)rzICollection of |ImagePart| objects corresponding to images in the package.cCs g|_dS©N)Ú _image_partsr rrrÚ__init__5szImageParts.__init__cCs |j |¡Sr )r!Ú __contains__©rÚitemrrrr#8szImageParts.__contains__cCs |j ¡Sr )r!Ú__iter__r rrrr&;szImageParts.__iter__cCs |j ¡Sr )r!Ú__len__r rrrr'>szImageParts.__len__cCs|j |¡dSr )r!rr$rrrrAszImageParts.appendrr rcCs,t |¡}| |j¡}|dk r"|S| |¡S)zºReturn |ImagePart| object containing image identified by `image_descriptor`. The image-part is newly created if a matching one is not present in the collection. N)rÚ from_fileÚ _get_by_sha1Úsha1Ú_add_image_part)rrÚimageZmatching_image_partrrrrDs   z ImageParts.get_or_add_image_partcCs&| |j¡}t ||¡}| |¡|S)z_Return an |ImagePart| instance newly created from image and appended to the collection.)Ú_next_image_partnameÚextr Z from_imager)rr,ÚpartnameÚ image_partrrrr+Ps   zImageParts._add_image_partcCs"|jD]}|j|kr|SqdS)zlReturn the image part in this collection having a SHA1 hash matching `sha1`, or |None| if not found.N)r!r*)rr*r0rrrr)Xs   zImageParts._get_by_sha1csV‡fdd„}dd„|Dƒ}tdt|ƒdƒD]}||kr,||ƒSq,|t|ƒdƒS)zþThe next available image partname, starting from ``/word/media/image1.{ext}`` where unused numbers are reused. The partname is unique by number, without regard to the extension. `ext` does not include the leading period. cstd|ˆfƒS)Nz/word/media/image%d.%sr)Ún©r.rrÚimage_partnamehsz7ImageParts._next_image_partname..image_partnamecSsg|] }|jj‘qSr)r/Úidx)Ú.0r0rrrÚ ksz3ImageParts._next_image_partname..é)ÚrangeÚlen)rr.r3Z used_numbersr1rr2rr-`s  zImageParts._next_image_partnameN) rrrrr"r#r&r'rrr+r)r-rrrrr2s rN)rÚ __future__rÚtypingrZdocx.image.imagerZdocx.opc.constantsrrZdocx.opc.packagerZdocx.opc.packurirZdocx.parts.imager Z docx.sharedr r rrrrrÚs        #