
    +hw                        d dl mZ d dlZd dlmZ d dlmZ  G d de      Z G d dej                        Z
dd	Zdd
Zy)    )annotationsN)Callable)overridec                  6     e Zd Zd fdZedd       ZeZ xZS )CancelledErrorc                2    || _         t        | 	  |       y N)msgsuper__init__)selfr
   	__class__s     G/var/www/html/venv/lib/python3.12/site-packages/openai/cli/_progress.pyr   zCancelledError.__init__	   s        c                    | j                   S r	   )r
   r   s    r   __str__zCancelledError.__str__   s    xxr   )r
   strreturnNone)r   r   )__name__
__module____qualname__r   r   r   __repr____classcell__r   s   @r   r   r      s#       Hr   r   c                  >     e Zd Zdd fdZddZeddd       Z xZS )	BufferReaderc                    t         |   |       t        |      | _        d| _        t        t        |      |      | _        y )Nr   )desc)r   r   len_len	_progressprogress	_callback)r   bufr    r   s      r   r   zBufferReader.__init__   s4    H	!#c(6r   c                    | j                   S r	   )r"   r   s    r   __len__zBufferReader.__len__   s    yyr   c                   t         j                  j                  | |      }| xj                  t	        |      z  c_        	 | j                  | j                         |S # t        $ r}t        dj                  |            d }~ww xY w)NzThe upload was cancelled: {})	ioBytesIOreadr#   r!   r%   	Exceptionr   format)r   nchunkes       r   r,   zBufferReader.read   sp    

a(#e*$	KNN4>>*   	K !?!F!Fq!IJJ	Ks    A 	B&B  B)r   N)r&   bytesr    
str | Noner   r   )r   int))r/   z
int | Noner   r2   )r   r   r   r   r(   r   r,   r   r   s   @r   r   r      s!    7 	 	r   r   c                H     dd l }|j                   d|      d fd}|S )Nr   T)total
unit_scaler    c                `    | _         | k(  rj                          y j                          y r	   )r/   closerefresh)r$   meterr7   s    r   incrzprogress.<locals>.incr0   s$    uKKMMMOr   )r$   floatr   r   )tqdm)r7   r    r?   r=   r<   s   `   @r   r$   r$   +   s&    IIEdI>E Kr   c                    t        | dz        S )Ni   )r4   )is    r   MBrB   :   s    qG|r   )r7   r>   r    r3   r   zCallable[[float], None])rA   r4   r   r4   )
__future__r   r*   typingr   typing_extensionsr   r-   r   r+   r   r$   rB    r   r   <module>rG      s7    " 	  &	Y 	2:: .r   