File: //home/arjun/projects/env/lib/python3.10/site-packages/docx/opc/__pycache__/package.cpython-310.pyc
o
$we� � @ s� d Z ddlmZ ddlmZmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZ dd lmZ G d
d � d �ZG d d
� d
�ZdS )z8Objects that implement reading and writing OPC packages.� )�RELATIONSHIP_TYPE)� PACKAGE_URI�PackURI)� PartFactory)�CorePropertiesPart)�
PackageReader)�
PackageWriter)�
Relationships)� lazypropertyc s� e Zd ZdZ� fdd�Zdd� Zedd� �Zdd � Zd
d � Z d!d
d�Z
edd� �Z dd� Z e
dd� �Zdd� Zedd� �Zdd� Zedd � �Zd d � Zed d � �Z� ZS )"�
OpcPackagez�Main 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.
c s t t| ��� d S �N)�superr �__init__��self�� __class__� �I/home/arjun/projects/env/lib/python3.10/site-packages/docx/opc/package.pyr s zOpcPackage.__init__c C s dS )z�Entry point for any post-unmarshaling processing.
May be overridden by subclasses without forwarding call to super.
Nr r r r r �after_unmarshal s zOpcPackage.after_unmarshalc C s | j jS )zl|CoreProperties| object providing read/write access to the Dublin Core
properties for this document.)�_core_properties_part�core_propertiesr r r r r s zOpcPackage.core_propertiesc # s( � d� fdd� � � | �D ]}|V q dS )z�Generate exactly one reference to each relationship in the package by
performing a depth-first traversal of the rels graph.Nc 3 sh � |d u rg n|}| j �� D ]#}|V |jrq|j}||v r q|�|� |}� ||�D ]}|V q+qd S r ��rels�values� is_external� target_part�append��source�visited�rel�part�
new_source�� walk_relsr r r% * s �
��z'OpcPackage.iter_rels.<locals>.walk_relsr r )r r! r r$ r � iter_rels& s
� �zOpcPackage.iter_relsc # s* � g f� fdd� � � | �D ]}|V q
dS )z�Generate exactly one reference to each of the parts in the package by
performing a depth-first traversal of the rels graph.c 3 sX � | j �� D ]#}|jr q|j}||v rq|�|� |V |}� ||�D ]}|V q#qd S r r r ��
walk_partsr r r( ? s �
��z)OpcPackage.iter_parts.<locals>.walk_partsNr )r r" r r' r �
iter_parts; s
�
�zOpcPackage.iter_partsFc C s | 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)r �reltype�target�rIdr r r r �load_relO s zOpcPackage.load_relc C s | � tj�S )a Return 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_by�RT�OFFICE_DOCUMENTr r r r �main_document_partY s z OpcPackage.main_document_partc C sJ dd� | � � D �}tdt|�d �D ]}|| }||vr"t|� S qdS )a� Return 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"
c S s h | ]}|j �qS r )�partname)�.0r" r r r � <setcomp>k s z+OpcPackage.next_partname.<locals>.<setcomp>� � N)r) �range�lenr )r �template� partnames�n�candidate_partnamer r r �
next_partnamec s ��zOpcPackage.next_partnamec C s" t �|�}| � }t�||t� |S )zGReturn an |OpcPackage| instance loaded with the contents of `pkg_file`.)r � from_file� Unmarshaller� unmarshalr )�cls�pkg_file�
pkg_reader�packager r r �openq s
zOpcPackage.openc C s | j �|�S )z�Return 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)r r+ r r r r/ y s zOpcPackage.part_related_byc C s t | �� �S )zJReturn a list containing a reference to each of the parts in this package.)�listr) r r r r �parts� s zOpcPackage.partsc C s | j �||�}|jS )z�Return rId key of relationship to `part`, from the existing relationship if
there is one, otherwise a newly created one.)r �
get_or_addr- )r r" r+ r! r r r � relate_to� s zOpcPackage.relate_toc C s
t tj�S )ztReturn a reference to the |Relationships| instance holding the collection of
relationships for this package.)r r �baseURIr r r r r � s
zOpcPackage.relsc C s* | j D ]}|�� qt�|| j| j � dS )zvSave this package to `pkg_file`, where `file` can be either a path to a file
(a string) or a file-like object.N)rI �before_marshalr �writer )r rC r" r r r �save� s
zOpcPackage.savec C s> z| � tj�W S ty t�| �}| �|tj� | Y S w )z�|CorePropertiesPart| object related to this package.
Creates a default core properties part if one is not present (not common).
)r/ r0 �CORE_PROPERTIES�KeyErrorr �defaultrK )r �core_properties_partr r r r � s
�z OpcPackage._core_properties_part)F)�__name__�
__module__� __qualname__�__doc__r r �propertyr r& r) r. r2 r> � classmethodrF r/ rI rK r
r rO r �
__classcell__r r r r r
s.
r c @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) r@ zHHosts static methods for unmarshalling a package from a |PackageReader|.c C s> t �| ||�}t �| ||� |�� D ]}|�� q|�� dS )z�Construct 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_relationshipsr r )rD rE � part_factoryrI r" r r r rA � s
zUnmarshaller.unmarshalc C s2 i }| � � D ]\}}}}||||||�||< q|S )z�Return 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)rD rE r] rI r3 � content_typer+ �blobr r r r[ � s
�z Unmarshaller._unmarshal_partsc C sV | � � D ]$\}}|dkr|n|| }|jr|jn||j }|�|j||j|j� qdS )z�Add 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_ref�target_partnamer. r+ r- )rD rE rI �
source_uri�srelr r, r r r r\ � s ��z%Unmarshaller._unmarshal_relationshipsN)rT rU rV rW � staticmethodrA r[ r\ r r r r r@ � s
r@ N)rW �docx.opc.constantsr r0 �docx.opc.packurir r �
docx.opc.partr �docx.opc.parts.corepropsr �docx.opc.pkgreaderr �docx.opc.pkgwriterr � docx.opc.relr �docx.opc.sharedr
r r@ r r r r �<module> s