U 42ig$@sdZddlmZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z d d l m Z d d lmZe rdd lmZddlmZddddZdddddZeGdddZdS)z4Helper functions for writing to terminals and files.) annotationsN)final)Literal)Sequence)TextIO) TYPE_CHECKING) assert_never)wcswidth) Formatter)LexerintreturncCs tjdd\}}|dkrd}|S)N)P)fallback(r)shutilget_terminal_size)width_r>/tmp/pip-unpacked-wheel-7n2p7kht/_pytest/_io/terminalwriter.pyget_terminal_widthsrrboolfilercCsjtjddkrdStjddkr(dStjdr8dStjdrHdSt|doh|ohtjd d kS) NZ PY_COLORS1T0FZNO_COLORZ FORCE_COLORisattyTERMZdumb)osenvirongethasattrr!)rrrrshould_do_markup$s   r'c@s4eZdZeddddddddd d d d d ddddddddZdKdddddZeddddZejddd d!dZeddd"d#Z d$d%d$d&d'd(Z dLd$d)d*d%dd+d,d-Z d.d/d$d%d%dd0d1d2Z dMd$d%dd4d5d6Z ddd7d8ZdNd:d:dd;dd?d@dAdBZdCddDdEZdOd$dGd$dHdIdJZdS)PTerminalWriter !"#$%r)*+,-./r r)ZblackZredZgreenZyellowZblueZpurpleZcyanZwhiteZBlackZRedZGreenZYellowZBlueZPurpleZCyanWhiteZboldlightZblinkinvertNz TextIO | NoneNonercCs|dkrtj}t|drd|rdtjdkrdz ddl}Wntk rJYnX||j}|dk sdt ||_ t ||_ d|_ d|_d|_dS)Nr!win32rT)sysstdoutr&r!platformcolorama ImportError AnsiToWin32streamAssertionError_filer' hasmarkup _current_line_terminal_widthcode_highlight)selfrrCrrr__init__Ks    zTerminalWriter.__init__rrcCs|jdk r|jStSN)rKrrMrrr fullwidth\s zTerminalWriter.fullwidth)valuercCs ||_dSrO)rK)rMrRrrrrQbscCs t|jS)z;Return an estimate of the width so far in the current line.)r rJrPrrrwidth_of_current_linefsz$TerminalWriter.width_of_current_linestrr)textmarkuprc sb|D]}|jkrtd|qjr^fdd|D}|r^ddd|D|d}|S)Nzunknown markup: csg|]\}}|rj|qSr) _esctable).0nameonrPrr psz)TerminalWriter.markup..r?css|]}d|dVqdS)z[mNr)rXZcodrrr rsz(TerminalWriter.markup..)rW ValueErrorrIitemsjoin)rMrUrVrYescrrPrrVks zTerminalWriter.markupz str | Nonez int | None)sepchartitlerQrVrcKs|dkr|j}tjdkr |d8}|dk rht|t|ddt|d}||}|d|d|}n||t|}t|t||kr||7}|j|f|dS)Nr>r r )rQr@rBmaxlenrstripline)rMrcrdrQrVNfillrirrrsepus " zTerminalWriter.sepF)flush)msgrmrVrcKs|r|ddd}d|kr$||_n|j|7_|j|f|}z|j|Wn0tk r|dd}|j|YnX|r|dS)N r zunicode-escapeascii) rsplitrJrVrHwriteUnicodeEncodeErrorencodedecoderm)rMrnrmrV current_linerrrrsszTerminalWriter.writer?)srVrcKs|j|f||ddS)Nro)rs)rMrxrVrrrriszTerminalWriter.linecCs|jdSrO)rHrmrPrrrrmszTerminalWriter.flushrz Sequence[str])linesindentsrcCs|r2t|t|kr2tdt|dt|d|sDdgt|}d|}||}t||D]\}}|||qfdS)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 ()r?roN)rgr_ra _highlight splitlineszipri)rMryrzsource new_linesindentnew_linerrr _write_sources zTerminalWriter._write_sourcezLiteral[('python', 'diff')]z Lexer | None)lexerrcCs`zD|dkrddlm}|WS|dkr:ddlm}|WSt|Wntk rZYdSXdS)Npythonr) PythonLexerZdiff) DiffLexer)Zpygments.lexers.pythonrZpygments.lexers.diffrr ModuleNotFoundError)rMrrrrrr_get_pygments_lexers   z"TerminalWriter._get_pygments_lexerzFormatter | Nonec Csz ddl}Wntk r"YdSXddlm}td}tdd}zddlm}|||dWS|jj k r}z|d|d |W5d}~XYn8|jj k r}z|d |d |W5d}~XYnXdS) Nr) UsageErrorZ PYTEST_THEMEZPYTEST_THEME_MODEZdark)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'.) Z pygments.utilrZ_pytest.config.exceptionsrr#getenvZpygments.formatters.terminalrutilZ ClassNotFound OptionError)rMpygmentsrZthemeZ theme_modererrr_get_pygments_formatters,       z&TerminalWriter._get_pygments_formatterrzLiteral[('diff', 'python')])rrrcCs|r|jr|js|S||}|dkr*|S|}|dkr>|Sddlm}||||}|ddkrz|ddkrz|dd}d|}|S)z5Highlight the given source if we have markup support.Nr) highlightrpror^)rIrLrrrr)rMrrZpygments_lexerZpygments_formatterrZ highlightedrrrr|s    zTerminalWriter._highlight)N)NN)r?)r)r)__name__ __module__ __qualname__dictrWrNpropertyrQsetterrSrVrlrsrirmrrrr|rrrrr(2sR %r()__doc__ __future__rr#rr@typingrrrrrcompatr Zwcwidthr Zpygments.formatterr Zpygments.lexerr rr'r(rrrrs$