o Zh"@sdZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z dd l m Z d e fd d Z d e d efddZeGdddZdS)z4Helper functions for writing to terminals and files.N)final)Literal)Optional)Sequence)TextIO)wcswidthreturncCs tjdd\}}|dkrd}|S)N)P)fallback(r )shutilget_terminal_size)width_rQ/var/www/html/lang_env/lib/python3.10/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthsrfilecCsjtjddkr dStjddkrdStjdrdStjdr$dSt|do4|o4tjd d kS) NZ PY_COLORS1T0FNO_COLOR FORCE_COLORisattyTERMdumb)osenvirongethasattrr)rrrrshould_do_markups   r!c @seZdZedIiddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(ZdTd*eed+d)fd,d-Zed+e fd.d/Z e j d0e d+d)fd1d/Z ed+e fd2d3Z d4e d5ed+e fd6d7Z ) )dUd8e d9ee d:ee d5ed+d)f d;d<Zd=d>d?e d@ed5ed+d)fdAdBZdVdDe d5ed+d)fdEdFZdWdGdHZdXdJee dKee d+d)fdLdMZ NdYdOe dPedQd+e fdRdSZd)S)ZTerminalWriterblackredgreen yellow!blue"purple#cyan$white%ZBlackr Red)ZGreen*ZYellow+ZBlue,ZPurple-ZCyan.ZWhite/boldrZlightblinkinvertNrr cCs|durtj}t|dr1|r1tjdkr1zddl}Wn ty$Yn w||j}|dus1J||_ t ||_ d|_ d|_ d|_dS)Nrwin32rT)sysstdoutr rplatformcolorama ImportErrorZ AnsiToWin32stream_filer! hasmarkup _current_line_terminal_widthcode_highlight)selfrrFrrr__init__Cs      zTerminalWriter.__init__cCs|jdur|jStSN)rLrrNrrr fullwidthTs zTerminalWriter.fullwidthvaluecCs ||_dSrP)rL)rNrSrrrrRZs cCs t|jS)z;Return an estimate of the width so far in the current line.)rrKrQrrrwidth_of_current_line^s z$TerminalWriter.width_of_current_linetextmarkupc sb|D]}|jvrtd|qjr/fdd|D}|r/ddd|D|d}|S)Nzunknown markup: csg|] \}}|rj|qSr) _esctable).0nameonrQrr hsz)TerminalWriter.markup..rBcss|]}d|VqdS)z[%smNr)rXZcodrrr jsz(TerminalWriter.markup..)rW ValueErrorrJitemsjoin)rNrUrVrYescrrQrrVcs zTerminalWriter.markupsepchartitlerRcKs|dur|j}tjdkr|d8}|dur4t|t|ddt|d}||}|d|d|}n||t|}t|t||krN||7}|j|fi|dS)NrArr< )rRrCrEmaxlenrstripline)rNrbrcrRrVNfillrhrrrsepms " zTerminalWriter.sepF)flushmsgrlcKs|rJ|ddd}d|vr||_n|j|7_|j|fi|}z|j|WntyA|dd}|j|Ynw|rL|dSdSdS)N rzunicode-escapeascii) rsplitrKrVrIwriteUnicodeEncodeErrorencodedecoderl)rNrmrlrV current_linerrrrrs   zTerminalWriter.writerBscKs |j|fi||ddS)Nrn)rr)rNrwrVrrrrhszTerminalWriter.linecCs|jdSrP)rIrlrQrrrrlszTerminalWriter.flushrlinesindentscCs|rt|t|krtdt|dt|d|s"dgt|}d|}||}t||D] \}}|||q3dS)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 ()rBrnN)rfr^r` _highlight splitlinesziprh)rNrxrysource new_linesindentnew_linerrr _write_sources zTerminalWriter._write_sourcepythonr~lexer)diffrc Cs<ddlm}|r|jr|js|Sz'ddlm}|dkr"ddlm}n |dkr,ddlm }ddl m }dd l }Wn t yB|YSwz)||||td d td d }|ddkrh|ddkrh|d d}d|WS|jjy} z |dtd | d } ~ w|jjy} z |dtd | d } ~ ww)z5Highlight the given source if we have markup support.r) UsageError)TerminalFormatterr) PythonLexerr) DiffLexer) highlightNZPYTEST_THEME_MODEZdarkZ PYTEST_THEME)bgstylerornr]zdPYTEST_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.exceptionsrrJrMZpygments.formatters.terminalrZpygments.lexers.pythonrZpygments.lexers.diffrpygmentsrZ pygments.utilrGrgetenvutil ClassNotFoundformat OptionError) rNr~rrrLexerrrZ highlightederrrr{s^         zTerminalWriter._highlightrP)NN)rB)r N)r)r)__name__ __module__ __qualname__dictrWrrrOpropertyintrRsetterrTstrboolrVrkrrrhrlrrrr{rrrrr"*s       %  r")__doc__rrrCtypingrrrrrZwcwidthrrrrr!r"rrrrs