a
    i=icR                     @   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rawZjpeg)r         c                 C   s   t t|  dd  S )N)i32PAD)c r   d/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/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 )IptcImageFileZIPTCzIPTC/NAAc                 C   s   t | j| S )N)r   info)selfkeyr   r   r   getint7   s    zIptcImageFile.getintc                 C   s   | j d}t|sdS |d |d f}|d dksL|d dk sL|d dkrTtd|d	 }|d
krntdn6|dkr|d}n(|dkrt| j |d }n
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sizer   r   r   field:   s    $

zIptcImageFile.fieldc           
   
   C   s  | j  }|  \}}|r|dkr$q|r6| j |}nd }|| jv r|t| j| trf| j| | q| j| |g| j|< q || j|< q t| jd d }t| jd d }d| jv rt| jd d d }nd}|dkr|sd| _	n2|dkr|rd| | _	n|d	kr|rd
| | _	| 
d| 
df| _zt| 
d }W n0 typ }	 ztd|	W Y d }	~	n
d }	~	0 0 |dkrd||fdd| jd | jd ffg| _d S )N   
   )r   <   r   r   )r   A   Lr   RGBr   Z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*   tagdataZlayers	componentidcompressioner   r   r   _openT   s>    



 
 zIptcImageFile._openc                 C   sj  t | jdks | jd d dkr,tj| S | jd \}}}|\}}| j| tjdd\}}t	|}|dkr|
d| j  |  \}}	|dkrq|	dkr| jt|	d	}
|
sq|
|
 |	t |
8 }	qq|  zdt| }|  |j| _W d    n1 s0    Y  W zt| W n ty<   Y n0 n&zt| W n tyb   Y n0 0 d S )
Nr   r   r6   F)textr	   zP5
%d %d
255
r,   i    )r$   r?   r   loadr"   seektempfilemkstemposfdopenwriter*   r+   r#   mincloser   openimunlinkr&   )r   typer?   boxencodingr@   Zo_fdoutfileor*   r(   Z_imr   r   r   rH      s>     

*zIptcImageFile.loadN)	__name__
__module____qualname__formatformat_descriptionr   r+   rF   rH   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rT| jd}|r|d}n6t| |j	rz| j
j|j }W n ttfy   Y n0 |du rdS G dd d}| } t| _i | _||| _z|   W n ttfy   Y n0 | jS )	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)rY   rZ   r[   r   r   r   r   	FakeImage   s   ra   )io r^   r_   r8   r   r   ZJpegImageFilegetZTiffImageFiler)   rA   ZIPTC_NAA_CHUNKAttributeErrorr>   	__class__BytesIOr"   rF   
IndexError)rR   rb   r^   r_   datar`   ra   r   r   r   getiptcinfo   s4    
rj   )rL   rJ   rc   r   r   _binaryr   r   r'   r   r   r   r=   r   r   r   r   Zregister_openr\   Zregister_extensionrj   r   r   r   r   <module>   s   
|