edZddlmZddlmZddlmZmZmZm Z ddl m Z ddl m Zddl mZddlmZdd lmZdd lmZerdd lmZe e j2e j4e j6e j8e j:e j<e j>e j@e jBe jDe jFe jHe jJe jLe jNe jPfZ)e e jTe jVe jXe jZfZ.ed e e)e jZe j^f Z0GddeZ GddeZGddejbZ2ddZ3y)z.Module to add McCabe checker class for pylint.) annotations)Sequence) TYPE_CHECKINGAnyTypeVarUnion)nodes) PathGraph)PathGraphingAstVisitor)checkers)only_required_for_messages)HIGH)PyLinter_AppendableNodeT)boundc eZdZdfd ZxZS)r c8t|ddd||_y)N)nameentitylineno)super__init__root)selfnode __class__s :/usr/lib/python3/dist-packages/pylint/extensions/mccabe.pyrzPathGraph.__init__1s bA6 )rz"_SubGraphNodes | nodes.FunctionDef)__name__ __module__ __qualname__r __classcell__rs@rr r 0s r r ceZdZd fd Zd dZd dZd dZeZddZexZ xZ xZ xZ xZ xZxZxZxZxZxZxZxZxZxZZddZeZddZ d ddZ dd ZxZS)r c>t|d|_d|_y)Nr)rr_bottom_countergraph)rrs rrzPathGraphingAstVisitor.__init__7s  '+ r cT|jD]}|j|g|yN) get_childrendispatch)rrargschilds rdefaultzPathGraphingAstVisitor.default<s-&&( (E DMM% '$ ' (r c||_|j}|jj|}|?|j}t |j d|z|j}||j|<||g|S)Nvisit)rr_cachegetr!getattrvisitorr0)rrr.klassmeth class_names rr-zPathGraphingAstVisitor.dispatch@sk {{u% <J4<<:)=t||LD!%DKK D 4  r c@|j|j|}||_|j|j|j }|xj dz c_|jj |j||jj ||||_yt||_||_|j|j|j|j|j|j<|jy)Nr) r) _append_nodetail dispatch_listbodyr(connectr graphs classnamerreset)rrpathnodebottoms rvisitFunctionDefz'PathGraphingAstVisitor.visitFunctionDefJs :: !((.H DI   tyy ),,-F  A % JJ  tyy& 1 JJ  tV ,DI"4DJDI   tyy ):>**DKK4>>*499+6 7 JJLr c&|j|yr+)r;rrs rvisitSimpleStatementz+PathGraphingAstVisitor.visitSimpleStatement^s $r c\|j||j|jyr+)r;r=r>rGs r visitWithz PathGraphingAstVisitor.visitWithys" $ 499%r c|jr |jsy|jj|j|||_|Sr+)r<r)r?rGs rr;z#PathGraphingAstVisitor._append_nodes6yy  499d+  r c|j[t||_|j||||j|j|j|<|j y|j ||j|||y)z@Create the subgraphs representing any `if` and `for` statements.N)r)r _subgraph_parser@rArBr;)rrr extra_blockss r _subgraphz PathGraphingAstVisitor._subgraphst :: "4DJ  t\ :59ZZDKK4>>*4&1 2 JJL   d #  t\ :r cxg}||_|j|j|j|j|D]?}||_|j|j|j|jA|jr>||_|j|j|j|jn|j||rZ|j rM|j }|xj dz c_|D]}|j j|| ||_yyy)zAParse the body and any `else` block of `if` and `for` statements.rN)r<r=r>appendorelser)r(r?)rrrCrN loose_endsextrarDends rrMz&PathGraphingAstVisitor._subgraph_parses    499%$))$! )EDI   uzz *   dii ( ) ;;DI   t{{ +   dii (   d # DJJ,,-F  A % ! 0 ""3/ 0DI 4r )returnNone)r nodes.NodeNGr.rrVrW)rrXr.rrVr)rznodes.FunctionDefrVrW)r_StatementNodesrVrW)rz nodes.WithrVrW)rrrVz_AppendableNodeT | None))r_SubGraphNodesrstrrNSequence[nodes.ExceptHandler]rVrW)rr[rCr[rNr]rVrW)r!r"r#rr0r-rEvisitAsyncFunctionDefrH visitAssert visitAssignvisitAugAssign visitDelete visitRaise visitYield visitImport visitCallvisitSubscript visitPass visitContinue visitBreak visitGlobal visitReturn visitExpr visitAwaitrJvisitAsyncWithr;rOrMr$r%s@rr r 6s8, (!$- 2>R-RKRR R  R R R R R R" #R& 'R* +R, -R,"-R,%. &N79 ;;;4 ;  ;"!4   r r cNeZdZdZdZddiZddddd d ffZed dd Zy )McCabeMethodCheckerzoChecks McCabe complexity cyclomatic threshold in methods and functions to validate a too complex code. designR1260)z*%s is too complex. The McCabe rating is %d too-complexzSUsed when a method or function is too complex based on McCabe Complexity Cyclomaticzmax-complexity intzz&McCabe complexity cyclomatic threshold)r0typemetavarhelprtct}|jD]}|j|||jj D]}|j }|j }t|drd|jd}n(d|jjjd}||jjjkr|jd|t ||fy)zVisit an astroid.Module node to check too complex rating and add message if is greater than max_complexity stored from options. r'zThis 'rt)r confidencer.N)r r>preorderr@values complexityrhasattrrrr!lowerlinterconfigmax_complexity add_messager)rrr6r/r)r node_names r visit_modulez McCabeMethodChecker.visit_modules )*YY -E   UG , -^^**, E))+J::DtV$ {!, $T^^%<%<%B%B%D$EQG T[[//>>>   DTJ@W   r N)rz nodes.ModulerVrW) r!r"r#__doc__rmsgsoptionsr rrZr rrqrqsZ D   D "@    G ./r rqc8|jt|yr+)register_checkerrq)rs rregisterrs /78r N)rrrVrW)4r __future__rcollections.abcrtypingrrrrastroidr mccaber Mccabe_PathGraphr Mccabe_PathGraphingAstVisitorpylintr pylint.checkers.utilsr pylint.interfacesr pylint.lintrAssertAssign AugAssignDeleteRaiseYieldImportCall SubscriptPassContinueBreakGlobalReturnExprAwaitrYIfTryForWhiler[ FunctionDefr BaseCheckerrqrrZr rrs- 5"$550J<"$ LL LL OO LL KK KK LL JJ OO JJ NN KK LL LL JJ KK&uxxEIIu{{BCeOU[[%BSBS$ST   {:{|.(...b9r