o
    Z#aR                     @   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   5/usr/lib/python3/dist-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   getint7   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*td|d	 }|d
kr6td|dkr@d}||fS |dkrRt| 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sizer   r   r   field:   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td|	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   _openT   sH   





 
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   r9   F)textr	   zP5
%d %d
255
Tr.   i    )r&   rB   r   loadr$   seektempfilemkstemposfdopenwriter,   r-   r%   mincloser   openimunlinkr(   )r   typerB   boxencodingrC   o_fdoutfileor,   r*   _imr   r   r   rL      sL    



zIptcImageFile.loadN)	__name__
__module____qualname__formatformat_descriptionr   r-   rJ   rL   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_   r`   ra   r   r   r   r   	FakeImage   s    rg   )io rd   re   r;   r   r   JpegImageFilegetTiffImageFiler+   rD   IPTC_NAA_CHUNKAttributeErrorrA   	__class__BytesIOr$   rJ   
IndexError)rV   rh   rd   re   datarf   rg   r   r   r   getiptcinfo   s<   


rs   )rP   rN   ri   r   r   _binaryr   r   r)   r   r   r   r@   r   r   r   r   register_openrb   register_extensionrs   r   r   r   r   <module>   s   
|