U `{jg$@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_rY/var/www/html/bwave/python/venv/lib/python3.8/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthsrrboolfilercCsjtjddkrdStjddkr(dStjdr8dStjdrHdSt|doh|ohtjd d kS) NZ PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)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)blackredgreenyellowbluepurplecyanwhiteZBlackRedZGreenZYellowZBlueZPurpleZCyanZWhiteboldZlightblinkinvertNz TextIO | NoneNonercCs|dkrtj}t|drd|rdtjdkrdz ddl}Wntk rJYnX||j}|dk sdt ||_ t ||_ d|_ d|_d|_dS)Nr#win32rT)sysstdoutr)r#platformcolorama ImportErrorZ AnsiToWin32streamAssertionError_filer* hasmarkup _current_line_terminal_widthcode_highlight)selfrrOrrr__init__Ks    zTerminalWriter.__init__rrcCs|jdk r|jStSN)rVrrXrrr fullwidth\s zTerminalWriter.fullwidth)valuercCs ||_dSrZ)rV)rXr]rrrr\bscCs t|jS)z;Return an estimate of the width so far in the current line.)r rUr[rrrwidth_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).0nameonr[rr psz)TerminalWriter.markup..rKcss|]}d|dVqdS)z[mNr)rcZcodrrr rsz(TerminalWriter.markup..)rb ValueErrorrTitemsjoin)rXr`rardescrr[rraks zTerminalWriter.markupz str | Nonez int | None)sepchartitler\rarcKs|dkr|j}tjdkr |d8}|dk rht|t|ddt|d}||}|d|d|}n||t|}t|t||kr||7}|j|f|dS)NrJr r )r\rLrNmaxlenrstripline)rXrnror\raNfillrtrrrsepus " zTerminalWriter.sepF)flush)msgrxrarcKs|r|ddd}d|kr$||_n|j|7_|j|f|}z|j|Wn0tk r|dd}|j|YnX|r|dS)N r zunicode-escapeascii) rsplitrUrarSwriteUnicodeEncodeErrorencodedecoderx)rXryrxra current_linerrrr~szTerminalWriter.writerK)srarcKs|j|f||ddS)Nrz)r~)rXrrarrrrtszTerminalWriter.linecCs|jdSrZ)rSrxr[rrrrxszTerminalWriter.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 ()rKrzN)rrrjrl _highlight splitlinesziprt)rXrrsource new_linesindentnew_linerrr _write_sources zTerminalWriter._write_sourcezLiteral[('python', 'diff')]z Lexer | None)lexerrcCs`zD|dkrddlm}|WS|dkr:ddlm}|WSt|Wntk rZYdSXdS)Npythonr) PythonLexerdiff) DiffLexer)Zpygments.lexers.pythonrZpygments.lexers.diffrr ModuleNotFoundError)rXrrrrrr_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.terminalrutil ClassNotFound OptionError)rXpygmentsrthemeZ 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) highlightr{rzri)rTrWrrrr)rXrrZpygments_lexerZpygments_formatterrZ highlightedrrrrs    zTerminalWriter._highlight)N)NN)rK)r)r)__name__ __module__ __qualname__dictrbrYpropertyr\setterr^rarwr~rtrxrrrrrrrrr+2sR %r+)__doc__ __future__rr&rrLtypingrrrrrcompatr Zwcwidthr Zpygments.formatterr Zpygments.lexerr rr*r+rrrrs$