Vpf<ddlmZddlmZddlZddlmZddlZddlm Z m Z ddl m Z ddl m Z  ddlZn #e$rdZYnwxYwe jde jd d d Zd\dZeZGdde jZGddeZeZGddeZGddeZGddeZGddeZGddeZeZGddeZ ej!d gd!Z"ej!d"gd#Z#Gd$d%eZ$ej!d&d'd(gZ%d]d.Z&d^d/Z'Gd0d1e Z(Gd2d3e Z)Gd4d5e Z*d_d:Z+d;Z,d`d@Z-dadAZ.dbdcdEZ/dddHZ0dedfdKZ1dgdLZ2dhdNZ3ddddOdidUZ4djdXZ5ee4e#j6e"j7YZ8ee4e#j9e"j:YZ;dkd[Z3:+<+<+>+>>s == A  A c6eZdZdZ dddddddZdZddZdS)DocPNz #  use_colorannotation_prefix source_mapwidthintr% bool | Noner&rr''list[list[tuple[int, int, Any]]] | NonercT|to tj}t|||||S)z Formats a pretty-printer document as a string. Args: source_map: for each line in the output, contains a list of (start column, end column, source) tuples. Each tuple associates a region of output text with a source. Nr$) CAN_USE_COLOR_PPRINT_USE_COLORvalue_format)selfr(r%r&r's rformatz Doc.formatHs>;$5$;i 4)%6: O O OOc*|SN)r2r1s r__str__z Doc.__str__Zs ;;==r3otherc$t||gSr5)concat)r1r8s r__add__z Doc.__add__]s 4-  r3)r#) r(r)r%r*r&rr'r+rr)r8r!rr!)r __module__ __qualname__ __slots__r2r7r;r"r3rr!r!Est)O7;":>OOOOOO$!!!!!!r3r!ceZdZdZdS)_NilDoccdS)Nnilr"r6s r__repr__z_NilDoc.__repr__asUUr3N)rr<r=rCr"r3rr@r@`s"""""r3r@c8eZdZUdZded<ded<d d dZdZdS) _TextDoc)text annotationrrF str | NonerGNct|ts J||t|ts J|||_||_dSr5) isinstancerrFrG)r1rFrGs r__init__z_TextDoc.__init__jsU dC &&$&&&  J!__annotations__rKrCr"r3rrErEesX$) )))!!!!! &&&&&r3rEc,eZdZUdZded<ddZdZdS) _ConcatDocchildrenz list[Doc]rR Sequence[Doc]ct||_td|jDsJ|jdS)Nc3@K|]}t|tVdSr5)rJr!).0docs r z&_ConcatDoc.__init__..|s,==z#s##======r3)listrRall)r1rRs rrKz_ConcatDoc.__init__zsDNNDM ==t}=== = =LLt}LLLLLr3cd|jdS)Nzconcat()rQr6s rrCz_ConcatDoc.__repr__~s7t}7777r3N)rRrSrMr"r3rrPrPvsF)MMMM87777r3rPc,eZdZUdZded<ddZdZdS) _BreakDocrFrrFcRt|ts J|||_dSr5)rJrrF)r1rFs rrKz_BreakDoc.__init__s+ dC &&$&&&DIIIr3cd|jdS)Nzbreak(r\r_r6s rrCz_BreakDoc.__repr__s2di2222r3N)rFrrMr"r3rr^r^s?) )))32222r3r^c,eZdZUdZded<ddZdZdS) _GroupDocchildr!recRt|ts J|||_dSr5)rJr!re)r1res rrKz_GroupDoc.__init__s+ eS ! !((5(((DJJJr3cd|jdS)Nzgroup(r\rdr6s rrCz_GroupDoc.__repr__s3dj3333r3N)rer!rMr"r3rrcrcs?) ***43333r3rcc6eZdZUdZded<ded<d dZdZdS) _NestDocnrer)rkr!rec`t|ts J|||_||_dSr5)rJr!rkre)r1rkres rrKz_NestDoc.__init__s2 eS ! !((5((( DFDJJJr3c&d|j|jfdS)Nznest(r\rjr6s rrCz_NestDoc.__repr__s:TVTZ$8::::r3N)rkr)rer!rMr"r3rririsH)&&& *** ;::::r3ric6eZdZUdZded<ded<d dZdZdS) _SourceMapDocresourcer!rerrqc`t|ts J|||_||_dSr5)rJr!rerq)r1rerqs rrKz_SourceMapDoc.__init__s2 eS ! !((5(((DJDKKKr3c(d|jd|jdS)Nzsource(z, r\rpr6s rrCz_SourceMapDoc.__repr__sCtzCCT[CCCCr3N)rer!rqrrMr"r3rrorosI!) *** +++ DCCCCr3ro_Color) BLACKREDGREENYELLOWBLUEMAGENTACYANWHITERESET _Intensity)DIMNORMALBRIGHTcNeZdZUdZded<ded<ded<ded<d d d d d d Zd S) _ColorDoc) foreground background intensityre Color | NonerrIntensity | Nonerr!reNrrrc|t|ts J|||_||_||_||_dSr5)rJr!rerrr)r1rerrrs rrKz_ColorDoc.__init__sB eS ! !((5(((DJ DO DODNNNr3)rer!rrrrrr)rr<r=r>rNrKr"r3rrrso@) ***?C*.-1r3r _BreakModeFLATBREAKrWr(r)agenda!list[tuple[int, _BreakMode, Doc]]c|dkrt|dkr|\}t|trnt|tr|t|jz}nZt|t r6|fdt|j Dnt|tr*tj krdS|t|jz}nt|tr&||jz|jfnt|t"r(|tj|jfnXt|t&st|t(r||jfnt+d||dkrt|dk|dkS)Nrc3"K|] }|fV dSr5r")rVdims rrXz_fits..s)>>!Q1I>>>>>>r3TInvalid document )lenpoprJr@rErFrPextendreversedrRr^rrriappendrkrercrrro ValueError)rWr(rrrs @@r_fitsrs  s6{{Q IAq##w1 C " "1 s38}}ee C $ $ 1 mm>>>>>x '='=>>>>>>> C # # 1 j  t s38}}ee C " "1 mmQY39-.... C # #1 mmQ 34444 C # #1z#}'E'E1 mmQ39%&&&& *C 0 00'  s6{{Q* !r3c(|g}d}d}t|dkrv|}t|trn7t|tr|j|dkr|rdS|dz }n t|t r(|t|j nt|trd}nt|tr| |j nt|tr| |j nUt|tst|t r| |j nt#d|t|dkvdS)NrFTr)rrrJr@rErGrPrrrRr^rirrercrror)rWrnum_annotations seen_breaks r_sparsers} 5&/* F a **,,C#w1 C " "1  # a  J 1 C $ $ 1 mmHS\**++++ C # # 1jj C " "1 mmCI C # #1 mmCI C # #1z#}'E'E1 mmCI *C 0 00) F a, r3c.eZdZUded<ded<ded<dS) _ColorStateColorrr IntensityrNrr<r=rNr"r3rrrs6r3rcBeZdZUded<ded<ded<ded<d ed <d S) _Stater)indentrmoder!rWrcolorrr'Nrr"r3rrr sC +++ (((/////r3rc.eZdZUded<ded<ded<dS)_LinerrFr)r(zstr | None | list[str]rNrr"r3rrrs0 ))) ***%%%%%%r3rr%stateupdatetuple[_ColorState, str]c|rt|dfSd}|j|jkr4|ttjt |jjz }|j|jkr4|ttjt |jjz }|j|jkrH|tj j z }|ttj t |jjz }||fS)N) coloramargetattrForernamerBackrStyler)r%rr color_strs r _update_colorrs h& 2:) *** F,=,B(C(CDDDI *** F,=,B(C(CDDDI _((( &&I V-=-B)C)CDDDI  r3c td|D}g}|D]-}t|jdkr*||dEt|jdkrI|||jd||jz zz|jd|||jd||jz zz|jd|jddD]/}|td|z|j|0/|S)Nc3$K|] }|jV dSr5)r()rVls rrXz%_align_annotations..(s$&&1qw&&&&&&r3rrr )rFr)rFr(r)maxrrr_replacerFr(r)linesmaxlenoutras r_align_annotationsr%so &&&&& & && #  K Ka 1=Q jj--.... Q]  q  jj#!'1A*B!B() a(8::;;;; jj#!'1A*B!B() a(8::;;;}QRR KK! 5cFl!'qIIIJJJJK *r3r&rr'r+c d g}ttjtjtj}ttjtjtj}|}d} t g} tdtj ||g} d} d} g}t| dkr!| \}}|@|kr:t| }| |kr!t ur| | |f||} t|trnt|tr]t!||\}}| |z } | |jz } |j| |j| t|jz } n(t|t&r8| fdt+|jDnt|t.rtj krt|dkrt!|||\}}| |z } | t1| | ||Ot| }| |kr!t ur| | |f| | g} } dz} g}} n t!||\}}| |z } | |jz } | t|jz } nt|t2r5| t|jz|jnt|t8rt;|r^t=||| z tj|jfgr7| ttj|jn| ttj |jnt|t@rat|j!pj!|j"pj"|j#pj#| t|jnVt|tHr1| t|j|j%ntMd|t| dk!t|dkrt!|||\}}| |z } |Kt| }| |kr!t ur| | |f| | | t1| | |tO|}d(fd|D}t!|||\}}||zS)Nrrc3>K|]}t|VdSr5)r)rVrrrrrqs rrXz_format..ZsG551aE622555555r3rr c3XK|]$}|j|jn|j|jV%dSr5)rrF)rVrr&s rrXz_format..sbGG;<m#AFF F 6% 6q} 6 6GGGGGGr3))rrr}rrr _NO_SOURCErrrrrrrJr@rErrFrGrPrrrRr^rrirkrercrrrrrrrrorqrrjoin)rWr(r%r&r'rdefault_colorsannotation_colors color_state source_startline_source_maprk line_textline_annotations agenda_sourceposrrrrrrqs ` @@@@rr0r09s %u{EK9IJJ.!%+u{IMJJ+, &/ 1j&^V D D E&!) F a&,jjll#Aq#um-6"9"9  NNc   z!9!9 c6:;;;fl#w41 C " "21,Y UKKk99i38i  #///3sx==aa C $ $+1 mm5555555%cl335555555 C # #(1 j    1 $ $#0K1B$D$D +y y ) U9a)9::;;;  !I# S V:%=%=  " "L#v#> ? ? ?   O , , ,/,!G  !.y+u!M!M YY SX  S]] C " "1 mmF1su9aE6BBCCCC C # #1 #,,MCa#)%D$EFFM fQ E6JJKKKK fQ 0#)UFKKLLLL C # #1#.>???? *C 0 00y F a| Q*9k+<>>K I i..CsvZ77lC8999o&&&,,uY#344555 U # #% GGGG@EGGG G G#)K)799+y yr3ctS)zAn empty document.)_nilr"r3rrBrBs +r3srGrHc"t||S)z Literal text.)rE)rrGs rrFrFs !Z  r3docsrSctt|}t|dkr|dSt|S)zConcatenation of documents.rr)rYrrP)rs rr:r:s3 d$YY!^^ 7N D  r3rrFc t|S)zYA break. Prints either as a newline or as `text`, depending on the enclosing group. )r^r_s rbrkrs 4r3c t|S)zLayout alternative groups. Prints the group with its breaks as their text (typically spaces) if the entire group would fit on the line when printed that way. Otherwise, breaks inside the group as printed as newlines. )rc)rWs rgrouprs 3r3rkc"t||S)z'Increases the indentation level by `n`.)ri)rkrWs rnestrs !S  r3rrrrrrc(t||||S)zANSI colors. Overrides the foreground/background/intensity of the text for the child doc. Requires use_colors=True to be set when printing and the `colorama` package to be installed; otherwise does nothing. r)r)rWrrrs rrrs$ 3:*& ( ( ((r3rqrc"t||S)aoSource mapping. A source map associates a region of the pretty-printer's text output with a source location that produced it. For the purposes of the pretty printer a ``source`` may be any object: we require only that we can compare sources for equality. A text region to source object mapping can be populated as a side output of the ``format`` method. )ro)rWrqs rr'r's sF # ##r3)rrsepct|}t|dkrtS|dg}|ddD],}||||-t |S)z(Concatenates `docs`, separated by `sep`.rrN)rYrrBrr:)rrxsrWs rrrss d$YY!^^ 55L Qy" !""XcIIcNNNIIcNNNN r3)rr)rWr!r(r)rrrr)rWr!rr)r%rrrrrrr) rWr!r(r)r%rr&rr'r+rr)rr!r5)rrrGrHrr!)rrSrr!)r)rFrrr!)rWr!rr!)rkr)rWr!rr!)rWr!rrrrrr)rWr!rqr)rr!rrSrr!)= __future__rcollections.abcrenum functoolsrrtypingrrjax._srcr r r ImportError bool_flagbool_envr.rr- StrictABCr!r@rrErPr^rcriobjectrroEnumrrrrrrrrrrrr0rBrFr:rrrrr'rrztype_annotationrrykeywordrr"r3rrs8#"""""$$$$$$  """"""""//// (((%F$FO*D11 J ????   !!!!!$.!!!6#####c###wyy&&&&&s&&&"888888883333333344444444 ; ; ; ; ;s ; ; ;VXX  D D D D DC D D D  (BBB C C DIl$?$?$? @ @ "TY|fg%6 7 7 88* Z&&&&&J&&&       (]]]]H!!!!! 37%)(, ( ( ( ( ( ( $ $ $ $'%9+;%*]444 '%9#3 K K K      s 5??