o `۷i"z(_ColorfulMock.__init__..)identitycolorful colormode NO_COLORS ANSI_8_COLORSselfrrr__init__ s  z_ColorfulMock.__init__cCdSr rrrrrdisable*z_ColorfulMock.disableccsGddd}|VdS)Nc@seZdZddZdS)z/_ColorfulMock.with_style..IdentityClasscSsddS)NcSr r r)yrrrr1rzM_ColorfulMock.with_style..IdentityClass.__getattr__..rrnamerrr __getattr__0sz;_ColorfulMock.with_style..IdentityClass.__getattr__N)__name__ __module__ __qualname__r!rrrr IdentityClass/s r%r)rrr%rrr with_style-s z_ColorfulMock.with_stylecCs|dkr|jS|jS)Nr&)r&rrrrrr!5sz_ColorfulMock.__getattr__N)r"r#r$rrrr&r!rrrrr s    r )ColorfulStringc@s eZdZdS)r'N)r"r#r$rrrrr'Fsr'c@seZdZgdZddZdS)_ColorfulProxy) rresetbolditalic underlineddimmed dodgerBlue limeGreenredorangeskyBluemagentayellowcCs0tt|}t|r|tjvrtd|d|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_cfcallabler(_proxy_allowlist ValueError)rr resrrrr!ds  z_ColorfulProxy.__getattr__N)r"r#r$r8r!rrrrr(Qs r(F)stripcCsNt}|}d}|jjtkr|j}|d7}|jjtks|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)framecallerlevelsrrr_external_caller_infots  rL) no_format_tags _numberedmsgargsrMrNrOkwargsc Os t|ts t|tryd}|durCg}|D]\}} | dur$||g7}q| dur)q||d| g7}q|rCttdd|}d} |dure|\} } } t| dt| d t| | d d } |rm| ||S| |j|i||S|rt d |g|}d 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/r< z&We do not support printing kwargs yet.cSsg|]}t|qSr)str.0rrrr sz_format_msg..) isinstancerXr'itemscfr)r-formatjoinr9)rPrMrNrOrQrRtags_str tags_listkv numbering_strcharsinr:rrr _format_msgs21  .   ricCs"ztjWStyYdSw)z/More robust check for interactive terminal/tty.F)sys __stdin__isatty Exceptionrrrr_isattys   rnc@seZdZUdZeed<eed<eed<dZeed<ddZ dcd d Z ddd d Z e ddZ ddZe ddZddZe ddZddZddZddZ   ded ed!ed"ed#efd$d%Zd&d'Zd ed(ed)efd*d+Zd ed(ed)efd,d-Zd.ed ed(ed)efd/d0Zd ed(ed)efd1d2Zd3d4Zd ed(ed)efd5d6Zd ed(ed)efd7d8Zd ed(ed)efd9d:Zd d;d ed(ed!ed)efdd?Z!d d;d ed(ed!ed)efd@dAZ"dBdCZ#dDdEZ$dd dFd ed(ed!ed#ed)ef dGdHZ%dfd efdIdJZ& dcd dKd e'ed(edLed)efdMdNZ(dOed ed(ed)efdPdQZ)e*+dRfdSe,edTefdUdVZ-dWdWd dXdYed ed(edZed[ed\e'e.d)efd]d^Z/d efd_d`Z0dadbZ1d S)g _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_colormodecCs@d|_d|_d|_d|_d|_d|_d|_tjj |_ | dS)NrFrsrt) rq _verbosity_verbosity_overriden _color_mode _log_stylerurrr^rrrv set_formatrrrrrs  z_CliLogger.__init__NcCs$|s ddlm}|}t||_dS)Nr) LOGGER_FORMAT)!ray.autoscaler._private.constantsr|logging Formatter _formatter)r format_tmplr|rrrr{s z_CliLogger.set_formatcCsB|dur |||dur|||dur|||dS)z*Configures the logger according to values.N)_set_log_style_set_color_mode_set_verbosity detect_colors)r log_stylerp verbosityrrr configure%s    z_CliLogger.configurecC|jSr )rzrrrrr2z_CliLogger.log_stylecCs^||_t|_|jdkrt|_dS|jdkr#d|_|ddS|jdkr-d|_dSdS)z(Configures interactivity and formatting.rsrtFfalseruTN)lowerrzrnrrrurrrrrrr6s      z_CliLogger._set_log_stylecCrr )ryrrrrrpCrz_CliLogger.color_modecCs||_|dSr )rryrrrrrrGs  z_CliLogger._set_color_modecCs|jr|jS|js dS|jS)Ni)rxrwrurrrrrKs z_CliLogger.verbositycCs||_d|_dS)NT)rwrxrrrrrSs z_CliLogger._set_verbositycCs`|jdkr|jtjkr|jt_dStjt_dS|jdkr"tdS|jdkr)dStd|j)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. trueNrrszInvalid log color setting: )rprvr^rrrrr9rrrrrWs    z_CliLogger.detect_colorscCs|ddS)zPrint a line feed.rSNprintrrrrnewlinemsz_CliLogger.newlineINFOTrP _level_str _linefeedendc Cs|jr d|j|}n$|dkrdSt}tjdd|d|d|idd}||_|j|}|d vr7t j }nt j }|sG| || 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 rSNclirr>r=)r levelpathnamer=rPrQexc_info)WARNINGERRORPANICrfile)rurqr;rLr~ LogRecord levelnamerr_rjstderrstdoutwriteflushr) rrPrrrrendered_message caller_infortstreamrRrrr_printqs2   z_CliLogger._printcs|Gfddd}|S)z8Context manager that starts an indented block of output.cs$eZdZfddZfddZdS)z3_CliLogger.indented..IndentedContextManagercsjd7_dSNr<rqr cli_loggerrr __enter__z=_CliLogger.indented..IndentedContextManager.__enter__csjd8_dSrrrtypevaluetbrrr__exit__rz<_CliLogger.indented..IndentedContextManager.__exit__Nr"r#r$rrrrrrIndentedContextManagers rr)rrrrrindentedsz_CliLogger.indentedrQrRcOs&|jt|g|Ri||S)zwPrint a group title in a special color and start an indented block. For arguments, see `_format_msg`. )rr^r.rrrPrQrRrrrgroupsz_CliLogger.groupcs"|Gfddd}|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`. cs*eZdZfddZfddZdS)zA_CliLogger.verbatim_error_ctx..VerbatimErorContextManagercs(jtddgRidS)Nz!!! z{}errorr^r*rrQrrRrPrrrs(zK_CliLogger.verbatim_error_ctx..VerbatimErorContextManager.__enter__cstddS)Nz!!!rrrrrrzJ_CliLogger.verbatim_error_ctx..VerbatimErorContextManager.__exit__NrrrrrVerbatimErorContextManagersrr)rrPrQrRrrrrverbatim_error_ctxs z_CliLogger.verbatim_error_ctxkeycOs4|t|dtt|g|Ri|dS)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)rr^r2rir*)rrrPrQrRrrr labeled_values4z_CliLogger.labeled_valuecO.|jdkr|j|g|Rddi|dSdS)z[Prints a message if verbosity is not 0. For arguments, see `_format_msg`. rrVINFONrrrrrrverbose  z_CliLogger.verbosecOr)zePrints a formatted warning if verbosity is not 0. For arguments, see `_format_msg`. rrVWARNN)r_warningrrrrverbose_warningrz_CliLogger.verbose_warningcOr)zXLogs an error if verbosity is not 0. For arguments, see `_format_msg`. rrVERRN)r_errorrrrr verbose_errorrz_CliLogger.verbose_errorcOr)zOPrints if verbosity is > 1. For arguments, see `_format_msg`. r<rVVINFONrrrrr very_verboserz_CliLogger.very_verbosecOs&|jt|g|Rddi|dS)zWPrints a formatted success message. For arguments, see `_format_msg`. rSUCCN)rr^r/rrrrsuccesss&z_CliLogger.success)rcO6|durtd|jt|g|Rd|i|dS)zWPrints a formatted warning message. For arguments, see `_format_msg`. NLog level not set.r)r9rr^r1rrPrrQrRrrrr&z_CliLogger._warningcO|j|ddi|dS)NrWARN)rrrQrRrrrwarning z_CliLogger.warningcOr)zUPrints a formatted error message. For arguments, see `_format_msg`. Nrr)r9rr^r0rrrrrrz_CliLogger._errorcOr)NrERRrrrrrrrz_CliLogger.errorcOr)Nrrrrrrrpanicrz_CliLogger.panicrrcOs&|jt|g|Ri|||ddS)zEPrints a message. For arguments, see `_format_msg`. rN)rri)rrPrrrQrRrrrr s& z_CliLogger.printcOs |j|g|Rd|i|dS)NrMr)rrPrMrQrRrrrinfo.s z_CliLogger.info)excrcOsT|dur|j|g|Rddi||dur|tj}|jr t}|dur&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())rclickClickExceptionruSilentClickException)rrPrrQrRexc_clsrrrabort1sz_CliLogger.abortvalcOs8|sd}|js t}|j|g|Rd|i|dSdS)zHandle assertion without throwing a scary exception. Args: val: Value to check. For other arguments, see `_format_msg`. Nr)ruAssertionErrorr)rrrPrQrRrrrrdoassertGs  z_CliLogger.doassertrUxs separatorcCs|dd|DS)z.)r`)rrrrrr render_listZsz_CliLogger.render_listF)_abort_default _timeout_syesrrrc Os|}|} |js|s|dtd| rd} nd} td| dd} t|g|Ri|} | r:|ds:| d7} t| dd } | | }|rW| |d t d d S|j |d dd}gd}gd}z |durtt j }nrt jdkrt}d} t||kr|d}n;trt}|dvr||d}n&|dkr|r|dd }tddd dn||}t|dd dntdqntt j ggg|\}}}|s|d}nt j }|}|dkr| }n7|}||vrd }n,||vrd }n$d| }|d|t|||d||d|j || d dqjWnty9|| }Ynw|sI|rI| dtd|S)aDisplay a confirmation dialog. Valid answers are "y/yes/true/1" and "n/no/false/0". Args: yes: If `yes` is `True` the dialog will default to "yes" and continue without waiting for user input. _abort (bool): If `_abort` is `True`, "no" means aborting the program. _default (bool): The default action to take if the user just presses enter with no input. _timeout_s (float): If user has no input within _timeout_s seconds, the default action is taken. None means no timeout. z^This command requires user confirmation. When running non-interactively, supply --yes to skip.z&Non-interactive confirm without --yes.zY/nzy/Nz Confirm [z]:rW zy z[automatic, due to --yes]TF)rN)rrr1)rhnor0r rS)r z )rrg?z'{}Invalid answer: {}. Expected {} or {}rVz Exiting...z:Exiting due to the response to confirm(should_abort=True).)rrrr9r^r,riendswithlensplitrr-rjstdinreadlineplatformtimermsvcrtkbhitgetwchrsleepselectrr;r*rKeyboardInterruptr)rrrPrrrrQrR should_abortdefaultyn_str confirm_strrmsg_len complete_strr: yes_answers no_answersans start_timechready_indentrrrconfirm^s          ;  z_CliLogger.confirmcOst|}t|g|Ri|}|r|ds|d7}|j|ddd}ztj}|}| }W|St yB| Y|Sw)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. rrWF)linefeedrS) r^r,rirrrjrrrr;rr)rrPrQrRrrr:r rrrprompts    z_CliLogger.promptcCstjtjdSr )rjrrrrrrrrs z_CliLogger.flushr )NNN)rTN)T)2r"r#r$__doc__rX__annotations__intboolVALID_LOG_STYLESrr{rpropertyrrrprrrrrrrrrrrrrrrrrrrrrrrrrrr^r)rrfloatrrrrrrrros       7            rocs0eZdZdZdeffdd ZdddZZS) ra;`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. messagecstt||dSr )superrr)rr __class__rrrrzSilentClickException.__init__NcCrr r)rrrrrshowrzSilentClickException.showr )r"r#r$rrXrr __classcell__rrrrrs rz --log-style)case_sensitiverszIf '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.)requiredrrhelpz --log-color)rsrrzAUse color logging. Auto enables color logging if stdout is a TTY.z-vz --verboseT)rcountfreturncs8ttD]}|qtddddfdd }|S)N)r log_colorrcst||||i|Sr )rr)rr%rrQrRr#rrwrapper4sz*add_click_logging_options..wrapper)reversedCLICK_LOGGING_OPTIONSr)r#optionr'rr&radd_click_logging_options0s  r+)0rr?r~rFrjr contextlibr functoolsrtypingrrrrrr rcoloramarayrrrr rr6 colorful.corer'use_8_ansi_colorsModuleNotFoundErrorr(r^initrLrXrrrirnrorrrr*Choicerr)r+rrrrs              \