o
    tBhX                     @   s   d dl Z d dlZd dlZd dlZddlmZ ddlmZ dZdZ	dZ
	 dd	d
dddZ	 dd Zdd ZG dd dejZG dd deZG dd dZd$ddZdd ZG dd dZG dd dZG d d! d!eZG d"d# d#eZdS )%    N   )Image)isPathi   i   Fzimage buffer overrun errorzdecoding errorzunknown errorzbad configurationzout of memory error)ic                 C   sH   zt j| }W n ty   t| }Y nw |sd|  }t|d )Nzdecoder error z when reading image file)r   coregetcodecstatusAttributeErrorERRORSgetOSError)errormessage r   d/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/PIL/ImageFile.pyraise_oserror@   s   
r   c                 C   s   | d S )N   r   )tr   r   r   	_tilesortJ      r   c                       sR   e Zd ZdZd fdd	Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
  ZS )	ImageFilez*Base class for image file format handlers.Nc                    s   t    d| _d | _d | _	 d| _d| _t| _t	|r)t
|d| _|| _d| _n	|| _|| _d | _z/z|   W n tttttjfyP } zt||d }~ww | jr[| jd dkr_tdW d S  tyq   | jrp| j   w )Nr   r   r   rbTznot identified by this driver)super__init__
_min_framecustom_mimetypetilereadonlydecoderconfigMAXBLOCKdecodermaxblockr   openfpfilename_exclusive_fp_open
IndexError	TypeErrorKeyErrorEOFErrorstructr   SyntaxErrormodesizeBaseExceptionclose)selfr$   r%   v	__class__r   r   r   W   sH   

	
zImageFile.__init__c                 C   s,   | j r| j S | jd urtj| j S d S N)r   formatr   MIMEr   upperr2   r   r   r   get_format_mimetype   s
   
zImageFile.get_format_mimetypec                 C   s   | j r| j  d| _dS )zCheck file integrityN)r&   r$   r1   r:   r   r   r   verify   s   

zImageFile.verifyc                 C   sj  | j du r	tdtj| }| j s|S d| _| jo t| j dk}|o(ttd }d}z| j	}d}W n t
y?   | jj}Y nw z| j}d}W n t
yT   | jj}Y nw |r| j d \}}}}	|dkrt|	dkr|	d | jkr|	d tjv rz=ddl}
t| j}|
j| d|
jd	| _W d   n1 sw   Y  tj| j| j|||	| _d}| jrd| j_W n t
ttfy   d| _Y nw |   d
}| js| j jtd z| j}W n t
y   d}Y nw dd t !| j dd D | _ | j D ]\}}}}	|| t"| j||	| j#}zr|$| j| |j%r'|&| j |'dd }nS|}	 z|| j(}W n t)t*j+fyO } zt,rFW Y d}~n4td|d}~ww |sat,rWn#tdt| d|| }|'|\}}|dk rrn||d }q*W |-  q|-  w g | _ || _.| /  | j0r| j1r| j2  d| _| jst,s|dk rt3| tj| S )z"Load image data based on tile listNzcannot load this imager   pypy_version_infor   Fraw   )accessr   key    c                 S   s   g | ]
\}}t |d  qS )r   )list).0_tilesr   r   r   
<listcomp>   s    
z"ImageFile.load.<locals>.<listcomp>c                 S   s   | d | d | d fS )Nr   r   r?   r   )r   r   r   r   <lambda>   s    z ImageFile.load.<locals>.<lambda>Tzimage file is truncatedzimage file is truncated (z bytes not processed))4r   r   r   loadmapr%   lenhasattrsys	load_readr   r$   read	load_seekseekr.   	_MAPMODESmmapr#   filenoACCESS_READr	   
map_bufferr/   impalettedirtyImportErrorload_preparesortr   tile_prefix	itertoolsgroupby_getdecoderr    setimagepulls_fdsetfddecoder"   r(   r,   r   LOAD_TRUNCATED_IMAGEScleanupr   load_endr&   !_close_exclusive_fp_after_loadingr1   r   )r2   pixeluse_mmapr   rP   rR   decoder_nameextentsoffsetargsrT   r$   err_codeprefixdecoderbsenr   r   r   rJ      s   






