o 'f%@sdZddlmZddlmZddlmZmZddl m Z ddl m Z GdddZ Gd d d ZGd d d ZGd ddZGdddZdS)zQLow-level, read-only API to a serialized Open Packaging Convention (OPC) package.)RELATIONSHIP_TARGET_MODE) parse_xml) PACKAGE_URIPackURI) PhysPkgReader)CaseInsensitiveDictcsbeZdZdZfddZeddZddZdd Zed d Z ed d Z edddZ Z S) PackageReaderzProvides access to the contents of a zip-format OPC package via its :attr:`serialized_parts` and :attr:`pkg_srels` attributes.cstt|||_||_dSN)superr__init__ _pkg_srels_sparts)self content_types pkg_srelssparts __class__J/home/ubuntu/flask/venv/lib/python3.10/site-packages/docx/opc/pkgreader.pyr s zPackageReader.__init__cCsBt|}t|j}t|t}t|||}|t|||S)zEReturn a |PackageReader| instance loaded with contents of `pkg_file`.) r_ContentTypeMapfrom_xmlcontent_types_xmlr _srels_forr_load_serialized_partsclose)pkg_file phys_readerrrrrrr from_files   zPackageReader.from_fileccs(|jD] }|j|j|j|jfVqdS)zuGenerate a 4-tuple `(partname, content_type, reltype, blob)` for each of the serialized parts in the package.N)r partname content_typereltypeblob)rsrrr iter_spartss zPackageReader.iter_spartsccs@|jD]}t|fVq|jD]}|jD]}|j|fVqqdS)z]Generate a 2-tuple `(source_uri, srel)` for each of the relationships in the package.N)r rr srelsr)rsrelspartrrr iter_srels%s    zPackageReader.iter_srelsc CsLg}t||}|D]\}}}}||} t|| |||} || q t|S)zReturn a list of |_SerializedPart| instances corresponding to the parts in `phys_reader` accessible by walking the relationship graph starting with `pkg_srels`.)r_walk_phys_parts_SerializedPartappendtuple) rrrr part_walkerrr"r!r%r r'rrrr.s  z$PackageReader._load_serialized_partscCs||}t|j|S)zvReturn |_SerializedRelationships| instance populated with relationships for source identified by `source_uri`.) rels_xml_for_SerializedRelationships load_from_xmlbaseURI)r source_urirels_xmlrrrr;s zPackageReader._srels_forNc cs|durg}|D]?}|jrq |j}||vrq |||j}t||}||}||||fVt|||}|D] \}}}}||||fVq:q dS)zGenerate a 4-tuple `(partname, blob, reltype, srels)` for each of the parts in `phys_reader` by walking the relationship graph rooted at srels.N) is_externaltarget_partnamer+r!rrblob_forr)) rr%visited_partnamesr&rr! part_srelsr" next_walkerrrrr)Bs*   zPackageReader._walk_phys_partsr ) __name__ __module__ __qualname____doc__r staticmethodrr$r(rrr) __classcell__rrrrr s     rcsDeZdZdZfddZddZeddZdd Zd d Z Z S) rzValue type providing dictionary semantics for looking up content type by part name, e.g. ``content_type = cti['/ppt/presentation.xml']``.cs"tt|t|_t|_dSr )r rr r _overrides _defaultsrrrrr ^s z_ContentTypeMap.__init__cCsZt|tsd}t|t|||jvr|j|S|j|jvr%|j|jSd}t||)z6Return content type for part identified by `partname`.z4_ContentTypeMap key must be , got %sz8no content type for partname '%s' in [Content_Types].xml) isinstancerKeyErrortyper@extrA)rrtmplrrr __getitem__cs      z_ContentTypeMap.__getitem__cCsJt|}t}|jD] }||j|jq |jD] }||j|jq|S)zcReturn a new |_ContentTypeMap| instance populated with the contents of `content_types_xml`.) rr overrides _add_overriderr defaults _add_default extension)r types_elmct_mapodrrrros  z_ContentTypeMap.from_xmlcC||j|<dS)z^Add the default mapping of `extension` to `content_type` to this content type mapping.N)rA)rrMr rrrrL{z_ContentTypeMap._add_defaultcCrR)z]Add the default mapping of `partname` to `content_type` to this content type mapping.N)r@)rrr rrrrJrSz_ContentTypeMap._add_override) r:r;r<r=r rHr>rrLrJr?rrrrrZs   rcs\eZdZdZfddZeddZeddZedd Zed d Z ed d Z Z S)r*zValue object for an OPC package part. Provides access to the partname, content type, blob, and serialized relationships for the part. cs0tt|||_||_||_||_||_dSr )r r*r _partname _content_type_reltype_blob_srels)rrr r!r"r%rrrr s  z_SerializedPart.__init__cC|jSr )rTrBrrrrz_SerializedPart.partnamecCrYr )rUrBrrrr rZz_SerializedPart.content_typecCrYr )rWrBrrrr"rZz_SerializedPart.blobcCrY)z-The referring relationship type of this part.rVrBrrrr!z_SerializedPart.reltypecCrYr )rXrBrrrr%rZz_SerializedPart.srels) r:r;r<r=r propertyrr r"r!r%r?rrrrr*s     r*csheZdZdZfddZeddZeddZedd Zed d Z ed d Z eddZ Z S)_SerializedRelationshipzValue object representing a serialized relationship in an OPC package. Serialized, in this case, means any target part is referred to via its partname rather than a direct link to an in-memory |Part| object. cs8tt|||_|j|_|j|_|j|_ |j |_ dSr ) r r^r _baseURIrId_rIdr!rV target_mode _target_mode target_ref _target_ref)rr1rel_elmrrrr s  z _SerializedRelationship.__init__cCs |jtjkS)z'True if target_mode is ``RTM.EXTERNAL``)rcRTMEXTERNALrBrrrr4s z#_SerializedRelationship.is_externalcCrY)z.Relationship type, like ``RT.OFFICE_DOCUMENT``r[rBrrrr!r\z_SerializedRelationship.reltypecCrY)zmRelationship id, like 'rId9', corresponds to the ``Id`` attribute on the ``CT_Relationship`` element.)rarBrrrr`z_SerializedRelationship.rIdcCrY)zwString in ``TargetMode`` attribute of ``CT_Relationship`` element, one of ``RTM.INTERNAL`` or ``RTM.EXTERNAL``.)rcrBrrrrbriz#_SerializedRelationship.target_modecCrY)zString in ``Target`` attribute of ``CT_Relationship`` element, a relative part reference for internal target mode or an arbitrary URI, e.g. an HTTP URL, for external target mode.)rerBrrrrdsz"_SerializedRelationship.target_refcCs4|jr d}t|t|dst|j|j|_|jS)z|PackURI| instance containing partname targeted by this relationship. Raises ``ValueError`` on reference if target_mode is ``'External'``. Use :attr:`target_mode` to check before referencing. zUtarget_partname attribute on Relationship is undefined where TargetMode == "External"_target_partname)r4 ValueErrorhasattrr from_rel_refr_rdrj)rmsgrrrr5s z'_SerializedRelationship.target_partname) r:r;r<r=r r]r4r!r`rbrdr5r?rrrrr^s      r^cs4eZdZdZfddZddZeddZZS)r/zRead-only sequence of |_SerializedRelationship| instances corresponding to the relationships item XML passed to constructor.cstt|g|_dSr )r r/r rXrBrrrr s z!_SerializedRelationships.__init__cCs |jS)z*Support iteration, e.g. 'for x in srels:'.)rX__iter__rBrrrros z!_SerializedRelationships.__iter__cCs8t}|durt|}|jD] }|jt||q|S)zReturn |_SerializedRelationships| instance loaded with the relationships contained in `rels_item_xml`. Returns an empty collection if `rels_item_xml` is |None|. N)r/rRelationship_lstrXr+r^)r1 rels_item_xmlr%rels_elmrfrrrr0s  z&_SerializedRelationships.load_from_xml) r:r;r<r=r ror>r0r?rrrrr/s  r/N)r=docx.opc.constantsrrg docx.opc.oxmlrdocx.opc.packurirrdocx.opc.phys_pkgrdocx.opc.sharedrrrr*r^r/rrrrs    P,%?