o i @sddlZddlZddlmZddlmZmZddlZddlm Z m Z m Z ddl m Z ddlmZmZddlmZddlmZmZd d lmZmZeeZGd d d eZddedefddZdefddZ dde deedefddZ!ej" ddedeededfddZ#dS)N) Generator)AnyOptional) HTTPErrorHTTPStatusError ReadTimeout)Segment) RichToolkitRichToolkitTheme)Progress) MinimalStyle TaggedStyle)Identitydelete_auth_configc s\eZdZd deffdd Z  d deeefdeded e e e efffd d Z Z S) FastAPIStyle tag_widthcstj|ddS)Nr)super__init__)selfr __class__O/home/ubuntu/.local/lib/python3.10/site-packages/fastapi_cloud_cli/utils/cli.pyrszFastAPIStyle.__init__Fmetadata is_animateddonereturncs\|s t|||Sgd}||jt|}|r|d}|jd}td|}t|g|fS)N)u🥚u🐣u🐤u🐥u🐓u🐔rr)r_get_tag_segmentsanimation_counterlenrmaxr)rrrremojistag left_paddingrrrr!s   zFastAPIStyle._get_tag_segments)r)FF)__name__ __module__ __qualname__intrdictstrrbooltuplelistrr! __classcell__rrrrrs rFminimalrc Cs>|rtntdd}t|ddddddd d d d d }t|dS)Nrrzwhite on #009485zwhite on #007166grey62whitez#007166grey85z on #007166redzindian_red italic) z tag.titler& placeholdertextselectedresultprogresserror cancelled)styletheme)r?)r rr r )r2r>r?rrrget_rich_toolkit3s r@cCs2d}t}|jdkrt|d7}|S|d7}|S)Nz"The specified token is not valid. userz,Use `fastapi login` to generate a new token.zMake sure to use a valid token.)r auth_moder)messageidentityrrrhandle_unauthorizedHs rEr<default_messagecCsbd}t|tr&|jj}|dkrt|jn|dkr t}n|dkr&d}|s/|p.d|}|S)Niiiz,You don't have permissions for this resourcezZSomething went wrong while contacting the FastAPI Cloud server. Please try again later. ) isinstancerresponse status_codeloggerdebugjsonrE)r<rFrCrIrrrhandle_http_errorWs rMr;)NNNc cszdVWdSty#}zt||dtddd}~wtyC}zt|t||}||tddd}~ww)NzJThe request to the FastAPI Cloud server timed out. Please try again later.r)rrJrK set_errortyperExitrrM)r;rFerCrrrhandle_http_errorsps"       rR)F)N)$ contextlibloggingcollections.abcrtypingrrrOhttpxrrr rich.segmentr rich_toolkitr r rich_toolkit.progressr rich_toolkit.stylesr r authrr getLoggerr(rJrr.r@r-rErMcontextmanagerrRrrrrs2    !