o gP@sdZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZGdd d eZGd d d eZGd d d e jZddZdS)z/Module to add McCabe checker class for pylint. )nodes) PathGraph)PathGraphingAstVisitor)checkers)check_messages)HIGHIAstroidCheckercseZdZfddZZS)rcstjdddd||_dS)N)nameentitylineno)super__init__rootselfnode __class__V/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pylint/extensions/mccabe.pyrs zPathGraph.__init__)__name__ __module__ __qualname__r __classcell__rrrrrsrcseZdZfddZddZddZddZeZd d ZeZ Z Z Z Z ZZZZZZZZZZZZd d ZeZd dZdddZddZZS)rcstd|_dS)Nr)rr_bottom_counter)rrrrrs  zPathGraphingAstVisitor.__init__cGs$|D] }|j|g|RqdSN) get_childrendispatch)rrargschildrrrdefault#s zPathGraphingAstVisitor.defaultcGsT||_|j}|j|}|dur"|j}t|jd||j}||j|<||g|RS)Nvisit)rr_cachegetrgetattrvisitorr")rrr klassmeth class_namerrrr's  zPathGraphingAstVisitor.dispatchcCs|jdur2||}||_||j|j}|jd7_|j|j||j||||_dSt||_||_||j|j|j|j |j <| dS)Nr ) graph _append_nodetail dispatch_listbodyrconnectrgraphs classnamer reset)rrpathnodebottomrrrvisitFunctionDef1s       z'PathGraphingAstVisitor.visitFunctionDefcCs||dSr)r,rrrrvisitSimpleStatementEsz+PathGraphingAstVisitor.visitSimpleStatementcCs||||jdSr)r,r.r/rrrr visitWithbs z PathGraphingAstVisitor.visitWithcCs$|jsdS|j|j|||_|Sr)r-r+r0rrrrr,hs z#PathGraphingAstVisitor._append_nodercCs`|jdur"t||_|||||j|j|j|<|dS||||||dS)z?create the subgraphs representing any `if` and `for` statementsN)r+r_subgraph_parser1r2r3r,)rrr extra_blocksrrr _subgraphos    z PathGraphingAstVisitor._subgraphcCsg}||_||j||j|D]}||_||j||jq|jr8||_||j||jn|||r[|j}|jd7_|D] }|j||qL||_dSdS)z@parse the body and any `else` block of `if` and `for` statementsr N)r-r.r/appendorelserr+r0)rrr4r: loose_endsextrar5endrrrr9{s(      z&PathGraphingAstVisitor._subgraph_parse)r) rrrrr"rr6visitAsyncFunctionDefr7 visitAssert visitAssignvisitAugAssign visitDelete visitPrint visitRaise visitYield visitImport visitCallvisitSubscript visitPass visitContinue visitBreak visitGlobal visitReturn visitExpr visitAwaitr8visitAsyncWithr,r;r9rrrrrrsV       rc@sPeZdZdZeZdZddiZddddd d ffZe d d e j d dfddZ dS)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)r"typemetavarhelprWrreturnNcCst}|jD]}|||q|jD]2}|}|j}t|dr*d|jd}n d|j j d}||j j kr;q|jd|t||fdqdS)zvisit an astroid.Module node to check too complex rating and add message if is greather than max_complexity stored from optionsr 'zThis 'rW)r confidencer N)rr/preorderr1values complexityrhasattrr rrlowerconfigmax_complexity add_messager)rrr'r!r+rb node_namerrr visit_modules    z McCabeMethodChecker.visit_module) rrr__doc__r__implements__r msgsoptionsrrModulerirrrrrTs   rTcCs|t|dS)zRequired method to auto register this checker. :param linter: Main interface object for Pylint plugins :type linter: Pylint object N)register_checkerrT)linterrrrregistersrqN)rjastroidrmccaberMccabe_PathGraphrMccabe_PathGraphingAstVisitorpylintrpylint.checkers.utilsrpylint.interfacesrr BaseCheckerrTrqrrrrs     u 1