a !f@sdZddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z e d d d Z eed d dZe GdddZdS)z4Helper functions for writing to terminals and files.N)Optional)Sequence)TextIO)wcswidth)finalreturncCs tjdd\}}|dkrd}|S)N)P)fallback(r )shutilget_terminal_size)width_r[/var/www/html/python-backend/venv/lib/python3.9/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthsrfiler cCsftjddkrdStjddkr(dSdtjvr6dSdtjvrDdSt|dod|odtjd d kS) NZ PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)osenvirongethasattrr)rrrrshould_do_markups   r"c@s&eZdZeddddddddd d d d d ddddddddZd9eeddddZee dddZ e j e ddddZ ee ddd Z e ee d!d"d#Zd:e ee ee edd$d%d&Zd'd(e eedd)d*d+Zd;e edd-d.d/Zddd0d1Zdfz)TerminalWriter.markup..rDcss|]}d|VqdS)z[%smNr)rZZcodrrr hr^z(TerminalWriter.markup..z)rY ValueErrorrLitemsjoin)rPrWrXr[escrrSrrXas zTerminalWriter.markup)sepchartitlerTrXr cKs|dur|j}tjdkr |d8}|durht|t|ddt|d}||}|d|d|}n||t|}t|t||kr||7}|j|fi|dS)NrCrr3 )rTrErGmaxlenrstripline)rPrdrerTrXNfillrjrrrsepks " zTerminalWriter.sepF)flush)msgrnrXr cKs|r|ddd}d|vr$||_n|j|7_|j|fi|}z|j|Wn.ty|dd}|j|Yn0|r|dS)N rzunicode-escapeascii) rsplitrMrXrKwriteUnicodeEncodeErrorencodedecodern)rPrornrX current_linerrrrts zTerminalWriter.writerD)srXr cKs |j|fi||ddS)Nrp)rt)rPryrXrrrrjszTerminalWriter.linecCs|jdSrR)rKrnrSrrrrnszTerminalWriter.flushr)linesindentsr cCs||r,t|t|kr,tdt|t||s>dgt|}d|}||}t||D]\}}|||q`dS)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. z5indents size ({}) should have same size as lines ({})rDrpN)rhr`formatrb _highlight splitlinesziprj)rPrzr{source new_linesindentnew_linerrr _write_sources  zTerminalWriter._write_source)rr c Csddlm}|jr|js|Sz0ddlm}ddlm}ddlm }ddl }Wnt yb|YS0z*||||t ddt d d }|WS|jjy|d t d Yn*|jjy|d t dYn0dS) z:Highlight the given source code if we have markup support.r) UsageError)TerminalFormatter) PythonLexer) highlightNZPYTEST_THEME_MODEZdarkZ PYTEST_THEME)bgstylezdPYTEST_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'.)Z_pytest.config.exceptionsrrLrOZpygments.formatters.terminalrZpygments.lexers.pythonrpygmentsrZ pygments.utilrIrgetenvutil ClassNotFoundr| OptionError)rPrrrrrrZ highlightedrrrr}s@         zTerminalWriter._highlight)N)NN)rD)r)__name__ __module__ __qualname__dictrYrrrQpropertyintrTsetterrVstrboolrXrmrtrjrnrrr}rrrrr#(sX  %r#)__doc__rrrEtypingrrrZwcwidthrZ_pytest.compatrrrrr"r#rrrrs