zImageFile.loadc                 C   sT   | j r| j j| jks| j j| jkrtj| j| j| _ | jdkr(tj|  d S d S )NP)rX   r.   r/   r   r	   newrJ   r:   r   r   r   r\     s
   "
zImageFile.load_preparec                 C   s   d S r6   r   r:   r   r   r   rh      s   zImageFile.load_endc                 C   sB   || j k st| dr| jd u s|| j| j  krtd|  |kS )N	_n_framesz attempt to seek outside sequence)r   rM   ry   n_framesr+   tell)r2   framer   r   r   _seek_check,  s   

zImageFile._seek_check)NN)__name__
__module____qualname____doc__r   r;   r<   rJ   r\   rh   r}   __classcell__r   r   r4   r   r   T   s    /	 r   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	StubImageFilez
    Base class for stub image loaders.

    A stub loader is an image loader that can identify files of a
    certain format, but relies on external code to load the file.
    c                 C      t d)Nz+StubImageFile subclass must implement _openNotImplementedErrorr:   r   r   r   r'   C  s   zStubImageFile._openc                 C   sP   |   }|d u rtd| j d|| }|d usJ |j| _|j| _| S )Nzcannot find loader for this z file)_loadr   r7   rJ   r5   __dict__)r2   loaderimager   r   r   rJ   F  s   
zStubImageFile.loadc                 C   r   )z (Hook) Find actual image loader.z+StubImageFile subclass must implement _loadr   r:   r   r   r   r   Q  r   zStubImageFile._loadN)r~   r   r   r   r'   rJ   r   r   r   r   r   r   ;  s
    r   c                   @   sP   e Zd ZdZdZdZdZdZdZdZ	dd Z
dd Zdd	 Zd
d Zdd ZdS )Parserzj
    Incremental image parser.  This class implements the standard
    feed/close consumer interface.
    Nr   c                 C   s   | j du s	J ddS )z
        (Consumer) Reset the parser.  Note that you can only call this
        method immediately after you've created a parser; parser
        instances cannot be reused.
        Nzcannot reuse parsers)datar:   r   r   r   resetc  s   zParser.resetc                 C   s  | j rdS | jdu r|| _n| j| | _| jrh| jdkr=tt| j| j}| j|d | _| j| | _| jdks;| js=dS | j| j\}}|dk r^d| _d| _ |dk r\d| _t| ndS | j|d | _dS | jrmdS zt	
| j}t|}W d   n1 sw   Y  W n
 ty   Y dS w t|dpt|d}|st|jdkrd| _n:|  |jd \}}}	}
g |_t|j||
|j| _| j|j| |	| _| jt| jkr| j| jd | _d| _|| _dS )z
        (Consumer) Feed data to the parser.

        :param data: A string buffer.
        :exception OSError: If the parser failed to parse the image file.
        Nr   r   rQ   rO   )finishedr   rr   rn   minrL   re   r   r   ioBytesIOr   r#   r   rM   r   r\   ra   r.   r    rb   rX   )r2   r   skiprv   ru   r$   rX   flagdoar   r   r   feedk  sX   	



zParser.feedc                 C   s   | S r6   r   r:   r   r   r   	__enter__  s   zParser.__enter__c                 G   s   |    d S r6   )r1   r2   ro   r   r   r   __exit__  s   zParser.__exit__c              	   C   s   | j r| d d | _| _ | jstd| jstd| jrKt| j}zt	|| _W | j
  n| j
  w W d   | jS 1 sFw   Y  | jS )a  
        (Consumer) Close the stream.

        :returns: An image object.
        :exception OSError: If the parser failed to parse the image file either
                            because it cannot be identified or cannot be
                            decoded.
        rC   Nzimage was incompletezcannot parse this image)rr   r   r   r   r   r   r   r   r   r#   rJ   )r2   r$   r   r   r   r1     s    


