a Ø þd‘ã@s‚dZddlmZddlmZddlmZddlmZdZ dZ dZ e hd £ƒZ d d „ZGd d „d ejƒZdd„Zdd„Zdd„ZdS)a;Calculate the number of blank lines between top-level entities. Calculates how many blank lines we need between classes, functions, and other entities at the same level. CalculateBlankLines(): the main function exported by this module. Annotations: newlines: The number of newlines required before the node. é)Útoken)Ú pytree_utils)Úpytree_visitor)Ústyleééé>Z assert_stmtZ exec_stmtZ with_stmtZ return_stmtZdel_stmtZ raise_stmtZ global_stmtZ continue_stmtZ simple_stmtZ while_stmtZfor_stmtZ small_stmtZtry_stmtZ nonlocal_stmtZ import_stmtZ expr_stmtZif_stmtZ async_stmtZ break_stmtZ yield_stmtZ print_stmtZ pass_stmtcCstƒ}| |¡dS)z¨Run the blank line calculator visitor over the tree. This modifies the tree in place. Arguments: tree: the top-level pytree node to annotate with subtypes. N)Ú_BlankLineCalculatorÚVisit)ÚtreeZblank_line_calculator©r új/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapf/pytree/blank_line_calculator.pyÚCalculateBlankLines,srcs`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Z‡fd d „Z dd„Z dd„Z dd„Z ‡Z S)r zB_BlankLineCalculator - see file-level docstring for a description.cCs"d|_d|_d|_d|_d|_dS)NrF)Ú class_levelÚfunction_levelÚlast_comment_linenoÚlast_was_decoratorÚlast_was_class_or_function)Úselfr r r Ú__init__;s z_BlankLineCalculator.__init__cCs.| |¡|jdjtjkr*|jdj|_dS©Nr)ÚDefaultNodeVisitÚchildrenÚtypeÚ grammar_tokenÚCOMMENTÚlinenor©rÚnoder r r ÚVisit_simple_stmtBs z&_BlankLineCalculator.Visit_simple_stmtcCsd|jr.|j|jdjdkr.t|jdtƒnt|jd| |¡ƒ|jD]}| |¡qJd|_dS)NrrT)rrrÚ_SetNumNewlinesÚ_NO_BLANK_LINESÚ_GetNumNewlinesr r)rrÚchildr r r ÚVisit_decoratorGsÿ  z$_BlankLineCalculator.Visit_decoratorcCsZd|_| |¡}d|_|jd7_|j|d…D]}| |¡q2|jd8_d|_dS)NFrT)rÚ(_SetBlankLinesBetweenCommentAndClassFuncrrrr ©rrÚindexr#r r r ÚVisit_classdefQs  z#_BlankLineCalculator.Visit_classdefcCsŒd|_| |¡}t|ƒr8| |jj¡}t|jddƒn | |¡}d|_|jd7_|j|d…D]}|  |¡qd|jd8_d|_dS)NFrrT) rr%Ú_AsyncFunctionÚ prev_siblingÚparentr rrrr r&r r r Ú Visit_funcdef[s ÿ  z"_BlankLineCalculator.Visit_funcdefcsH|jr.t |¡tvr.t |¡}t|| |¡ƒd|_tt|ƒ  |¡dS)zÄOverride the default visitor for Node. This will set the blank lines required if the last entity was a class or function. Arguments: node: (pytree.Node) The node to visit. FN) rrZNodeNameÚ_PYTHON_STATEMENTSÚ FirstLeafNoder r"Úsuperr r)rrZleaf©Ú __class__r r rks  z%_BlankLineCalculator.DefaultNodeVisitcCsÆd}t |j|¡rP| |j|jd¡|jsFt|j|jdtƒ|d7}q|rŒ|j|jd|j|djdjkrŒt|j|tƒn6|j d|j|jkr¨t}n |  |¡}t|j||ƒ|S)aLSet the number of blanks between a comment and class or func definition. Class and function definitions have leading comments as children of the classdef and functdef nodes. Arguments: node: (pytree.Node) The classdef or funcdef node. Returns: The index of the first child past the comment nodes. rr) rZIsCommentStatementrr rr Ú_ONE_BLANK_LINErr!rr")rrr'Ú num_newlinesr r r r%{s  ÿ z=_BlankLineCalculator._SetBlankLinesBetweenCommentAndClassFunccCs&|jr tS| |¡r"dt d¡StS)NrZ'BLANK_LINES_AROUND_TOP_LEVEL_DEFINITION)rr!Ú _IsTopLevelrZGetr2rr r r r"šs  z$_BlankLineCalculator._GetNumNewlinescCs|jp |j ot|ƒS©N)rrÚ_StartsInZerothColumnrr r r r4¡sÿz _BlankLineCalculator._IsTopLevel)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr$r(r,rr%r"r4Ú __classcell__r r r0r r 8s   r cCst |tjj|¡dSr5)rZSetNodeAnnotationZ AnnotationZNEWLINES)rr3r r r r ¦s ÿr cCs$t |¡jdkp"t|ƒo"|jjdkSr)rr.Úcolumnr)r*©rr r r r6«sÿr6cCs|jo|jjtjkSr5)r*rrÚASYNCr=r r r r)°sr)N)r:Z yapf_third_party._ylib2to3.pgen2rrZ yapf.pytreerrZ yapf.yapflibrr!r2Z_TWO_BLANK_LINESÚ frozensetr-rZ PyTreeVisitorr r r6r)r r r r Ús      n