File: //home/arjun/projects/env/lib/python3.10/site-packages/docx/__pycache__/package.cpython-310.pyc
o
$we � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl
m Z ddl m
Z
ddlmZ dd lmZ G d
d � d e �ZG d d
� d
�ZdS )z3WordprocessingML Package class and related objects.� )� annotations)�IO)�Image)�RELATIONSHIP_TYPE)�
OpcPackage��PackURI)� ImagePart)� lazypropertyc @ s8 e Zd ZdZdd� Zddd �Zedd d ��Zd
d� ZdS )�Packagez6Customizations specific to a WordprocessingML package.c C s | � � dS )z�Called by loading code after all parts and relationships have been loaded.
This method affords the opportunity for any required post-processing.
N)�_gather_image_parts��self� r �E/home/arjun/projects/env/lib/python3.10/site-packages/docx/package.py�after_unmarshal s zPackage.after_unmarshal�image_descriptor�str | IO[bytes]�returnr c C � | j �|�S )z�Return |ImagePart| containing image specified by `image_descriptor`.
The image-part is newly created if a matching one is not already present in the
collection.
)� image_parts�get_or_add_image_part)r r r r r r s z Package.get_or_add_image_part�
ImagePartsc C s t � S )z0|ImageParts| collection object for this package.)r r
r r r r ! s zPackage.image_partsc C sD | � � D ]}|jr
q|jtjkrq|j| jv rq| j�|j� qdS )zCLoad the image part collection with all the image parts in package.N)� iter_rels� is_external�reltype�RT�IMAGE� target_partr �append)r �relr r r r & s �zPackage._gather_image_partsN�r r r r )r r ) �__name__�
__module__� __qualname__�__doc__r r r
r r r r r r r s
r c @ sZ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zddd�Z dd� Z
dd� Z dd� Z dS )r zICollection of |ImagePart| objects corresponding to images in the package.c C s
g | _ d S �N)� _image_partsr
r r r �__init__5 �
zImageParts.__init__c C r r& )r' � __contains__�r �itemr r r r* 8 s zImageParts.__contains__c C �
| j �� S r& )r' �__iter__r
r r r r. ; r) zImageParts.__iter__c C r- r& )r' �__len__r
r r r r/ > r) zImageParts.__len__c C s | j �|� d S r& )r' r r+ r r r r A � zImageParts.appendr r r r c C s, t �|�}| �|j�}|dur|S | �|�S )z�Return |ImagePart| object containing image identified by `image_descriptor`.
The image-part is newly created if a matching one is not present in the
collection.
N)r � from_file� _get_by_sha1�sha1�_add_image_part)r r �image�matching_image_partr r r r D s
z ImageParts.get_or_add_image_partc C s& | � |j�}t�||�}| �|� |S )z_Return an |ImagePart| instance newly created from image and appended to the
collection.)�_next_image_partname�extr �
from_imager )r r5 �partname�
image_partr r r r4 P s
zImageParts._add_image_partc C s" | j D ] }|j|kr| S qdS )zlReturn the image part in this collection having a SHA1 hash matching `sha1`,
or |None| if not found.N)r' r3 )r r3 r; r r r r2 X s
�zImageParts._get_by_sha1c sV � fdd�}dd� | D �}t dt| �d �D ] }||vr"||� S q|t| �d �S )z�The next available image partname, starting from ``/word/media/image1.{ext}``
where unused numbers are reused.
The partname is unique by number, without regard to the extension. `ext` does
not include the leading period.
c s t d| � f �S )Nz/word/media/image%d.%sr )�n�r8 r r �image_partnameh r0 z7ImageParts._next_image_partname.<locals>.image_partnamec S s g | ]}|j j�qS r )r: �idx)�.0r; r r r �
<listcomp>k s z3ImageParts._next_image_partname.<locals>.<listcomp>� )�range�len)r r8 r> � used_numbersr< r r= r r7 ` s �z ImageParts._next_image_partnameNr! )
r" r# r$ r% r( r* r. r/ r r r4 r2 r7 r r r r r 2 s
r N)r% �
__future__r �typingr �docx.image.imager �docx.opc.constantsr r �docx.opc.packager �docx.opc.packurir �docx.parts.imager � docx.sharedr
r r r r r r �<module> s #