zParser.close)r~   r   r   r   incrementalr   r   rr   rn   r   r   r   r   r   r1   r   r   r   r   r   V  s    Pr   c              
   C   sf  |    t| dsd| _|jtd tt|| jd d }z| }|	  d}W n t
tjfy@ } z|}W Y d}~nd}~ww |D ]b\}}}}	|dkrR|| t| j||	| j}
zD|
| j| |
jrs|
| |
 \}}n|r	 |
|\}}}|| |rnqvn|
||}|dk rtd| d	|W |
  qC|
  w t|d
r|	  dS dS )zHelper to save image based on tile list

    :param im: Image object.
    :param fp: File object.
    :param tile: Tile list.
    :param bufsize: Optional buffer size
    encoderconfigr   rA   r      NTzencoder error z when writing image fileflush)rJ   rM   r   r]   r   maxr!   r/   rU   r   r   r   UnsupportedOperationrR   r   _getencoderr.   rb   rX   	pushes_fdrd   encode_to_pyfdencodewriteencode_to_filer   rg   )rX   r$   r   bufsizefhexcru   rs   r   r   encoderlrt   r   r   r   r   _save  sL   	




r   c                 C   s   |dkrdS |t kr| |}t||k rtd|S g }|}|dkr=| t|t }|s.n|| |t|8 }|dks#tdd |D |k rLtdd|S )a  
    Reads large blocks in a safe way.  Unlike fp.read(n), this function
    doesn't trust the user.  If the requested size is larger than
    SAFEBLOCK, the file is read block by block.

    :param fp: File handle.  Must implement a <b>read</b> method.
    :param size: Number of bytes to read.
    :returns: A string containing <i>size</i> bytes of data.

    Raises an OSError if the file is truncated and the read cannot be completed

    r   rC   zTruncated File Readc                 s   s    | ]}t |V  qd S r6   )rL   )rE   r   r   r   r   	<genexpr>1  s    z_safe_read.<locals>.<genexpr>)	SAFEBLOCKrP   rL   r   r   appendsumjoin)r$   r/   r   remaining_sizeblockr   r   r   
_safe_read  s&   


r   c                   @   s   e Zd Zdd Zdd ZdS )PyCodecStatec                 C   s   d| _ d| _d| _d| _d S )Nr   )xsizeysizexoffyoffr:   r   r   r   r   7  s   
zPyCodecState.__init__c                 C   s    | j | j| j | j | j| j fS r6   )r   r   r   r   r:   r   r   r   rm   =  s    zPyCodecState.extentsN)r~   r   r   r   rm   r   r   r   r   r   6  s    r   c                   @   s6   e Zd Zdd Zdd Zdd Zdd Zdd
dZd	S )PyCodecc                 G   s(   d | _ t | _d | _|| _| | d S r6   )rX   r   statefdr.   init)r2   r.   ro   r   r   r   r   B  s
   zPyCodec.__init__c                 C   
   || _ dS )z
        Override to perform codec specific initialization

        :param args: Array of args items from the tile entry
        :returns: None
        N)ro   r   r   r   r   r   I     
