a ]j"@sdZddlmZddlmZmZmZmZddlm Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZerdd lmZdd lmZddlmZddlmZGdddZ GdddZ!dS)z8Objects that implement reading and writing OPC packages.) annotations)IO TYPE_CHECKINGIteratorcast)RELATIONSHIP_TYPE) PACKAGE_URIPackURI) PartFactory)CorePropertiesPart) PackageReader) PackageWriter) Relationships) lazyproperty)Self)CoreProperties)Part) _Relationshipc@seZdZdZddZeddddZddd d Zd dd d Zd3dddddddZ eddZ dddddZ e dddddZ dd d!d"d#Zed$dd%d&Zd dd'd(d)Zed*d+Zdd,d-d.Zed/dd0d1Zd2S)4 OpcPackagezMain API class for |python-opc|. A new instance is constructed by calling the :meth:`open` class method with a path to a package file or file-like object containing one. cCsdS)zEntry point for any post-unmarshaling processing. May be overridden by subclasses without forwarding call to super. NselfrrL/var/www/html/assistant/venv/lib/python3.9/site-packages/docx/opc/package.pyafter_unmarshalszOpcPackage.after_unmarshalr)returncCs|jjS)zl|CoreProperties| object providing read/write access to the Dublin Core properties for this document.)_core_properties_partcore_propertiesrrrrr(szOpcPackage.core_propertiesIterator[_Relationship]c#s0dddddfdd |D] }|Vq dS) zGenerate exactly one reference to each relationship in the package by performing a depth-first traversal of the rels graph.NzOpcPackage | Partzlist[Part] | Noner)sourcevisitedrc3sf|dur gn|}|jD]F}|V|jr,q|j}||vr.walk_rels)Nr)rr'rr)r iter_rels.s zOpcPackage.iter_relszIterator[Part]c#s(gffdd |D] }|VqdS)zGenerate exactly one reference to each of the parts in the package by performing a depth-first traversal of the rels graph.c3sV|jD]F}|jrq |j}||vr&q |||V|}||D] }|VqDq dSr r!r& walk_partsrrr-Is z)OpcPackage.iter_parts..walk_partsNr)rr(rr,r iter_partsEs zOpcPackage.iter_partsFstrz Part | strbool)reltypetargetrIdr$cCs|j||||S)a}Return newly added |_Relationship| instance of `reltype` between this part and `target` with key `rId`. Target mode is set to ``RTM.EXTERNAL`` if `is_external` is |True|. Intended for use during load from a serialized package, where the rId is well known. Other methods exist for adding a new relationship to the package during processing. )r"Zadd_relationship)rr1r2r3r$rrrload_relYszOpcPackage.load_relcCs |tjS)aReturn a reference to the main document part for this package. Examples include a document part for a WordprocessingML package, a presentation part for a PresentationML package, or a workbook part for a SpreadsheetML package. )part_related_byRTZOFFICE_DOCUMENTrrrrmain_document_partcszOpcPackage.main_document_partr )templatercCsJdd|D}tdt|dD] }||}||vr$t|Sq$dS)aReturn a |PackURI| instance representing partname matching `template`. The returned part-name has the next available numeric suffix to distinguish it from other parts of its type. `template` is a printf (%)-style template string containing a single replacement item, a '%d' to be used to insert the integer portion of the partname. Example: "/word/header%d.xml" cSsh|] }|jqSr)partname).0r(rrr uz+OpcPackage.next_partname..N)r.rangelenr )rr8Z partnamesnZcandidate_partnamerrr next_partnamems zOpcPackage.next_partnamezstr | IO[bytes]r)pkg_filercCs"t|}|}t||t|S)zGReturn an |OpcPackage| instance loaded with the contents of `pkg_file`.)r from_file Unmarshaller unmarshalr )clsrC pkg_readerpackagerrropen{s zOpcPackage.openr)r1rcCs |j|S)zReturn part to which this package has a relationship of `reltype`. Raises |KeyError| if no such relationship is found and |ValueError| if more than one such relationship is found. )r"Zpart_with_reltype)rr1rrrr5szOpcPackage.part_related_byz list[Part]cCs t|S)zJReturn a list containing a reference to each of the parts in this package.)listr.rrrrpartsszOpcPackage.parts)r(r1cCs|j||}|jS)zReturn rId key of new or existing relationship to `part`. If a relationship of `reltype` to `part` already exists, its rId is returned. Otherwise a new relationship is created and that rId is returned. )r"Z get_or_addr3)rr(r1r'rrr relate_toszOpcPackage.relate_tocCs ttjS)ztReturn a reference to the |Relationships| instance holding the collection of relationships for this package.)rrZbaseURIrrrrr"szOpcPackage.rels)rCcCs*|jD] }|qt||j|jdS)znSave this package to `pkg_file`. `pkg_file` can be either a file-path or a file-like object. N)rLZbefore_marshalr writer")rrCr(rrrsaves  zOpcPackage.saver cCsHztt|tjWStyBt|}||tj|YS0dS)z|CorePropertiesPart| object related to this package. Creates a default core properties part if one is not present (not common). N)rr r5r6ZCORE_PROPERTIESKeyErrordefaultrM)rZcore_properties_partrrrrs   z OpcPackage._core_properties_partN)F)__name__ __module__ __qualname____doc__rpropertyrr+r.r4r7rB classmethodrJr5rLrMrr"rOrrrrrrs*      rc@s4eZdZdZeddZeddZeddZdS) rEzHHosts static methods for unmarshalling a package from a |PackageReader|.cCs>t|||}t||||D] }|q$|dS)zConstruct graph of parts and realized relationships based on the contents of `pkg_reader`, delegating construction of each part to `part_factory`. Package relationships are added to `pkg`. N)rE_unmarshal_parts_unmarshal_relationshipsr#r)rHrI part_factoryrLr(rrrrFs   zUnmarshaller.unmarshalcCs2i}|D] \}}}}||||||||<q |S)zReturn a dictionary of |Part| instances unmarshalled from `pkg_reader`, keyed by partname. Side-effect is that each part in `pkg_reader` is constructed using `part_factory`. )Z iter_sparts)rHrIrZrLr9 content_typer1ZblobrrrrXszUnmarshaller._unmarshal_partscCsV|D]H\}}|dkr|n||}|jr0|jn||j}||j||j|jqdS)zAdd a relationship to the source object corresponding to each of the relationships in `pkg_reader` with its target_part set to the actual target part in `parts`./N)Z iter_srelsr$Z target_refZtarget_partnamer4r1r3)rHrIrLZ source_uriZsrelrr2rrrrYsz%Unmarshaller._unmarshal_relationshipsN)rRrSrTrU staticmethodrFrXrYrrrrrEs  rEN)"rU __future__rtypingrrrrZdocx.opc.constantsrr6Zdocx.opc.packurirr Z docx.opc.partr Zdocx.opc.parts.corepropsr Zdocx.opc.pkgreaderr Zdocx.opc.pkgwriterr Z docx.opc.relrZ docx.sharedrZtyping_extensionsrZdocx.opc.corepropsrrrrrErrrrs$