o h"@sdZddlmZddlmZmZmZmZddlm Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZer`dd 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@seZdZdZddZed4ddZd5d d Zd6d d Zd7d8ddZ eddZ d9ddZ e d:d!d"Z d;d$d%Zedd.d/Zed?d1d2Zd3S)@ 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. NselfrrP/var/www/html/premium_crap/venv/lib/python3.10/site-packages/docx/opc/package.pyafter_unmarshalszOpcPackage.after_unmarshalreturnrcCs|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#s, d d fdd |D]}|VqdS) zGenerate exactly one reference to each relationship in the package by performing a depth-first traversal of the rels graph.NsourceOpcPackage | Partvisitedlist[Part] | Nonerrc3sh|durgn|}|jD]#}|V|jrq|j}||vrq|||}||D]}|Vq+qdSNrelsvalues is_external target_partappendrr relpart new_source walk_relsrrr.2s z'OpcPackage.iter_rels..walk_relsr")rrr r!rrr)rr*rr-r iter_rels.s  zOpcPackage.iter_relsIterator[Part]c#s*gffdd |D]}|Vq dS)zGenerate exactly one reference to each of the parts in the package by performing a depth-first traversal of the rels graph.c3sX|jD]#}|jr q|j}||vrq|||V|}||D]}|Vq#qdSr"r#r) walk_partsrrr2Is z)OpcPackage.iter_parts..walk_partsNr)rr+rr1r iter_partsEs  zOpcPackage.iter_partsFreltypestrtarget Part | strrIdr&boolcCs|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$add_relationship)rr4r6r8r&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_byRTOFFICE_DOCUMENTrrrrmain_document_partcs zOpcPackage.main_document_parttemplater cCsJdd|D}tdt|dD]}||}||vr"t|SqdS)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 usz+OpcPackage.next_partname..N)r3rangelenr )rr@ partnamesncandidate_partnamerrr next_partnamems zOpcPackage.next_partnamepkg_filestr | IO[bytes]rcCs"t|}|}t||t|S)zGReturn an |OpcPackage| instance loaded with the contents of `pkg_file`.)r from_file Unmarshaller unmarshalr )clsrL pkg_readerpackagerrropen{s zOpcPackage.openrcCs |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$part_with_reltype)rr4rrrr<s zOpcPackage.part_related_by list[Part]cCs t|S)zJReturn a list containing a reference to each of the parts in this package.)listr3rrrrpartss zOpcPackage.partsr+cCs|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$ get_or_addr8)rr+r4r*rrr relate_toszOpcPackage.relate_tocCs ttjS)ztReturn a reference to the |Relationships| instance holding the collection of relationships for this package.)rrbaseURIrrrrr$s zOpcPackage.relscCs*|jD]}|qt||j|jdS)znSave this package to `pkg_file`. `pkg_file` can be either a file-path or a file-like object. N)rXbefore_marshalr writer$)rrLr+rrrsaves  zOpcPackage.saver cCsDz tt|tjWSty!t|}||tj|YSw)z|CorePropertiesPart| object related to this package. Creates a default core properties part if one is not present (not common). )rr r<r=CORE_PROPERTIESKeyErrordefaultrZ)rcore_properties_partrrrrs  z OpcPackage._core_properties_partN)rr)rr)rr0)F)r4r5r6r7r8r5r&r9)r@r5rr )rLrMrr)r4r5rr)rrV)r+rr4r5)rLrM)rr )__name__ __module__ __qualname____doc__rpropertyrr/r3r;r?rK classmethodrTr<rXrZrr$r^rrrrrrs,             rc@s4eZdZdZeddZeddZeddZdS) rOzHHosts 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)rO_unmarshal_parts_unmarshal_relationshipsr%r)rRrS part_factoryrXr+rrrrPs    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`. ) iter_sparts)rRrSrkrXrA content_typer4blobrrrriszUnmarshaller._unmarshal_partscCsV|D]$\}}|dkr|n||}|jr|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) iter_srelsr& target_reftarget_partnamer;r4r8)rRrSrX source_urisrelrr6rrrrjs z%Unmarshaller._unmarshal_relationshipsN)rcrdrerf staticmethodrPrirjrrrrrOs   rON)"rf __future__rtypingrrrrdocx.opc.constantsrr=docx.opc.packurirr docx.opc.partr docx.opc.parts.corepropsr docx.opc.pkgreaderr docx.opc.pkgwriterr docx.opc.relr docx.sharedrtyping_extensionsrdocx.opc.corepropsrrrrrOrrrrs&