zPyCodec.initc                 C   s   dS )zT
        Override to perform codec specific cleanup

        :returns: None
        Nr   r:   r   r   r   rg   R  s   zPyCodec.cleanupc                 C   r   )z
        Called from ImageFile to set the Python file-like object

        :param fd: A Python file-like object
        :returns: None
        N)r   )r2   r   r   r   r   rd   Z  r   zPyCodec.setfdNc                 C   s   || _ |r|\}}}}nd\}}}}|dkr%|dkr%| j j\| j_| j_n|| j_|| j_|| | j_|| | j_| jjdksE| jjdkrItd| jj| jj | j jd kse| jj| jj | j jd kritddS )z
        Called from ImageFile to set the core output image for the codec

        :param im: A core image object
        :param extents: a 4 tuple of (x0, y0, x1, y1) defining the rectangle
            for this tile
        :returns: None
        )r   r   r   r   r   zSize cannot be negativer   z Tile cannot extend outside imageN)rX   r/   r   r   r   r   r   
ValueError)r2   rX   rm   x0y0x1y1r   r   r   rb   c  s    zPyCodec.setimager6   )r~   r   r   r   r   rg   rd   rb   r   r   r   r   r   A  s    		r   c                   @   s2   e Zd ZdZdZedd Zdd Zd
dd	ZdS )	PyDecoderz
    Python implementation of a format decoder. Override this class and
    add the decoding logic in the :meth:`decode` method.

    See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
    Fc                 C      | j S r6   )	_pulls_fdr:   r   r   r   rc        zPyDecoder.pulls_fdc                 C      t  )a=  
        Override to perform the decoding process.

        :param buffer: A bytes object with the data to be decoded.
        :returns: A tuple of ``(bytes consumed, errcode)``.
            If finished with decoding return -1 for the bytes consumed.
            Err codes are from :data:`.ImageFile.ERRORS`.
        r   )r2   bufferr   r   r   re        	zPyDecoder.decodeNc                 C   sd   |s| j }t| j d|}|| j| j  ||}|d dkr&td|d dkr0tddS )a  
        Convenience method to set the internal image from a stream of raw data

        :param data: Bytes to be set
        :param rawmode: The rawmode to be used for the decoder.
            If not specified, it will default to the mode of the image
        :returns: None
        r>   r   znot enough image datar   zcannot decode image dataN)	r.   r   ra   rb   rX   r   rm   re   r   )r2   r   rawmoder   rt   r   r   r   
set_as_raw  s   

zPyDecoder.set_as_rawr6   )	r~   r   r   r   r   propertyrc   re   r   r   r   r   r   r     s    
r   c                   @   s8   e Zd ZdZdZedd Zdd Zdd Zd	d
 Z	dS )	PyEncoderz
    Python implementation of a format encoder. Override this class and
    add the decoding logic in the :meth:`encode` method.

    See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
    Fc                 C   r   r6   )
_pushes_fdr:   r   r   r   r     r   zPyEncoder.pushes_fdc                 C   r   )a   
        Override to perform the encoding process.

        :param bufsize: Buffer size.
        :returns: A tuple of ``(bytes encoded, errcode, bytes)``.
            If finished with encoding return 1 for the error code.
            Err codes are from :data:`.ImageFile.ERRORS`.
        r   )r2   r   r   r   r   r     r   zPyEncoder.encodec                 C   s2   | j sdS | d\}}}|r| j| ||fS )z
        If ``pushes_fd`` is ``True``, then this method will be used,
        and ``encode()`` will only be called once.

        :returns: A tuple of ``(bytes consumed, errcode)``.
            Err codes are from :data:`.ImageFile.ERRORS`.
        )r   r   r   )r   r   r   r   )r2   bytes_consumederrcoder   r   r   r   r     s   zPyEncoder.encode_to_pyfdc                 C   sB   d}|dkr|  |\}}}|dkr|||d  |dks|S )z
        :param fh: File handle.
        :param bufsize: Buffer size.

        :returns: If finished successfully, return 0.
            Otherwise, return an error code. Err codes are from
            :data:`.ImageFile.ERRORS`.
        r   N)r   r   )r2   r   r   r   statusbufr   r   r   r     s   	zPyEncoder.encode_to_fileN)
r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s    
r   )r   )r   r_   r,   rN    r   _utilr   r!   r   rf   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>   s<   

 h 
4!F/