[Fi"dZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z dd l m Z d d l m Z d efd Zde d efdZeGddZy)z4Helper functions for writing to terminals and files.N)final)Literal)Optional)Sequence)TextIO) assert_never)wcswidthreturncFtjd\}}|dkrd}|S)N)P)fallback(r)shutilget_terminal_size)width_s l/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthrs)'':HE1 rz Lfilectjjddk(rytjjddk(rytjjdrytjjdryt|dxr4|j xr"tjjd d k7S) N PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)osenvirongethasattrr )rs rshould_do_markupr's zz~~k"c) zz~~k"c) zz~~j! zz~~m$hVDKKMVbjjnnV6LPV6Vrc eZdZed@iddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(ZdHd*eed+d)fd,Zed+e fd-Z e jd.e d+d)fd/Z ed+e fd0Z d1e d2ed+e fd3Z dId4e d5ee d6ee d2ed+d)f d7Zd8d9d:e d;ed2ed+d)fd<ZdJd=e d2ed+d)fd>ZdKd?ZdLdAee dBee d+d)fdCZ dMdDe dEedFd+e fdGZy))NTerminalWriterblackredgreen yellow!blue"purple#cyan$white%BlackrRed)Green*Yellow+Blue,Purple-Cyan.White/boldr lightrblinkinvertNrr cD|tj}t|drG|jr7tjdk(r$ ddl}|j |j}|J||_ t||_ d|_ d|_ d|_y#t$rY8wxYw)Nr win32rT)sysstdoutr&r platformcolorama AnsiToWin32stream ImportError_filer' hasmarkup _current_line_terminal_widthcode_highlight)selfrrUs r__init__zTerminalWriter.__init__Ds <::D 4 "t{{}9P ( ++D188''' )$/.2"  sB BBcF|j |jStSN)r\rr^s r fullwidthzTerminalWriter.fullwidthUs#    +'' '!##rvaluec||_yra)r\)r^rds rrczTerminalWriter.fullwidth[s $rc,t|jS)z;Return an estimate of the width so far in the current line.)r r[rbs rwidth_of_current_linez$TerminalWriter.width_of_current_line_s**++rtextmarkupc |D]}||jvstd||jrR|jDcgc]\}}|s |j|}}}|rdj d|D|zdz}|Scc}}w)Nzunknown markup: rQc3&K|] }d|z yw)z[%smN).0cods r z(TerminalWriter.markup..ks?CzC/?s) _esctable ValueErrorrZitemsjoin)r^rhrinameonescs rrizTerminalWriter.markupds >D4>>) #3D8!<== > >>8> MHD""4>>$'MCMww?3??$FR Ns  BBsepchartitlercc | |j}tjdk(r|dz}|:t|t |z dz dt |zzd}||z}|d|d|}n||t |zz}t |t |j z|kr||j z }|j |fi|y)NrPr r )rcrRrTmaxlenrstripline)r^rxryrcriNfillrs rsepzTerminalWriter.sepns  I <<7 " NI   YU+a/QW5EFJAQ;DV1UG1TF+Di3w<78D t9s7>>+, , 9 GNN$ $D $!&!rF)flushmsgrc |ry|jddd}d|vr||_n|xj|z c_|j|fi|} |jj ||r|jyyy#t $r>|j djd}|jj |Y[wxYw)N r zunicode-escapeascii) rsplitr[rirYwriteUnicodeEncodeErrorencodedecoder)r^rrri current_lines rrzTerminalWriter.writes ::dA.r2Ls{%1"""l2"$++c,V,C &   % + & &jj!1299'B   % &s A==ACCsc L|j|fi||jdy)Nr)r)r^rris rrzTerminalWriter.lines! 1 4rc8|jjyra)rYrrbs rrzTerminalWriter.flushs rrllinesindentscP|r;t|t|k7r$tdt|dt|d|sdgt|z}dj|}|j|j }t ||D]\}}|j ||zy)aIWrite lines of source code possibly highlighted. Keeping this private for now because the API is clunky. We should discuss how to evolve the terminal writer so we can have more precise color support, for example being able to write part of a line in one color and the rest in another, and so on. zindents size (z") should have same size as lines ()rQrN)r}rrrt _highlight splitlineszipr)r^rrsource new_linesindentnew_lines r _write_sourcezTerminalWriter._write_sources s7|s5z1 W.PQTUZQ[P\\]^ dSZ'G5!OOF+668 #GY 7 ) FH IIfx' ( )rrlexer)diffpythonc ddlm}|r|jr |js|S ddlm}|dk(rddlm}n|dk(rddlm }n t|ddl m }dd l } |||tjd d tjd  }|ddk(r |ddk7r|d d}d|zS#|j j"$r0} |dj%tjd | d } ~ w|j j&$r0} |dj%tjd | d } ~ wwxYw#t($r|cYSwxYw)z5Highlight the given source if we have markup support.r) UsageError)TerminalFormatterr) PythonLexerr) DiffLexer) highlightNPYTEST_THEME_MODEdark PYTEST_THEME)bgstylerrrpzdPYTEST_THEME environment variable had an invalid value: '{}'. Only valid pygment styles are allowed.zrPYTEST_THEME_MODE environment variable had an invalid value: '{}'. The only allowed values are 'dark' and 'light'.)_pytest.config.exceptionsrrZr]pygments.formatters.terminalrpygments.lexers.pythonrpygments.lexers.diffrr pygmentsr pygments.utilr#getenvutil ClassNotFoundformat OptionErrorrX) r^rrrrLexerrr highlightedes rrzTerminalWriter._highlightsU 9T^^43F3FM-  F G&CU# *  #,G%99%8&A ii7$ r?d*vbzT/A"-cr"2K !;..==..  ==CV .1>   ==,,  FFLf "56G   9 M s63EAB00E +C44E+D;;E EEra)NN)rQ)r N)rl)r)__name__ __module__ __qualname__dictrqrrr_propertyintrcsetterrgstrboolrirrrrrrrrrlrrr)r)+s7        !"#$%&'()I.#Xf-##"$3$$ %s%t%%,s,,3$3 $#' #"#"}#"C= #"  #"  #"J052c$4)8C=)8C=)RV)&?G66")*:";6 6rr))__doc__r#rrRtypingrrrrrcompatr wcwidthr rrrr'r)rlrrrsa: ! C 6 d OOOr