U  W[ @spdZddlZddlmZGdddZGdddZdedjejZ iZ ejD]Z e D] Z e e e <qZqR[ dS) zGModule to parse ANSI escape sequences Maintainer: Jean-Paul Calderone N)logc@sNeZdZdZdZeddeDZeee\Z Z Z Z Z ZZZdZddZdS) ColorTextzb Represents an element of text along with the texts colors and additional attributes. )brgylmcwcCsg|] }|qS)upper).0xr r 7/usr/lib/python3/dist-packages/twisted/conch/ui/ansi.py szColorText.)ZBlackZRedZGreenZYellowZBlueZMagentaZCyanZWhitec CsP||||_|_|_||_||_||_||_||_|jrL|j|j|_|_dSN)textfgbgdisplaybold underlineflashreverse) selfrrrrrrrrr r r__init__szColorText.__init__N)__name__ __module__ __qualname____doc__COLORStuple BOLD_COLORSrangelenZBLACKZREDZGREENZYELLOWZBLUEZMAGENTAZCYANZWHITEZ COLOR_NAMESrr r r rrs 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)rr:r;r r rr@s 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. r9r8rN)r%_setsfind)rstringresultZshowiLnr r r stripEscapesHs   zAnsiParser.stripEscapescCsdSrr )rZcolorstrr r r writeString^szAnsiParser.writeStringc Cs||jr|j|}d|_|d}t|dkrD|||dn4|||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||_qdtd ||f||dd}|||qddS) zK Turn a string input into a list of L{ColorText} elements. r9r?r8rNz[;?[zUnhandled ANSI control type: %c)r>splitr%rI formatTextrCZdigits_setmapgetr& COLOR_SET parseColor CURSOR_SET parseCursor ERASE_SET parseEraseMODE_SETrmsg) rstrpartsr.rFrEtypecursoreraser r r parseStringasb         "  "    zAnsiParser.parseStringcCs|dd}|sd}ztt|d}Wn>tk rftdt||f|j|j|_ |_ YdSX|D]p}|dkr|j|j|_ |_ d\|_ |_ |_ |_d|_ql|dkrd|_ qld |krd krnn |d |_ qld |krd krnn |d |_ ql|d kr|j|_ ql|dkr,|j|_ ql|dkr>d|_ ql|dkrPd|_ ql|dkrbd|_ql|dkrtd|_ql|dkrd|_ ql|dkrd|_ ql|dkrd|_ql|dkrd|_ql|dkrd|_qltd|fqldS)z5 Handle a single ANSI color sequence Nr@0;z$Invalid ANSI color sequence (%d): %srr7r8%(/'1z#Unrecognised ANSI color command: %d)mapintrK ValueErrorrrVr%r:r;r<r=rrrrrZblink)rrWrXrr r rrPsV                 zAnsiParser.parseColorcCsdSrr )rrZr r rrRszAnsiParser.parseCursorcCsdSrr )rr[r r rrTszAnsiParser.parseErasecCs*|rtj|S|jr||p$tj|SdSr)rr!r)rvaluemodeZBOLDr r r pickColors zAnsiParser.pickColorc Cs4t|||jd||jd|j|j|j|j|jS)Nrr8) rrsr<r=rrrrr)rrr r rrLs  zAnsiParser.formatTextN)rrrr rQrSrUZ ASSIGN_SETrOSETSrrHrIr\rPrRrTrr#rsrLr r r rr&*s =5r&r9) r rCZtwisted.pythonrrr&joinrnrtrArMr.rr r r rs A