U lj@svdZddlZddlmZeZGdddZGdddZdedjej Z iZ ej D]Z e D] Z e e e <q`qX[ dS) zGModule to parse ANSI escape sequences Maintainer: Jean-Paul Calderone N)Loggerc@sNeZdZdZdZeddeDZeee\Z Z Z Z Z ZZZdZddZdS) ColorTextzb Represents an element of text along with the texts colors and additional attributes. )brgylmcwccs|]}|VqdSN)upper).0xr9/tmp/pip-unpacked-wheel-aghyhbke/twisted/conch/ui/ansi.py szColorText.)ZBlackZRedZGreenZYellowZBlueZMagentaZCyanWhitec CsP||||_|_|_||_||_||_||_||_|jrL|j|j|_|_dSr )textfgbgdisplaybold underlineflashreverse) selfrrrrrrrrrrr__init__)szColorText.__init__N)__name__ __module__ __qualname____doc__COLORStuple BOLD_COLORSrangelenBLACKREDGREENYELLOWBLUEMAGENTACYANWHITEZ COLOR_NAMESrrrrrrs  rc@seZdZdZdZdZdZdZdZeeeeefZ ddZ d d Z d d Z d dZ ddZddZddZejfddZddZdS) AnsiParserz& Parser class for ANSI codes. ) HfABCDRsudG)JKP)hr)p)r cCsD|||_|_|j|j|_|_d\|_|_|_|_d|_d|_ dS)Nrrrr) defaultFG defaultBG currentFG currentBGrrrrrprepend)rrCrDrrrrJs zAnsiParser.__init__cCsd}d}d}t|}||kr|dkr6||tkr6d}n@|rv|d|}|dkr^|||dS||||}|}d}|d}q|S)zF Remove all ANSI color escapes from the given string. rBrArN)r&_setsfind)rstringresultshowiLnrrr stripEscapesQs   zAnsiParser.stripEscapescCsdSr r)rZcolorstrrrr writeStringgszAnsiParser.writeStringc Csz|jr|j|}d|_|d}t|dkrD|||dn2|||d|ddD]}t|}d}d}||kr||tjdkrq|d7}qz|sd|_dS|ddkr||||ddqdn|dd}|d8}||dkr d|_dSt||d}|dkr&qd|t j krj| |d|d||dd}|||qd|t j kr|d|d||dd}}| ||||qd|t jkr|d|d||dd}}|||||qd|t jkr,||dd}|||qd||krBd||_qdtjd ||d ||dd}|||qddS) zK Turn a string input into a list of L{ColorText} elements. rBrHrArNz[;?[z+Unhandled ANSI control type: {control_type})Z control_type)rGsplitr&rS formatTextrLdigits_setmapgetr/ COLOR_SET parseColor CURSOR_SET parseCursor ERASE_SET parseEraseMODE_SET_logwarn) rstrpartsr7rPrOtypecursoreraserrr parseStringjsh         "  "    zAnsiParser.parseStringcCs|dd}|sd}ztt|d}Wn6tk r^tjd|d|j|j|_|_ YdSX|D]l}|dkr|j|j|_|_ d\|_ |_ |_ |_ d |_qd|d krd |_ qdd |krd krnn |d |_qdd |krd krnn |d |_ qd|dkr|j|_qd|dkr"|j|_ qd|dkr4d |_ qd|dkrFd |_ qd|dkrXd |_ qd|dkrjd|_qd|dkr|d|_ qd|dkrd|_ qd|dkrd|_qd|dkrd|_ qd|dkrd |_qdtjd|dqddS)z5 Handle a single ANSI color sequence NrI0;z)Invalid ANSI color sequence: {sequence!r})sequencerr@rA%(/'1z*Unrecognised ANSI color command: {command})command)mapintrU ValueErrorraerrorrCrDrErFrrrrrZblink)rrcrdrrrrr[sV                 zAnsiParser.parseColorcCsdSr r)rrfrrrr]szAnsiParser.parseCursorcCsdSr r)rrgrrrr_szAnsiParser.parseErasecCs*|rtj|S|jr||p$tj|SdSr )rr"r)rvaluemodeZBOLDrrr pickColors zAnsiParser.pickColorc Cs4t|||jd||jd|j|j|j|j|jS)NrrA) rrrErFrrrrr)rrrrrrVs  zAnsiParser.formatTextN)rrr r!r\r^r`Z ASSIGN_SETrZSETSrrRrSrhr[r]r_rr$rrVrrrrr/4s ?5r/rB)r!rLZtwisted.loggerrrarr/joinr|rrJrXr7rrrrrs "D