"Zj0ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZmZdd lmZmZmZmZdd lmZ dd lm!Z!m"Z"m#Z#Gd d e$Z%dS)N)Counter)contextmanager)Process)Any CollectionDictNoReturnOptionalUnioncastoverload)Literal)rowtable)COLORSICONSMESSAGES can_render)color) locale_escape supports_ansiwrapcpeZdZ d3ded ed eed eed ed edededededefdZe de fdZ d4de de dededeef dZ e d4de de dededeedf dZe d5de de dedededdef d Z d4de de dededeeeefdeedeff d!Z d4de de dededeef d"Z d4de de dededeef d#Z d6de de d$eeeefd%eeeefd&eededed edeefd'Z d7ded)eded&eefd*Zd+eeeffd,Zd+efd-Zed8defd/Zd8defd0Z d9de de d1eedededeef d2ZdS):PrinterTFNP⠙⠹⠸⠼⠴⠦⠧⠇⠏|/-\WASABIprettyno_printcolorsiconsline_max animationanimation_asciihide_animationignore_warnings env_prefix timestampc Ztjd| d} tjd| d} t|_|o| |_||_to| |_|p| |_ | |_ ||_ tt|_tt|_| |_|r|j||r|j|t'|r|n||_dS)aInitialize the command-line printer. pretty (bool): Pretty-print output (colors, icons). no_print (bool): Don't actually print, just return. colors (Optional[Dict]): Optional color values to add or overwrite, name mapped to value. icons (Optional[Dict]): Optional icons to add or overwrite. Name mapped to unicode icon. line_max (int): Maximum line length (for divider). animation (str): Steps of loading animation for loading() method. animation_ascii (str): Alternative animation for ASCII terminals. hide_animation (bool): Don't display animation, e.g. for logs. ignore_warnings (bool): Do not output messages of type MESSAGE.WARN. env_prefix (str): Prefix for environment variables, e.g. WASABI_LOG_FRIENDLY. timestamp (bool): Print a timestamp (default False). RETURNS (Printer): The initialized printer. z{}_LOG_FRIENDLYFz {}_NO_PRETTYN)osgetenvformatr_countsr r!r show_colorr'r(r$dictrr"rr#r*updateranim)selfr r!r"r#r$r%r&r'r(r)r*env_log_friendly env_no_prettys Z/Users/jameslopez/projects/MentorCore/.venv/lib/python3.11/site-packages/wasabi/printer.py__init__zPrinter.__init__s<9%6%=%=j%I%I5QQ ."7"7 "C"CUKK '  2]!2   '//B2B.B,@0@.  6ll %[[ "  ' K  v & & &  % J  e $ $ $!+I!6!6KIIO returnc|jS)zGet the counts of how often the special printers were fired, e.g. MESSAGES.GOOD. Can be used to print an overview like "X warnings". )r/)r4s r7countszPrinter.countsDs |r9titletextshowspacedexitscL|||tj|||S)aPrint a success message. title (Any): The main text to print. text (Any): Optional additional text to print. show (bool): Whether to print or not. Can be used to only output messages under certain condition, e.g. if --verbose flag is set. spaced (bool): Whether to add newlines around the output. exits (Optional[int]): Optional toggle to perform a system exit. styler@rArB)_get_msgrGOODr4r>r?r@rArBs r7goodz Printer.goodK."}} 4x}4e   r9)rFcdSNrHs r7failz Printer.fail`  r9)rTcdSrLrMrHs r7rNz Printer.failkrOr9cL|||tj|||S)aPrint an error message. title (Any): The main text to print. text (Any): Optional additional text to print. show (bool): Whether to print or not. Can be used to only output messages under certain condition, e.g. if --verbose flag is set. spaced (bool): Whether to add newlines around the output. exits (Optional[int]): Optional toggle to perform a system exit. rD)rFrFAILrHs r7rNz Printer.failvrJr9cL|||tj|||S)aPrint a warning message. title (Any): The main text to print. text (Any): Optional additional text to print. show (bool): Whether to print or not. Can be used to only output messages under certain condition, e.g. if --verbose flag is set. spaced (bool): Whether to add newlines around the output. exits (Optional[int]): Optional toggle to perform a system exit. rD)rFrWARNrHs r7warnz Printer.warnrJr9cL|||tj|||S)aPrint an informational message. title (Any): The main text to print. text (Any): Optional additional text to print. show (bool): Whether to print or not. Can be used to only output messages under certain condition, e.g. if --verbose flag is set. spaced (bool): Whether to add newlines around the output. exits (Optional[int]): Optional toggle to perform a system exit. rD)rFrINFOrHs r7infoz Printer.inforJr9rbg_coloriconc |sdS|jr|jtt||}|jtt||}|jtt|}|r5t d||}|j rt|||}t|d}|r%d|t|d}|j rGtj d} d| |}| |rd |}|js|st#|| t$jt$j|js|r| dkr t-|#t.$rV} d t3jdd } t7d| | d} ~ wwxYwt%j| |js|r|SdS) aPrint a message. title (Any): The main text to print. text (Any): Optional additional text to print. color (Optional[Union[str, int]]): Optional foreground color. bg_color (Optional[Union[str, int]]): Optional background color. icon (Optional[str]): Optional name of icon to add. spaced (bool): Whether to add newlines around the output. show (bool): Whether to print or not. Can be used to only output messages under certain condition, e.g. if --verbose flag is set. no_print (bool): Don't actually print, just return. exits (Optional[int]): Perform a system exit. Optional. N{} {})fgbgr)indentz{} {}z%Y-%m-%d %H:%M:%Sz{} {}z {}  )r r"getr strr#rr.stripr0_colorrr*datetimenowstrftimer!printsysstdoutflushstderr RuntimeError Exceptionjoin traceback format_stack SystemExitexit) r4r>r?rrYrZrAr@r!rBrgetbs r7r?z Printer.textsd2  F ; *KOODe$4$4e<>$sD//22D K%gnnT5&A&ABBHHJJ =u8<<<q)))E  AOOE4Q+?+?+?@@E > 0#''))223FGGCOOC//E   OOE**E} X  %LLL   J      J     } = =UaZZ=&u{{}}555 ===9#9#;#;CRC#@AAB$X__R%;%;<<<= HUOOO = H L  s*!H I+AI&&I+=charc |sdSt|dkr"td||jr|jt t|}|r5td|| }|tt|j t|z dz dz z}|rd|nd}td|| d }t||j kr|||j t|z zz}|jr|St|dS) agPrint a divider with a headline: ============================ Headline here =========================== text (str): Headline text. If empty, only the line is printed. char (str): Line character to repeat, e.g. =. show (bool): Whether to print or not. icon (Optional[str]): Optional icon to display with title. Nrz9Divider chars need to be one character long. Received: {}r\z {} r=z {deco}{text}{deco})decor?T)bold)len ValueErrorr.r r#rbr rcrrdintroundr$rer!ri)r4r?rxr@rZr{s r7dividerzPrinter.dividersa  F t99>>%vd||  ; A:>>$sD//22D I$W^^D$%?%?@@FFHH3udmc$ii&?AAAEFFJKD*.66==&&&BD&--4d-CC$D4yy4=((ddmc$ii&?@@ = K d r9datac |dd}t|fi|}|r|||jr|St |dS)zPrint data as a table. data (Union[Collection, Dict]): The data to render. Either a list of lists (one per row) or a dict for two-column tables. kwargs: Table settings. See tables.table for details. r>N)poprrr!ri)r4rkwargsr>r?s r7rz Printer.tablesd 7D))T$$V$$  LL    = K d r9c Pt|fi|}|jr|St|dS)zPrint a table row. data (Collection): The individual columns to format. kwargs: Row settings. See tables.row for details. N)rr!ri)r4rrr?s r7rz Printer.row$s7 4""6"" = K d r9 Loading...c#*K|jrdVdS|jrt|dVdStjt |j|f}| dVnG#t$r:}| tj d|d}~wwxYw| tj dtjdS)N)targetargsr`z ) r!r'rirjrkrlr_spinnerstartro terminatewrite)r4r?trus r7loadingzPrinter.loading/s  =  EEEEE    $KKK EEEEE J     t}D7;;;A GGIII        &&&   KKMMM J  [ ) ) ) J       s6A;; B?5B::B?ctj|jD]g}tjd||tjtj dhdS)Nz {} {}g?) itertoolscycler3rjrkrr.rltimesleep)r4r?rxs r7rzPrinter._spinnerEspODI..  D J  Y--dD99 : : : J      JsOOOO  r9rEc |jr|tjkrd}|j|xxdz cc<||||||||S)NFr)rrZr@rArB)r(rrTr/r?)r4r>r?rEr@rArBs r7rFzPrinter._get_msgKsi   EX]$:$:D Uq yy 4u5tFRW   r9) TFNNrrrFFrF)r=r=TFN)r=r=TFT) r=r=NNNFTFN)r=rwTN)r)NFFN)__name__ __module__ __qualname__boolr rrrcr8propertyrr<rrIr rrNr r rUrXr?rrrrrrrrFrMr9r7rrsq!% $6&$ %".L.L.L.L .L ~ .L  .L.L.L.L.L.L.L.L.L.L`X#         }     *-1         )*    X "&         w      X ,0         c4i()   sD(" #    .#         }     .#         }     .+/.2"#9999c3h( 9 5c?+ 9 sm 9999}9999z" #### # sm ####J % D 01         C^*S $#    }      }      r9r)&rfrr,rjrrq collectionsr contextlibrmultiprocessingrtypingrrrr r r r r compatrtablesrrutilrrrrrrerrrobjectrrMr9r7rso %%%%%%######SSSSSSSSSSSSSSSSSSSS555555555555!!!!!!4444444444F F F F F fF F F F F r9