o
    e~#g                     @  s   d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl	Z	d dl
mZ d dlmZ d dlmZ d dlmZmZ d d	lmZ d d
lmZmZmZ edZG dd deZG dd deZG dd deZG dd deZdS )    )annotationsN)
NamedTuple)quote)stat)Headers)HTTPException)Response
guess_type)StaticFiles)ReceiveScopeSendz#^bytes=(?P<start>\d+)-(?P<end>\d*)$c                   @  s2   e Zd ZU ded< ded< dddZddd	Zd
S )ClosedRangeintstartendreturnc                 C  s   | j | j d S )N   )r   r   self r   {/var/www/static.ux5.de/https/Moving-Object-Detection-with-OpenCV/env/lib/python3.10/site-packages/gradio/ranged_response.py__len__   s   zClosedRange.__len__boolc                 C  s   t | dkS )Nr   )lenr   r   r   r   __bool__   s   zClosedRange.__bool__N)r   r   )r   r   )__name__
__module____qualname____annotations__r   r   r   r   r   r   r      s
   
 
r   c                   @  s,   e Zd ZU ded< dZded< d
dd	ZdS )	OpenRanger   r   Nz
int | Noner   r   r   c                 C  sB   t | j|}tdd | j|fD }t||}t ||}t||S )Nc                 s  s    | ]}|r|V  qd S )Nr   ).0xr   r   r   	<genexpr>'   s    z"OpenRange.clamp.<locals>.<genexpr>)maxr   minr   r   )r   r   r   beginr   r   r   clamp%   s
   


zOpenRange.clamp)r   r   r   r   r   r   )r   r   r   r   r   r'   r   r   r   r   r    !   s   
 r    c                   @  s:   e Zd ZdZ					ddddZd ddZd!ddZdS )"RangedFileResponsei   Npathstr | os.PathLikeranger    headersdict[str, str] | None
media_type
str | Nonefilenamestat_resultos.stat_result | Nonemethodr   Nonec           
      C  s   t d u rtd|| _|| _|| _d | _|d uo| dk| _|d u r-t|p'|d p,d}|| _	| 
|p5i  | jd urZt| j}|| jkrLd| }	nd| j d}	| jd|	 || _d S )	Nz0'aiofiles' must be installed to use FileResponseHEADr   z
text/plainzattachment; filename*=utf-8''zattachment; filename=""zcontent-disposition)aiofilesModuleNotFoundErrorr)   r+   r0   
backgrounduppersend_header_onlyr	   r.   init_headersr   r,   
setdefaultr1   )
r   r)   r+   r,   r.   r0   r1   r3   content_disposition_filenamecontent_dispositionr   r   r   __init__2   s*   




zRangedFileResponse.__init__r   c                 C  sP   | j std| j j}t|}d|j d|j d| | jd< t|| jd< d S )Nz(No stat result to set range headers withzbytes -/zcontent-rangezcontent-length)r1   
ValueErrorst_sizer   r   r   r,   str)r   r+   total_lengthcontent_lengthr   r   r   set_range_headersT   s   z$RangedFileResponse.set_range_headersscoper   receiver   sendr   c              
     s  | j d u r;zt| jI d H }|| _ W n ty) } z
td| j d|d }~ww |j}t|s;td| j d| j	d| j j
}| | tj| jdd4 I d H v}||jI d H  |dd| jd	I d H  | jrz|d
dddI d H  nFt|}	|s|d
dddI d H  	 W d   I d H  d S |	dkrt| j|	}
||
I d H }|	t|8 }	|d
||	dkdI d H  |	dksW d   I d H  d S 1 I d H sw   Y  d S )NzFile at path z does not exist.z is not a file.r   rb)modezhttp.response.start   )typestatusr,   zhttp.response.body    F)rO   body	more_body)r1   aio_statr)   FileNotFoundErrorRuntimeErrorst_moder   S_ISREGr+   r'   rD   rH   r7   openseekr   raw_headersr;   r   r%   
chunk_sizeread)r   rI   rJ   rK   r1   fnferM   
byte_rangefileremaining_bytesr\   chunkr   r   r   __call___   sb   









.zRangedFileResponse.__call__)NNNNN)r)   r*   r+   r    r,   r-   r.   r/   r0   r/   r1   r2   r3   r/   r   r4   )r+   r   r   r4   )rI   r   rJ   r   rK   r   r   r4   )r   r   r   r\   r@   rH   rc   r   r   r   r   r(   /   s    
"r(   c                      s,   e Zd Z	dd fddZdddZ  ZS )RangedStaticFiles   	full_pathr*   r1   os.stat_resultrI   r   status_coder   r   r   c                   sH   t |d}|dr| j|||d}n
t j||||d}d|jd< |S )NrI   r+   )r1   rI   )r1   rI   rh   byteszaccept-ranges)r   getranged_file_responsesuperfile_responser,   )r   rf   r1   rI   rh   request_headersresponse	__class__r   r   rn      s   


zRangedStaticFiles.file_responsec                 C  sp   |d }t |d}|d }t|}|std|d|d}}	tt||	r-t|	nd }
t||
||dS )Nr3   ri   r+   i  r   r   )r1   r3   )r   RANGE_REGEXsearchr   groupr    r   r(   )r   rf   r1   rI   r3   ro   range_headermatchr   r   r+   r   r   r   rl      s   

z&RangedStaticFiles.ranged_file_response)re   )
rf   r*   r1   rg   rI   r   rh   r   r   r   )rf   r*   r1   rg   rI   r   r   r   )r   r   r   rn   rl   __classcell__r   r   rq   r   rd      s    rd   )
__future__r   osrer   typingr   urllib.parser   r7   aiofiles.osrT   starlette.datastructuresr   starlette.exceptionsr   starlette.responsesr   r	   starlette.staticfilesr
   starlette.typesr   r   r   compilers   r   r    r(   rd   r   r   r   r   <module>   s$   
d