s j"#dZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl Z ddl mZdd lmZdd lmZdd lmZd d lmZddlmZddZddZe GddZdS)z4Helper functions for writing to terminals and files.) annotations)SequenceN)final)Literal)TextIO)TerminalFormatter)Lexer) DiffLexer) PythonLexer) assert_never)wcswidthreturnintcFtjd\}}|dkrd}|S)N)P)fallback(r)shutilget_terminal_size)width_s JC:\Projects\Muse2API\.venv\Lib\site-packages\_pytest/_io/terminalwriter.pyget_terminal_widthrs.':::HE1 rzz LfilerboolctjddkrdStjddkrdStjdrdStjdrdSt|do6|o"tjd d kS) N PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)osenvirongethasattrr&)rs rshould_do_markupr-$s z~~k""c))t z~~k""c))u z~~j!!u z~~m$$thVDKKMMVbjnnV6L6LPV6VrcXeZdZedIiddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(ZdXdYd.ZedZd0Zejd[d2ZedZd3Z d\d8Z d]d^d>Z d?d@d_dCZ d?d@d`dDZ dadbdGZdcdHZdddedMZdfdQZdgdSZ dhdidWZd)S)jTerminalWriterblackredgreen yellow!blue"purple#cyan$white%BlackrRed)Green*Yellow+Blue,Purple-Cyan.White/boldrlightr blinkinvertNr TextIO | NonerNonecR| tj}t|drX|rDtjdkr4 ddl}||j}|Jn#t$rYnwxYw||_ t||_ d|_ d|_ d|_dS)Nr&win32rT)sysstdoutr,r&platformcolorama AnsiToWin32stream ImportError_filer- hasmarkup _current_line_terminal_widthcode_highlight)selfrr]s r__init__zTerminalWriter.__init__Ks <:D 4 " " (t{{}} (9P9P ( ++D118''''       )$//+/"sA'' A43A4rc:|j|jStSN)rdrrfs r fullwidthzTerminalWriter.fullwidth\s   +' '!###rvaluec||_dSri)rd)rfrls rrkzTerminalWriter.fullwidthbs$rc*t|jS)z;Return an estimate of the width so far in the current line.)rrcrjs rwidth_of_current_linez$TerminalWriter.width_of_current_linefs*+++rtextstrmarkuprc |D]}|jvrtd|jrGfd|D}|r%dd|D|zdz}|S)Nzunknown markup: c4g|]\}}|j|S) _esctable).0nameonrfs r z)TerminalWriter.markup..ps)MMMHD""M4>$'MMMrrYc3"K|] }d|dV dS)z[mNru)rwcods r z(TerminalWriter.markup..rs*==#~s~~~======r)rv ValueErrorrbitemsjoin)rfrprrrxescs` rrrzTerminalWriter.markupks > >D4>)) ! QMMMM MMMC Qww=======DyP rsepchartitle str | Nonerk int | Nonec ||j}tjdkr|dz}|Ft|t |z dz dt |zzd}||z}|d|d|}n||t |zz}t |t |z|kr||z }|j|fi|dS)NrXrr  )rkrZr\maxlenrstripline)rfrrrkrrNfillrs rsepzTerminalWriter.sepus  I <7 " " NI   YU+a/QW5EFJJAQ;D++U++T++DDi3w<<78D t99s7>>++,, , 9 9 GNN$$ $D $!!&!!!!!rFflushmsgrc |r_|ddd}d|vr||_n|xj|z c_|j|fi|}|||dSdS)N rr)rsplitrcrr write_raw)rfrrrr current_lines rwritezTerminalWriter.writes  -::dA..r2Ls{{%1""""l2""$+c,,V,,C NN3eN , , , , , - -rc |j|nR#t$rE|dd}|j|YnwxYw|r|dSdS)Nzunicode-escapeascii)rarUnicodeEncodeErrorencodedecoder)rfrrs rrzTerminalWriter.write_raws " J  S ! ! ! !! " " "**-..55g>>C J  S ! ! ! ! ! "   JJLLLLL  sA A,+A,rYsc L|j|fi||ddS)Nr)r)rfrrrs rrzTerminalWriter.lines4 1 4rc8|jdSri)rarrjs rrzTerminalWriter.flushs rrulines Sequence[str]indentsc|rPt|t|kr0tdt|dt|d|sdgt|z}d|}||}t ||dD]\}}|||zdS) 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 ()rYrF)strictN)rrr _highlight splitlineszipr)rfrrsource new_linesindentnew_lines r _write_sourcezTerminalWriter._write_sources  s7||s5zz11^W^^QTUZQ[Q[^^^  (dSZZ'G5!!OOF++6688 #GYu E E E ) ) FH IIfx' ( ( ( ( ) )rlexerLiteral['python', 'diff']r ct|dkrtS|dkrtSt|dS)Npythondiff)r r r )rfrs r_get_pygments_lexerz"TerminalWriter._get_pygments_lexers> H  == f__;;       rrc*ddlm}tjd}tjdd} t ||S#t jj$r}|d|d|d}~wt jj$r}|d |d |d}~wwxYw) Nr) UsageError PYTEST_THEMEPYTEST_THEME_MODEdark)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'.) _pytest.config.exceptionsrr)getenvrpygmentsutil ClassNotFound OptionError)rfrtheme theme_modees r_get_pygments_formatterz&TerminalWriter._get_pygments_formatters888888 .))Y2F;;  $ %@@@ @}*   *SESSS }(   *GQ[GGG  s#ABA&&B=B  BrrLiteral['diff', 'python']c|r|jr|js|S||}|}t j|||}|ddkr|ddkr |dd}d|z}|S)z5Highlight the given source if we have markup support.rrNr)rbrerrr highlight)rfrrpygments_lexerpygments_formatter highlighteds rrzTerminalWriter._highlights T^ 43F M11%88!99;;#- N$6  r?d " "vbzT'9'9%crc*K  +- rri)rrUrrVrr)rlrrrV)rprqrrrrrq)NN) rrqrrrkrrrrrrV)rrqrrrrrrrV)rrqrrrrV)rY)rrqrrrrrV)rrV)ru)rrrrrrV)rrrr )rr)r)rrqrrrrq)__name__ __module__ __qualname__dictrvrgpropertyrksetterrorrrrrrrrrrrrurrr/r/2sb Bbr  R  r Rbb BbrRrR b!"Q#$a%&a'(q)I.#####"$$$X$ %%%%,,,X,! $ #"#"#"#"#"J05 - - - - - -49 )))))&    (?Grr/r)rrrr)__doc__ __future__rcollections.abcrr)rrZtypingrrrrpygments.formatters.terminalrpygments.lexerr pygments.lexers.diffr pygments.lexers.pythonr compatr wcwidthrrr-r/rurrrse::""""""$$$$$$ :::::: ******......!!!!!!     OOOOOOOOOOr