j0ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZmZdd lmZmZmZmZdd lmZ dd lm!Z!m"Z"m#Z#Gd d e$Z%y)N)Counter)contextmanager)Process)Any CollectionDictNoReturnOptionalUnioncastoverload)Literal)rowtable)COLORSICONSMESSAGES can_render)color) locale_escape supports_ansiwrapceZdZ d*dededeedeedededed ed ed ed efd Ze de fdZ d+de de dededeef dZ e d+de de dededeedf dZe d,de de dedededdef dZ d+de de dededeeeefdeedeff dZ d+de de dededeef dZ d+de de dededeef dZ d-de de deeeefdeeeefdeededededeefd Z d.ded!ededeefd"Zd#eeeffd$Zd#efd%Zed/defd&Zd/defd'Z d0de de d(eedededeef d)Zy)1PrinterNprettyno_printcolorsiconsline_max animationanimation_asciihide_animationignore_warnings env_prefix timestampc <tjdj| d} tjdj| d} t|_|xr| |_||_txr| |_|xs| |_ | |_ ||_ tt|_tt|_| |_|r|jj%||r|j j%|t'|r||_y||_y)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_countsrrr show_colorr#r$r dictrrrrr&updateranim)selfrrrrr r!r"r#r$r%r&env_log_friendly env_no_prettys X/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/wasabi/printer.py__init__zPrinter.__init__s<99%6%=%=j%I5Q ."7"7 "CUK '  2]!2   '/B2B.B,@0@.  6l %[ "  KK  v &  JJ  e $!+I!6I O 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+)r0s r3countszPrinter.countsDs ||r5titletextshowspacedexitscL|j||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;r<r=)_get_msgrGOODr0r9r:r;r<r=s r3goodz Printer.goodK+"}} 4x}}4e  r5)rFcyNrCs r3failz Printer.fail` r5)rTcyrGrHrCs r3rIz Printer.failkrJr5cL|j||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. r?)rArFAILrCs r3rIz Printer.failvrEr5cL|j||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. r?)rArWARNrCs r3warnz Printer.warnrEr5cL|j||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. r?)rArINFOrCs r3infoz Printer.inforEr5rbg_coloriconc |sy|jr|jjtt||}|jjtt||}|j jtt|}|r)t dj||j}|jrt|||}t|d}|rdj|t|d}|jr?tjjjd} dj| |}| |rd j|}|j s |s t#|| ~t$j&j)t$j*j)|j s|r| dk7r t-|jt%j8| |j s|r|Sy#t.$rF} d j1t3j4dd } t7dj| | d} ~ wwxYw) 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 {}  )rrgetr strrrr*stripr,_colorrr&datetimenowstrftimerprintsysstdoutflushstderr RuntimeError Exceptionjoin traceback format_stack SystemExitexit) r0r9r:rrTrUr<r;rr=rbetbs r3r:z Printer.texts2  ;;KKOODe$4e>$sD/2D%gnnT5&ABHHJu8<q)E OOE4Q+?@E >>##'')223FGCOOC/E  OOE*E}}X %L   JJ    JJ   }}UaZ=&u{{}55 HHUO ==HL% !=9#9#9#;CR#@AB$X__R%;<<=s H IAIIcharc |syt|dk7rtdj||jr|jj t t|}|r)tdj||j}|tt|jt|z dz dz z}|rdj|nd}tdj|| d }t||jkr|||jt|z zz}|jr|St|y) 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: {}rWz {} z {deco}{text}{deco})decor:T)bold)len ValueErrorr*rrr]r r^rr_introundr r`rrd)r0r:rrr;rUrvs r3dividerzPrinter.dividers  t9>%vd|  ;;::>>$sD/2D$W^^D$%?@FFH3udmmc$i&?AAEFJKD*.6==&BD&--4d-C$D4y4==(ddmmc$i&?@@ ==K d r5datac |jdd}t|fi|}|r|j||jr|St |y)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. r9N)poprr|rrd)r0r}kwargsr9r:s r3rz Printer.tablesE 7D)T$V$  LL  ==K d r5c Nt|fi|}|jr|St|y)zPrint a table row. data (Collection): The individual columns to format. kwargs: Row settings. See tables.row for details. N)rrrd)r0r}rr:s r3rz Printer.row$s& 4"6" ==K d r5c#K|jrdy|jrt|dytjj t |j|f}|j d|jtjjdtjj y#t$r6}|jtjjd|d}~wwxYww)N)targetargsr[z ) rr#rdrerfrgr_spinnerstartrj terminatewrite)r0r:trps r3loadingzPrinter.loading/s ==   $K  JJ   t}}D7;A GGI  KKM JJ  [ ) JJ        &  s+A4D 7C ;AD  D1DDD ctj|jD]d}tjj dj ||tjjtjdfy)Nz {} {}g?) itertoolscycler/rerfrr*rgtimesleep)r0r:rrs r3rzPrinter._spinnerEsXOODII. D JJ  Y--dD9 : JJ    JJsO r5r@c |jr|tjk(rd}|j|xxdz cc<|j |||||||S)NFr)rrUr;r<r=)r$rrOr+r:)r0r9r:r@r;r<r=s r3rAzPrinter._get_msgKsW   EX]]$:D Uq yy 4u5tFRW  r5) TFNNPu⠙⠹⠸⠼⠴⠦⠧⠇⠏z|/-\FFWASABIF)ruruTFN)ruruTFT) ruruNNNFTFN)ru=TN)z Loading...)NFFN)__name__ __module__ __qualname__boolr rrzr^r4propertyrr8rrDr rrIr r rPrSr:r|rrrrrrrArHr5r3rrsJ!% $6&$ %".L.L.L .L ~ .L  .L.L.L.L.L.L.L`#        }  *-1        )*   "&        w     ,0        c4i()   sD(" # .#        }  .#        }  .+/.2"#999c3h( 9 5c?+ 9 sm 9999}9z" ### # sm #J % D 01   C*S $#   }      } r5r)&rarr(rerrl collectionsr contextlibrmultiprocessingrtypingrrrr r r r r compatrtablesrrutilrrrrrr`rrrobjectrrHr5r3rsL %#SSS55!44F fF r5