o ڭj@sdZddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z d e fd d Z d ed efd 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_rW/var/www/html/CropPilot/venv/lib/python3.10/site-packages/_pytest/_io/terminalwriter.pyget_terminal_widthsrfilecCsftjddkr dStjddkrdSdtjvrdSdtjvr"dSt|do2|o2tjd d kS) N 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(ZdQd*eed+d)fd,d-Zed+e fd.d/Z e j d0e d+d)fd1d/Z ed+e fd2d3Z d4e d5ed+e fd6d7Z ) )dRd8e d9ee d:ee d5ed+d)f d;d<Zd=d>d?e d@ed5ed+d)fdAdBZdSdDe d5ed+d)fdEdFZdTdGdHZdUdJee dKee d+d)fdLdMZdNe d+e fdOdPZd)S)VTerminalWriterblackredgreen yellow!blue"purple#cyan$white%Blackr Red)Green*Yellow+Blue,Purple-Cyan.White/boldrlightblinkinvertNrrcCs|durtj}t|dr1|r1tjdkr1zddl}Wn ty$Yn w||j}|dus1J||_ t ||_ d|_ d|_ d|_dS)Nrwin32rT)sysstdoutr rplatformcolorama ImportError AnsiToWin32stream_filer! hasmarkup _current_line_terminal_widthcode_highlight)selfrrNrrr__init__As      zTerminalWriter.__init__cCs|jdur|jStSN)rUrrWrrr fullwidthRs zTerminalWriter.fullwidthvaluecCs ||_dSrY)rU)rWr\rrrr[Xs cCs t|jS)z;Return an estimate of the width so far in the current line.)rrTrZrrrwidth_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).0nameonrZrr fsz)TerminalWriter.markup..rJcss|]}d|VqdS)z[%smNr)racodrrr hsz(TerminalWriter.markup..z)r` ValueErrorrSitemsjoin)rWr^r_rbescrrZrr_as zTerminalWriter.markupsepchartitler[cKs|dur|j}tjdkr|d8}|dur4t|t|ddt|d}||}|d|d|}n||t|}t|t||krN||7}|j|fi|dS)NrIrrD )r[rKrMmaxlenrstripline)rWrkrlr[r_Nfillrqrrrsepks " zTerminalWriter.sepF)flushmsgrucKs|rJ|ddd}d|vr||_n|j|7_|j|fi|}z|j|WntyA|dd}|j|Ynw|rL|dSdSdS)N rzunicode-escapeascii) rsplitrTr_rRwriteUnicodeEncodeErrorencodedecoderu)rWrvrur_ current_linerrrr{s   zTerminalWriter.writerJscKs |j|fi||ddS)Nrw)r{)rWrr_rrrrqszTerminalWriter.linecCs|jdSrY)rRrurZrrrruszTerminalWriter.flushrlinesindentscCs||rt|t|krtdt|t||sdgt|}d|}||}t||D] \}}|||q0dS)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 ({})rJrwN)rorgformatri _highlight splitlinesziprq)rWrrsource new_linesindentnew_linerrr _write_sources  zTerminalWriter._write_sourcerc Csddlm}|jr |js|Szddlm}ddlm}ddlm }ddl }Wn t y1|YSwz||||t ddt d d }|WS|jjyZ|d t d |jjyl|d t dw) z:Highlight the given source code if we have markup support.r) UsageError)TerminalFormatter) PythonLexer) highlightNPYTEST_THEME_MODEdark 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'.)_pytest.config.exceptionsrrSrVpygments.formatters.terminalrpygments.lexers.pythonrpygmentsr pygments.utilrOrgetenvutil ClassNotFoundr OptionError)rWrrrrrr highlightedrrrrsH        zTerminalWriter._highlightrY)NN)rJ)rN)r)__name__ __module__ __qualname__dictr`rrrXpropertyintr[setterr]strboolr_rtr{rqrurrrrrrrr"(s       %  r")__doc__rr rKtypingrrrwcwidthr_pytest.compatrrrrr!r"rrrrs