o i0@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZmZdd lmZmZmZmZdd lmZ dd lm!Z!m"Z"m#Z#Gd d d e$Z%dS)N)Counter)contextmanager)Process)Any CollectionDictNoReturnOptionalUnioncastoverload)Literal)rowtable)COLORSICONSMESSAGES can_render)color) locale_escape supports_ansiwrapc@seZdZ           d@ded ed eed eed ed edededededefddZe de fddZ     dAde de dededeef ddZ e     dAde de dededeed f d!d"Ze     dBde de dededed#def d$d"Z     dAde de dededeeeefdeedeff d%d"Z     dAde de dededeef d&d'Z     dAde de dededeef d(d)Z         dCde de d*eeeefd+eeeefd,eededed edeefd-d.Z  /  dDded0eded,eefd1d2Zd3eeeffd4d5Zd3efd6d7ZedEdefd9d:ZdEdefd;d<Z    dFde de d=eedededeef d>d?ZdS)GPrinterTFNP⠙⠹⠸⠼⠴⠦⠧⠇⠏|/-\WASABIprettyno_printcolorsiconsline_max animationanimation_asciihide_animationignore_warnings env_prefix timestampc Cstd| d} td| d} t|_|o| |_||_to$| |_|p)| |_ | |_ ||_ t t |_t t|_| |_|rF|j||rN|j|t|rW||_dS||_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_countsrrr show_colorr%r&r"dictrr rr!r(updateranim)selfrrr r!r"r#r$r%r&r'r(env_log_friendly env_no_prettyr4B/home/ubuntu/.local/lib/python3.10/site-packages/wasabi/printer.py__init__s"      zPrinter.__init__returncCs|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,)r1r4r4r5countsDszPrinter.countstitletextshowspacedexitscC|j||tj|||dS)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_msgrGOODr1r:r;r<r=r>r4r4r5goodKz Printer.good)rFcCdSNr4rDr4r4r5fail` z Printer.fail)r TcCrGrHr4rDr4r4r5rIkrJcCr?)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@)rBrFAILrDr4r4r5rIvrFcCr?)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@)rBrWARNrDr4r4r5warnrFz Printer.warncCr?)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@)rBrINFOrDr4r4r5inforFz Printer.inforbg_coloriconc Csx|sdS|jr@|jtt||}|jtt||}|jtt|}|r0td||}|j r:t |||d}t |dd}|rLd|t |dd}|j r]t j d} d| |}| dusc|rhd |}|jsq|sqt|| durtjtj|js|r| dkrzt|ty} zd tdd } td| | d} ~ wwt| |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 {}  )rr getr strr!rr+stripr-_colorrr(datetimenowstrftimerprintsysstdoutflushstderr RuntimeError Exceptionjoin traceback format_stack SystemExitexit) r1r:r;rrPrQr=r<rr>r]etbr4r4r5r;sF          z Printer.text=charcCs|sdSt|dkrtd||jr`|jtt|}|r)td|| }|t t |j t|dd}|rAd|nd}t dj||d d d }t||j kr`|||j t|}|jre|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. Nr z9Divider chars need to be one character long. Received: {}rRz {} r9z {deco}{text}{deco})decor;T)bold)len ValueErrorr+rr!rXr rYrrZintroundr"r[rr_)r1r;rnr<rQrpr4r4r5dividers* " zPrinter.dividerdatacKs@|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)poprrvrr_)r1rwkwargsr:r;r4r4r5rs   z Printer.tablecKs&t|fi|}|jr |St|dS)zPrint a table row. data (Collection): The individual columns to format. kwargs: Row settings. See tables.row for details. N)rrr_)r1rwryr;r4r4r5r$s z Printer.row Loading...c cs|jr dVdS|jrt|dVdStjt|j|fd}|zdVWnt yC}z | tj d|d}~ww| tj dtjdS)N)targetargsrVz ) rr%r_r`rarbr_spinnerstartre terminatewrite)r1r;trkr4r4r5loading/s(      zPrinter.loadingcCs>t|jD]}tjd||tjt dqdS)Nz {} {}g?) itertoolscycler0r`rarr+rbtimesleep)r1r;rnr4r4r5r}Es   zPrinter._spinnerrAc Cs>|jr |tjkr d}|j|d7<|j|||||||dS)NFr )rrQr<r=r>)r&rrLr,r;)r1r:r;rAr<r=r>r4r4r5rBKs  zPrinter._get_msg) TFNNrrrFFrF)r9r9TFN)r9r9TFT) r9r9NNNFTFN)r9rmTN)rz)NFFN)__name__ __module__ __qualname__boolr rrtrYr6propertyrr8rrEr rrIrr rMrOr;rvrrrrrr}rBr4r4r4r5rs    0            = %  r)&r\rr)r`rrg collectionsr contextlibrmultiprocessingrtypingrrrrr r r r compatrtablesrrutilrrrrrr[rrrobjectrr4r4r4r5s    (