gg$dZddlmZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z d d l m Z d d lmZe r dd lmZddlmZddZddZeGddZy)z4Helper functions for writing to terminals and files.) annotationsN)final)Literal)Sequence)TextIO) TYPE_CHECKING) assert_never)wcswidth) Formatter)LexercFtjd\}}|dkrd}|S)N)P)fallback(r)shutilget_terminal_size)width_s O/var/www/openai/venv/lib/python3.12/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthrs)'':HE1 rz Lctjjddk(rytjjddk(rytjjdrytjjdryt|dxr4|j xr"tjjd d k7S) N PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)osenvirongethasattrr!)files rshould_do_markupr)$s zz~~k"c) zz~~k"c) zz~~j! zz~~m$hVDKKMVbjjnnV6LPV6Vrc\eZdZed5iddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(Zd:d;d*Zedd.Z d? d@d/Z d0d1dAd2Z dBdCd3Z dDd4ZdEdFd6ZdGd7ZdHd8Z dI dJd9Zy))KTerminalWriterblackredgreen yellow!blue"purple#cyan$white%BlackrRed)Green*Yellow+Blue,Purple-Cyan.White/boldr lightr blinkinvertNcD|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)selfr(rWs r__init__zTerminalWriter.__init__Ks <::D 4 "t{{}9P ( ++D188''' )$/+/"  sB BBcF|j |jStSN)r^rr`s r fullwidthzTerminalWriter.fullwidth\s#    +'' '!##rc||_yrc)r^)r`values rrezTerminalWriter.fullwidthbs $rc,t|jS)z;Return an estimate of the width so far in the current line.)r r]rds rwidth_of_current_linez$TerminalWriter.width_of_current_linefs**++rc |D]}||jvstd||jrR|jDcgc]\}}|s |j|}}}|rdj d|D|zdz}|Scc}}w)Nzunknown markup: rSc3(K|] }d|d yw)z[mN).0cods r z(TerminalWriter.markup..rs=#se1~s) _esctable ValueErrorr\itemsjoin)r`textmarkupnameonescs rrwzTerminalWriter.markupksD4>>) #3D8!<== >>8> MHD""4>>$'CMww===DyP Ns  BBc | |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)NrRr r  )rerTrVmaxlenrstripline)r`sepchartitlererwNfillrs rsepzTerminalWriter.sepus  I <<7 " NI   YU+a/QW5EFJAQ;DV1UG1TF+Di3w<78D t9s7>>+, , 9 GNN$ $D $!&!rF)flushc |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]rwr[writeUnicodeEncodeErrorencodedecoder)r`msgrrw current_lines rrzTerminalWriter.writes ::dA.r2Ls{%1"""l2"$++c,V,C &   % + & &jj!1299'B   % &s A==ACCc L|j|fi||jdy)Nr)r)r`srws rrzTerminalWriter.lines! 1 4rc8|jjyrc)r[rrds rrzTerminalWriter.flushs rrmcP|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 ()rSrN)r~rsru _highlight splitlineszipr)r`linesindentssource new_linesindentnew_lines r _write_sourcezTerminalWriter._write_sources s7|s5z1 W.PQTUZQ[P\\]^ dSZ'G5!OOF+668 #GY 7 FH IIfx' (!8rc |dk(r ddlm}|S|dk(r ddlm}|St |y#t $rYywxYw)Npythonr) PythonLexerdiff) DiffLexer)pygments.lexers.pythonrpygments.lexers.diffrr ModuleNotFoundError)r`lexerrrs r_get_pygments_lexerz"TerminalWriter._get_pygments_lexersF  >"}$&: {"U#"  s22 2 >>c\ ddl}ddlm}t j d}t j dd} ddlm}|||S#t$rYywxYw#|jj$r}|d|d |d}~w|jj$r}|d |d |d}~wwxYw) Nr) UsageError PYTEST_THEMEPYTEST_THEME_MODEdark)TerminalFormatter)bgstylez9PYTEST_THEME environment variable has an invalid value: 'zC'. Hint: See available pygments styles with `pygmentize -L styles`.z>PYTEST_THEME_MODE environment variable has an invalid value: 'z7'. The allowed values are 'dark' (default) and 'light'.) pygments.utilr_pytest.config.exceptionsrr$getenvpygments.formatters.terminalrutil ClassNotFound OptionError)r`pygmentsrtheme theme_moderes r_get_pygments_formatterz&TerminalWriter._get_pygments_formatters   9 .)YY2F;   F$ %@ @#  }}** KE7SSS }}(( PQ[P\]GG  s4AA AAB+0 A==B+ B&&B+c|r|jr |js|S|j|}||S|j}||Sddlm}||||}|ddk(r |ddk7r|dd}d|z}|S)z5Highlight the given source if we have markup support.Nr) highlightrrrq)r\r_rrrr)r`rrpygments_lexerpygments_formatterr highlighteds rrzTerminalWriter._highlightsT^^43F3FM11%8  !M!99;  %M&$V^=OP  r?d "vbzT'9%cr*K  +- rrc)r(z TextIO | NonereturnNonerint)rgrrr)rvstrrwboolrr)NN) rrrz str | Nonerez int | Nonerwrrr)rrrrrwrrr)rS)rrrwrrr)rr)rm)r Sequence[str]rrrr)rzLiteral['python', 'diff']rz Lexer | None)rzFormatter | None)r)rrrzLiteral['diff', 'python']rr)__name__ __module__ __qualname__dictrrrapropertyresetterrirwrrrrrrrrrmrrr+r+2s        !"#$%&'()I.#"$$ %%,,! $ #"#"#" #"  #"  #"J052)$ 8?G"; rr+r)r(rrr)__doc__ __future__rr$rrTtypingrrrrrcompatr wcwidthr pygments.formatterr pygments.lexerrrr)r+rmrrrsZ:"  !,$  ___r