o 'f@sdZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZGd d d ZGd d d ZdS)z8Objects that implement reading and writing OPC packages.)RELATIONSHIP_TYPE) PACKAGE_URIPackURI) PartFactory)CorePropertiesPart) PackageReader) PackageWriter) Relationships) lazypropertycseZdZdZfddZddZeddZdd Zd d Z d!d dZ eddZ ddZ e ddZddZeddZddZeddZddZedd ZZS)" 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. cstt|dSN)superr __init__self __class__H/home/ubuntu/flask/venv/lib/python3.10/site-packages/docx/opc/package.pyrszOpcPackage.__init__cCsdS)zEntry point for any post-unmarshaling processing. May be overridden by subclasses without forwarding call to super. Nrrrrrafter_unmarshalszOpcPackage.after_unmarshalcCs|jjS)zl|CoreProperties| object providing read/write access to the Dublin Core properties for this document.)_core_properties_partcore_propertiesrrrrr szOpcPackage.core_propertiesc#s(dfdd |D]}|Vq dS)zGenerate exactly one reference to each relationship in the package by performing a depth-first traversal of the rels graph.Nc3sh|durgn|}|jD]#}|V|jrq|j}||vrq|||}||D]}|Vq+qdSr relsvalues is_external target_partappendsourcevisitedrelpart new_source walk_relsrrr%*s z'OpcPackage.iter_rels..walk_relsr r)rr!rr$r iter_rels&s  zOpcPackage.iter_relsc#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 rr walk_partsrrr(?s z)OpcPackage.iter_parts..walk_partsNr)rr"rr'r iter_parts;s  zOpcPackage.iter_partsFcCs|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. )radd_relationship)rreltypetargetrIdrrrrload_relOszOpcPackage.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_partYs zOpcPackage.main_document_partcCsJdd|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 ksz+OpcPackage.next_partname..N)r)rangelenr)rtemplate partnamesncandidate_partnamerrr next_partnamecs zOpcPackage.next_partnamecCs"t|}|}t||t|S)zGReturn an |OpcPackage| instance loaded with the contents of `pkg_file`.)r from_file Unmarshaller unmarshalr)clspkg_file pkg_readerpackagerrropenqs zOpcPackage.opencCs |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. )rpart_with_reltype)rr+rrrr/ys zOpcPackage.part_related_bycCs t|S)zJReturn a list containing a reference to each of the parts in this package.)listr)rrrrpartss zOpcPackage.partscCs|j||}|jS)zReturn rId key of relationship to `part`, from the existing relationship if there is one, otherwise a newly created one.)r get_or_addr-)rr"r+r!rrr relate_toszOpcPackage.relate_tocCs ttjS)ztReturn a reference to the |Relationships| instance holding the collection of relationships for this package.)r rbaseURIrrrrrs zOpcPackage.relscCs*|jD]}|qt||j|jdS)zvSave this package to `pkg_file`, where `file` can be either a path to a file (a string) or a file-like object.N)rIbefore_marshalrwriter)rrCr"rrrsaves  zOpcPackage.savecCs>z|tjWStyt|}||tj|YSw)z|CorePropertiesPart| object related to this package. Creates a default core properties part if one is not present (not common). )r/r0CORE_PROPERTIESKeyErrorrdefaultrK)rcore_properties_partrrrrs  z OpcPackage._core_properties_part)F)__name__ __module__ __qualname____doc__rrpropertyrr&r)r.r2r> classmethodrFr/rIrKr rrOr __classcell__rrrrr s.        r c@s4eZdZdZeddZeddZeddZdS) r@zHHosts 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)r@_unmarshal_parts_unmarshal_relationshipsrr)rDrE part_factoryrIr"rrrrAs    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)rDrEr]rIr3 content_typer+blobrrrr[s   zUnmarshaller._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.r+r-)rDrErI source_urisrelrr,rrrr\s z%Unmarshaller._unmarshal_relationshipsN)rTrUrVrW staticmethodrAr[r\rrrrr@s  r@N)rWdocx.opc.constantsrr0docx.opc.packurirr docx.opc.partrdocx.opc.parts.corepropsrdocx.opc.pkgreaderrdocx.opc.pkgwriterr docx.opc.relr docx.opc.sharedr r r@rrrrs