o
    g                     @   s   d dl Z d dlZddlmZmZ ddlmZ ddlmZ ddlm	Z
 ddlmZ dd	d
Zed d Zdd Zdd ZG dd dejZeeje eejd dd ZdS )    N   )Image	ImageFile)i8)i16be)i32be)o8rawjpeg)r         c                 C   s   t t|  dd  S )N)i32PAD)c r   Q/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/PIL/IptcImagePlugin.pyi#   s   r   c                 C   s(   | D ]}t dt| dd qt   d S )Nz%02x )end)printr   )r   r   r   r   r   dump'   s   
r   c                   @   s4   e Zd ZdZdZdd Zdd Zdd Zd	d
 ZdS )IptcImageFileIPTCzIPTC/NAAc                 C   s   t | j| S )N)r   info)selfkeyr   r   r   getint6   s   zIptcImageFile.getintc                 C   s   | j d}t|sdS |d |d f}|d dks&|d dk s&|d dkr,d}t||d	 }|d
kr:d}t||dkrDd}||fS |dkrVt| j |d }||fS t|d	}||fS )Nr   )Nr   r      r      	   zinvalid IPTC/NAA file      z%illegal field length in IPTC/NAA file   )fpreadlenSyntaxErrorOSErrorr   i16)r   stagmsgsizer   r   r   field9   s&   $
zIptcImageFile.fieldc              
   C   s  	 | j  }|  \}}|r|dkrn2|r| j |}nd }|| jv r?t| j| tr4| j| | n| j| |g| j|< n|| j|< qt| jd d }t| jd d }d| jv rht| jd d d }nd}|dkrt|std| _	n|dkr|rd	| | _	n|d
kr|rd| | _	| 
d| 
df| _z	t| 
d }W n ty }	 zd}
t|
|	d }	~	ww |dkrd||fdd| jd | jd ffg| _d S d S )NT   
   )r!   <   r   r   )r!   A   Lr!   RGBr   CMYK)r!      )r!      )r!   x   zUnknown IPTC image compressioniptc)r$   tellr.   r%   r   
isinstancelistappendr   moder   _sizeCOMPRESSIONKeyErrorr(   r-   tile)r   offsetr+   r-   tagdatalayers	componentidcompressioner,   r   r   r   _openU   sJ   





 
zIptcImageFile._openc                 C   sh  t | jdks| jd d dkrtj| S | jd \}}}|\}}| j| tjdd\}}t	|}|dkrA|
d| j  	 |  \}}	|d	krMn |	dkrl| jt|	d
}
|
s]n|
|
 |	t |
8 }	|	dksQqB|  z1t|}|  |j| _W d    n1 sw   Y  W zt| W d S  ty   Y d S w zt| W w  ty   Y w w )Nr   r   r:   F)textr	   zP5
%d %d
255
Tr/   i    )r&   rC   r   loadr$   seektempfilemkstemposfdopenwriter-   r.   r%   mincloser   openimunlinkr(   )r   typerC   boxencodingrD   o_fdoutfileor-   r*   _imr   r   r   rM      sL    



zIptcImageFile.loadN)	__name__
__module____qualname__formatformat_descriptionr   r.   rK   rM   r   r   r   r   r   2   s    1r   z.iimc              	   C   s   ddl }ddlm}m} d}t| tr| jS t| |jr*| jd}|r)|d}nt| |j	rEz	| j
j|j }W n ttfyD   Y nw |du rKdS G dd d}| } t| _i | _||| _z|   W | jS  ttfyv   Y | jS w )	z
    Get IPTC information from TIFF, JPEG, or IPTC file.

    :param im: An image containing IPTC data.
    :returns: A dictionary containing IPTC information, or None if
        no IPTC information block was found.
    r   Nr   )JpegImagePluginTiffImagePlugin	photoshopi  c                   @   s   e Zd ZdS )zgetiptcinfo.<locals>.FakeImageN)r`   ra   rb   r   r   r   r   	FakeImage   s    rh   )io re   rf   r<   r   r   JpegImageFilegetTiffImageFiler+   rE   IPTC_NAA_CHUNKAttributeErrorrB   	__class__BytesIOr$   rK   
IndexError)rW   ri   re   rf   datarg   rh   r   r   r   getiptcinfo   s<   


rt   )rQ   rO   rj   r   r   _binaryr   r   r)   r   r   r   rA   r   r   r   r   register_openrc   register_extensionrt   r   r   r   r   <module>   s   
|