
    tni              	       *   d dl Z d dlZd dlmZ d dlmZmZ d dlZd dlm	Z	m
Z
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 d	d
lmZ  ej        e          Z G d de          ZddedefdZe j        	 ddedee         ded         fd            Z dS )    N)	Generator)AnyOptional)	HTTPErrorHTTPStatusErrorReadTimeout)Segment)RichToolkitRichToolkitTheme)Progress)MinimalStyleTaggedStyle   )Identityc                   t     e Zd Zd
def fdZ	 	 ddeeef         dedede	e
e         ef         f fd	Z xZS )FastAPIStyle   	tag_widthc                 L    t                                          |           d S )Nr   )super__init__)selfr   	__class__s     X/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/fastapi_cloud_cli/utils/cli.pyr   zFastAPIStyle.__init__   s$    9-----    Fmetadatais_animateddonereturnc                     |s#t                                          |||          S g d}|| j        t          |          z           }|r|d         }| j        dz
  }t          d|          }t          |          g|fS )N)u   🥚u   🐣u   🐤u   🐥u   🐓u   🐔r   r   )r   _get_tag_segmentsanimation_counterlenr   maxr	   )r   r   r   r   emojistagleft_paddingr   s          r   r#   zFastAPIStyle._get_tag_segments   s      	J77,,X{DIII
 
 
 T+c&kk9: 	*C~)1l++~|++r   )r   )FF)__name__
__module____qualname__intr   dictstrr   booltuplelistr	   r#   __classcell__)r   s   @r   r   r      s        . .# . . . . . . "	, ,sCx., , 	,
 
tG}c!	", , , , , , , , , ,r   r   Fminimalr    c                     | rt                      nt          d          }t          |dddddddd	d
          }t          |          S )Nr   r   zwhite on #009485zwhite on #007166grey85whitez#007166z
on #007166red)z	tag.titler(   placeholdertextselectedresultprogresserror)styletheme)r@   )r   r   r   r
   )r4   r?   r@   s      r   get_rich_toolkitrA   3   sk    %ELNNN<"+E+E+EE+%#!$	
 	
  E U####r   r=   message)NNNc              #     K   	 d V  d S # t           $ rI}t                              |           |                     d           t	          j        d          d d }~wt          $ r}t                              |           t          |t                    rA|j	        j
        dk    r1t                              |j	                                                   t          |t                    r5|j	        j
        dv r'd}t                      }|j        dk    r|dz  }n|dz  }n|pd	| }|                     |           t	          j        d          d d }~ww xY w)
NzJThe request to the FastAPI Cloud server timed out. Please try again later.r   i  )i  i  z"The specified token is not valid. userz,Use `fastapi login` to generate a new token.zMake sure to use a valid token.zZSomething went wrong while contacting the FastAPI Cloud server. Please try again later. 

)r   loggerdebug	set_errortyperExitr   
isinstancer   responsestatus_codejsonr   	auth_mode)r=   rB   eidentitys       r   handle_http_errorsrQ   G   sr     
#& & & &QX	
 	
 	
 jmm% & & &Q a)) 	,aj.D.K.KLL**+++a)) 	aj.D
.R.R:GzzH!V++II<<  vursuu 
 	7###jmm%3&s"   
 
EAAE%C+EE)F)N)!
contextlibloggingcollections.abcr   typingr   r   rH   httpxr   r   r   rich.segmentr	   rich_toolkitr
   r   rich_toolkit.progressr   rich_toolkit.stylesr   r   authr   	getLoggerr*   rE   r   r0   rA   contextmanagerr/   rQ    r   r   <module>r_      s        % % % % % %                  9 9 9 9 9 9 9 9 9 9             6 6 6 6 6 6 6 6 * * * * * * 9 9 9 9 9 9 9 9      		8	$	$, , , , ,; , , ,B$ $d ${ $ $ $ $(  "'& '&'&c]'&  '& '& '& '& '& '&r   