~_]UddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z m Z mZddlmZddlmZmZd Zeed fed <ed d fde j0dedededef dZde j0dedeededeededeefdZde j0dedeededeededeefdZdeededefdZdeedeefdZGddeZd Zde j0d!ed"eedeefd#Z y)$N)partial)IterableListSetTuple)format_simplified)parsesortingwrap) add_to_line)DEFAULT_CONFIGConfig)zdef zcdef zcpdef zclass @z async def.STATEMENT_DECLARATIONSpyimportparsedconfig extension import_typereturnc $|jdk(r t|j|jS|jj }j Dcgc] }t |}}tj|jj}jriid|jd<d}|D]$$dk(rd} |jddj|j$jdi|jddj|j$jdi|d z}g} t} d } |D]$|j$d} j st#j$| $fd  } |j$d} j st#j$| $fd  } t'|| $||}t)|| $||}dg| r| r j*ndz}j,r ||z|z}n||z|z}j.rg}g}|D]Y}|s|j1dr|j3|)|r|j3t5||g}I|j3|[t#j$|t7t"j8j:j<j>j@jBjD }g}|D]<}tG|dd}|r|jI||j3tK|>$}|jLv}|r||jNvr||jN|<jPj|jSd}|rI|| vrEjTr| jW|d|}||jddvr|jYd|| s|s| dgjZzz } | |z } d } | xs| } j\r t_| } | r?| djadk(r)| jc| r| djadk(r)| r@| djadk(r*| jcd| r| djadk(r*jdr:je|jjg| |ji} d}|j|jjkr |j}| ||d|tm| z}|||dzDcgc]}|jac}dgk(r:|jc||||dzDcgc]}|jac}dgk(r:tm||kDr d}||d}to|D]\}}tqjr|dtm|jtd ^} }!}"| s[|jarK|jaj1dr(tm||dzkDr||dzjar|}n|!s|}njvdk7r'tyjvDcgc]}dc}||dn(|dk7r|j1tzrddg||dndg||d|jNrg}#to|D]\}}|#j3|||j|vs&|#jI|jN|j||tm||dzks||dzjadk7s||#j3d|#}t||jScc}wcc}wcc}wcc}w)zAdds the imports back to the file. (at the index of the first import) sorted alphabetically and split between groups )straightfrom no_sectionsFUTURE)rrr)rFc6tj|dS)NT) section_namestraight_importr module_keykeyrsections ./usr/lib/python3/dist-packages/isort/output.pyz sorted_imports..7s 2 2gt!r&c4tj|S)Nr!r#r%s r(r)z sorted_imports..?sg.@.@f[b.cr*r#) order_by_type force_to_toplexicographical length_sortreverse_relativegroup_by_packagecommentsz# r N)in_quoteindexsection_comments needs_importpyi)? import_index_output_as_stringlines_without_importsline_separatorcopyremove_importsr itertoolschainsectionsforced_separaterimportsupdategetset only_sectionsr naturally_with_straight_imports_with_from_importslines_between_types from_firstforce_sort_within_sections startswithappend_LineWithCommentsr section_keyr0r1r2r3r4r5getattrextendstrno_lines_before place_importsimport_headingslowerdedup_headingsaddinsertlines_between_sectionsensure_newline_before_comments_ensure_newline_before_commentstrippopformatting_functionjoin splitlinesoriginal_line_countlen enumerater skip_liner9lines_after_importsrangerimport_placements)%rrrrformatted_outputremovalrArD base_sectionsoutput seen_headingspending_lines_beforestraight_modules from_modulesstraight_imports from_imports lines_betweensection_outputcomments_abovenew_section_outputliner6r!rX section_titlesection_comment output_at imports_tail characternext_constructtailr8 should_skipr7_ new_out_linesr's% ` @r(sorted_importsrsb  !=!=v?T?TUU"(">">"C"C"E@F@U@UVW'0VNV'oofoov?U?UVH 57(D}%)+  bG(" +  NN= )* 5 < <w'++J;  NN= )& 1 8 89P9T9TU[]_9` a b!#33F!eM ZO!>>'2:>##&00   ~~g.v6 ##",,"cL2 F,g~{ * FL'>; *6;KF & &QR    )M9!>>> n $F#( #7#N;N uZOx,,/7 VBZ%%'2-  VBZ%%'2- VAY__&", 1  VAY__&",!!++  ! ! & &v . 6 *, I V777'' $*Yq!s6{*L+;LT$U02Bx \! ,02t \! ,  $%56 -KE4   &v///$$V%9%9&:R:RSW:X%YZ()eai8' 288:b@!((, -) -v/D/D EEkWF  >0bs]*8]/1]42 ]9rur'rAc g}|D ]}||vr d|d|dtjd|}jrjrB|jvr4j j stj|fd}|r|D cgc]} |d| |vs | }} |D cgc] } |d|  } } t|| D cic]<\} } | jdvr&| jd| D cgc] } | d|  c} >}} } } jrd|vr jsjs/|D]*}j rtj||||<,tj|D]b} | |vs|j| }jd|| r|j| ||d z|d zL|j| |||d zdd }j dj|d }j d dj|d}|s&|r|j#|d}d|vrjrut%j&t)t+|t|xsd dj,j. j0}|D cgc] } | |vs|  }} d}nGjr|jvrd}|r&|jd} t)|| zj,j. }j dj3|ij| d}|r||xrdxs j.d|z }| |vrΉjd|| r2|s0|j5t%j&|j0j dj3|d| j s4|j#fdtj|| DnQ|j#fd|| Dn0|j5t%j&|j0d}|rn|r|d|vr|jd} j stj|| || <j dj3|d| xsgjd|| r|sj dj3|ij| d}|rj5||j5t%j&t)| zj,j. j0g|| D]}j dj3|ij|d}|rj5||j5t%j&t)|zj,j. j0g|r |d|vrd|vr`|j5t)t+|t|xsd dj,j. |j7dd}tj|D]} j dj3|ij| d}|s5t)|| zj,j. }||xrdxs j.d|z }|j5t%j&|j0|j7| d}g}|rz|d|vs$jrgjd| rO|j5|jd|r-|d|vr*jrjd| rOjr2|xsgtj dj|dd z}t)|dj9|zj,j. }|sd}d }j:}|rt=||k\rd}t=|j>kDrt=|d kDrd}t=|j>kDrPt=|dkDrBj@t$jBjDt$jBjFfvrd}|rt%jH||j0}j@t$jBjDk(rmt%jH||j0t$jBjJ}tMd|jOdDj>kDr|}|s9t=|j>kDr!t%j&|j0}|r|j5||r |Scc} wcc} wcc} wcc} } } wcc} w)Nzfrom  rc8tj|dS)NTr-r#)r&r ignore_caser's r(r)z$_with_from_imports..s!G$6$6#%, %r*r+. as *r Fraboveremovedcomment_prefixTr.rnested;rc 3K|]I}ttj|zjjj KywrN with_commentsr r|r?ignore_commentsr.0 as_importr from_comments import_startrs r( z%_with_from_imports..D[ *%.!.$1$(II(4y(@&BWBWY_%&-3,B,B393H3H !"!" *AAc 3K|]I}ttj|zjjj Kywrrrs r(rz%_with_from_imports..Qrrz.__combined_as__z, )rrwr6r?r)rrwr6r?rmulti_line_outputc32K|]}t|ywN)rh)rxs r(rz%_with_from_imports..sL!s1vLs )(listrFno_inline_sortforce_single_linesingle_line_exclusions'force_alphabetical_sort_within_sectionsrJr rKzipas_mapcombine_as_imports combine_starr@r8rccategorized_commentsrVr r|r_with_star_commentsrrr?rHrRremovereforce_grid_wraprh line_lengthrModesGRIDVERTICALimport_statement VERTICAL_GRIDmaxsplit)rrrur'rArrqmodulerwr| from_import sub_modules sub_module as_module as_importsridxonly_show_as_importsr6above_commentsrsingle_import_linecommentspecific_commentfrom_import_sectiondo_multiline_reformatrother_import_statementrrrs`` ` @@@r(rMrMs FX0 ^ # vha }A6 FNN73F;FCD $$  $ $v7T7T)T HHK''&00    !-F81TF1C>1YLEQQ[&;-0Q Q ,/|[+I   ( ZV]]622 AGvAVWaAb4=;-tI;/     $ $c\.AfFYFY((!+YI!//070A0A*YBW0X 9-Y $yy6 T *,&,,[9C~~g.v6v>{K>Hnn[>Y cAg#';8B{8S SC!G5  T %..v6::62F44W=fEII&RVW n-!%l"v':':#'99!+FFDRS>SU[ \)/(C(CJ(O(S(S%ha }5)  &33"MM *291B1B:kCZ1[ * #MM *2,2,B,B393H3H !" !' 5 5 &  )+ %/ %<+ "77A S_ SD1) ,)001AB  II -$1$09$<,2,B,B393H3H !" !' 5 5 &  )+ -+C#|A*'Dr,&MM%/X^QS@TU+nA.$*$:$:+1+@+@ !'',#H#'99\#:(K33H=AA&"MQQR]_cd-:$(;6$*$:$:+1+@+@ .* +'/CH63H3HIwiX* dii0BFDYDY[a&bc$++K8#'!($')#" O:511"NN73F;FCKP(..|/?/?/BC# O:511"NN73F;FCKP,, (B$33F;??6(JZ@[]_`3 H$1 D;;/B#CC"22#)#8#8 $ +')$(-%"("8"8"s+>'??'R,0)'(6+=+==#FYBZ]^B^,0) ()F,>,>>/01400 JJOOTZZ-@-@AB-1)('+'<'<%1%8!)'-'<'<% ($//4::??B151F1F)5)<%-+1+@+@#).2jj.F.F 2.L/?/E/Ed/KLLvOaOaa/E,,5E1FI[I[1['+yy1A6CXCXZ`'a$ ./KgX0r MER N s6h.*h.4h3.h= h8h= 7 ii8h= rtc g}|D] |vrg} jdvrWj|d r|jd |j fdjd Dn|jd jddj d}|r|j||j fd|D|S)Nrrc32K|]}dd|yw)rrNr)rrrrs r(rz)_with_straight_imports.. s*%-qYK8%src3K|]C}tjdj|jjEyw)rrN)rrrHrr)ridefrrrs r(rz)_with_straight_imports..sS  ++J7;;FC..%44    sA A )rrFrRrVrrc) rrrtr'rArrqimport_definitionrzrs `` ` @r(rLrLsF"  ^ #  V]]:. .~~g&z26:!((K=&)BC  $ $%!'z!:6!B%   $ $ }AfX%> ?44W=jIMMfVZ[  MM. )  *  % 8 Mr*linesr?c6|jt|Sr)re_normalize_empty_lines)rr?s r(r=r=#s   5e< ==r*c|r@|djdk(r*|jd|r|djdk(r*|jd|S)Nrr.)rbrcrR)rs r(rr'sL E"IOO%+ "  E"IOO%+ LL Lr*ceZdZfdZxZS)rSc6t|||}||_|Sr)super__new__r6)clsvaluer6instance __class__s r(rz_LineWithComments.__new__0s 7?3.$r*)__name__ __module__ __qualname__r __classcell__)rs@r(rSrS/s r*rScg}d}t|dg|zD]<\}}||r|dk7r||s|jd|j|>|S)Nc,|xr|jdS)Nr/)rQ)r|s r( is_commentz2_ensure_newline_before_comment..is_comment9s,,,r*r.)rrR)rq new_outputrr| prev_lines r(rara6sgJ-vv7 i d  R 98M   b !$  r*rr6ct|jdj|ijdd}|r||gzS|S)Nrr)rrHrc)rrr6 star_comments r(rrCsB..x8<rs --*""2,*ac3ha $ DF   DF DFDF DF  DFNb   b b3-b b I b  b #YbJ %   % %sm% % I %  % #Y%P>T#Y>>>$s)S   3 3SDQTIZ^_bZcr*