U
    %d                     @   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   7/tmp/pip-unpacked-wheel-vp0hp2u_/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rXd}t||d	 }|d
krvd}t|n6|dkrd}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msgsizer   r   r   field:   s"    $

zIptcImageFile.fieldc              
   C   s  | j  }|  \}}|r|dkr$q|r6| j |}nd }|| jkr|t| j| trf| j| | q| j| |g| j|< q || j|< q t| jd d }t| jd d }d| jk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 n2 tk
rr }	 zd}
t|
|	W 5 d }	~	X Y nX |dkrd||fdd| jd | jd ffg| _d S )N   
   )r   <   r   r   )r   A   Lr   Z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   r   _openV   s@    




 zIptcImageFile._openc                 C   s2  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|  z*t|}|  |j| _W 5 Q R X W 5 zt| W n tk
r*   Y nX X 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closeunlinkr&   r   openim)r   typer?   Zboxencodingr@   Zo_fdoutfileor+   r(   Z_imr   r   r   rH      s6     

zIptcImageFile.loadN)	__name__
__module____qualname__formatformat_descriptionr   r,   rF   rH   r   r   r   r   r   2   s   2r   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}n8t| |j	rz| j
j|j }W n ttfk
r   Y nX |dkrdS G dd d}| } t| _i | _||| _z|   W n ttfk
r   Y nX | 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)rX   rY   rZ   r   r   r   r   	FakeImage   s   r`   )io r]   r^   r8   r   r   ZJpegImageFilegetZTiffImageFiler)   rA   ZIPTC_NAA_CHUNKAttributeErrorr>   	__class__BytesIOr"   rF   
IndexError)rS   ra   r]   r^   datar_   r`   r   r   r   getiptcinfo   s4    
ri   )rL   rJ   rb   r   r   _binaryr   r   r'   r   r   r   r=   r   r   r   r   Zregister_openr[   Zregister_extensionri   r   r   r   r   <module>   s   
