B
    zbR                 @   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   2lib/python3.7/site-packages/PIL/IptcImagePlugin.pyi#   s    r   c             C   s,   x | D ]}t dt| dd qW 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  x| j  }|  \}}|r$|dkr&P |r8| j |}nd }|| jkr~t| j| trh| j| | q| j| |g| j|< q|| j|< qW t| jd d }t| jd d }d| jkrt| jd d d }nd}|dkr|sd| _	n6|dkr|rd| | _	n|d	kr |r d
| | _	| 
d| 
df| _yt| 
d }W n. tk
rv }	 ztd|	W d 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   Z_sizeCOMPRESSIONKeyErrorr&   r*   tile)
r   offsetr)   r*   tagdataZlayersZ	componentidZcompressioner   r   r   _openT   s>    




zIptcImageFile._openc             C   s:  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  xV|  \}}	|dkrP x:|	dkr| jt|	d	}
|
sP |
|
 |	t |
8 }	qW qW |  z*t|}|  |j| _W d Q R X W d yt| W n tk
r2   Y nX X d S )
Nr   r   r4   F)textr	   zP5
%d %d
255
)r,   r-   i    )r$   r<   r   loadr"   seektempfileZmkstemposfdopenwriter*   r+   r#   mincloser   openimunlinkr&   )r   typer<   Zboxencodingr=   Zo_fdZoutfileor*   r(   Z_imr   r   r   rC      s8     


zIptcImageFile.loadN)	__name__
__module____qualname__formatZformat_descriptionr   r+   rA   rC   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}n8t| |j	ry| j
j|j }W n ttfk
r   Y nX |dkrdS G dd d}| } t| _i | _||| _y|   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)rQ   rR   rS   r   r   r   r   	FakeImage   s   rX   )io rU   rV   r6   r   r   ZJpegImageFilegetZTiffImageFiler)   r>   ZIPTC_NAA_CHUNKAttributeErrorr;   	__class__BytesIOr"   rA   
IndexError)rL   rY   rU   rV   datarW   rX   r   r   r   getiptcinfo   s4    
ra   )rF   rE   rZ   r   r   Z_binaryr   r   r'   r   r   r   r:   r   r   r   r   Zregister_openrT   Zregister_extensionra   r   r   r   r   <module>   s   
|