&`iz(_ColorfulMock.__init__.."s!)identitycolorful colormode NO_COLORS ANSI_8_COLORSselfs r__init__z_ColorfulMock.__init__ s.#   !rcdSrrrs rdisablez_ColorfulMock.disable* rc#<KGdd}|VdS)NceZdZdZdS)/_ColorfulMock.with_style..IdentityClasscdS)Nc|Srr)ys rrzM_ColorfulMock.with_style..IdentityClass.__getattr__..1srrrnames r __getattr__z;_ColorfulMock.with_style..IdentityClass.__getattr__0s "{"rN)__name__ __module__ __qualname__r)rrr IdentityClassr#/s# # # # # #rr-r)rrr-s r with_stylez_ColorfulMock.with_style-sH # # # # # # # #moorc*|dkr|jS|jS)Nr.)r.rr's rr)z_ColorfulMock.__getattr__5s <  ? "}rN)r*r+r,rrrr.r)rrrr r s\"""   ^rr )ColorfulStringceZdZdS)r0N)r*r+r,rrrr0r0Fs rr0ceZdZgdZdZdS)_ColorfulProxy) rresetbolditalic underlineddimmed dodgerBlue limeGreenredorangeskyBluemagentayellowctt|}t|r#|tjvrt d|zdz|S)NzUsage of the colorful method 'zh' is forbidden by the proxy to keep a consistent color scheme. Check `cli_logger.py` for allowed methods)getattr_cfcallabler3_proxy_allowlist ValueError)rr(ress rr)z_ColorfulProxy.__getattr__dsWc4   C== T)HHH047;<<   rN)r*r+r,rDr)rrrr3r3Qs5$rr3F)stripctj}|}d}|jjtkr!|j}|dz }|jjtk!|jtj |jjdS)zGet the info from the caller frame. Used to override the logging function and line number with the correct ones. See the comment on _patched_makeRecord for more info. r)linenofilename) inspect currentframef_code co_filename__file__f_backf_linenoospathbasename)framecallerlevelss r_external_caller_inforYts|  " "E F F - #x / /!  - #x / //G$$V]%>??  r) no_format_tags _numberedmsgargsrZr[r\kwargsc t|tst|tr d}|g}|D]!\}} | dur||gz }| dur||dz| zgz }"|rXttdd|}d} |U|\} } } t| dt| zd zt| z| d zd z} |r| |z|zS| |j|i|z|zS|rtd |g|}d |D}d|S)awFormats a message for printing. Renders `msg` using the built-in `str.format` and the passed-in `*args` and `**kwargs`. Args: *args (Any): `.format` arguments for `msg`. no_format (bool): If `no_format` is `True`, `.format` will not be called on the message. Useful if the output is user-provided or may otherwise contain an unexpected formatting string (e.g. "{}"). _tags (Dict[str, Any]): key-value pairs to display at the end of the message in square brackets. If a tag is set to `True`, it is printed without the value, the presence of the tag treated as a "flag". E.g. `_format_msg("hello", _tags=dict(from=mom, signed=True))` `hello [from=Mom, signed]` _numbered (Tuple[str, int, int]): `(brackets, i, n)` The `brackets` string is composed of two "bracket" characters, `i` is the index, `n` is the total. The string `{i}/{n}` surrounded by the "brackets" is prepended to the message. This is used to number steps in a procedure, with different brackets specifying different major tasks. E.g. `_format_msg("hello", _numbered=("[]", 0, 5))` `[0/5] hello` Returns: The formatted message. NTF=z [{}], r/rI z&We do not support printing kwargs yet.c,g|]}t|Sr)str.0rs r z_format_msg..s   a3q66   r) isinstancergr0itemscfr4r8formatjoinrE)r]rZr[r\r^r_tags_str tags_listkv numbering_strcharsinrFs r _format_msgrxsb#sFz#~>>F  I  + +199!$I::a#gk]*  U88BIIgnnTYYy=Q=Q.R.R$S$STT  #KE1aIIeAhQ&7#&=A&Fq&QRRUXXM  2 3&1 1zsz4:6:::XEE CABBB ,,C  3   C 99S>>rcb tjS#t$rYdSwxYw)z/More robust check for interactive terminal/tty.F)sys __stdin__isatty Exceptionrrr_isattyr~sA}##%%% uus  ..ceZdZUdZeed<eed<eed<dZeed<dZ dAd Z dBd Z e d Z d Ze d ZdZe dZdZdZdZ dCdedededefdZdZdededefdZdededefdZdedededefd Zdededefd!Zd"Zdededefd#Zdededefd$Zdededefd%Zdd&dedededefd'Z d(Z!dd&dedededefd)Z"d*Z#d+Z$ddd,dededededef d-Z%dDdefd.Z& dAdd/de'eded0edefd1Z(d2edededefd3Z)e*+d4fd5e,ed6efd7Z-d8d8dd9d:ededed;edZ/defd?Z0d@Z1dS)E _CliLoggeravSingleton class for CLI logging. Without calling 'cli_logger.configure', the CLILogger will default to 'record' style logging. Attributes: color_mode (str): Can be "true", "false", or "auto". Enables or disables `colorful`. If `color_mode` is "auto", is set to `not stdout.isatty()` indent_level (int): The current indentation level. All messages will be indented by prepending `" " * indent_level` vebosity (int): Output verbosity. Low verbosity will disable `verbose` and `very_verbose` messages. color_mode indent_level interactive)autorecordpretty_autodetected_cf_colormodecd|_d|_d|_d|_d|_d|_d|_tjj |_ | dS)NrFrr) r _verbosity_verbosity_overriden _color_mode _log_stylerrrmrrr set_formatrs rrz_CliLogger.__init__s\$)!!"   +-+*?' rNcL|sddlm}|}tj||_dS)Nr) LOGGER_FORMAT)!ray.autoscaler._private.constantsrlogging Formatter _formatter)r format_tmplrs rrz_CliLogger.set_formats7 ( G G G G G G'K!+K88rc||||||||||dS)z*Configures the logger according to values.N)_set_log_style_set_color_mode_set_verbosity detect_colors)r log_styler verbositys r configurez_CliLogger.configure%sh     * * *  !   , , ,     * * * rc|jSr)rrs rrz_CliLogger.log_style2s rc||_t|_|jdkrt|_dS|jdkrd|_|ddS|jdkr d|_dSdS)z(Configures interactivity and formatting.rrFfalserTN)lowerrr~rrrrrs rrz_CliLogger._set_log_style6s''))"99 ?f $ $!))DKKK _ ( (DK   ) ) ) ) ) _ ( (DKKK) (rc|jSr)rrs rrz_CliLogger.color_modeCs rc`||_|dSr)rrrrs rrz_CliLogger._set_color_modeGs*7799 rc>|jr|jS|jsdS|jS)Ni)rrrrs rrz_CliLogger.verbosityKs,  $ ? " 3rc"||_d|_dS)NT)rrrs rrz_CliLogger._set_verbositySs$(!!!rc*|jdkr?|jtjkr|jt_ntjt_dS|jdkrtdS|jdkrdStd|jz)zUpdate color output settings. Parse the `color_mode` string and optionally disable or force-enable color output (8-color ANSI if no terminal detected to be safe) in colorful. trueNrrzInvalid log color setting: )rrrmrrrrrErs rrz_CliLogger.detect_colorsWs ?f $ $.",>>#> !/ F ?g % % JJLLL F ?f $ $ F6HIIIrc0|ddS)zPrint a line feed.raNprintrs rnewlinez_CliLogger.newlinems 2rINFOTr] _level_str _linefeedendc |jrd|jz|z}np|dkrdSt}t jdd|d|d|id}||_|j|}|d vr tj }n tj }|s+| || dSd |i} t|fd |i| dS) zProxy for printing messages. Args: msg: Message to print. linefeed (bool): If `linefeed` is `False` no linefeed is printed at the end of the message. z raNclirrKrJ)r(levelpathnamerJr]r^exc_info)WARNINGERRORPANICrfile)rrrGrYr LogRecord levelnamerrnrzstderrstdoutwriteflushr) rr]rrrrendered_message caller_inforstreamr_s r_printz_CliLogger._printqs ; >#d&77#=  yy{{b  /11K& %Z0"8,F *F #55f==  6 6 6ZFFZF  LL) * * * LLNNN F 66V6v66666rc:|Gfdd}|S)z8Context manager that starts an indented block of output.c$eZdZfdZfdZdS)3_CliLogger.indented..IndentedContextManagerc(xjdz c_dSNrIr)r cli_loggers r __enter__z=_CliLogger.indented..IndentedContextManager.__enter__''1,''''rc(xjdzc_dSrrrtypevaluetbrs r__exit__z<_CliLogger.indented..IndentedContextManager.__exit__rrNr*r+r,rr)rsrIndentedContextManagerrsG - - - - - - - - - - - -rrr)rrrs @rindentedz_CliLogger.indentedsK  - - - - - - - - - -&%'''rr^r_c||jt|g|Ri||S)zwPrint a group title in a special color and start an indented block. For arguments, see `_format_msg`. )rrmr9rrr]r^r_s rgroupz_CliLogger.groups@  2==%%7777777}}rcF|Gfdd}|S)aContext manager for printing multi-line error messages. Displays a start sequence "!!! {optional message}" and a matching end sequence "!!!". The string "!!!" can be used as a "tombstone" for searching. For arguments, see `_format_msg`. c*eZdZfdZfdZdS)A_CliLogger.verbatim_error_ctx..VerbatimErorContextManagercbjtddzgRidS)Nz!!! z{}errorrmr5)rr^rr_r]s rrzK_CliLogger.verbatim_error_ctx..VerbatimErorContextManager.__enter__s=  4!7NtNNNvNNNNNrcbtddS)Nz!!!rrs rrzJ_CliLogger.verbatim_error_ctx..VerbatimErorContextManager.__exit__s'  00000rNr)r^rr_r]srVerbatimErorContextManagerrs^ O O O O O O O O 1 1 1 1 1 1 1rrr)rr]r^r_rrs ``` @rverbatim_error_ctxz_CliLogger.verbatim_error_ctxs_  1 1 1 1 1 1 1 1 1 1 1 1 1*)+++rkeyc|t|dztt|g|Ri|zdS)zDisplays a key-value pair with special formatting. Args: key: Label that is prepended to the message. For other arguments, see `_format_msg`. z: N)rrmr=rxr5)rrr]r^r_s r labeled_valuez_CliLogger.labeled_valuesS BJJsOOd*[-W-W-W-WPV-W-WWXXXXXrcF|jdkr|j|g|Rddi|dSdS)z[Prints a message if verbosity is not 0. For arguments, see `_format_msg`. rrVINFONrrrs rverbosez_CliLogger.verbosesF >A   DJs @T @ @ @g @ @ @ @ @ @  rcF|jdkr|j|g|Rddi|dSdS)zePrints a formatted warning if verbosity is not 0. For arguments, see `_format_msg`. rrVWARNN)r_warningrs rverbose_warningz_CliLogger.verbose_warningsF >A   DM# C C C C CF C C C C C  rcF|jdkr|j|g|Rddi|dSdS)zXLogs an error if verbosity is not 0. For arguments, see `_format_msg`. rrVERRN)r_errorrs r verbose_errorz_CliLogger.verbose_errorsF >A   DK @d @ @ @v @ @ @ @ @ @  rcF|jdkr|j|g|Rddi|dSdS)zOPrints if verbosity is > 1. For arguments, see `_format_msg`. rIrVVINFONrrs r very_verbosez_CliLogger.very_verbosesF >A   DJs AT A A Ah A& A A A A A  rc\|jt|g|Rddi|dS)zWPrints a formatted success message. For arguments, see `_format_msg`. rSUCCN)rrmr:rs rsuccessz_CliLogger.successs=  2<<$$ItIIII&IIIIIr)rc~|td|jt|g|Rd|i|dS)zWPrints a formatted warning message. For arguments, see `_format_msg`. NLog level not set.r)rErrmr<rr]rr^r_s rrz_CliLogger._warningsP  122 2 299S>>JDJJJZJ6JJJJJrc$|j|ddi|dS)NrWARN)rrr^r_s rwarningz_CliLogger.warning s$ t99&99999rc~|td|jt|g|Rd|i|dS)zUPrints a formatted error message. For arguments, see `_format_msg`. Nrr)rErrmr;rs rrz_CliLogger._errorsP  122 2 266#;;GGGG*GGGGGGrc$|j|ddi|dS)NrERRrrs rrz_CliLogger.errors$ T6e6v66666rc$|j|ddi|dS)Nrrrrs rpanicz_CliLogger.panics$ T8g888888rrrcR|t|g|Ri|||dS)zEPrints a message. For arguments, see `_format_msg`. rN)rrx)rr]rrr^r_s rrz_CliLogger.print s; K5d555f55*RU VVVVVrc,|j|g|Rd|i|dS)NrZr)rr]rZr^r_s rinfoz_CliLogger.info.s/ 3=d===)=f=====r)excrc||j|g|Rddi|||tj}|jrt}|d}||)zPrints an error and aborts execution. Print an error and throw an exception to terminate the program (the exception will not print a message). Nrrz!Exiting due to cli_logger.abort())rclickClickExceptionrSilentClickException)rr]rr^r_exc_clss rabortz_CliLogger.abort1si ? DK Ad A A Aw A& A A A ?I& ; +*G ;5Cgcllrvalcb|s,d}|jst}|j|g|Rd|i|dSdS)zHandle assertion without throwing a scary exception. Args: val: Value to check. For other arguments, see `_format_msg`. Nr)rAssertionErrorr )rrr]r^r_rs rdoassertz_CliLogger.doassertGs_ 6C; '$&& DJs 5T 5 5 5s 5f 5 5 5 5 5 6 6rrcxs separatorc@|d|DS)z.\s(;;;1s2771::;;;r)ro)rrrs r render_listz_CliLogger.render_listZs#~~;;;;;<< : C%),,..CC\W,,!%JC, IKK*4CC LLNNN"&C!#\^^,!'B!\11 $ &)Dj %!#t#&!G*-crc(C$)'r$F$F$F$F&)Bh %bb = = = = = JsOOO),,#)- RZ"P"PKE1a 3 "!i0022iikk$;;!Ciikk+%%C*$$Cw =GGCIIKK(($$[#66$$Z55  F[0E BBBu: Cv!    LLNNNCCC  |  KK % % %&L  sH9M M+*M+ct|}t|g|Ri|}|r|ds|dz }||dd} t j}|}| }n$#t$r| YnwxYw|S)zPrompt the user for some text input. Args: msg: The mesage to display to the user before the prompt. Returns: The string entered by the user. rreF)linefeedra) rmr7rxr#rrzr&r'rrGr/r)rr]r^r_r5rrFr8s rpromptz_CliLogger.prompts}}S)) &|EdEEEfEE  $CLL$6$6 $  #  $u 555 )$$&&C))++C))++CC     LLNNNNN  s!AB((C C c~tjtjdSr)rzrrrrs rrz_CliLogger.flushs0  rr)NNN)rTN)T)2r*r+r,__doc__rg__annotations__intboolVALID_LOG_STYLESrrrpropertyrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrmr4rrfloatr>rArrrrrrs,OOO3 #### 9999    X     X X)))JJJ,! 5757 5757 57  57575757n ( ( (SC,c,#,,,,,*YY3YsYcYYYYA3AsAcAAAADDDAASACAAAABBCB3BBBBJ3JsJcJJJJ@DKKKCKKKsKKKK:::>BHHH#HcHsHSHHHH777999! W W W  W W W  W  W W W W>>>>>>$(@DC=03:=PS,6D6s636#6666&;=((4..==d3i=C====&*FFF FF F  F  FUOFFFFFP#4rrc0eZdZdZdeffd ZddZxZS)r a;`ClickException` that does not print a message. Some of our tooling relies on catching ClickException in particular. However the default prints a message, which is undesirable since we expect our code to log errors manually using `cli_logger.error()` to allow for colors and other formatting. messagecXtt||dSr)superr r)rrK __class__s rrzSilentClickException.__init__s( "D))227;;;;;rNcdSrr)rrs rshowzSilentClickException.showr rr)r*r+r,rCrgrrP __classcell__)rNs@rr r sb<<<<<<<        rr z --log-style)case_sensitiverzIf 'pretty', outputs with formatting and color. If 'record', outputs record-style without formatting. 'auto' defaults to 'pretty', and disables pretty logging if stdin is *not* a TTY.)requiredrr1helpz --log-color)rrrzAUse color logging. Auto enables color logging if stdout is a TTY.z-vz --verboseT)r1countfreturncttD] }|tddddfd }|S)N)r log_colorrcLt||||i|Sr)rr)rrYrr^r_rVs rwrapperz*add_click_logging_options..wrapper4s0Y 7;;;q$!&!!!r)reversedCLICK_LOGGING_OPTIONSr)rVoptionr[s` radd_click_logging_optionsr_0sk011 F1II 1XX!%t""""""X" Nr)0rCrLrrSrzr) contextlibr functoolsrtypingrrrrr r r coloramarayr(r*r.r rrB colorful.corer0use_8_ansi_colorsModuleNotFoundErrorr3rminitrYrgrFrErxr~rr r rr^ChoicerGr]r_rrrrjs %%%%%%================  <7MMMMMMM: ,,,,,,C          -//CCC <^  E, &* OOO O OO S> O S#s]# O  OOOOx   SSSSSSSSl     5/   "Z\\ EL U\*5e L L L '   EL U\333E J J JQ EL{D===)0  h      s!A;;BB